diff options
Diffstat (limited to 'tpblite')
| -rw-r--r-- | tpblite/models/torrents.py | 1 | ||||
| -rw-r--r-- | tpblite/models/utils.py | 2 | ||||
| -rw-r--r-- | tpblite/tpblite.py | 11 |
3 files changed, 11 insertions, 3 deletions
diff --git a/tpblite/models/torrents.py b/tpblite/models/torrents.py index 8567280..9fd40c4 100644 --- a/tpblite/models/torrents.py +++ b/tpblite/models/torrents.py @@ -1,5 +1,6 @@ import re from bs4 import BeautifulSoup +from .utils import Query def fileSizeStrToInt(size_str): diff --git a/tpblite/models/utils.py b/tpblite/models/utils.py index 72409b8..977672b 100644 --- a/tpblite/models/utils.py +++ b/tpblite/models/utils.py @@ -8,7 +8,7 @@ class Query(object): Query object capable of getting html response given a search query and other parameters. ''' - def __init__(self, query, base_url='https://tpb.party', page=0, order=99, category=0): + def __init__(self, query, base_url, page, order, category): self.base_url = base_url segments = ('search', query, str(page), str(order), str(category)) self.url = URL(base_url, segments) diff --git a/tpblite/tpblite.py b/tpblite/tpblite.py index c987e49..c99bfd2 100644 --- a/tpblite/tpblite.py +++ b/tpblite/tpblite.py @@ -1,8 +1,15 @@ from .models.torrents import Torrents, Torrent from .models.utils import Query - - +class TPB(object): + + def __init__(self, base_url='https://tpb.party'): + self.base_url = base_url + + def search(self, query, base_url, page=0, order=99, category=0): + webpage = Query(query, base_url, page=0, order=99, category=0) + return Torrents(webpage) + def run(): q = Query('avengers endgame 1080p') print(q.url) |