diff --git a/FAAPI/FAAPI.py b/FAAPI/FAAPI.py index b2403ea..26642e5 100644 --- a/FAAPI/FAAPI.py +++ b/FAAPI/FAAPI.py @@ -52,6 +52,7 @@ class Submission: self._download_url: str = None self._faved: bool = None self._fave_url: str = None + self._recommended: List[Submission] = None @property def url(self): @@ -165,6 +166,13 @@ class Submission: return self._faved + @property + def recommended(self): + if self._recommended is None: + self._get_submission() + + return self._recommended + @staticmethod def new_or_cached(api, id: int): if id not in Submission._submissioncache: @@ -240,6 +248,12 @@ class Submission: self._comments.append(comment) + # Parse recommended submissions + self._recommended = [] + for recommended in soup.select('.preview-gallery-container a'): + id = int(recommended.get('href')[6:-1]) # e.g. /view/35992314/ + self._recommended.append(Submission.new_or_cached(self._api, id)) + def _toggle_fave(self): soup = self._api.get_soup(self._fave_url) diff --git a/setup.py b/setup.py index d99e349..abb8a73 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ from setuptools import setup, find_packages setup( name='FAAPI', - version='0.1.0', + version='0.2.0', packages=find_packages(), url='https://tem.party/gitea/tom/FAAPI', license='WTFPL',