From 55300aef48335969f5615f44e6ac5e1bf05667ec Mon Sep 17 00:00:00 2001 From: Loic Coyle Date: Thu, 20 Feb 2020 01:10:57 +0100 Subject: added top --- tpblite/models/utils.py | 8 ++++++++ tpblite/tpblite.py | 16 ++++++++++++++++ 2 files changed, 24 insertions(+) (limited to 'tpblite') diff --git a/tpblite/models/utils.py b/tpblite/models/utils.py index 6c479d4..8043707 100644 --- a/tpblite/models/utils.py +++ b/tpblite/models/utils.py @@ -42,6 +42,14 @@ class QueryParser: segments = ("browse", str(category), str(page), str(order), "0") return cls(base_url, segments) + @classmethod + def top(cls: Type[T], base_url: str, category: int, last_48: bool) -> T: + if last_48: + segments = ("top", '48h' + str(category)) + else: + segments = ("top", str(category)) + return cls(base_url, segments) + def _sendRequest(self): req = Request(self.url, headers=headers()) return urlopen(req).read() diff --git a/tpblite/tpblite.py b/tpblite/tpblite.py index 1ec3991..abd59ae 100644 --- a/tpblite/tpblite.py +++ b/tpblite/tpblite.py @@ -61,3 +61,19 @@ class TPB: q = QueryParser.browse(self.base_url, category, page, order) self._search_url = q.url return Torrents(q.html_source) + + def top(self, category: int=0, last_48=False): + """Get the top torrents of a category and return a list of Torrents + + Args: + category: Restrict search to specific category, for list of categories see + tpblite.models.constants.CATEGORIES + last_48: wether to fetch the top torrent in the last 48 hours or the overall top + + Return: + Torrent object + """ + q = QueryParser.top(self.base_url, category, last_48) + self._search_url = q.url + return Torrents(q.html_source) + -- cgit v1.2.3 From 85adf1b1126de2091828d5ed0309d037e66b3232 Mon Sep 17 00:00:00 2001 From: Loic Coyle Date: Sat, 22 Aug 2020 03:37:04 +0200 Subject: add 0 check --- tpblite/models/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'tpblite') diff --git a/tpblite/models/utils.py b/tpblite/models/utils.py index 8043707..b9126e9 100644 --- a/tpblite/models/utils.py +++ b/tpblite/models/utils.py @@ -44,8 +44,10 @@ class QueryParser: @classmethod def top(cls: Type[T], base_url: str, category: int, last_48: bool) -> T: + if category == 0: + category = "all" if last_48: - segments = ("top", '48h' + str(category)) + segments = ("top", "48h" + str(category)) else: segments = ("top", str(category)) return cls(base_url, segments) -- cgit v1.2.3 From e2a941fbfc032e1aa57994879c43ab0585324346 Mon Sep 17 00:00:00 2001 From: Loic Coyle Date: Sat, 22 Aug 2020 03:40:37 +0200 Subject: PEP8 and remove unused import --- tpblite/tpblite.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'tpblite') diff --git a/tpblite/tpblite.py b/tpblite/tpblite.py index abd59ae..b4028b6 100644 --- a/tpblite/tpblite.py +++ b/tpblite/tpblite.py @@ -1,11 +1,10 @@ from typing import Optional -from .models.torrents import Torrents, Torrent +from .models.torrents import Torrents from .models.utils import QueryParser class TPB: - def __init__(self, base_url: str = "https://tpb.party"): """ThePirateBay Object @@ -62,7 +61,7 @@ class TPB: self._search_url = q.url return Torrents(q.html_source) - def top(self, category: int=0, last_48=False): + def top(self, category: int = 0, last_48: bool = False): """Get the top torrents of a category and return a list of Torrents Args: -- cgit v1.2.3