#!/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 # Used fragments of code from enigma2-plugin-tv3play by Taapat (https://github.com/Taapat/enigma2-plugin-tv3play) # import os,time,sys,os,os.path import datetime,re import urllib2 from enigma import ePicLoad, eServiceReference, eTimer, getDesktop from Components.ActionMap import ActionMap,HelpableActionMap from Screens.InfoBarGenerics import InfoBarShowHide from Components.AVSwitch import AVSwitch from Components.Label import Label from Components.MenuList import MenuList from Components.Pixmap import Pixmap from Components.Sources.List import List from Components.Sources.StaticText import StaticText from Components.Button import Button from Plugins.Plugin import PluginDescriptor from Screens.InfoBar import MoviePlayer from Screens.MessageBox import MessageBox from Tools import Notifications from Screens.LocationBox import LocationBox #from Screens.InputBox import InputBox from Screens.ChoiceBox import ChoiceBox from Screens.VirtualKeyBoard import VirtualKeyBoard from Components.Input import Input from Screens.Screen import Screen from Tools.BoundFunction import boundFunction from Tools.Directories import resolveFilename, SCOPE_PLUGINS from Tools.LoadPixmap import LoadPixmap from twisted.web.client import downloadPage,defer,reactor from Components.Task import job_manager from Components.config import config, ConfigSubsection, ConfigText, ConfigInteger, ConfigLocations, ConfigDirectory, ConfigSet, ConfigYesNo, ConfigSelection, getConfigListEntry, ConfigSelectionNumber import ContentSources import util from sources.SourceBase import stream0 from VideoDownload import downloadJob, HLSDownloadJob,VideoDownloadList #import enigma2_api TMPDIR = "/tmp/playstream/" e2 = None config.playstream = ConfigSubsection() config.playstream.locations = ConfigLocations(default=["/media/hdd/movie/"]) config.playstream.download_dir = ConfigText(default="/media/hdd/movie/") def make_service(stream): url = stream["url"] headers = [] if stream.has_key("headers"): for h in stream["headers"]: headers.append("%s=%s"%(h,stream["headers"][h])) if headers: headers ="&".join(headers) url = url+"#"+headers print "stream_url with headers=",headers url = url.encode("utf8") if "|" in url: url = url.replace("|","#") #url = url.replace(":","%3a") print "stream_url=",url service = eServiceReference(4097, 0, url) service.setName( stream["name"]) return service ##################################################################################################################### class PSSubs(Screen): def __init__(self, session): desktopWidth = getDesktop(0).size().width() desktopHeight = getDesktop(0).size().height() offset = 20 screenWidth = desktopWidth - (2 * offset) widgetWidth = screenWidth / 2 - 5 self.skin = """ """ % (offset, desktopHeight-offset-140, screenWidth, screenWidth) self['subtitle'] = Label() Screen.__init__(self, session) ##################################################################################################################### class PSPlayer(MoviePlayer): def __init__(self, session, streams): print "PSPlayer init: ",streams self.session = session self.streams = streams self.cur_stream = self.streams[0] # TODO self.selected = 0 self.resume_pos = 0 service = make_service(self.cur_stream) MoviePlayer.__init__(self, session, service) self.skinName = "MoviePlayer" self["actions"] = ActionMap(["MediaPlayerActions","MediaPlayerSeekActions","MoviePlayerActions","InfobarSeekActions","MovieSelectionActions","ColorActions"], { "stop":self.leavePlayer, "leavePlayer":self.leavePlayer, "audio":self.select_stream, "AudioSelection":self.select_stream, "green":self.select_stream, "subtitles":self.select_captions, "text":self.select_captions, "yellow_key":self.select_captions, "yellow":self.select_captions, "pauseServiceYellow":self.select_captions, "pause":self.select_captions, "showEventInfo":self.service_info, "info":self.service_info }) self.stimer = eTimer() if "callback" in dir(self.stimer): self.stimer.callback.append(self.update_subtitles) elif "timeout": self.stimer_conn = self.stimer.timeout.connect(self.update_subtitles) else: self.stimer = None self.stimer_step = 500 if self.cur_stream["subs"]: self.subs = self.cur_stream["subs"] self.cur_subs = 0 # TODO - no konfiguracijas self.svisible = True # TODO - no konfiguracija self.sind = 0 self.get_subs_current() else: self.subs = [] self.cur_subs = 0 self.svisible = False self.subtitle_window = self.session.instantiateDialog (PSSubs) self.onLayoutFinish.append(self.start_subtitles_timer) def start_subtitles_timer(self): if self.stimer: self.subtitle_window.show() print "start_subtitles_timer" self.stimer.start(self.stimer_step) def get_sub_pts(self,pts): sc = self.get_sub_ind(self.sind) # current subbtitle while True: if not sc: return "Error - no subs find" # TODO if pts > sc["end"]: self.sind += 1 sc = self.get_sub_ind(self.sind) continue else: if pts { "template": [MultiContentEntryText(pos=(10, 1), size=(560, 30), \ font=0, flags=RT_HALIGN_LEFT, text=0)], "fonts": [gFont("Regular", 20)], "itemHeight": 30 } """ def __init__(self, session): Screen.__init__(self, session) #self.setTitle2("Home") self.session = session self.e2 = None self["key_red"] = Button(_("Back")) self["key_green"] = Button(_("Select")) self["key_yellow"] = Button(_("Options")) self["key_menu"] = Button("Menu" ) self["actions"] = ActionMap(["OkCancelActions", "ColorActions","MenuActions", "NumberActions"], { "cancel": self.Cancel, "ok": self.Ok, "green": self.Ok, "red": self.Back, "yellow": self.options_screen, "blue": self.download_list, "menu": self.item_menu, }) self["list"] = List([]) self["list"].onSelectionChanged.append(self.SelectionChanged) self["pic"] = Pixmap() self["cur"] = Label() self["title"] = Label() self.downloading = 0 self.activeDownloads = 0 if not os.path.exists(TMPDIR): os.mkdir(TMPDIR) self.onLayoutFinish.append(self.LayoutFinish) def LayoutFinish(self): self.cur_directory = os.path.dirname(os.path.realpath(__file__)) self.defimage = LoadPixmap(os.path.join(self.cur_directory,"PlayStream.png")) #self.defimage = None #sc = AVSwitch().getFramebufferScale() #self.defimage0 = ePicLoad() #self.defimage0.PictureData.get().append(boundFunction(self.FinishDecodeDef, os.path.join(self.cur_directory,"PlayStream.png"))) #self.defimage0.setPara((self["pic"].instance.size().width(),self["pic"].instance.size().height(),sc[0], sc[1], False, 0, "#00000000")) #self.defimage0.startDecode("default") self.activeDownloads = 0 self.images = {} self.images_url = {} self.picloads = {} self.history = [] reload(ContentSources) self.sources = ContentSources.ContentSources(os.path.join(self.cur_directory,"sources")) self.config = self.sources.plugins["config"] self.cur_menu = ("Home","config::home","","Sākums") # self.content = self.sources.get_content(self.cur_menu[1]) #print self.content self["list"].setList(self.content) self["cur"].setText(self.content[0][3]) self.setTitle2(self.cur_menu[0]) self.ShowPic(self.content[0][2]) def SelectionChanged(self): current = self["list"].getCurrent() print "[PlayStream] SelectionChanged: current=",current if not current: return self["cur"].setText(current[3]) if current[3] else self["cur"].setText("") if current[2]: self.ShowPic(current[2]) else: self.ShowDefPic() def setTitle2(self,title): #print self.keys() self["title"].setText(title) def ShowDefPic(self): if self.defimage: self["pic"].instance.setPixmap(self.defimage) def ShowPic(self,image_url): if image_url == "default": self.ShowDefPic() return elif self.images.has_key(image_url): if self.images[image_url] in (-1,-2): return elif self.images[image_url] == -3: self.ShowDefPic() return else: self["pic"].instance.setPixmap(self.images[image_url]) else: if image_url.startswith("http"): fname = image_url.replace(":","-").replace("/","_") image_path = os.path.join(TMPDIR, fname) self.download_image(image_path, image_url) else: # local file image_path = os.path.join(self.cur_directory,image_url) self.start_decode(image_path,image_url) def start_decode(self,image_path,image_url): self.images[image_url] = -2 self.images_url[image_path] = image_url sc = AVSwitch().getFramebufferScale() if not self.picloads.has_key(image_path): self.picloads[image_path] = ePicLoad() self.picloads[image_path].PictureData.get().append(boundFunction(self.FinishDecode, image_path)) self.picloads[image_path].setPara((self["pic"].instance.size().width(), self["pic"].instance.size().height(), sc[0], sc[1], True, 0, "#00000000")) #print image_path,image_url self.picloads[image_path].startDecode(image_path) def FinishDecode(self, image_path,picInfo = None): image_url = self.images_url[image_path] del self.images_url[image_path] #III self.images[image_url] = self.picloads[image_path].getData() self["pic"].instance.setPixmap(self.images[image_url]) del self.picloads[image_path] if len(self.images)>30: del self.images[self.images.keys()[0]] # self.images.pop() #def FinishDecodeDef(self, image_path,picInfo = None): # self.defimage = self.defimage0.getData() # del self.defimage0 # self["pic"].instance.setPixmap(self.defimage) def download_image(self,image_path,image_url): #print "Image download started",self.downloading,image_path,image_url self.downloading += 1 self.images[image_url] = -1 downloadPage(image_url, image_path).addCallback(boundFunction(self.downloadFinished, image_path,image_url)).addErrback(boundFunction(self.downloadFailed, image_path,image_url)) def downloadFinished(self, image_path, image_url, result): self.downloading -= 1 #print "[ Play] Image downloaded finished ",self.downloading,image_path, image_url,result self.start_decode(image_path,image_url) def downloadFailed(self, image_path, image_url,result): self.downloading -= 1 print "[TV Play] Image downloaded failed ",self.downloading,image_path, image_url,result self.images[image_url] = -3 def Ok(self): current = self["list"].getCurrent() self.current = current index = self["list"].getIndex() self.index = index print "[PlayStream] - menu selected ", current data = current[1].split("::")[1] if "::" in current[1] else current[1] if not data: return elif self.sources.is_video(current[1]): if self.sources.stream_type(current[1]): stream = stream0.copy() stream["url"] = current[1] stream["name"] = current[0] streams = [stream] else: try: streams = self.sources.get_streams(current[1]) except Exception,e: print str(e) self.session. open(MessageBox, "Error - %s"%str(e) , MessageBox.TYPE_INFO) return if streams: #print streams for s in streams: if not s["name"]: s["name"] = current[0] if not s["img"]: s["img"] = current[2] if not s["desc"]: s["desc"] = current[3] try: self.session.open(PSPlayer, streams) except Exception as e: self.msg2("Error launching player - " + str(e)) return else: self.msg("No stream found - %s"%(self.current[1])) return elif current[1] == "back": cur_menu_old = self.cur_menu self.cur_menu = self.history.pop() new_content = self.sources.get_content(self.cur_menu[1]) try: index = zip(*new_content)[1].index(cur_menu_old[1]) except: index = 0 self.setTitle2(self.cur_menu[0]) self.show_content(new_content,index) else: print "selected=",current if "{0}" in current[1]: self.session.openWithCallback(self.cb_input,VirtualKeyBoard, title="Enter value", text="") #a = raw_input("Enter value:") #a = "big bang" #current = (current[0],current[1].format(a),current[2],current[3]) #self.get_content(current) else: self.get_content(current) def cb_input(self,value): if not value: return current = self.current current = (current[0],current[1].format(value),current[2],current[3]) self.get_content(current) def get_content(self,current): self.history.append(self.cur_menu) self.cur_menu = current try: new_content = self.sources.get_content(self.cur_menu[1]) except Exception,e: self.cur_menu = self.history.pop() self.session. open(MessageBox, "Error - %s"%str(e) , MessageBox.TYPE_INFO) return self.setTitle2(self.cur_menu[0]) self.show_content(new_content) def Back(self): self["list"].setIndex(0) self.Ok() def show_content(self,content,index=0): self["list"].setList(content) self["list"].setIndex(index) self.SelectionChanged() def Cancel(self): #if os.path.exists(TMPDIR): #for name in os.listdir(TMPDIR): #os.remove(os.path.join(TMPDIR, name)) #os.rmdir(TMPDIR) self.close() def item_menu(self): print "\n[PlayStream] options\n" self.current = self["list"].getCurrent() self.index = self["list"].getIndex() #self.session. open(MessageBox, "Item options - %s"%current[0] , MessageBox.TYPE_INFO) #args=[current,self.cur_menu] #self.session.open(ItemMenuScreen,current,index,self) lst = [ ("Aditional information","info","Display additional information about item"), ("Add to bouquet","bouquet","Add current item to Enigma2 bouquet"), ("Add to favorites","favorites","Add current item to PlayStrem favorites"), ("Show active downloads","download_list","Show active downloads list"), ("Set download folder","download_folder","Set download folder") ] if self.sources.is_video(self.current[1]): lst.extend([ ("Download video","download","Download video in background"), ]) else: lst.extend([ ("Download videos in folder","download","Download videos in folder (if any)"), ]) if "config::" in self.cur_menu[1]: lst.extend([ ("Rename item","rename","Rename list item"), ("Move item","move","Move list item"), ("Delete item","delete","Delete list item"), ("Add submenu","add_list","Add submenu before selected item"), ]) title = self.current[0] self.session.openWithCallback(self.cb_item_menu, ChoiceBox, title = title, list = lst) #TODO def cb_item_menu(self,answer): #print "item_menu_selected",answer if not answer: return if answer[1] == "info": self.session.open(MessageBox, "Not yet implemented!", MessageBox.TYPE_INFO) pass # TODO parada papildus info elif answer[1] == "bouquet": #if not e2: #e2 = enigma2_api.DBServices() #print "load_buuquets - ",e2._load_bouquets() self.session.open(MessageBox, "Not yet implemented!", MessageBox.TYPE_INFO) elif answer[1] == "favorites": lists = self.config.get_lists() lists2 = [(l,l) for l in lists] self.session.openWithCallback(self.cb_favorites, ChoiceBox, title="Selected menu item will be added",list = lists2) elif answer[1] == 'download': current = self.current if not self.sources.is_video(current[1]): #self.msg("Can not download listst (yet) - %s"%(current[1])) n = 0 for current2 in self.sources.get_content(current[1]): if self.sources.is_video(current2[1]): n += 1 self.download_video(current2) if n>0: self.msg("%s videos download started"%n) else: self.msg("No videos to download") else: if self.download_video(current): self.msg("Video download started") elif answer[1] == 'download_list': self.download_list() elif answer[1] == 'download_folder': #downloadDir = "/media/hdd/movie" #config.plugins.playstream.downloadDir.value TODO self.session.openWithCallback(self.select_download_dir, LocationBox,"Select download folder","",config.playstream.download_dir.value,config.playstream.locations,False,"Select folder",None,True,True) elif answer[1] == "delete": lst = self.cur_menu[1].replace("config::","") #print lst self.config.del_item(lst,self.index) self.config.write_streams() txt = "'%s' deleted from favourite stream list '%s'"%(self.current[0],lst) self.session.open(MessageBox, txt, MessageBox.TYPE_INFO,timeout=5) elif answer[1] == "rename": #name2 = "Renamed" self.session.openWithCallback(self.cb_rename,VirtualKeyBoard, title="Enter new item name", text=self.current[0]) elif answer[1] == "add_list": self.session.open(MessageBox, "Not yet implemented!", MessageBox.TYPE_INFO,timeout=5) pass #TODO return def select_download_dir(self, downloadDir, select=None): if not downloadDir: return print "Folder selected - %s"%downloadDir config.playstream.download_dir.setValue(downloadDir) config.playstream.download_dir.save() config.playstream.locations.save() config.save() def download_list(self): self.session.open(VideoDownloadList) def download_video(self,current): if self.sources.stream_type(current[1]): stream = stream0.copy() stream["url"] = current[1] stream["name"] = current[0] streams = [stream] else: try: streams = self.sources.get_streams(current[1]) except Exception,e: print "Error - %s"%str(e) self.session. open(MessageBox, "Error - %s"%str(e) , MessageBox.TYPE_INFO) return if not streams: self.msg("No stream found to download - %s"%(self.current[1])) return for s in streams: if not s["name"]: s["name"] = current[0] if not s["img"]: s["img"] = current[2] if not s["desc"]: s["desc"] = current[3] if len(streams)>1: stream = streams[0] # TODO iespeja izvelēties strīmu, ja to ir vairāki else: stream = streams[0] return self.download_stream(stream) def download_stream(self,stream): print "download stream",stream #self.msg("Start downloading..") self.stream = stream stream_type = self.stream["type"] #self.sources.stream_type(stream["url"]) if not stream_type: # print "Not supported stream type found to download - %s"%(self.current[1]) self.msg("Not supported stream type found to download - %s"%(self.current[1])) return title = self.stream["name"].strip() url = self.stream["url"] stream_type = self.stream["type"] #self.sources.stream_type(stream["url"]) downloadDir = config.playstream.download_dir.value if not os.path.exists(downloadDir): print 'Sorry, download directory "%s" not exist!\nPlease specify in the settings existing directory'%downloadDir self.msg0(_('Sorry, download directory "%s" not exist!\nPlease specify in the settings existing directory'%downloadDir)) return fname0 = re.sub("[/\n\r\t,:]"," ",title) fname0 = re.sub("['""]","",fname0) fname = fname0 +".mp4" outputfile = os.path.join(downloadDir, fname) #print "Trying to download - ", current if self.stream["subs"]: suburl = self.stream["subs"][0]["url"] print "\n**Download subtitles %s - %s"%(title,suburl) subs = urllib2.urlopen(suburl).read() if subs: #fname0 = re.sub("[/\n\r\t,:]"," ",title) subext = ".srt" subfile = os.path.join(downloadDir,fname0+subext) if ".xml" in suburl: subs = util.ttaf2srt(subs) with open(subfile,"w") as f: f.write(subs) else: print "\n Error downloading subtitle %s"%suburl if os.path.exists(outputfile): self.msg( _('Sorry, this file already exists:\n%s') % outputfile) return False #os.remove(outputfile) if stream_type in ("http","https"): print "\n**Download %s - %s"%(title,url) #reload(downloadJob) job_manager.AddJob(downloadJob(url, outputfile, title[:20], self.video_download_stop)) self.activeDownloads += 1 #self.msg(_('Video download started!')) return True elif stream_type == "hls": #self.msg("HLS stream download not yet implemented!") print "\n**Download %s - %s"%(title,url) #reload(HLSDownloadJob) print "HLSDownload", url,outputfile job_manager.AddJob(HLSDownloadJob(url, outputfile, title[:20], self.video_download_stop)) self.activeDownloads += 1 #self.msg(_('Video download started!')) return True elif stream_type == "rstp": self.msg("RSTP stream download not yet implemented!") return False else: self.msg("Unkown stream type!") return False def cb_rename(self,value): if not value: return lst = self.cur_menu[1].replace("config::","") pos = self.index print value item2 = list(self.current) item2[0]=value item2 = tuple(item2) self.config.replace_item(lst,item2,pos) self.config.write_streams() txt = "'%s' renamed to '%s'"%(self.current[0],value) self.session.open(MessageBox, txt, MessageBox.TYPE_INFO,timeout=5) def cb_favorites(self,answer): print "cb_favorites",answer,self.current if not answer: return value = answer[1] self.config.add_item(value,self.current) self.config.write_streams() txt = "'%s' added to favourite stream list '%s'"%(self.current[0],value) self.session.open(MessageBox, txt, MessageBox.TYPE_INFO,timeout=3) #self.session.openWithCallback(self.callMyMsg, MessageBox, _("Do you want to exit the plugin?"), MessageBox.TYPE_INFO) def options_screen(self): source = self.cur_menu[1].split("::")[0] options = self.sources.options_read(source) print source if not options: self.session. open(MessageBox, "No options available for source %s (%s)"%(self.cur_menu[0],source) , MessageBox.TYPE_INFO) else: self.session.open(OptionsScreen,self) def video_download_stop(self,title): #self.activeDownloads -= 1 #self.msg("Download '%s'finished!"%title) print "video_download_stop ", title def msg2(self,msg,timeout=10,mtype = None): mtype=mtype if mtype else MessageBox.TYPE_INFO Notifications.AddPopup(text = msg, type=mtype, timeout=timeout) def msg(self,msg,timeout=10): self.session.open(MessageBox, msg, MessageBox.TYPE_INFO, timeout) ########################################################################## from Components.config import config, ConfigSubsection, ConfigYesNo,\ getConfigListEntry, ConfigSelection, ConfigNumber, ConfigDirectory,ConfigText, ConfigSubDict from Components.ConfigList import ConfigListScreen #from Screens.LocationBox import LocationBox config.plugins.playstream = ConfigSubDict() class OptionsScreen(ConfigListScreen,Screen): skin = """ """ def __init__(self, session,*args): self.session = session Screen.__init__(self, session) self.main = args[0] self.setTitle(self.main.cur_menu[0]+" options") self.source = self.main.cur_menu[1].split("::")[0] self.cfg = config.plugins.playstream self.list = [] self.options = self.main.sources.options_read(self.source) if not self.options: #self.session. open(MessageBox, "No options available for source %s (%s)"%(self.main.cur_menu[0],self.source) , MessageBox.TYPE_INFO) self.close(False,self.session) for k in self.options: self.cfg[k]=ConfigText(default=self.options[k],fixed_size=False) self.list.append(getConfigListEntry(k, self.cfg[k])) ConfigListScreen.__init__(self, self.list, session = self.session) self["key_red"] = Button(_("Cancel")) self["key_green"] = Button("Save") self["key_yellow"] = Button("") self["key_blue"] = Button("") self["setupActions"] = ActionMap(["SetupActions", "ColorActions"], { "red": self.cancel, "green": self.save, "save": self.save, "cancel": self.cancel, "ok": self.ok, }, -2) def getCurrentEntry(self): return self["config"].getCurrent()[0] def getCurrentValue(self): return str(self["config"].getCurrent()[1].getText()) def ok(self): self.save() #if self["config"].getCurrent()[1] == config.plugins.getpicons.folder: #folder = config.plugins.getpicons.folder.value #self.session.openWithCallback(self.change_dir, LocationBox,"Select Folder") #else: #def change_dir(self, folder, select=None): #if folder: ##print "change_dir to %s"%folder #config.plugins.getpicons.folder.value = folder def save(self): print "saving" #self.saveAll() for k in self.options.keys(): self.options[k]=self.cfg[k].value print "%s=%s"%(k,self.cfg[k].value) self.main.sources.options_write(self.source,self.options) self.close(True,self.session) def cancel(self): print "cancel" self.close(False,self.session)