From 223f6c7a5f41b7918f0c20fbd1f0ac505023e575 Mon Sep 17 00:00:00 2001
From: Ruud <ruud@crashdummy.nl>
Date: Tue, 27 Sep 2011 23:45:55 +0200
Subject: [PATCH] Trailer provider

---
 .../providers/trailer/hdtrailers/__init__.py  |   6 +
 .../core/providers/trailer/hdtrailers/main.py | 111 ++++++++++++++++++
 2 files changed, 117 insertions(+)
 create mode 100644 couchpotato/core/providers/trailer/hdtrailers/__init__.py
 create mode 100644 couchpotato/core/providers/trailer/hdtrailers/main.py

diff --git a/couchpotato/core/providers/trailer/hdtrailers/__init__.py b/couchpotato/core/providers/trailer/hdtrailers/__init__.py
new file mode 100644
index 00000000..016db7a2
--- /dev/null
+++ b/couchpotato/core/providers/trailer/hdtrailers/__init__.py
@@ -0,0 +1,6 @@
+from .main import HDTrailers
+
+def start():
+    return HDTrailers()
+
+config = []
diff --git a/couchpotato/core/providers/trailer/hdtrailers/main.py b/couchpotato/core/providers/trailer/hdtrailers/main.py
new file mode 100644
index 00000000..8085ecce
--- /dev/null
+++ b/couchpotato/core/providers/trailer/hdtrailers/main.py
@@ -0,0 +1,111 @@
+from beautifulsoup import SoupStrainer, BeautifulSoup
+from couchpotato.core.logger import CPLog
+from couchpotato.core.providers.trailer.base import TrailerProvider
+from string import letters, digits
+from urllib import urlencode
+import re
+
+log = CPLog(__name__)
+
+
+class HDTrailers(TrailerProvider):
+
+    urls = {
+        'api': 'http://www.hd-trailers.net/movie/%s/',
+        'backup': 'http://www.hd-trailers.net/blog/',
+    }
+    providers = ['apple.ico', 'yahoo.ico', 'moviefone.ico', 'myspace.ico', 'favicon.ico']
+
+    def find(self, movie):
+
+        movie_name = movie['library']['titles'][0]['title']
+
+        url = self.url['api'] % self.movieUrlName(movie_name)
+        data = self.urlopen(url)
+
+        p480 = []
+        p720 = []
+        p1080 = []
+        did_alternative = False
+        for provider in self.providers:
+            results = self.findByProvider(data, provider)
+
+            # Find alternative
+            if results.get('404') and not did_alternative:
+                results = self.findViaAlternative(movie_name)
+                did_alternative = True
+
+            p480.extend(results.get('480p'))
+            p720.extend(results.get('720p'))
+            p1080.extend(results.get('1080p'))
+
+        return {'480p':p480, '720p':p720, '1080p':p1080}
+
+    def findViaAlternative(self, movie):
+        results = {'480p':[], '720p':[], '1080p':[]}
+
+        url = "%s?%s" % (self.url['backup'], urlencode({'s':movie}))
+
+        data = self.urlopen(url)
+
+        try:
+            tables = SoupStrainer('div')
+            html = BeautifulSoup(data, parseOnlyThese = tables)
+            result_table = html.findAll('h2', text = re.compile(movie))
+
+            for h2 in result_table:
+                if 'trailer' in h2.lower():
+                    parent = h2.parent.parent.parent
+                    trailerLinks = parent.findAll('a', text = re.compile('480p|720p|1080p'))
+                    try:
+                        for trailer in trailerLinks:
+                            results[trailer].insert(0, trailer.parent['href'])
+                    except:
+                        pass
+
+
+        except AttributeError:
+            log.debug('No trailers found in via alternative.')
+
+        return results
+
+    def findByProvider(self, data, provider):
+
+        results = {'480p':[], '720p':[], '1080p':[]}
+        try:
+            tables = SoupStrainer('table')
+            html = BeautifulSoup(data, parseOnlyThese = tables)
+            result_table = html.find('table', attrs = {'class':'bottomTable'})
+
+
+            for tr in result_table.findAll('tr'):
+                trtext = str(tr).lower()
+                if 'clips' in trtext:
+                    break
+                if 'trailer' in trtext and not 'clip' in trtext and provider in trtext:
+                    nr = 0
+                    resolutions = tr.findAll('td', attrs = {'class':'bottomTableResolution'})
+                    #sizes = tr.findNext('tr').findAll('td', attrs = {'class':'bottomTableFileSize'})
+                    for res in resolutions:
+                        results[str(res.a.contents[0])].insert(0, res.a['href'])
+                        #int(sizes[nr].contents[0].replace('MB', ''))
+                        nr += 1
+
+            return results
+
+        except AttributeError:
+            log.debug('No trailers found in provider %s.' % provider)
+            results['404'] = True
+
+        return results
+
+    def movieUrlName(self, string):
+        safe_chars = letters + digits + ' '
+        r = ''.join([char if char in safe_chars else ' ' for char in string])
+        name = re.sub('\s+' , '-', r).lower()
+
+        try:
+            int(name)
+            return '-' + name
+        except:
+            return name