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.
82 lines
2.1 KiB
82 lines
2.1 KiB
from beautifulsoup import BeautifulSoup
|
|
from couchpotato.core.providers.userscript.base import UserscriptBase
|
|
import re
|
|
|
|
|
|
class IMDB(UserscriptBase):
|
|
|
|
includes = ['http*://*.imdb.com/title/tt*', 'http*://imdb.com/title/tt*']
|
|
|
|
def getMovie(self, url):
|
|
|
|
data = self.urlopen(url)
|
|
|
|
html = BeautifulSoup(data)
|
|
headers = html.findAll('h5')
|
|
|
|
# Don't add TV show
|
|
for head in headers:
|
|
if 'seasons' in head.lower():
|
|
return False
|
|
|
|
return {
|
|
'id': re.search('(?P<id>tt[0-9{7}]+)', url).group('id'),
|
|
'year': self.getYear(html)
|
|
}
|
|
|
|
def getYear(self, html):
|
|
|
|
headers = html.findAll('h1')
|
|
|
|
tv_pattern = '/^\((TV|Video) ([0-9]+)\)$/';
|
|
|
|
try:
|
|
year = headers[0].findAll('a').text
|
|
return year
|
|
except:
|
|
pass
|
|
|
|
for head in headers:
|
|
match = re.search(tv_pattern, head)
|
|
if match:
|
|
return match[1]
|
|
|
|
#CouchPotato['imdb.com'] = (function(){
|
|
#
|
|
# function isMovie(){
|
|
# var series = document.getElementsByTagName('h5')
|
|
# for (var i = 0; i < series.length; i++) {
|
|
# if (series[i].innerHTML == 'Seasons:') {
|
|
# return false;
|
|
# }
|
|
# }
|
|
# return true;
|
|
# }
|
|
#
|
|
# function getId(){
|
|
# return 'tt' + location.href.replace(/[^\d+]+/g, '');
|
|
# }
|
|
#
|
|
# function getYear(){
|
|
# try {
|
|
# return document.getElementsByTagName('h1')[0].getElementsByTagName('a')[0].text;
|
|
# } catch (e) {
|
|
# var spans = document.getElementsByTagName('h1')[0].getElementsByTagName('span');
|
|
# var pattern = /^\((TV|Video) ([0-9]+)\)$/;
|
|
# for (var i = 0; i < spans.length; i++) {
|
|
# if (spans[i].innerHTML.search(pattern)) {
|
|
# return spans[i].innerHTML.match(pattern)[1];
|
|
# }
|
|
# }
|
|
# }
|
|
# }
|
|
#
|
|
# var constructor = function(){
|
|
# if(isMovie()){
|
|
# lib.osd(getId(), getYear());
|
|
# }
|
|
# }
|
|
#
|
|
# return constructor;
|
|
#
|
|
#})();
|
|
|