From 9befc5afdd74d584e818f95ceeaefce7c835e612 Mon Sep 17 00:00:00 2001
From: Janez Troha <janez.troha@gmail.com>
Date: Sun, 1 Jul 2012 01:51:36 +0200
Subject: [PATCH] PublicHD torrent site

---
 .../providers/torrent/publichd/__init__.py    |  23 ++++
 .../core/providers/torrent/publichd/main.py   | 129 ++++++++++++++++++
 2 files changed, 152 insertions(+)
 create mode 100644 couchpotato/core/providers/torrent/publichd/__init__.py
 create mode 100644 couchpotato/core/providers/torrent/publichd/main.py

diff --git a/couchpotato/core/providers/torrent/publichd/__init__.py b/couchpotato/core/providers/torrent/publichd/__init__.py
new file mode 100644
index 00000000..c28781e3
--- /dev/null
+++ b/couchpotato/core/providers/torrent/publichd/__init__.py
@@ -0,0 +1,23 @@
+from .main import PublicHD
+
+def start():
+    return PublicHD()
+
+config = [{
+    'name': 'publichd',
+    'groups': [
+        {
+            'tab': 'searcher',
+            'subtab': 'providers',
+            'name': 'PublicHD',
+            'description': 'Public Torrent site with only HD content.',
+            'options': [
+                {
+                    'name': 'enabled',
+                    'type': 'enabler',
+                    'default': False,
+                },
+            ],
+        },
+    ],
+}]
\ No newline at end of file
diff --git a/couchpotato/core/providers/torrent/publichd/main.py b/couchpotato/core/providers/torrent/publichd/main.py
new file mode 100644
index 00000000..04b1fb8f
--- /dev/null
+++ b/couchpotato/core/providers/torrent/publichd/main.py
@@ -0,0 +1,129 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+from bs4 import BeautifulSoup
+from couchpotato.core.event import fireEvent
+from couchpotato.core.helpers.variable import getTitle
+from couchpotato.core.logger import CPLog
+from couchpotato.core.providers.torrent.base import TorrentProvider
+import re
+from urlparse import parse_qs
+from urllib import quote_plus
+import urllib2
+log = CPLog(__name__)
+
+
+class PublicHD(TorrentProvider):
+
+    urls = {
+        'test': 'http://publichd.eu',
+        'download': 'http://publichd.eu/%s',
+        'detail': 'http://publichd.eu/index.php?page=torrent-details&id=%s',
+        'search': 'http://publichd.eu/index.php?page=torrents&search=%s&active=1&category=%d',
+        }
+
+    cat_ids = [([2], ['720p']), ([5], ['1080p']), ([15], ['bdrip']),
+               ([16], ['brrip']), ([16], ['blue-ray'])]
+
+    def search(self, movie, quality):
+
+        results = []
+        if self.isDisabled():
+            return results
+
+        movie_name = re.sub("\W", " ", getTitle(movie['library']))
+        movie_name = re.sub("  ", " ", movie_name)
+        log.info('Cleaned Name: %s', movie_name)
+        cache_key = 'publichd.%s.%s' % (movie['library']['identifier'], quality.get('identifier'))
+        searchUrl = self.urls['search'] \
+            % (quote_plus(movie_name + ' '
+               + quality['identifier']),
+               self.getCatId(quality['identifier'])[0])
+
+        data = self.getCache(cache_key, searchUrl)
+
+        try:
+            soup = BeautifulSoup(data)
+
+            resultsTable = soup.find('table', attrs={'id': 'bgtorrlist2'
+                    })
+            entries = resultsTable.findAll('tr')
+            for result in entries[1:]:
+                info_url = result.find(href=re.compile('torrent-details'
+                        ))
+                download = result.find(href=re.compile('\.torrent'))
+                date_uploaded = result.findAll('td')[3].string
+                if info_url and download and date_uploaded:
+                    new = {
+                        'type': 'torrent',
+                        'check_nzb': False,
+                        'description': '',
+                        'provider': self.getName(),
+                        }
+                    log.info('Name: %s', result.findAll('td')[1].string)
+                    log.info('Date uploaded: %s', date_uploaded)
+                    log.info('Seeds: %s', result.findAll('td'
+                             )[4].string)
+                    log.info('Leaches: %s', result.findAll('td'
+                             )[5].string)
+                    log.info('Size: %s', result.findAll('td')[7].string)
+
+                    url = parse_qs(info_url['href'])
+
+                    new['name'] = info_url.string
+                    new['id'] = url['id'][0]
+                    new['url'] = self.urls['download'] % download['href'
+                            ]
+                    new['size'] = self.parseSize(result.findAll('td'
+                            )[7].string)
+                    new['seeders'] = int(result.findAll('td')[4].string)
+                    new['leechers'] = int(result.findAll('td'
+                            )[5].string)
+                    new['imdbid'] = movie['library']['identifier']
+
+                    new['extra_score'] = self.extra_score
+                    new['score'] = fireEvent('score.calculate', new,
+                            movie, single=True)
+                    is_correct_movie = fireEvent(
+                        'searcher.correct_movie',
+                        nzb=new,
+                        movie=movie,
+                        quality=quality,
+                        imdb_results=True,
+                        single_category=False,
+                        single=True,
+                        )
+
+                    if is_correct_movie:
+                        new['download'] = self.download
+                        results.append(new)
+                        self.found(new)
+            return results
+        except:
+
+            log.info('No results found at PublicHD')
+            return []
+
+    def extra_score(self, req):
+        url = self.urls['detail'] % req['id']
+        imdbId = req['imdbid']
+        return self.imdbMatch(url, imdbId)
+
+    def imdbMatch(self, url, imdbId):
+        try:
+            data = urllib2.urlopen(url).read()
+            pass
+        except IOError:
+            log.error('Failed to open %s.' % url)
+            return ''
+
+        imdbIdAlt = re.sub('tt[0]*', 'tt', imdbId)
+        data = unicode(data, errors='ignore')
+        if 'imdb.com/title/' + imdbId in data or 'imdb.com/title/' \
+            + imdbIdAlt in data:
+            return 50
+        return 0
+
+    def download(self, url='', nzb_id=''):
+        torrent = self.urlopen(url)
+        return torrent