Merge 27a2bb7391 into b538f9a08d

couchpotato/core/providers/torrent/torrentbytes/__init__.py

@@ -0,0 +1,42 @@
+from .main import TorrentBytes
+
+def start():
+    return TorrentBytes()
+
+config = [{
+    'name': 'torrentbytes',
+    'groups': [
+        {
+            'tab': 'searcher',
+            'subtab': 'providers',
+            'list': 'torrent_providers',
+            'name': 'TorrentBytes',
+            'description': 'See <a href="http://torrentbytes.net">TorrentBytes</a>',
+            'wizard': True,
+            'options': [
+                {
+                    'name': 'enabled',
+                    'type': 'enabler',
+                    'default': False,
+                },
+                {
+                    'name': 'username',
+                    'default': '',
+                },
+                {
+                    'name': 'password',
+                    'default': '',
+                    'type': 'password',
+                },
+                {
+                    'name': 'extra_score',
+                    'advanced': True,
+                    'label': 'Extra Score',
+                    'type': 'int',
+                    'default': 20,
+                    'description': 'Starting score for each release found via this provider.',
+                }
+            ],
+        },
+    ],
+}]

couchpotato/core/providers/torrent/torrentbytes/main.py

@@ -0,0 +1,88 @@
+from bs4 import BeautifulSoup
+from couchpotato.core.helpers.encoding import tryUrlencode
+from couchpotato.core.helpers.variable import tryInt
+from couchpotato.core.logger import CPLog
+from couchpotato.core.providers.torrent.base import TorrentProvider
+import traceback
+
+
+log = CPLog(__name__)
+
+
+class TorrentBytes(TorrentProvider):
+
+    urls = {
+        'test' : 'http://www.torrentbytes.net/',
+        'login' : 'http://www.torrentbytes.net/takelogin.php',
+        'detail' : 'http://www.torrentbytes.net/details.php?id=%s',
+        'search' : 'http://www.torrentbytes.net/browse.php?search=%s&cat=%d',
+        'download' : 'http://www.torrentbytes.net/download.php?id=%s&name=%s',
+    }
+
+    cat_ids = [
+        ([5], ['720p', '1080p']),
+        ([19], ['cam']),
+        ([19], ['ts', 'tc']),
+        ([19], ['r5', 'scr']),
+        ([19], ['dvdrip']),
+        ([5], ['brrip']),
+        ([20], ['dvdr']),
+    ]
+
+    http_time_between_calls = 1 #seconds
+    cat_backup_id = None
+
+    def _searchOnTitle(self, title, movie, quality, results):
+
+        url = self.urls['search'] % (tryUrlencode('%s %s' % (title.replace(':', ''), movie['library']['year'])), self.getCatId(quality['identifier'])[0])
+        data = self.getHTMLData(url, opener = self.login_opener)
+
+        if data:
+            html = BeautifulSoup(data)
+
+            try:
+		result_table = html.find('table', attrs = {'border' : '1'})
+		if not result_table:
+                    return
+				
+		entries = result_table.find_all('tr')
+				
+		for result in entries[1:]:
+                    cells = result.find_all('td')
+					
+                    link = cells[1].find('a', attrs = {'class' : 'index'})
+					
+                    full_id = link['href'].replace('details.php?id=', '')
+                    torrent_id = full_id[:6]
+                    name = link.contents[0]
+                    url = self.urls['download'] % (torrent_id, name)
+                    detail_url = self.urls['detail'] % torrent_id
+                    size = cells[6].contents[0] + cells[6].contents[2]
+                    seeders = cells[8].find('span').contents[0]
+                    leechers = cells[9].find('span').contents[0]
+					
+                    #log.debug('id: %s  name: %s  url: %s  details: %s  size: %s  seeders: %s  leechers: %s', (torrent_id, name, url, detail_url, size, seeders, leechers))
+					
+                    results.append({
+			'id': torrent_id,
+                        'name': name,
+                        'url': self.urls['download'] % (torrent_id, name),
+                        'detail_url': self.urls['detail'] % torrent_id,
+                        'download': self.loginDownload,
+                        'size': self.parseSize(size),
+                        'seeders': tryInt(seeders),
+                        'leechers': tryInt(leechers),
+                    })
+                    
+            except:
+                log.error('Failed to parsing %s: %s', (self.getName(), traceback.format_exc()))
+
+    def getLoginParams(self):
+        return tryUrlencode({
+            'username': self.conf('username'),
+            'password': self.conf('password'),
+            'login': 'submit',
+        })
+
+    def loginSuccess(self, output):
+        return 'logout.php' in output.lower() or 'Welcome' in output.lower()