-
Notifications
You must be signed in to change notification settings - Fork 11
/
default.py
111 lines (91 loc) · 3.71 KB
/
default.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
# -*- coding: UTF-8 -*-
#/*
# * Copyright (C) 2013 Maros Ondrasek
# *
# *
# * This Program is free software; you can redistribute it and/or modify
# * it under the terms of the GNU General Public License as published by
# * the Free Software Foundation; either version 2, or (at your option)
# * any later version.
# *
# * This Program is distributed in the hope that it will be useful,
# * but WITHOUT ANY WARRANTY; without even the implied warranty of
# * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# * GNU General Public License for more details.
# *
# * You should have received a copy of the GNU General Public License
# * along with this program; see the file COPYING. If not, write to
# * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
# * http://www.gnu.org/copyleft/gpl.html
# *
# */
import os
sys.path.append( os.path.join ( os.path.dirname(__file__),'resources','lib') )
import markiza
from epg_sms import get_sms_epg
import xbmcprovider,xbmcaddon
import util
from urllib.parse import urlencode, parse_qsl
import json
import socket
class IPTVManager:
"""Interface to IPTV Manager"""
def __init__(self, port, channels, epgdays=3, epgdaysback=1):
"""Initialize IPTV Manager object"""
self.port = port
self.channels = channels
self.epgdays = epgdays
self.epgdaysback = epgdaysback
def via_socket(func):
"""Send the output of the wrapped function to socket"""
def send(self):
"""Decorator to send over a socket"""
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect(('127.0.0.1', self.port))
try:
sock.sendall(json.dumps(func(self)).encode())
finally:
sock.close()
return send
@via_socket
def send_channels(self):
"""Return JSON-STREAMS formatted python datastructure to IPTV Manager"""
return dict(version=1, streams=self.channels)
@via_socket
def send_epg(self):
"""Return JSON-EPG formatted python data structure to IPTV Manager"""
epg_data=get_sms_epg(self.epgdays,self.epgdaysback)
return dict(version=1, epg=epg_data)
__scriptid__ = 'plugin.video.markiza.sk'
__scriptname__ = 'markiza.sk'
__addon__ = xbmcaddon.Addon(id=__scriptid__)
__language__ = __addon__.getLocalizedString
#import web_pdb; web_pdb.set_trace()
settings = {'downloads':__addon__.getSetting('downloads'),'quality':__addon__.getSetting('quality'),'epgdays':__addon__.getSetting('epgdays')}
provider = markiza.markizaContentProvider()
if len (sys.argv)>1 and 'port' in sys.argv[2][1:]:
channels=[]
if 'iptv/channels' in sys.argv[0]:
paramstring=sys.argv[2][1:]
params = dict(parse_qsl(paramstring))
port=int(params['port'])
result=provider.categories()
for item in result:
if item['type']!='video':
continue
item2=provider.resolve(item)
item['url']=item2[0]['url']
channels.append(dict(id=item['title'],name=item['title'],logo=item['img'],stream=item2[0]['url']))
IPTVManager(port,channels).send_channels()
if 'iptv/epg' in sys.argv[0]:
paramstring=sys.argv[2][1:]
params = dict(parse_qsl(paramstring))
port=int(params['port'])
epgdays=int(settings['epgdays'])
IPTVManager(port,channels,epgdays).send_epg()
else:
params = util.params()
class XBMCMarkizaContentProvider(xbmcprovider.XBMCMultiResolverContentProvider):
def render_default(self, item):
self.render_dir(item)
XBMCMarkizaContentProvider(provider,settings,__addon__).run(params)