You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
979 B
36 lines
979 B
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):
|
|
|
|
if not self.conf('automation_username'):
|
|
log.error('Please fill in your username')
|
|
return []
|
|
|
|
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
|
|
|