diff options
Diffstat (limited to 'tpblite/models/torrents.py')
| -rw-r--r-- | tpblite/models/torrents.py | 103 |
1 files changed, 103 insertions, 0 deletions
diff --git a/tpblite/models/torrents.py b/tpblite/models/torrents.py new file mode 100644 index 0000000..8567280 --- /dev/null +++ b/tpblite/models/torrents.py @@ -0,0 +1,103 @@ +import re +from bs4 import BeautifulSoup + + +def fileSizeStrToInt(size_str): + '''Converts file size given in *iB format to bytes integer''' + + unit_dict = {'KiB':(2**10), + 'MiB':(2**20), + 'GiB':(2**30), + 'TiB':(2**40)} + try: + num = float(size_str[:-3]) + unit = size_str[-3:] + return int(num * unit_dict[unit]) + except Exception as e: + raise AttributeError('Cannot determine filesize: {0}, error: {1}'.format(size_str,e)) + +class Torrent(object): + ''' + Abstract class to contain info about torrent + magnet link, file size, number of seeds, number of leeches etc. + ''' + def __init__(self, html_row): + self.html_row = html_row + self.title = self._getTitle() + self.seeds, self.leeches = self._getPeers() + self.uploaded, self.filesize, self.byte_size, self.uploader = self._getFileInfo() + self.filesize_int = fileSizeStrToInt(self.filesize) + self.magnetlink = self._getMagnetLink() + + def __str__(self): + return '{0}, S: {1}, L: {2}, {3}'.format(self.title, + self.seeds, + self.leeches, + self.filesize) + + def _getTitle(self): + return self.html_row.find('a', class_='detLink').string + + def _getPeers(self): + taglist = self.html_row.find_all('td', align='right') + return int(taglist[0].string), int(taglist[1].string) + + def _getFileInfo(self): + text = self.html_row.find('font', class_='detDesc').get_text() + t = text.split(',') + uptime = t[0].replace('Uploaded ','') + size = t[1].replace('Size ', '') + byte_size = fileSizeStrToInt(size) + uploader = t[2].replace('ULed by ', '').strip() + return uptime, size, byte_size, uploader + + def _getMagnetLink(self): + tag = self.html_row.find('a', href=(re.compile('magnet'))) + link = tag.get('href') + return link + +class Torrents(object): + ''' + Torrent object, takes query response and parses into + torrent list or dict. Has methods to select items from + torrent list. + ''' + def __init__(self, webpage): + self.webpage = webpage + self.list = self._createTorrentList() + + def __str__(self): + return 'Torrents Object: {0} torrents'.format(len(self.list)) + + def __iter__(self): + return iter(self.list) + + def _createTorrentList(self): + soup = BeautifulSoup(self.webpage, features='html.parser') + rows = self.__getRows(soup) + torrents = [] + for row in rows: + torrents.append(Torrent(row)) + return torrents + + def __getRows(self, soup): + rows = soup.body.find_all('tr') + # remove first and last entries + del rows[0] + del rows[-1] + return rows + + def getBestTorrent(self, min_seeds=30, min_filesize='1 GiB', max_filesize='4 GiB'): + if not type(min_filesize) == 'int': + min_filesize = fileSizeStrToInt(min_filesize) + if not type(max_filesize) == 'int': + max_filesize = fileSizeStrToInt(max_filesize) + filtered_list = filter(lambda x: self._filterTorrent(x, min_seeds, min_filesize, max_filesize), self.list) + sorted_list = sorted(filtered_list, key=lambda x: x.seeds, reverse=True) + return sorted_list[0] + + def _filterTorrent(self, torrent, min_seeds, min_filesize, max_filesize): + if (torrent.seeds < min_seeds) or (torrent.filesize_int < min_filesize) or (torrent.filesize_int > max_filesize): + return False + else: + return True
\ No newline at end of file |