Userscript
This commit is contained in:
@@ -1,6 +1,4 @@
|
||||
from couchpotato.api import addApiView
|
||||
from couchpotato.core.event import addEvent, fireEvent
|
||||
from couchpotato.core.helpers.request import getParam
|
||||
from couchpotato.core.logger import CPLog
|
||||
from couchpotato.core.plugins.base import Plugin
|
||||
from urlparse import urlparse
|
||||
@@ -20,30 +18,26 @@ class UserscriptBase(Plugin):
|
||||
addEvent('userscript.get_excludes', self.getExclude)
|
||||
addEvent('userscript.get_version', self.getVersion)
|
||||
|
||||
addApiView('userscript.add_via_url', self.addViaUrl)
|
||||
addEvent('userscript.get_movie_via_url', self.belongsTo)
|
||||
|
||||
def search(self, name, year = None):
|
||||
|
||||
movie = fireEvent('movie.search', q = '%s %s' % (name, year), limit = 1)
|
||||
|
||||
return movie
|
||||
|
||||
def addViaUrl(self):
|
||||
def belongsTo(self, url):
|
||||
|
||||
url = getParam('url')
|
||||
host = urlparse(url).hostname
|
||||
if host.startswith('www.'):
|
||||
host = host[4:]
|
||||
|
||||
# Check if the url matches the provider
|
||||
is_provider = False
|
||||
for include in self.includes:
|
||||
if host in include:
|
||||
is_provider = True
|
||||
break
|
||||
if not is_provider:
|
||||
return False
|
||||
return self.getMovie(url)
|
||||
|
||||
return
|
||||
|
||||
fireEvent()
|
||||
def getMovie(self, url):
|
||||
return None
|
||||
|
||||
def getInclude(self):
|
||||
return self.includes
|
||||
|
||||
Reference in New Issue
Block a user