/addons/plugin.video.tvisio.tv/default.py

http://seppius-xbmc-repo.googlecode.com/ · Python · 340 lines · 224 code · 65 blank · 51 comment · 41 complexity · 0facd6fcfbbe5bb02e87c5236e31b4ae MD5 · raw file 

    

  1. #!/usr/bin/python
    2. 

  2. # -*- coding: utf-8 -*-
    3. 

  3. #/*
    4. 

  4. # *      Copyright (C) 2011 Silen
    5. 

  5. # *
    6. 

  6. # *
    7. 

  7. # *  This Program is free software; you can redistribute it and/or modify
    8. 

  8. # *  it under the terms of the GNU General Public License as published by
    9. 

  9. # *  the Free Software Foundation; either version 2, or (at your option)
    10. 

  10. # *  any later version.
    11. 

  11. # *
    12. 

  12. # *  This Program is distributed in the hope that it will be useful,
    13. 

  13. # *  but WITHOUT ANY WARRANTY; without even the implied warranty of
    14. 

  14. # *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    15. 

  15. # *  GNU General Public License for more details.
    16. 

  16. # *
    17. 

  17. # *  You should have received a copy of the GNU General Public License
    18. 

  18. # *  along with this program; see the file COPYING.  If not, write to
    19. 

  19. # *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
    20. 

  20. # *  http://www.gnu.org/copyleft/gpl.html
    21. 

  21. # */
    22. 

  22. import re, os, urllib, urllib2, cookielib, time, sys, urlparse
    23. 

  23. from time import gmtime, strftime
    24. 

  24. 

  25. try:
    26. 

  26.     import json
    27. 

  27. except ImportError:
    28. 

  28.     try:
    29. 

  29.         import simplejson as json
    30. 

  30.         #xbmc.log( '[%s]: Error import json. Uses module simplejson' % addon_id, 2 )
    31. 

  31.     except ImportError:
    32. 

  32.         try:
    33. 

  33.             import demjson3 as json
    34. 

  34.             #xbmc.log( '[%s]: Error import simplejson. Uses module demjson3' % addon_id, 3 )
    35. 

  35.         except ImportError:
    36. 

  36.             xbmc.log( '[%s]: Error import demjson3. Sorry.' % addon_id, 4 )
    37. 

  37. 

  38. import xbmc, xbmcgui, xbmcplugin, xbmcaddon
    39. 

  39. 

  40. Addon = xbmcaddon.Addon(id='plugin.video.tvisio.tv')
    41. 

  41. icon = xbmc.translatePath(os.path.join(Addon.getAddonInfo('path'),'icon.png'))
    42. 

  42. fcookies = xbmc.translatePath(os.path.join(Addon.getAddonInfo('path'), r'cookies.txt'))
    43. 

  43. 

  44. # load XML library
    45. 

  45. try:
    46. 

  46.     sys.path.append(os.path.join(Addon.getAddonInfo('path'), r'resources', r'lib'))
    47. 

  47.     from BeautifulSoup  import BeautifulSoup
    48. 

  48. except:
    49. 

  49.     try:
    50. 

  50.         sys.path.insert(0, os.path.join(Addon.getAddonInfo('path'), r'resources', r'lib'))
    51. 

  51.         from BeautifulSoup  import BeautifulSoup
    52. 

  52.     except:
    53. 

  53.         sys.path.append(os.path.join(os.getcwd(), r'resources', r'lib'))
    54. 

  54.         from BeautifulSoup  import BeautifulSoup
    55. 

  55.         icon = xbmc.translatePath(os.path.join(os.getcwd().replace(';', ''),'icon.png'))
    56. 

  56. 

  57. import HTMLParser
    58. 

  58. hpar = HTMLParser.HTMLParser()
    59. 

  59. 

  60. h = int(sys.argv[1])
    61. 

  61. 

  62. def showMessage(heading, message, times = 3000):
    63. 

  63.     xbmc.executebuiltin('XBMC.Notification("%s", "%s", %s, "%s")'%(heading, message, times, icon))
    64. 

  64. 

  65. #---------- get web page -------------------------------------------------------
    66. 

  66. def get_HTML(url, post = None, ref = None):
    67. 

  67.     request = urllib2.Request(url, post)
    68. 

  68. 

  69.     host = urlparse.urlsplit(url).hostname
    70. 

  70.     if ref==None:
    71. 

  71.         ref='http://'+host
    72. 

  72. 

  73.     request.add_header('User-Agent', 'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0; Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1) ; .NET CLR 1.1.4322; .NET CLR 2.0.50727; .NET CLR 3.0.4506.2152; .NET CLR 3.5.30729; .NET4.0C)')
    74. 

  74.     request.add_header('Host',   host)
    75. 

  75.     request.add_header('Accept', '*/*')
    76. 

  76.     request.add_header('Accept-Language', 'ru-RU')
    77. 

  77.     request.add_header('Referer',             ref)
    78. 

  78. 

  79.     try:
    80. 

  80.         f = urllib2.urlopen(request)
    81. 

  81.     except IOError, e:
    82. 

  82.         if hasattr(e, 'reason'):
    83. 

  83.            xbmc.log('We failed to reach a server.')
    84. 

  84.         elif hasattr(e, 'code'):
    85. 

  85.            xbmc.log('The server couldn\'t fulfill the request.')
    86. 

  86. 

  87.     html = f.read()
    88. 

  88. 

  89.     return html
    90. 

  90. 

  91. #---------- get Moscow Time ----------------------------------------------------
    92. 

  92. def MSK_time():
    93. 

  93.     try:
    94. 

  94.         #-- get MSK time page from Time&Date server
    95. 

  95.         url = 'http://www.timeanddate.com/worldclock/city.html?n=166'
    96. 

  96.         html = get_HTML(url)
    97. 

  97. 

  98.         MSK_TOff =  re.compile('<td>Current time zone offset:<\/td><td><strong>UTC\/GMT \+(.+?) hours<\/strong>', re.MULTILINE|re.DOTALL).findall(html)
    99. 

  99.         TOff = int(MSK_TOff[0])
    100. 

  100.     except:
    101. 

  101.         TOff = 3
    102. 

  102. 

  103.     #--- return MSK time
    104. 

  104.     return gmtime(time.time()+TOff*60*60)
    105. 

  105. 

  106. 

  107. #---------- get list of TV channels --------------------------------------------
    108. 

  108. def Get_TV_Channels():
    109. 

  109.     url = 'http://tvisio.tv'
    110. 

  110.     html = get_HTML(url)
    111. 

  111. 

  112.     try:
    113. 

  113.         #-- get authenticity token
    114. 

  114.         token = re.compile('<input name="authenticity_token" type="hidden" value="(.+?)" \/>', re.MULTILINE|re.DOTALL).findall(html)[0]
    115. 

  115.         #-- login to tvisio.tv
    116. 

  116.         login       = Addon.getSetting('Login')
    117. 

  117.         password    = Addon.getSetting('Password')
    118. 

  118. 

  119.         values = {
    120. 

  120.               'user[email]' : login,
    121. 

  121.               'user[password]' : password,
    122. 

  122.               'user[remember_me]' : 1,
    123. 

  123.               'authenticity_token' : token,
    124. 

  124.               'commit' : '?????',
    125. 

  125.               'utf8' : '?'
    126. 

  126.             }
    127. 

  127. 

  128.         post = urllib.urlencode(values)
    129. 

  129. 

  130.         url = 'http://tvisio.tv/users/login'
    131. 

  131.         html = get_HTML(url, post)
    132. 

  132.     except:
    133. 

  133.         pass
    134. 

  134. 

  135.     # -- parsing web page ------------------------------------------------------
    136. 

  136.     html = re.compile('<body>(.+?)<\/body>', re.MULTILINE|re.DOTALL).findall(html)[0]
    137. 

  137.     soup = BeautifulSoup(html)
    138. 

  138. 

  139.     nav = soup.find('div', { 'id':"channels_list"})
    140. 

  140.     for ch in nav.findAll("a"):
    141. 

  141.         name    = unescape(ch.find('img')['alt']).encode('utf-8')
    142. 

  142.         img     = 'http://tvisio.tv'+ch.find('img')['src']
    143. 

  143.         ch_url  = 'http://tvisio.tv'+ch['href']
    144. 

  144.         i = xbmcgui.ListItem(name, iconImage=img, thumbnailImage=img)
    145. 

  145.         u = sys.argv[0] + '?mode=EPG_DATE'
    146. 

  146.         u += '&name=%s'%urllib.quote_plus(name)
    147. 

  147.         u += '&url=%s'%urllib.quote_plus(ch_url)
    148. 

  148.         u += '&img=%s'%urllib.quote_plus(img)
    149. 

  149.         xbmcplugin.addDirectoryItem(h, u, i, True)
    150. 

  150. 

  151.     xbmcplugin.endOfDirectory(h)
    152. 

  152. #-------------------------------------------------------------------------------
    153. 

  153. 

  154. def Get_EPG_Date(params):
    155. 

  155.     # -- parameters
    156. 

  156.     url  = urllib.unquote_plus(params['url'])
    157. 

  157.     img = urllib.unquote_plus(params['img'])
    158. 

  158. 

  159.     #-- get online TV link
    160. 

  160.     print url
    161. 

  161.     html = get_HTML(url)
    162. 

  162.     html = re.compile('<body>(.+?)<\/body>', re.MULTILINE|re.DOTALL).findall(html)[0]
    163. 

  163.     soup = BeautifulSoup(html)
    164. 

  164. 

  165.     print html
    166. 

  166. 

  167.     for online in soup.find('div',{'class':'head'}).findAll("a"):
    168. 

  168.         prg = 'http://tvisio.tv'+online['href']
    169. 

  169. 

  170.     name = '[COLOR FF3BB9FF]???????? ??????[/COLOR]'
    171. 

  171. 

  172.     prg_name = '<< ???????? ?????? >>'
    173. 

  173. 

  174.     i = xbmcgui.ListItem(name, iconImage=img, thumbnailImage=img)
    175. 

  175.     u = sys.argv[0] + '?mode=PLAY'
    176. 

  176.     u += '&name=%s'%urllib.quote_plus(name)
    177. 

  177.     u += '&url=%s'%urllib.quote_plus(prg)
    178. 

  178.     u += '&prg=%s'%urllib.quote_plus(prg_name)
    179. 

  179.     u += '&img=%s'%urllib.quote_plus(img)
    180. 

  180.     xbmcplugin.addDirectoryItem(h, u, i, False)
    181. 

  181. 

  182. 

  183.     #-- get MSK time
    184. 

  184.     MSK = time.mktime(MSK_time())
    185. 

  185.     #-- fill up ETR data list
    186. 

  186.     for day_off in range(0, 16):    #-- tvisio.tv keeps 2 weeks of TV data
    187. 

  187.         ETR_date = time.localtime(MSK-day_off*24*60*60)
    188. 

  188.         name = unescape(strftime("%a, %d %b %Y", ETR_date)).encode('utf-8')
    189. 

  189.         id   = url + '?date=' + strftime("%Y-%m-%d", ETR_date)
    190. 

  190.         i = xbmcgui.ListItem(name, iconImage=img, thumbnailImage=img)
    191. 

  191.         u = sys.argv[0] + '?mode=EPG'
    192. 

  192.         u += '&name=%s'%urllib.quote_plus(name)
    193. 

  193.         u += '&url=%s'%urllib.quote_plus(id)
    194. 

  194.         u += '&img=%s'%urllib.quote_plus(img)
    195. 

  195.         xbmcplugin.addDirectoryItem(h, u, i, True)
    196. 

  196. 

  197.     xbmcplugin.endOfDirectory(h)
    198. 

  198. 

  199. #---------- get EPG for selected channel ---------------------------------------
    200. 

  200. def Get_EPG(params):
    201. 

  201.     # -- parameters
    202. 

  202.     url = urllib.unquote_plus(params['url'])
    203. 

  203.     img = urllib.unquote_plus(params['img'])
    204. 

  204. 

  205.     html = get_HTML(url)
    206. 

  206. 

  207.     # -- parsing web page ------------------------------------------------------
    208. 

  208.     html = re.compile('<body>(.+?)<\/body>', re.MULTILINE|re.DOTALL).findall(html)[0]
    209. 

  209.     soup = BeautifulSoup(html)
    210. 

  210. 

  211.     nav = soup.findAll("div", { "class" : "broadcast" })
    212. 

  212.     for ch in nav:
    213. 

  213.         try:
    214. 

  214.             prg = 'http://tvisio.tv'+ch.find('a')['href']
    215. 

  215.         except:
    216. 

  216.             prg = '*'
    217. 

  217. 

  218.         name = '[COLOR FF3BB9FF]'+unescape(ch.find("span", { "class" : "time" }).text).encode('utf-8')+'[/COLOR]'+' '
    219. 

  219.         if prg == '*':
    220. 

  220.             name += '[COLOR FFFF0000]'+unescape(ch.find("span", { "class" : "title" }).text).encode('utf-8')+'[/COLOR]'
    221. 

  221.         else:
    222. 

  222.             name += '[COLOR FFC3FDB8]'+unescape(ch.find("span", { "class" : "title" }).text).encode('utf-8')+'[/COLOR]'
    223. 

  223. 

  224.         prg_name = unescape(ch.find("span", { "class" : "title" }).text).encode('utf-8')
    225. 

  225. 

  226.         i = xbmcgui.ListItem(name, iconImage=img, thumbnailImage=img)
    227. 

  227.         u = sys.argv[0] + '?mode=PLAY'
    228. 

  228.         u += '&name=%s'%urllib.quote_plus(name)
    229. 

  229.         u += '&url=%s'%urllib.quote_plus(prg)
    230. 

  230.         u += '&prg=%s'%urllib.quote_plus(prg_name)
    231. 

  231.         u += '&img=%s'%urllib.quote_plus(img)
    232. 

  232.         xbmcplugin.addDirectoryItem(h, u, i, False)
    233. 

  233. 

  234.     xbmcplugin.endOfDirectory(h)
    235. 

  235. #-------------------------------------------------------------------------------
    236. 

  236. 

  237. 

  238. #-------------------------------------------------------------------------------
    239. 

  239. 

  240. def PLAY(params):
    241. 

  241.     # -- parameters
    242. 

  242.     url  = urllib.unquote_plus(params['url'])
    243. 

  243.     img  = urllib.unquote_plus(params['img'])
    244. 

  244.     name = urllib.unquote_plus(params['prg'])
    245. 

  245. 

  246.     if url == '*':
    247. 

  247.         return False
    248. 

  248. 

  249.     # -- check if video available
    250. 

  250.     html = get_HTML(url)
    251. 

  251. 

  252.     # -- parsing web page ----------------------------------------------------------
    253. 

  253.     var = re.compile('flashvars.(.+?) = "(.+?)";', re.MULTILINE|re.DOTALL).findall(html)
    254. 

  254.     for rec in var:
    255. 

  255.         if rec[0].find('stream') > -1:
    256. 

  256.             v_stream = rec[1]
    257. 

  257.         elif rec[0].find('start') > -1:
    258. 

  258.             v_start = rec[1]
    259. 

  259.         elif rec[0].find('server') > -1:
    260. 

  260.             v_server = rec[1]
    261. 

  261.         elif rec[0].find('session') > -1:
    262. 

  262.             v_session = rec[1]
    263. 

  263. 

  264.     swf = re.compile('swfobject.embedSWF\("(.+?)"', re.MULTILINE|re.DOTALL).findall(html)
    265. 

  265.     v_swf = swf[0]
    266. 

  266. 

  267.     # -- assemble RTMP link ----------------------------------------------------
    268. 

  268.     if name <> '<< ???????? ?????? >>':
    269. 

  269.         video = 'rtmp://%s/rtmp app=rtmp swfUrl=http://tvisio.tv%s pageUrl=%s playpath=%s?start=%s conn=S:%s' % (v_server, v_swf, url, v_stream, v_start, v_session)
    270. 

  270.     else:
    271. 

  271.         video = 'rtmp://%s/rtmp app=rtmp swfUrl=http://tvisio.tv%s pageUrl=%s playpath=%s conn=S:%s' % (v_server, v_swf, url, v_stream, v_session)
    272. 

  272. 

  273.     i = xbmcgui.ListItem(name, path = urllib.unquote(video), thumbnailImage=img)
    274. 

  274.     xbmc.Player().play(video, i)
    275. 

  275. 

  276. #-------------------------------------------------------------------------------
    277. 

  277. 

  278. def unescape(text):
    279. 

  279.     try:
    280. 

  280.         text = hpar.unescape(text)
    281. 

  281.     except:
    282. 

  282.         text = hpar.unescape(text.decode('utf8'))
    283. 

  283. 

  284.     try:
    285. 

  285.         text = unicode(text, 'utf-8')
    286. 

  286.     except:
    287. 

  287.         text = text
    288. 

  288. 

  289.     return text
    290. 

  290. 

  291. def get_url(url):
    292. 

  292.     return "http:"+urllib.quote(url.replace('http:', ''))
    293. 

  293. 

  294. #-------------------------------------------------------------------------------
    295. 

  295. def get_params(paramstring):
    296. 

  296. 	param=[]
    297. 

  297. 	if len(paramstring)>=2:
    298. 

  298. 		params=paramstring
    299. 

  299. 		cleanedparams=params.replace('?','')
    300. 

  300. 		if (params[len(params)-1]=='/'):
    301. 

  301. 			params=params[0:len(params)-2]
    302. 

  302. 		pairsofparams=cleanedparams.split('&')
    303. 

  303. 		param={}
    304. 

  304. 		for i in range(len(pairsofparams)):
    305. 

  305. 			splitparams={}
    306. 

  306. 			splitparams=pairsofparams[i].split('=')
    307. 

  307. 			if (len(splitparams))==2:
    308. 

  308. 				param[splitparams[0]]=splitparams[1]
    309. 

  309. 	return param
    310. 

  310. #-------------------------------------------------------------------------------
    311. 

  311. params=get_params(sys.argv[2])
    312. 

  312. 

  313. # get cookies from last session
    314. 

  314. cj = cookielib.MozillaCookieJar(fcookies)
    315. 

  315. try:
    316. 

  316.     cj.load()
    317. 

  317. except:
    318. 

  318.     pass
    319. 

  319. hr  = urllib2.HTTPCookieProcessor(cj)
    320. 

  320. opener = urllib2.build_opener(hr)
    321. 

  321. urllib2.install_opener(opener)
    322. 

  322. 

  323. mode = None
    324. 

  324. 

  325. try:
    326. 

  326. 	mode = urllib.unquote_plus(params['mode'])
    327. 

  327. except:
    328. 

  328. 	Get_TV_Channels()
    329. 

  329. 

  330. if mode == 'EPG_DATE':
    331. 

  331. 	Get_EPG_Date(params)
    332. 

  332. elif mode == 'EPG':
    333. 

  333. 	Get_EPG(params)
    334. 

  334. elif mode == 'PLAY':
    335. 

  335. 	PLAY(params)
    336. 

  336. 

  337. #-- store cookies
    338. 

  338. cj.save()
    339.