diff options
Diffstat (limited to 'tpblite/models')
| -rw-r--r-- | tpblite/models/torrents.py | 6 | ||||
| -rw-r--r-- | tpblite/models/utils.py | 3 |
2 files changed, 5 insertions, 4 deletions
diff --git a/tpblite/models/torrents.py b/tpblite/models/torrents.py index 9fd40c4..7852c28 100644 --- a/tpblite/models/torrents.py +++ b/tpblite/models/torrents.py @@ -63,8 +63,8 @@ class Torrents(object): torrent list or dict. Has methods to select items from torrent list. ''' - def __init__(self, webpage): - self.webpage = webpage + def __init__(self, html_source): + self.html_source = html_source self.list = self._createTorrentList() def __str__(self): @@ -74,7 +74,7 @@ class Torrents(object): return iter(self.list) def _createTorrentList(self): - soup = BeautifulSoup(self.webpage, features='html.parser') + soup = BeautifulSoup(self.html_source, features='html.parser') rows = self.__getRows(soup) torrents = [] for row in rows: diff --git a/tpblite/models/utils.py b/tpblite/models/utils.py index 977672b..200982f 100644 --- a/tpblite/models/utils.py +++ b/tpblite/models/utils.py @@ -10,9 +10,10 @@ class Query(object): ''' def __init__(self, query, base_url, page, order, category): self.base_url = base_url + print('category is {} {}'.format(category,str(category))) segments = ('search', query, str(page), str(order), str(category)) self.url = URL(base_url, segments) - self.webpage = self._sendRequest() + self.html_source = self._sendRequest() def _sendRequest(self): req = Request(self.url, headers=headers()) |