# -*- coding: utf8 -*- import os, time import base64 import requests import aes as EN import StringIO import gzip import xmltv import urllib import simplejson as json def Mbox(title, text, style): return ctypes.windll.user32.MessageBoxW(0, text, title, style) class dodat(): def __init__(self, base, login, path, cachetime=1, dbg=True, dump_name='', timeout=0.5, ver='0.0.0', xxx=False, os_id='pcweb', agent_id='pcweb', app_ver='0.01', force_group_name=False, gen_m3u=True, gen_epg=False, compress=True, map_url=None, ext_logo=True, logo_color=False, proc_cb=None): self.__UA = { 'Host': 'api.iptv.bulsat.com', 'User-Agent': 'Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:36.0) Gecko/20100101 Firefox/36.0', 'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8', 'Accept-Language': 'en-US,en;q=0.5', 'Accept-Encoding': 'gzip, deflate', 'Referer': 'https://test.iptv.bulsat.com/iptv-login.php', 'Origin': 'https://test.iptv.bulsat.com', 'Connection': 'keep-alive', 'Pragma': 'no-cache', } self.__log_in = {} self.__p_data = { 'user': [None, ''], 'device_id': [None, os_id], 'device_name': [None, os_id], 'os_version': [None, os_id], 'os_type': [None, os_id], 'app_version': [None, app_ver], 'pass': [None, ''], } self.__path = path self.__refresh = int(cachetime * 60 * 60) self.__p_data['user'][1] = login['usr'] self.__log_in['pw'] = login['pass'] self.__DEBUG_EN = dbg self.__t = timeout self.__BLOCK_SIZE = 16 self.__URL_LIST = base + '/tv/samsungtv/live' self.__URL_EPG = base + '/epg/short' self.__js = None self.__app_version = ver self.__x = xxx self.__en_group_ch = force_group_name self.__gen_m3u = gen_m3u self.__gen_epg = gen_epg self.__compress = compress self.__cb = proc_cb self.__MAP_URL = map_url self.__gen_jd = False self.__ext_logo = ext_logo self.__logo_color = logo_color self.__s = requests.Session() if agent_id != 'pcweb': self.__URL_LOGIN = base + '/?auth' else: self.__URL_LOGIN = base + '/auth' if agent_id != 'pcweb': self.__UA['User-Agent'] = agent_id if not os.path.exists(self.__path): os.makedirs(self.__path) def __log_dat(self, d): if self.__DEBUG_EN is not True: return print '--------- BEGIN ---------' if type(d) is str: print d elif type(d) is dict or type(d).__name__ == 'CaseInsensitiveDict': for k, v in d.iteritems(): print k + ' : ' + str(v) elif type(d) is list: for l in d: print l else: print 'Todo add type %s' % type(d) print '--------- END -----------' def __store_data(self): with open(os.path.join(self.__path, '', 'data.dat'), 'wb+') as f: f.write(unicode(json.dumps(self.__js, sort_keys=True, indent=1, ensure_ascii=False))) if self.__DEBUG_EN is True: import io with io.open(os.path.join(self.__path, '', 'src.dump'), 'w+', encoding=self.__char_set) as f: f.write(unicode(json.dumps(self.__tv_list, sort_keys=True, indent=1, ensure_ascii=False))) def __restore_data(self): with open(os.path.join(self.__path, '', 'data.dat'), 'r') as f: self.__js = json.load(f) def __log_out(self): r = self.__s.post(self.__URL_LOGIN, timeout=self.__t, headers=self.__UA, files={'logout': [None, '1']}) self.__log_dat(r.request.headers) self.__log_dat(r.request.body) if r.status_code == requests.codes.ok and r.json()['Logged'] == 'false': self.__log_dat('Logout ok') if self.__cb: self.__cb({'pr': 100, 'str': 'Logout ok'}) else: self.__log_dat('Logout Fail') raise Exception("LogoutFail") def __goforit(self): if self.__cb: self.__cb({'pr': 10, 'str': 'Session'}) r = self.__s.post(self.__URL_LOGIN, timeout=self.__t, headers=self.__UA) if r.status_code == requests.codes.ok: if self.__cb: self.__cb({'pr': 20, 'str': 'Session start'}) self.__log_in['key'] = r.headers['challenge'] self.__log_in['session'] = r.headers['ssbulsatapi'] self.__s.headers.update({'SSBULSATAPI': self.__log_in['session']}) _text = self.__log_in['pw'] + (self.__BLOCK_SIZE - len(self.__log_in['pw']) % self.__BLOCK_SIZE) * '\0' enc = EN.AESModeOfOperationECB(self.__log_in['key']) agent_id = self.__UA['User-Agent'] if agent_id == 'Bulsatcom for android': self.__p_data['pass'][1] = 'E0HqAKzOTeXMmStDe5jsiRk1vmfhtlRrNikCdYl0IVOKl6mTiEpj0WnEjoGzCJ7u' else: self.__p_data['pass'][1] = base64.b64encode(enc.encrypt(_text)) self.__log_dat(self.__log_in) self.__log_dat(self.__p_data) if self.__cb: self.__cb({'pr': 30, 'str': 'Login start'}) r = self.__s.post(self.__URL_LOGIN, timeout=self.__t, headers=self.__UA, files=self.__p_data) self.__log_dat(r.request.headers) self.__log_dat(r.request.body) if r.status_code == requests.codes.ok: data = r.json() if data['Logged'] == 'true': self.__log_dat('Login ok') if self.__cb: self.__cb({'pr': 50, 'str': 'Login ok'}) self.__s.headers.update({'Access-Control-Request-Method': 'POST'}) self.__s.headers.update({'Access-Control-Request-Headers': 'ssbulsatapi'}) r = self.__s.options(self.__URL_LIST, timeout=self.__t, headers=self.__UA) self.__log_dat(r.request.headers) self.__log_dat(r.headers) self.__log_dat(str(r.status_code)) if self.__cb: self.__cb({'pr': 70, 'str': 'Fetch data'}) r = self.__s.post(self.__URL_LIST, timeout=self.__t, headers=self.__UA) self.__log_dat(r.request.headers) self.__log_dat(r.headers) if r.status_code == requests.codes.ok: self.__char_set = r.headers['content-type'].split('charset=')[1] self.__log_dat('get data ok') self.__tv_list = r.json() self.__js = {} self.__log_dat(self.__js) if self.__cb: self.__cb({'pr': 90, 'str': 'Fetch data done'}) if self.__gen_epg: for i, ch in enumerate(self.__tv_list): if self.__cb: self.__cb( { 'pr': int((i * 100) / len(self.__tv_list)), 'str': 'Fetch: %s' % ch['epg_name'].encode('utf-8'), 'idx': i, 'max': len(self.__tv_list) } ) if ch.has_key('program'): r = self.__s.post(self.__URL_EPG, timeout=self.__t, headers=self.__UA, data={ # 'epg': 'nownext', 'epg': '1week', # 'epg': '1day', 'channel': ch['epg_name'] } ) if r.status_code == requests.codes.ok: ch['program'] = r.json().items()[0][1]['programme'] from HTMLParser import HTMLParser as h self.__js = json.loads(h().unescape(json.dumps(self.__js).decode(self.__char_set))) # Disable LogOut for now since the new changes in Fusion platform requires user to be logged in so to can open RTMP streams # self.__log_out() # if r.status_code != requests.codes.ok: # self.__log_dat('Error status code: %d' % (r.status_code, )) # raise Exception("FetchFail") else: raise Exception("LoginFail") def __data_fetch(self, f): self.__tv_list = None if os.path.exists(os.path.join(self.__path, '', 'data.dat')) and f is False: self.__restore_data() if time.time() - self.__js['ts'] < self.__refresh and self.__js['os_type'] == self.__p_data['os_type'][1]: self.__log_dat('Use cache file') else: self.__log_dat('Use site') self.__js = None if self.__js is None: self.__goforit() self.__log_dat('Len: %d' % len(self.__tv_list)) self.__js['ts'] = divmod(time.time(), self.__refresh)[0] * self.__refresh self.__js['app_version'] = self.__app_version self.__js['os_type'] = self.__p_data['os_type'][1] self.__log_dat('Base time: %s' % time.ctime(self.__js['ts'])) self.__store_data() def gen_all(self, force_refresh=False): ret = False self.__data_fetch(force_refresh) if self.__tv_list: ret = True map = None if self.__gen_epg: w = xmltv.Writer(encoding=self.__char_set.upper(), date=str(time.time()), source_info_url="", source_info_name="", generator_info_name="", generator_info_url="") else: if self.__MAP_URL: try: m = requests.get(self.__MAP_URL, timeout=self.__t, headers={'User-Agent': 'fusion_tv'}) self.__log_dat(m.request.headers) self.__log_dat(m.headers) self.__log_dat(m.status_code) if m.status_code == requests.codes.ok: map = m.json() self.__log_dat(map) except: pass pl = u'#EXTM3U\n' if self.__gen_jd: jdump = {} dat = [x for x in self.__tv_list] for i, ch in enumerate(dat): if self.__cb: self.__cb( { 'pr': int((i * 100) / len(dat)), 'str': 'Sync: %s' % ch['epg_name'].encode('utf-8'), 'idx': i, 'max': len(dat) } ) if self.__en_group_ch: ch_group_name = self.__en_group_ch else: ch_group_name = ch['genre'] if self.__gen_m3u: if not map: if self.__ext_logo: if self.__logo_color: pl = pl + '#EXTINF:-1 radio="%s" group-title="%s" tvg-logo="http://sintez66.com/TVLogos/BulsatColor/%s.png" tvg-id="%s",%s\n%s\n' % ( ch['radio'], ch_group_name, ch['epg_name'], ch['epg_name'], ch['title'], ch['sources']) else: pl = pl + '#EXTINF:-1 radio="%s" group-title="%s" tvg-logo="http://sintez66.com/TVLogos/BulsatWhite/%s.png" tvg-id="%s",%s\n%s\n' % ( ch['radio'], ch_group_name, ch['epg_name'], ch['epg_name'], ch['title'], ch['sources']) else: pl = pl + '#EXTINF:-1 radio="%s" group-title="%s" tvg-logo="%s" tvg-id="%s",%s\n%s\n%s\n' % ( ch['radio'], ch_group_name, ch['epg_name'], ch['epg_name'], ch['title'], ch['sources']) else: e_map = map.get(ch['epg_name'], { ch['epg_name']: {'id': ch['epg_name'], 'offset': '0', 'ch_logo': ch['epg_name']}}) gid = e_map.get('id', ch['epg_name']) offset = e_map.get('offset', '0') logo = e_map.get('ch_logo', ch['epg_name']) if self.__ext_logo: if self.__logo_color: pl = pl + '#EXTINF:-1 radio="%s" group-title="%s" tvg-logo="http://sintez66.com/TVLogos/BulsatColor/%s.png" tvg-id="%s",%s\n%s\n' % ( ch['radio'], ch_group_name, ch['epg_name'], ch['epg_name'], ch['title'], ch['sources']) else: pl = pl + '#EXTINF:-1 radio="%s" group-title="%s" tvg-logo="http://sintez66.com/TVLogos/BulsatWhite/%s.png" tvg-id="%s",%s\n%s\n' % ( ch['radio'], ch_group_name, ch['epg_name'], ch['epg_name'], ch['title'], ch['sources']) else: pl = pl + '#EXTINF:-1 radio="%s" group-title="%s" tvg-logo="%s" tvg-id="%s",%s\n%s\n%s\n' % ( ch['radio'], ch_group_name, ch['epg_name'], ch['epg_name'], ch['title'], ch['sources']) if self.__gen_jd: jdump[ch['epg_name']] = ch['epg_name'] if self.__gen_epg: w.addChannel( {'display-name': [(ch['title'], u'bg')], 'id': ch['epg_name'], 'url': ['https://test.iptv.bulsat.com']} ) if ch.has_key('program'): for p in ch['program']: w.addProgramme( {'start': p['start'], 'stop': p['stop'], 'title': [(p['title'], u'')], 'desc': [(p['desc'], u'')], 'category': [(ch['genre'], u'')], 'channel': ch['epg_name']} ) agent_id = self.__UA['User-Agent'] if agent_id == 'Bulsatcom for android': if self.__gen_m3u: f_m3u = open(os.path.join(self.__path, '', 'bulsat_android.m3u'), 'wb+') f_m3u.write(pl.encode(self.__char_set, 'replace')) f_m3u.close() if self.__gen_jd: with open(os.path.join(self.__path, '', 'map.json'), 'wb+') as f: f.write(unicode(json.dumps(jdump, sort_keys=True, indent=1, ensure_ascii=False))) if self.__gen_epg: if self.__compress: out = StringIO.StringIO() w.write(out, pretty_print=True) f_lmx = gzip.open(os.path.join(self.__path, '', 'bulsat_android.xml.gz'), 'w+', 9) f_lmx.write(out.getvalue()) f_lmx.close() out.close() else: out = open(os.path.join(self.__path, '', 'bulsat_adnroid.xml'), 'wb+') w.write(out, pretty_print=True) out.close() else: if self.__gen_m3u: f_m3u = open(os.path.join(self.__path, '', 'bsc.m3u'), 'wb+') f_m3u.write(pl.encode(self.__char_set, 'replace')) f_m3u.close() if self.__gen_jd: with open(os.path.join(self.__path, '', 'map.json'), 'wb+') as f: f.write(unicode(json.dumps(jdump, sort_keys=True, indent=1, ensure_ascii=False))) if self.__gen_epg: if self.__compress: out = StringIO.StringIO() w.write(out, pretty_print=True) f_lmx = gzip.open(os.path.join(self.__path, '', 'bulsat_samsung.xml.gz'), 'w+', 9) f_lmx.write(out.getvalue()) f_lmx.close() out.close() else: out = open(os.path.join(self.__path, '', 'bulsat_samsung.xml'), 'wb+') w.write(out, pretty_print=True) out.close() return ret