def _connect(self):
if self.quiet is False:
kodi.log("Connecting to " + self.db_file)
try:
from sqlite3 import dbapi2 as database
if self.quiet is False:
kodi.log("%s loading sqlite3 as DB engine" % kodi.get_name())
except:
from pysqlite2 import dbapi2 as database
if self.quiet is False:
kodi.log("%s loading pysqlite2 as DB engine" % kodi.get_name())
if self.quiet is False:
kodi.log("Connecting to SQLite on: " + self.db_file)
directory = kodi.vfs.dirname(self.db_file)
if not kodi.vfs.exists(directory):
kodi.vfs.mkdir(directory)
self.DBH = database.connect(self.db_file, check_same_thread=False)
try:
self.DBC = self.DBH.cursor()
self.__connected = True
except Exception as e:
self.handel_error(
DatabaseException("SQLite Database Error: %s" % e))
kodi.log("SQLite Database Error: %s" % e)
def github_install():
import re
from libs import github_installer
from libs import github
c = kodi.dialog_confirm("Confirm Install",
kodi.arg('file'),
yes="Install",
no="Cancel")
if not c: return
addon_id = re.sub("-[\d\.]+zip$", "", kodi.arg('file'))
github_installer.GitHub_Installer(
addon_id, kodi.arg('url'), kodi.arg('full_name'),
kodi.vfs.join("special://home", "addons"))
r = kodi.dialog_confirm(kodi.get_name(),
'Click Continue to install more addons or',
'Restart button to finalize addon installation',
yes='Restart',
no='Continue')
if r:
import sys
import xbmc
if sys.platform in ['linux', 'linux2', 'win32']:
xbmc.executebuiltin('RestartApp')
else:
xbmc.executebuiltin('ShutDown')
def get_installed_resources():
global search_path
import os
main = kodi.vfs.join(os.path.dirname(os.path.abspath(__file__)),
'scrapers')
results = [{
'path': main,
'name': 'ScrapeCore',
'addonid': 'script.module.scrapecore'
}]
test_path = kodi.vfs.join(kodi.get_path(), 'resources/scrapers/')
if kodi.vfs.exists(test_path):
results += [{
'path': kodi.vfs.join(kodi.get_path(), 'resources'),
'name': kodi.get_name(),
'addonid': kodi.get_id()
}]
temp = kodi.kodi_json_request(
"Addons.GetAddons", {
"installed": True,
'type': 'kodi.resource.images',
"properties": ["path", "name"]
})
for a in temp['result']['addons']:
if a['type'] == 'kodi.resource.images' and a['addonid'].startswith(
'resource.scrapecore.'):
del a['type']
results.append(a)
return results
def master_control():
options = [
"Send to Master Control", "Stream with Master Control",
"Master Control Queue"
]
c = kodi.dialog_select("Master Control Menu", options)
if c is False: return
if c == 2:
kodi.execute_url("plugin://master.control?mode=queue")
elif c == 1:
from scrapecore import scrapers
resolved_url = scrapers.get_scraper_by_name(
kodi.args['service']).resolve_url(kodi.args['raw_url'])
if not resolved_url: return
from mastercontrol import api as master_control
stream_url = master_control.get_streaming_url(resolved_url)
kodi.play_url(stream_url)
elif c == 0:
ids = kodi.arg('ids', decode='json')
from commoncore import trakt
from mastercontrol import api as master_control
if kodi.args['media'] == 'movie':
media = 'movie'
title = "%s (%s)" % (kodi.args['title'], kodi.args['year'])
filename = kodi.vfs.clean_file_name(title)
destination = ''
else:
media = 'tvshow'
destination = kodi.vfs.join(kodi.args['title'],
"Season %s" % kodi.args['season'])
title = "%s - S%02dE%02d" % (kodi.args['title'],
int(kodi.args['season']),
int(kodi.args['episode']))
filename = kodi.vfs.clean_file_name(title)
from scrapecore import scrapers
resolved_url = scrapers.get_scraper_by_name(
kodi.args['service']).resolve_url(kodi.args['raw_url'])
if not resolved_url: return
video = {
"type": media,
"filename": filename,
"url": resolved_url,
"title": title,
"addon": kodi.get_id(),
"destination": destination,
"trakt_id": kodi.args['trakt_id']
}
response = master_control.enqueue(video)
kodi.log(response)
message = 'Failed Adding to Queue %s.' % (title)
try:
if response['status'] == 200:
message = 'Added to Queue %s.' % (title)
except:
pass
kodi.notify(kodi.get_name(), message)
def create_custom_list(title, description=None, private=True, allow_comments=False):
uri = '/users/me/lists'
post_dict = {
"name": title,
"description": "Created by %s" % kodi.get_name() if description is None else description,
"privacy": "private",
"display_numbers": private,
"allow_comments": allow_comments
}
return trakt.request(uri, data=post_dict, auth=True)
def github_install():
import re
from libs import github_installer
from libs import github
c = kodi.dialog_confirm("Confirm Install", kodi.arg('file'), yes="Install", no="Cancel")
if not c: return
addon_id = re.sub("-[\d\.]+zip$", "", kodi.arg('file'))
github_installer.GitHub_Installer(addon_id, kodi.arg('url'), kodi.arg('full_name'), kodi.vfs.join("special://home", "addons"))
r = kodi.dialog_confirm(kodi.get_name(), 'Click Continue to install more addons or', 'Restart button to finalize addon installation', yes='Restart', no='Continue')
if r:
import sys
import xbmc
if sys.platform in ['linux', 'linux2', 'win32']:
xbmc.executebuiltin('RestartApp')
else:
xbmc.executebuiltin('ShutDown')
def create_custom_list(title,
description=None,
private=True,
allow_comments=False):
uri = '/users/me/lists'
post_dict = {
"name":
title,
"description":
"Created by %s" %
kodi.get_name() if description is None else description,
"privacy":
"private",
"display_numbers":
private,
"allow_comments":
allow_comments
}
return trakt.request(uri, data=post_dict, auth=True)
def install_batch():
import xbmcgui
from libs import github
from libs import github_installer
if kodi.mode == 'install_batch':
url = kodi.arg('url')
xml, zip_ref = github.batch_installer(url)
else:
url = kodi.dialog_file_browser('Select a install file', mask='.zip')
if not github.re_installer.search(url): return
xml, zip_ref = github.batch_installer(url, True)
if not kodi.dialog_confirm('Batch Installer?', "Click YES to proceed.", "This will install a list of addons.", "Some configuration files and settings may be overwritten."): return
if not xml: return
# Install each addon as instructed
installed_list = []
count = 0
for a in xml.findAll('addon'):
count +=1
PB = kodi.ProgressBar()
PB.new('Batch Installer - Progress', count)
for a in xml.findAll('addon'):
addon_id = a.find('addon_id')
username = a.find('username')
if addon_id is None or username is None: continue
username = username.text
addon_id = addon_id.text
PB.next(addon_id)
if not kodi.get_condition_visiblity("System.HasAddon(%s)"% addon_id):
if PB.is_canceled(): return
kodi.log("Batch install " + addon_id)
url, filename, full_name, version = github.find_zip(username, addon_id)
installed_list += github_installer.GitHub_Installer(addon_id, url, full_name, kodi.vfs.join("special://home", "addons"), quiet=True, batch=True, installed_list=installed_list).installed_list
kodi.sleep(1000)
# Look for config files.
# Need to add error checking for missing config files
configs= xml.find('configs')
if configs is not None and 'dir' in configs.attrs[0]:
config_dir = configs['dir']
for config in configs.findAll('config'):
source = config.find('source')
destination = config.find('destination')
if source is None or destination is None: continue
source = source.text
destination = destination.text
if not kodi.vfs.exists(destination): kodi.vfs.mkdir(destination, True)
kodi.vfs.write_file(kodi.vfs.join(destination, source), zip_ref.read(config_dir + '/' + source))
# Now look for individual setting key and value pairs
# Set them as instructed
settings= xml.find('settings')
if settings is not None:
for setting in settings.findAll('setting'):
if 'addon_id' in setting.attrs[0]:
addon_id = setting['addon_id']
k = setting.find('key')
v = setting.find('value')
if k is None or v is None: continue
kodi.set_setting(k.text, v.text, addon_id)
builtins= xml.find('builtins')
if builtins is not None:
for cmd in builtins.findAll('command'):
cmd = cmd.text
kodi.run_command(cmd)
jsonrpc= xml.find('jsonrpc')
if jsonrpc is not None:
from ast import literal_eval
for cmd in jsonrpc.findAll('command'):
method = cmd.find('method').text
params = literal_eval(cmd.find('params').text)
id = cmd.find('id').text
kodi.kodi_json_request(method, params, id)
# Now clean up
zip_ref.close()
PB.close()
r = kodi.dialog_confirm(kodi.get_name(), 'Click Continue to install more addons or', 'Restart button to finalize addon installation', yes='Restart', no='Continue')
if r:
import sys
import xbmc
if sys.platform in ['linux', 'linux2', 'win32']:
xbmc.executebuiltin('RestartApp')
else:
xbmc.executebuiltin('ShutDown')
def movie_list():
from commoncore import trakt
from coreplayback import in_progress
api = kodi.arg('api', decode='json')
if api['method'] == "search":
query = core.get_query('Movies', api['args'][0])
api['args'] = (query, 'movie')
results = core.execute_api(locals(), api)
if not results:
kodi.handel_error(kodi.get_name(), 'Not results found')
watched = trakt.get_watched_history(
'movies') if trakt.is_authorized() else []
@kodi.map_directory(results['items'])
def make_movie_directory(record):
ids = record['movie']['ids'] if 'movie' in record else record['ids']
infoLabel = core.make_infolabel('movie', record, watched=watched)
infoLabel['display'] = "%s (%s)" % (infoLabel['title'],
infoLabel['year'])
menu = kodi.ContextMenu()
inprogress = in_progress('movie', infoLabel['trakt_id'])
menu.add(
'Toggle Watched', {
"mode": "toggle_watched",
"api": {
"name":
"trakt",
"method":
"set_watched_state",
"args":
("movie", infoLabel['trakt_id'], infoLabel['playcount']
== 0),
"refresh":
True
}
})
if kodi.mode in ['shows_watchlist', 'shows_custom_list']:
menu.add(
'Remove From List', {
"mode": "execute_api",
"api": {
"name": "trakt",
"method": "remove_from_list",
"args":
("movie", api['args'][0], infoLabel['trakt_id']),
"refresh": True,
"confirm": "Confirm Remove from List?",
"message": api['args'][0]
}
})
else:
menu.add(
'Add to List', {
"mode": "execute_api",
"api": {
"name": "trakt",
"method": "add_to_list",
"args": ("movie", infoLabel['trakt_id'])
}
})
menu.add(
'Set Default View', {
"mode": "set_default_view",
"api": {
"name": "kodi",
"method": "set_default_view",
"args": ("movie", )
}
})
kodi.add_menu_item(
{
"mode": "search_streams",
"media": "movie",
"trakt_id": infoLabel['trakt_id'],
"title": infoLabel['title'],
"year": infoLabel["year"],
"ids": ids
},
infoLabel,
in_progress=inprogress,
menu=menu,
image=infoLabel['poster'])
if 'total_pages' in results and results['total_pages'] > 1 and results[
'current_page'] < results['total_pages']:
kodi.add_menu_item(
{
"mode": kodi.mode,
"api": api,
"page": results['current_page'] + 1
}, {"title": core.format_color("Next Page >>", 'green')},
icon="next_page.png")
kodi.eod(DEFAULT_VIEWS.MOVIES)
def tv_list():
from commoncore import trakt
api = kodi.arg('api', decode='json')
if api['method'] == "search":
query = core.get_query('Shows', api['args'][0])
api['args'] = (query, 'show')
results = core.execute_api(locals(), api)
if not results:
kodi.handel_error(kodi.get_name(), 'Not results found')
@kodi.map_directory(results['items'])
def make_show_directory(record):
infoLabel = core.make_infolabel('show', record)
ids = record['show']['ids'] if 'show' in record else record['ids']
infoLabel['display'] = "%s (%s)" % (infoLabel['title'],
infoLabel['year'])
menu = kodi.ContextMenu()
menu.add(
'Set Default View', {
"mode": "set_default_view",
"api": {
"name": "kodi",
"method": "set_default_view",
"args": ("show", )
}
})
if kodi.mode in ['shows_watchlist', 'shows_custom_list']:
menu.add(
'Remove From List', {
"mode": "execute_api",
"api": {
"name": "trakt",
"method": "remove_from_list",
"args":
("show", api['args'][0], infoLabel['trakt_id']),
"refresh": True,
"confirm": "Confirm Remove from List?",
"message": api['args'][0]
}
})
else:
menu.add(
'Add to List', {
"mode": "execute_api",
"api": {
"name": "trakt",
"method": "add_to_list",
"args": ("show", infoLabel['trakt_id'])
}
})
kodi.add_menu_item(
{
"mode": "list_seasons",
"trakt_id": infoLabel['trakt_id'],
"ids": ids
},
infoLabel,
menu=menu,
image=infoLabel['poster'])
if 'total_pages' in results and results['total_pages'] > 1 and results[
'current_page'] < results['total_pages']:
kodi.add_menu_item(
{
"mode": kodi.mode,
"api": api,
"page": results['current_page'] + 1
}, {"title": core.format_color("Next Page >>", 'green')},
icon="next_page.png")
kodi.eod(DEFAULT_VIEWS.SHOWS)
def in_progress():
import json
from commoncore import trakt
import coreplayback
api = kodi.arg('api', decode='json')
results = core.execute_api(locals(), api)
if not results:
kodi.handel_error(kodi.get_name(), 'Not results found')
return
def make_show_directory(record):
ids = json.loads(record[0])
metadata = json.loads(record[1])
episode = metadata['episode']
show = metadata['show']['items']
infoLabel = metadata['episode']
infoLabel['display'] = "%sx%s. %s - %s" % (
infoLabel['season'], infoLabel['episode'], show['title'],
infoLabel['title'])
menu = kodi.ContextMenu()
menu.add(
'Set Default View', {
"mode": "set_default_view",
"api": {
"name": "kodi",
"method": "set_default_view",
"args": ("show", )
}
})
menu.add(
'Add to List', {
"mode": "execute_api",
"api": {
"name": "trakt",
"method": "add_to_list",
"args": ("show", infoLabel['trakt_id'])
}
})
kodi.add_menu_item(
{
"mode": "search_streams",
"media": "episode",
"trakt_id": infoLabel['trakt_id'],
"title": show['title'],
"year": show['year'],
"season": infoLabel['season'],
"episode": infoLabel['episode'],
"ids": ids
},
infoLabel,
menu=menu,
in_progress=True,
image=infoLabel['poster'])
def make_movie_directory(record):
ids = json.loads(record[0])
infoLabel = json.loads(record[1])
infoLabel['display'] = "%s (%s)" % (infoLabel['title'],
infoLabel['year'])
menu = kodi.ContextMenu()
menu.add(
'Set Default View', {
"mode": "set_default_view",
"api": {
"name": "kodi",
"method": "set_default_view",
"args": ("movie", )
}
})
menu.add(
'Add to List', {
"mode": "execute_api",
"api": {
"name": "trakt",
"method": "add_to_list",
"args": ("movie", infoLabel['trakt_id'])
}
})
kodi.add_menu_item(
{
"mode": "search_streams",
"media": "movie",
"trakt_id": infoLabel['trakt_id'],
"title": infoLabel['title'],
"year": infoLabel['year'],
"ids": ids
},
infoLabel,
menu=menu,
in_progress=True,
image=infoLabel['poster'])
if kodi.mode == 'shows_inprogress':
map(make_show_directory, results)
else:
map(make_movie_directory, results)
kodi.eod(DEFAULT_VIEWS.SHOWS)
def search_streams():
from scrapecore import scrapers
results = scrapers.search(kodi.args['media'],
kodi.args['title'],
episode_title=kodi.arg('episode_title'),
year=kodi.arg('year'),
season=kodi.arg('season'),
episode=kodi.arg('episode'),
trakt_id=kodi.arg('trakt_id'))
if len(results) == 0:
kodi.handel_error(kodi.get_name(), 'Not results found')
return
ids = kodi.arg('ids', decode='json')
ids['season'] = kodi.arg('season')
ids['episode'] = kodi.arg('episode')
@kodi.map_directory(results)
def make_stream_directory(r):
menu = kodi.ContextMenu()
menu.add(
'Set Default View', {
"mode": "set_default_view",
"api": {
"name": "kodi",
"method": "set_default_view",
"args": ("stream", )
}
})
infoLabel = {"title": r['title'], "display": r['display']}
menu.add(
'Master Control', {
"mode": "master_control",
"media": kodi.args['media'],
"raw_url": r['raw_url'],
"service": r['service'],
"title": kodi.args['title'],
"year": kodi.arg('year'),
"season": kodi.arg('season'),
"episode": kodi.arg('episode'),
"trakt_id": kodi.arg('trakt_id')
})
if r['torrent']:
menu.add(
'Add to Cloud', {
"mode": "execute_api",
"api": {
"name": "premiumize",
"method": "create_transfer",
"args": (r['raw_url'], ),
"notify": True,
"confirm": "Add to Premiumize Cloud?",
"message": r['title']
}
})
kodi.add_video_item(
{
"mode": "play_stream",
"raw_url": r['raw_url'],
"title": kodi.args['title'],
"service": r['service'],
"ids": ids,
"media": kodi.args['media'],
"trakt_id": kodi.args['trakt_id']
},
infoLabel,
menu=menu,
icon=r['icon'])
kodi.eod(DEFAULT_VIEWS.STREAMS)
def install_batch():
import xbmcgui
from libs import github
from libs import github_installer
if kodi.mode == 'install_batch':
url = kodi.arg('url')
xml, zip_ref = github.batch_installer(url)
else:
url = kodi.dialog_file_browser('Select a install file', mask='.zip')
if not github.re_installer.search(url): return
xml, zip_ref = github.batch_installer(url, True)
if not kodi.dialog_confirm('Batch Installer?', "Click YES to proceed.", "This will install a list of addons.", "Some configuration files and settings may be overwritten."): return
if not xml: return
# Install each addon as instructed
installed_list = []
count = 0
for a in xml.findAll('addon'):
count +=1
PB = kodi.ProgressBar()
PB.new('Batch Installer - Progress', count)
for a in xml.findAll('addon'):
addon_id = a.find('addon_id')
username = a.find('username')
if addon_id is None or username is None: continue
username = username.text
addon_id = addon_id.text
PB.next(addon_id)
if not kodi.get_condition_visiblity("System.HasAddon(%s)"% addon_id):
if PB.is_canceled(): return
kodi.log("Batch install " + addon_id)
url, filename, full_name, version = github.find_zip(username, addon_id)
installed_list += github_installer.GitHub_Installer(addon_id, url, full_name, kodi.vfs.join("special://home", "addons"), quiet=True, batch=True, installed_list=installed_list).installed_list
kodi.sleep(1000)
# Look for config files.
# Need to add error checking for missing config files
configs= xml.find('configs')
if configs is not None and 'dir' in configs.attrs[0]:
config_dir = configs['dir']
for config in configs.findAll('config'):
source = config.find('source')
destination = config.find('destination')
if source is None or destination is None: continue
source = source.text
destination = destination.text
if not kodi.vfs.exists(destination): kodi.vfs.mkdir(destination, True)
kodi.vfs.write_file(kodi.vfs.join(destination, source), zip_ref.read(config_dir + '/' + source))
# Now look for individual setting key and value pairs
# Set them as instructed
settings= xml.find('settings')
if settings is not None:
for setting in settings.findAll('setting'):
if 'addon_id' in setting.attrs[0]:
addon_id = setting['addon_id']
k = setting.find('key')
v = setting.find('value')
if k is None or v is None: continue
kodi.set_setting(k.text, v.text, addon_id)
builtins= xml.find('builtins')
if builtins is not None:
for cmd in builtins.findAll('command'):
cmd = cmd.text
kodi.run_command(cmd)
jsonrpc= xml.find('jsonrpc')
if jsonrpc is not None:
from ast import literal_eval
for cmd in jsonrpc.findAll('command'):
method = cmd.find('method').text
params = literal_eval(cmd.find('params').text)
id = cmd.find('id').text
kodi.kodi_json_request(method, params, id)
# Now clean up
zip_ref.close()
PB.close()
r = kodi.dialog_confirm(kodi.get_name(), 'Click Continue to install more addons or', 'Restart button to finalize addon installation', yes='Restart', no='Continue')
if r:
import sys
import xbmc
if sys.platform in ['linux', 'linux2', 'win32']:
xbmc.executebuiltin('RestartApp')
else:
xbmc.executebuiltin('ShutDown')