Merge ee2df5aff2 into b538f9a08d

couchpotato/core/providers/automation/goodfilms/__init__.py

@@ -0,0 +1,28 @@
+from .main import Goodfilms
+
+def start():
+    return Goodfilms()
+
+config = [{
+    'name': 'goodfilms',
+    'groups': [
+        {
+            'tab': 'automation',
+            'list': 'watchlist_providers',
+            'name': 'goodfilms_automation',
+            'label': 'Goodfilms',
+            'description': 'import movies from your <a href="http://goodfil.ms">Goodfilms</a> queue',
+            'options': [
+                {
+                    'name': 'automation_enabled',
+                    'default': False,
+                    'type': 'enabler',
+                },
+                {
+                    'name': 'automation_username',
+                    'label': 'Username',
+                },
+            ],
+        },
+    ],
+}]

couchpotato/core/providers/automation/goodfilms/main.py

@@ -0,0 +1,32 @@
+from couchpotato.core.logger import CPLog
+from couchpotato.core.providers.automation.base import Automation
+from bs4 import BeautifulSoup
+
+log = CPLog(__name__)
+
+
+class Goodfilms(Automation):
+
+    url = 'http://goodfil.ms/%s/queue'
+
+    def getIMDBids(self):
+
+        movies = []
+
+        for movie in self.getWatchlist():
+            imdb_id = self.search(movie.get('title'), movie.get('year'), imdb_only = True)
+            movies.append(imdb_id)
+
+        return movies
+
+    def getWatchlist(self):
+
+        url = self.url % self.conf('automation_username')
+        soup = BeautifulSoup(self.getHTMLData(url))
+
+        movies = []
+
+        for movie in soup.find_all('div', attrs={ 'class': 'movie', 'data-film-title': True }):
+            movies.append({ 'title': movie['data-film-title'], 'year': movie['data-film-year'] })
+
+        return movies