PlayStream/ContentSources.py

#!/usr/bin/env python
# coding=utf8
#
# This file is part of PlayStream - enigma2 plugin to play video streams from various sources
# Copyright (c) 2016 ivars777 (ivars777@gmail.com)
# Distributed under the GNU GPL v3. For full terms see http://www.gnu.org/licenses/gpl-3.0.en.html
#

import sys, os, re
import glob, traceback
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from sources.SourceBase import stream_type
import util


class ContentSources(object):
    """Wrapper for content sources plugin"""

    #----------------------------------------------------------------------
    def __init__(self, plugin_path):
        self.plugins = {}
        self.error_content = [("..atpakaļ", "back", None, "Kļūda, atgriezties atpakaļ")]
        sys.path.insert(0, plugin_path)
        #for f in os.listdir(plugin_path):
        #for f in next(os.walk(plugin_path))[2]:
        print "ContentSources: Importing sources from "+plugin_path
        files = glob.glob(os.path.join(plugin_path, "*.py"))
        for f in files:
            fname, ext = os.path.splitext(f)
            fname = os.path.split(fname)[1]
            if fname == "__init__": continue
            if ext == '.py':
                print "Importing %s"%fname
                mod = __import__(fname)
                reload(mod)
                if "Source" in dir(mod):
                    self.plugins[fname] = mod.Source()
                    print fname+ " imported"
                else:
                    pass
                    #print fname+ "skipped"
        sys.path.pop(0)
        if not "config" in self.plugins:
            raise Exception("Problem importing content sources")
        cfg = self.plugins["config"]
        for pl in self.plugins.keys():
            found = False
            for lst in cfg.get_lists():
                for item in cfg.lists[lst]:
                    if item[1].split("::")[0]==pl:
                        found = True
                        break
                    if found: break
            if not found:
                title = self.plugins[pl].title if "title" in dir(self.plugins[pl]) else pl
                img = self.plugins[pl].img if "img" in dir(self.plugins[pl]) else ""
                desc = self.plugins[pl].desc if "desc" in dir(self.plugins[pl]) else title
                cfg.add_item("home",(title,"%s::home"%pl,img,desc))
                cfg.write_streams()

    def get_content(self,data):
        source = data.split("::")[0]
        if source in self.plugins:
            content = self.plugins[source].get_content(data)
            if content:
                if isinstance(content,list):
                    for i,item in enumerate(content):
                        item2=[]
                        for el in item:
                            if isinstance(el,unicode):
                                el = el.encode("utf8")
                            item2.append(el)
                        content[i]=tuple(item2)
                else:
                    item2=[]
                    for el in content:
                        if isinstance(el,unicode):
                            el = el.encode("utf8")
                        item2.append(el)
                    content=tuple(item2)
                return content
            else:
                return self.error_content
        else:
            return self.error_content

    def get_streams(self,data):
        if stream_type(data):
            if "::" in data:
                data = data.split("::")[1]
            content = self.get_content(data)
            stream = util.item()
            stream["name"] = data
            stream["url"] = data
            stream["type"] = stream_type(data)
            #stream["img"] = ""
            #stream["desc"] = ""
            return[stream]

        if not self.is_video(data):
            return []
        source = data.split("::")[0]
        if source in self.plugins:
            streams = self.plugins[source].get_streams(data)
            for s in streams:
                for k in s:
                    if isinstance(s[k],unicode):
                        s[k] = s[k].encode("utf8")
                if not "resolver" in s:
                    s["resolver"] = source
                if not "surl" in s or not s["surl"]:
                    s["surl"] = data
                if not "nfo" in s:
                    s["nfo"]={"movie":{"title":s["name"],"thumb":s["img"],"plot":s["desc"]}}
            return streams
        else:
            return []

    def get_info(self,data):
        nfo = {}
        if self.is_video(data):
            source = data.split("::")[0]
            if source in self.plugins:
                if "get_info" in dir(self.plugins[source]):
                    streams = self.get_streams(data)
                    nfo = streams[0]["nfo"] if streams and "nfo" in streams[0] else {}
                else:
                    nfo = self.plugins[source].get_info(data)
        else:
            pass    # TODO create nfo for listing
        return nfo


    def stream_type(self,data):
        return stream_type(data)

    def is_video(self,data):
        if self.stream_type(data):
            return True
        source = data.split("::")[0]
        if source in self.plugins:
            return self.plugins[source].is_video(data)
        else:
            return False

    def options_read(self,source):
        if source in self.plugins:
            options = self.plugins[source].options_read()
            if options:
                return options
            else:
                return None
        else:
            return None

    def options_write(self,source,options):
        if source in self.plugins:
            return self.plugins[source].options_write(options)
        else:
            return None

if __name__ == "__main__":

    sources = ContentSources("sources")
    if len(sys.argv)>1:
        data= sys.argv[1]
    else:
        data = "config::home"

    #options = sources.options_read("ltc")
    #print options
    history = []
    cur = ("Home",data,None,None)
    content = sources.get_content(cur[1])

    exit_loop = False
    while True:
        print
        for i,item in enumerate(content):
            s = "%i: %s - %s %s"%(i,item[0],item[1],item[2])
            print s #.encode(sys.stdout.encoding,"replace")

        while True:
            a = raw_input("Enter number, (-) for download, q for exit: ")
            if a in ("q","Q","x","X"):
                exit_loop = True
                print "Exiting"
                break
            try:
                n = int(a)
                break
            except:
                print "Not number!"
        if exit_loop: break
        download = False
        if n<0:
            n = abs(n)
            download = True
        cur2 = content[n]
        data0 = cur2[1].split("::")[1] if "::" in cur2[1] else cur2[1]
        if not data0:
            pass
        elif cur2[1] == "back":
            cur = history.pop()
        elif sources.is_video(cur2[1]):
            if sources.stream_type(cur2[1]):
                stream = util.item()
                stream["url"] = cur2[1]
                stream["name"] = cur2[0]
                streams = [stream]
            else:
                try:
                    streams = sources.get_streams(cur2[1])
                except Exception as e:
                    print unicode(e)
                    traceback.print_exc()
                    streams = []
            if streams:
                if not download:
                    util.play_video(streams)
                else:
                    Downloader.download_video(streams)
            else:
                print "**No stream to play - %s "%(
                    cur2[1])
                raw_input("Press any key")
            #import os
            #os.system('"c:\Program Files (x86)\VideoLAN\VLC\vlc.exe" "%s"'%cur2[1])

        else:
            if "{0}" in cur2[1]:
                a = raw_input("Enter value:")
                cur2 = (cur2[0],cur2[1].format(a),cur2[2],cur2[3])
            history.append(cur)
            cur = cur2
        try:
            content = sources.get_content(cur[1])
        except Exception as e:
            print unicode(e)
            traceback.print_exc()
            raw_input("Continue?")