From b4afac95fd6dbf7cf56fe3528b7b4e5636ebc3df Mon Sep 17 00:00:00 2001 From: Thomas Wade Date: Tue, 21 Apr 2020 20:33:55 +0930 Subject: [PATCH] Get recommended submissions --- FAAPI/FAAPI.py | 14 ++++++++++++++ setup.py | 2 +- 2 files changed, 15 insertions(+), 1 deletion(-) 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',