initial commit

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