# -*- coding: utf-8 -*- import os,os.path,sys, glob, shutil, re import urllib, traceback try: import cPickle as pickle except: import pickle import pickle from kodiswift import Plugin, ListItem, storage from kodiswift import xbmc, xbmcgui, xbmcplugin, xbmcvfs, xbmcaddon, CLI_MODE from resources.lib.content import ContentSources, util cur_directory = os.path.dirname(__file__) icon_folder = os.path.join(cur_directory, "resources", "picons") icong_url = xbmcaddon.Addon().getAddonInfo("path") + "/resources/picons/" sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__)),"resources","lib", "content", "sources")) plugin = Plugin() plugin.load_addon_settings() use_storage = plugin.get_setting("general_use_storage",bool) # TODO vajag nočekot vai nav labāk lietot pickle storage_ttl = plugin.get_setting("general_ttl",int) use_proxy = plugin.get_setting("general_proxy_use",bool) proxy_url = plugin.get_setting("general_proxy_url",str) playlist = plugin.get_setting("general_playlist",str) #storage_path = os.path.join(plugin.storage_path,"sources.p") if use_storage: try: storage = plugin.get_storage("playstream","pickle",ttl=30) except Exception as e: os.remove(os.path.join(plugin.storage_path, "playstream")) try: storage = plugin.get_storage("playstream", "pickle", ttl=30) except Exception as e: print "[playstream] error opening storage", plugin.storage_path print "Got Exception: ", str(e) import traceback traceback.print_exc() plugin.notify("Error opening permament storage", "Info", 10000, xbmcgui.NOTIFICATION_INFO) storage = None os.remove(os.path.join(plugin.storage_path, "playstream")) cur_directory = os.path.dirname(__file__) sources_directory = os.path.join(cur_directory,"resources","lib", "content", "sources") cfg_list = glob.glob(os.path.join(sources_directory,"*.cfg")) for cf in cfg_list: cf2 = os.path.join(plugin.storage_path,os.path.split(cf)[1]) if not os.path.exists(cf2): shutil.copyfile(cf,cf2) if use_storage and storage is not None and "sources" in storage: print "[playstream] Restore sources from storage" sources = storage["sources"] # if use_storage and os.path.exists(storage_path): #"sources" in storage: #sources = pickle.load(open(storage_path,"rb")) else: print "[playstream] Create sources objects" sources = ContentSources.ContentSources(sources_directory) for source in sources.plugins: if not ("options" in dir(sources.plugins[source]) and sources.plugins[source].options): continue options = sources.plugins[source].options if not options: continue for option in options: key="%s_%s"%(source,option) if key in ("viaplay_device"): continue # exception list, value = plugin.get_setting(key) options[option] = value sources.plugins[source].options_write(options) prefix = "" @plugin.route(".+" ) def main(): global prefix prefix = "%s://%s/"%(plugin.request.scheme,plugin.request.netloc) plugin.set_content("movies") data = plugin.request.url.replace(prefix,"") data = urllib.unquote(data) sources.plugins["config"].read_streams() if not data: data = u"config::home" print "[playstream] processing data=%s"%data if sources.is_video(data): try: streams = sources.get_streams(data) except Exception,e: #xbmcgui.Dialog().ok("Error",unicode(e)) plugin.notify(unicode(e),"Error",10000, xbmcgui.NOTIFICATION_ERROR) traceback.print_exc() return plugin.set_resolved_url(None) if streams: return play_video(streams) else: plugin.notify("No streams found!","Error",10000,xbmcgui.NOTIFICATION_ERROR) return plugin.set_resolved_url(None) else: if "{0}" in data: q = plugin.keyboard(default=None, heading="Search for", hidden=False) if isinstance(q,str): q = q.decode("utf8") if isinstance(data,str): data = data.decode("utf8") data = data.format(q) try: items = get_list(data) except Exception,e: plugin.notify(unicode(e),"Error",10000,xbmcgui.NOTIFICATION_ERROR) traceback.print_exc() return [] #xbmc.executebuiltin('Container.SetViewMode(500)') plugin.set_view_mode(50) if use_storage and storage is not None: print "[playstream] Save sources to storage" storage["sources"] = sources storage.sync() return items def get_list(data): if isinstance(data,unicode): data = data.encode("utf8") content = sources.get_content(data) print "[playstream] %s items returned"%len(content) items = [] for item in content: if item[1] == "back": continue title = item[0].decode("utf8") if isinstance(item[0],str) else item[0] data2 = item[1].decode("utf8") if isinstance(item[1],str) else item[1] is_playable = True if sources.is_video(item[1]) else False img = item[2].decode("utf8") if isinstance(item[2],str) else item[2] desc = item[3].decode("utf8") if isinstance(item[3],str) else item[3] context_menu = [ ("Add to PlayStream playlist", u'RunScript(special://home/addons/%s/context_menu.py,"playlist","%s","%s","%s","%s")' % ( plugin.id, title, data2, playlist, proxy_url)), ("Add to PlayStream favorites", u'RunScript(special://home/addons/%s/context_menu.py,"add","%s","%s","%s","%s")'%( plugin.id, title, data2 ,img, desc)), ("Delete from PlayStream favorites", u'RunScript(special://home/addons/%s/context_menu.py,"delete","%s","%s","%s","%s")' % ( plugin.id, title, data2, img, desc)), ("Download", u'RunScript(special://home/addons/%s/context_menu.py,"download","%s","%s","%s","%s")' % ( plugin.id, title, data2, img, desc)), ] items.append({ "label": title, "path": prefix+data2, "thumbnail":thumb_data(img, is_playable), "info":{"plot":desc}, "is_playable":is_playable, "context_menu": context_menu, }) return items def play_video(streams): if len(streams)>1: slist = [] for s in streams: slist.append("%s [%s,%s]"%(s["name"],s["quality"],s["lang"])) res = xbmcgui.Dialog().select("Select stream",slist) if not CLI_MODE else 0 #res = xbmcgui.Dialog().contextmenu(slist) if not CLI_MODE else 0 stream = streams[res] else: stream = streams[0] subfiles = [] #stream = util.stream_chamge(stream) if use_proxy: if "resolver" in stream and stream["resolver"] in ("hqq","filmas") or \ "surl" in stream and re.search("http*://(hqq|goo\.gl)",stream["surl"]) or \ "lattelecom.tv/mobile-vod/" in stream["url"]: # TODO #re.search(r"http*://.+?lattelecom\.tv/.+?auth_token=[^=]+=", stream["url"]): stream["url"] = util.streamproxy_encode(stream["url"],stream["headers"],proxy_url) stream["headers"] = {} if stream["headers"]: hh = [] for k in stream["headers"]: h = "%s=%s"%(k,urllib.quote(stream["headers"][k])) hh.append(h) hh = "&".join(hh) stream["url"] = stream["url"] +"|"+hh print "[playstream] play_video ", stream["url"] if "subs" in stream and stream["subs"]: for sub in stream["subs"]: suburl = sub["url"] subs = util.Captions(suburl) srt = subs.get_srt() #subfile = plugin.temp_fn("subtitles.srt") subfile = os.path.join(os.path.dirname(__file__),sub["lang"]+".srt") f = open(subfile, "w") f.write(srt) f.close() subfiles.append(subfile) item = ListItem(label=stream["name"], thumbnail=thumb_data(stream["img"], True), path=stream["url"]) item.set_info("video",{"plot":stream["desc"]}) item.set_is_playable(True) return plugin.set_resolved_url(item,subfiles) #return plugin.play_video(item) def thumb_data(img, video=False): default = "video.png" if video else "folder.png" if img in ("default", ""): img = default if not img.startswith("http"): img = icong_url + img return img if __name__ == '__main__': if CLI_MODE: from kodiswift.cli.cli import main as start start() else: plugin.run() if use_storage and storage is not None: print "[playstream] Save sources to storage" storage["sources"] = sources storage.sync() print "Save sources to storage" #pickle.dump(sources,open(storage_path,"wb"),pickle.HIGHEST_PROTOCOL)