PageRenderTime 64ms CodeModel.GetById 22ms RepoModel.GetById 0ms app.codeStats 0ms

/IPTVPlayer/hosts/hostwrestlingnetwork.py

https://gitlab.com/zatoshi/iptvplayer-for-e2
Python | 337 lines | 303 code | 17 blank | 17 comment | 8 complexity | 4b962b96978c19157925f58b7d6ea7fe MD5 | raw file
    

  1. # -*- coding: utf-8 -*-

    2. 

  2. 

    3. 

  3. ###################################################

    4. 

  4. # LOCAL import

    5. 

  5. ###################################################

    6. 

  6. from Plugins.Extensions.IPTVPlayer.components.iptvplayerinit import TranslateTXT as _, SetIPTVPlayerLastHostError

    7. 

  7. from Plugins.Extensions.IPTVPlayer.components.ihost import CHostBase, CBaseHostClass, CDisplayListItem, ArticleContent, RetHost, CUrlItem

    8. 

  8. from Plugins.Extensions.IPTVPlayer.tools.iptvtools import CSelOneLink, printDBG, printExc, CSearchHistoryHelper, GetLogoDir, GetCookieDir

    9. 

  9. from Plugins.Extensions.IPTVPlayer.libs.urlparserhelper import getDirectM3U8Playlist

    10. 

  10. from Plugins.Extensions.IPTVPlayer.libs.urlparser import urlparser

    11. 

  11. from Plugins.Extensions.IPTVPlayer.libs.youtube_dl.utils import clean_html

    12. 

  12. from Plugins.Extensions.IPTVPlayer.libs.crypto.hash.sha1Hash import SHA1

    13. 

  13. ###################################################

    14. 

  14. 

    15. 

  15. ###################################################

    16. 

  16. # FOREIGN import

    17. 

  17. ###################################################

    18. 

  18. from Components.config import config, ConfigInteger, ConfigSelection, ConfigYesNo, ConfigText, getConfigListEntry

    19. 

  19. from datetime import timedelta

    20. 

  20. from binascii import hexlify

    21. 

  21. import re

    22. 

  22. import urllib

    23. 

  23. import time

    24. 

  24. import random

    25. 

  25. try:    import simplejson as json

    26. 

  26. except: import json

    27. 

  27. ###################################################

    28. 

  28. 

    29. 

  29. 

    30. 

  30. ###################################################

    31. 

  31. # E2 GUI COMMPONENTS 

    32. 

  32. ###################################################

    33. 

  33. from Plugins.Extensions.IPTVPlayer.components.asynccall import MainSessionWrapper

    34. 

  34. from Screens.MessageBox import MessageBox

    35. 

  35. from Components.config import config, ConfigSelection, getConfigListEntry

    36. 

  36. ###################################################

    37. 

  37. 

    38. 

  38. 

    39. 

  39. ###################################################

    40. 

  40. # Config options for HOST

    41. 

  41. ###################################################

    42. 

  42.  

    43. 

  43. def GetConfigList():

    44. 

  44.     optionList = []

    45. 

  45.     return optionList

    46. 

  46. ###################################################

    47. 

  47. 

    48. 

  48. def gettytul():

    49. 

  49.     return 'http://wrestlingnetwork.tv/'

    50. 

  50. 

    51. 

  51. class WrestlingNet(CBaseHostClass):

    52. 

  52.     MAIN_URL   = 'http://wrestlingnetwork.tv/'

    53. 

  53.     

    54. 

  54.     MAIN_CAT_TAB = [{'category':'search',                'title':_('Search'), 'search_item':True},

    55. 

  55.                     {'category':'search_history',        'title':_('Search history')} ]

    56. 

  56.     

    57. 

  57.     def __init__(self):

    58. 

  58.         printDBG("WrestlingNet.__init__")

    59. 

  59.         CBaseHostClass.__init__(self, {'history':'wrestling-network.net'})

    60. 

  60.         self.DEFAULT_ICON = 'http://static-server1.wrestling-network.net/images/layout/wnhd_5years_bug.png'

    61. 

  61.         self.cacheSubCategory = {}

    62. 

  62.         

    63. 

  63.     def _getFullUrl(self, url):

    64. 

  64.         if 0 < len(url) and not url.startswith('http'):

    65. 

  65.             url =  self.MAIN_URL + url

    66. 

  66.         return url

    67. 

  67. 

    68. 

  68.     def listsTab(self, tab, cItem):

    69. 

  69.         printDBG("WrestlingNet.listsMainMenu")

    70. 

  70.         for item in tab:

    71. 

  71.             params = dict(cItem)

    72. 

  72.             params.update(item)

    73. 

  73.             params['name']  = 'category'

    74. 

  74.             self.addDir(params)

    75. 

  75.             

    76. 

  76.     def listsMainMenu(self, cItem, subNextCategory, nextCategory):

    77. 

  77.         printDBG("listsMainMenu")

    78. 

  78.         self.cacheSubCategory = {}

    79. 

  79.         

    80. 

  80.         sts, data = self.cm.getPage(cItem['url'])

    81. 

  81.         if not sts: return

    82. 

  82.         

    83. 

  83.         subCatsUrls = []

    84. 

  84.         catsData = self.cm.ph.getAllItemsBeetwenMarkers(data, '<ul class="sub-menu">', '</ul>', withMarkers=False)

    85. 

  85.         for catItem in catsData:

    86. 

  86.             tmp = re.compile('<a[^>]*?href="([^"]+?)"[^>]*?>([^<]+?)<').findall(catItem)

    87. 

  87.             for item in tmp:

    88. 

  88.                 url   = self._getFullUrl(item[0])

    89. 

  89.                 title = self.cleanHtmlStr(item[1])

    90. 

  90.                 catUrl = '/'.join( url.split('/')[:-1] )

    91. 

  91.                 if catUrl not in self.cacheSubCategory: 

    92. 

  92.                     self.cacheSubCategory[catUrl] = []

    93. 

  93.                 self.cacheSubCategory[catUrl].append({'url':url, 'title':title})

    94. 

  94.                 subCatsUrls.append(url)

    95. 

  95.         

    96. 

  96.         data = self.cm.ph.getDataBeetwenMarkers(data, '<div class="menu-main-menu', '</div>')[1]

    97. 

  97.         data = re.compile('<a[^>]*?href="([^"]+?)"[^>]*?>([^<]+?)<').findall(data)

    98. 

  98.         prevUrl = ''

    99. 

  99.         categories = []

    100. 

  100.         for item in data:

    101. 

  101.             url   = self._getFullUrl(item[0])

    102. 

  102.             if url in subCatsUrls: continue

    103. 

  103.             title = self.cleanHtmlStr(item[1])

    104. 

  104.             if url in self.cacheSubCategory:

    105. 

  105.                 category = subNextCategory

    106. 

  106.             else:

    107. 

  107.                 category = nextCategory

    108. 

  108.             params = dict(cItem)

    109. 

  109.             params.update({'url':url, 'title':title})

    110. 

  110.             params['category'] = category

    111. 

  111.             self.addDir(params)

    112. 

  112.         

    113. 

  113.     def listSubCategory(self, cItem, nextCategory):

    114. 

  114.         printDBG("listSubCategory")

    115. 

  115.         tab =  self.cacheSubCategory.get(cItem['url'], [])

    116. 

  116.         params = dict(cItem)

    117. 

  117.         params['category'] = nextCategory

    118. 

  118.         self.listsTab(tab, params)

    119. 

  119.         

    120. 

  120.     def listVideosByCategory(self, cItem, category):

    121. 

  121.         printDBG("WrestlingNet.listVideosByCategory")

    122. 

  122.         self._listVideos(cItem)

    123. 

  123.         

    124. 

  124.     def listSearchResult(self, cItem, searchPattern, searchType):

    125. 

  125.         printDBG("WrestlingNet.listSearchResult cItem[%s], searchPattern[%s] searchType[%s]" % (cItem, searchPattern, searchType))

    126. 

  126.         searchPattern = urllib.quote_plus(searchPattern)

    127. 

  127.         cItem.update({'url':self.MAIN_URL, 'srch':'?s=%s&submit=Search' % searchPattern})

    128. 

  128.         self._listVideos(cItem)

    129. 

  129.     

    130. 

  130.     def _listVideos(self, cItem):

    131. 

  131.         printDBG("WrestlingNet._listVideos")

    132. 

  132.         

    133. 

  133.         page = cItem.get('page', 1)

    134. 

  134.         url = cItem.get('url', '')

    135. 

  135.         if page > 1:

    136. 

  136.             url += '/page/%s' % page

    137. 

  137.         url += cItem.get('srch', '')

    138. 

  138.         

    139. 

  139.         sts, data = self.cm.getPage(url)

    140. 

  140.         if not sts: return

    141. 

  141.         

    142. 

  142.         if ('/page/%s?' % (page+1)) in data or ('/page/%s/' % (page+1)) in data:

    143. 

  143.             nextPage = True

    144. 

  144.         else: nextPage = False

    145. 

  145.         

    146. 

  146.         data = self.cm.ph.getAllItemsBeetwenMarkers(data, '<div class="content-container">', '<!-- close .content-container -->', withMarkers=False)

    147. 

  147.         for item in data:

    148. 

  148.             url    = self.cm.ph.getSearchGroups(item, 'href="([^"]+?)"')[0]

    149. 

  149.             if '/contact' in url: continue

    150. 

  150.             icon   = self.cm.ph.getSearchGroups(item, 'src="([^"]+?)"')[0]

    151. 

  151.             title  = self.cm.ph.getDataBeetwenMarkers(item, '<h3', '</h3>')[1]

    152. 

  152.             if '' != url and '' != title:

    153. 

  153.                 params = dict(cItem)

    154. 

  154.                 params.update( {'title':self.cleanHtmlStr( title ), 'url':self._getFullUrl(url), 'desc':self.cleanHtmlStr( item ), 'icon':self._getFullUrl(icon)} )

    155. 

  155.                 self.addVideo(params)

    156. 

  156.         

    157. 

  157.         if nextPage:

    158. 

  158.             params = params = dict(cItem)

    159. 

  159.             params.update({'title':_("Next page"), 'page':page+1})

    160. 

  160.             self.addDir(params)

    161. 

  161.     

    162. 

  162.     def getLinksForVideo(self, cItem):

    163. 

  163.         printDBG("WrestlingNet.getLinksForVideo [%s]" % cItem)

    164. 

  164.         urlTab = []

    165. 

  165.         sts, data = self.cm.getPage(cItem['url'])

    166. 

  166.         if not sts: return urlTab

    167. 

  167.         

    168. 

  168.         sts, data = self.cm.ph.getDataBeetwenMarkers(data, '<div class="solar-main-content">', '<!-- close .content-container -->', True)

    169. 

  169.         if not sts:

    170. 

  170.             SetIPTVPlayerLastHostError(_("Please visit '%s' from using web-browser form the PC. If links are available please report this problem.\nEmail: samsamsam@o2.pl") % cItem['url'])

    171. 

  171.             return urlTab

    172. 

  172.             

    173. 

  173.         data = data.split('<h2 style="text-align: center;">')

    174. 

  174.         if len(data): del data[0]

    175. 

  175.         re_links = re.compile('<a[^>]+?href="([^"]+?)"[^>]*?>([^<]+?)</a>')

    176. 

  176.         for item in data:

    177. 

  177.             type = self.cleanHtmlStr( item.split('</h2>')[0] )

    178. 

  178.             

    179. 

  179.             links = re_links.findall(item)

    180. 

  180.             for link in links:

    181. 

  181.                 urlTab.append({'name':type + ' ' + link[1], 'url':link[0]})

    182. 

  182.             

    183. 

  183.         return urlTab

    184. 

  184.         

    185. 

  185.     def getResolvedURL(self, url):

    186. 

  186.         printDBG("WrestlingNet.getResolvedURL [%s]" % url)

    187. 

  187.         urlTab = []

    188. 

  188.         sts, data = self.cm.getPage(url)

    189. 

  189.         if not sts: return urlTab

    190. 

  190.         

    191. 

  191.         videoUrl = self.cm.ph.getSearchGroups(data, '<iframe[^>]+?src="([^"]+?)"')[0]

    192. 

  192.         if not videoUrl:

    193. 

  193.             videoUrl = self.cm.ph.getSearchGroups(data, "file: '([^']+?)'")[0]

    194. 

  194.             if '.m3u8' in videoUrl:

    195. 

  195.                 urlTab = getDirectM3U8Playlist(videoUrl, checkExt=True)

    196. 

  196.             

    197. 

  197.         urlTab = self.up.getVideoLinkExt( videoUrl )

    198. 

  198.         return urlTab

    199. 

  199.         

    200. 

  200.     def getFavouriteData(self, cItem):

    201. 

  201.         return cItem['url']

    202. 

  202.         

    203. 

  203.     def getLinksForFavourite(self, fav_data):

    204. 

  204.         return self.getLinksForVideo({'url':fav_data})

    205. 

  205.     

    206. 

  206.     def handleService(self, index, refresh=0, searchPattern='', searchType=''):

    207. 

  207.         printDBG('WrestlingNet.handleService start')

    208. 

  208.         CBaseHostClass.handleService(self, index, refresh, searchPattern, searchType)

    209. 

  209.         name     = self.currItem.get("name", None)

    210. 

  210.         category = self.currItem.get("category", '')

    211. 

  211.         printDBG( "WrestlingNet.handleService: ---------> name[%s], category[%s] " % (name, category) )

    212. 

  212.         searchPattern = self.currItem.get("search_pattern", searchPattern)

    213. 

  213.         self.currList = [] 

    214. 

  214. 

    215. 

  215.     #MAIN MENU

    216. 

  216.         if None == name:

    217. 

  217.             cItem = {'name':'category', 'url':self.MAIN_URL, 'icon':self.DEFAULT_ICON}

    218. 

  218.             self.listsMainMenu(cItem, 'list_subcategory', 'list_category')

    219. 

  219.             self.listsTab(WrestlingNet.MAIN_CAT_TAB, cItem)

    220. 

  220.         elif 'list_subcategory' == category:

    221. 

  221.             self.listSubCategory(self.currItem, 'list_category')

    222. 

  222.     #CATEGORY

    223. 

  223.         elif 'list_category' == category:

    224. 

  224.             self.listVideosByCategory(self.currItem, '')

    225. 

  225.     #WYSZUKAJ

    226. 

  226.         elif category in ["search", "search_next_page"]:

    227. 

  227.             cItem = dict(self.currItem)

    228. 

  228.             cItem.update({'search_item':False, 'name':'category'}) 

    229. 

  229.             self.listSearchResult(cItem, searchPattern, searchType)

    230. 

  230.     #HISTORIA WYSZUKIWANIA

    231. 

  231.         elif category == "search_history":

    232. 

  232.             self.listsHistory({'name':'history', 'category': 'search'}, 'desc', _("Type: "))

    233. 

  233.         else:

    234. 

  234.             printExc()

    235. 

  235.         CBaseHostClass.endHandleService(self, index, refresh)

    236. 

  236. 

    237. 

  237. class IPTVHost(CHostBase):

    238. 

  238. 

    239. 

  239.     def __init__(self):

    240. 

  240.         CHostBase.__init__(self, WrestlingNet(), True)

    241. 

  241. 

    242. 

  242.     def getLogoPath(self):

    243. 

  243.         return RetHost(RetHost.OK, value = [GetLogoDir('wrestlingnetworklogo.png')])

    244. 

  244.     

    245. 

  245.     def getLinksForVideo(self, Index = 0, selItem = None):

    246. 

  246.         retCode = RetHost.ERROR

    247. 

  247.         retlist = []

    248. 

  248.         if not self.isValidIndex(Index): return RetHost(retCode, value=retlist)

    249. 

  249.         

    250. 

  250.         urlList = self.host.getLinksForVideo(self.host.currList[Index])

    251. 

  251.         for item in urlList:

    252. 

  252.             need_resolve = 1

    253. 

  253.             retlist.append(CUrlItem(item["name"], item["url"], need_resolve))

    254. 

  254. 

    255. 

  255.         return RetHost(RetHost.OK, value = retlist)

    256. 

  256.     # end getLinksForVideo

    257. 

  257.     

    258. 

  258.     def getResolvedURL(self, url):

    259. 

  259.         # resolve url to get direct url to video file

    260. 

  260.         retlist = []

    261. 

  261.         urlList = self.host.getResolvedURL(url)

    262. 

  262.         for item in urlList:

    263. 

  263.             need_resolve = 0

    264. 

  264.             retlist.append(CUrlItem(item["name"], item["url"], need_resolve))

    265. 

  265. 

    266. 

  266.         return RetHost(RetHost.OK, value = retlist)

    267. 

  267. 

    268. 

  268.     def convertList(self, cList):

    269. 

  269.         hostList = []

    270. 

  270.         searchTypesOptions = [] # ustawione alfabetycznie

    271. 

  271.         #searchTypesOptions.append(("Filmy",  "filmy"))

    272. 

  272.         #searchTypesOptions.append(("Seriale","seriale"))

    273. 

  273.         

    274. 

  274.         for cItem in cList:

    275. 

  275.             hostLinks = []

    276. 

  276.             type = CDisplayListItem.TYPE_UNKNOWN

    277. 

  277.             possibleTypesOfSearch = None

    278. 

  278. 

    279. 

  279.             if 'category' == cItem['type']:

    280. 

  280.                 if cItem.get('search_item', False):

    281. 

  281.                     type = CDisplayListItem.TYPE_SEARCH

    282. 

  282.                     possibleTypesOfSearch = searchTypesOptions

    283. 

  283.                 else:

    284. 

  284.                     type = CDisplayListItem.TYPE_CATEGORY

    285. 

  285.             elif cItem['type'] == 'video':

    286. 

  286.                 type = CDisplayListItem.TYPE_VIDEO

    287. 

  287.             elif 'more' == cItem['type']:

    288. 

  288.                 type = CDisplayListItem.TYPE_MORE

    289. 

  289.             elif 'audio' == cItem['type']:

    290. 

  290.                 type = CDisplayListItem.TYPE_AUDIO

    291. 

  291.                 

    292. 

  292.             if type in [CDisplayListItem.TYPE_AUDIO, CDisplayListItem.TYPE_VIDEO]:

    293. 

  293.                 url = cItem.get('url', '')

    294. 

  294.                 if '' != url:

    295. 

  295.                     hostLinks.append(CUrlItem("Link", url, 1))

    296. 

  296.                 

    297. 

  297.             title       =  self.host.cleanHtmlStr( cItem.get('title', '') )

    298. 

  298.             description =  self.host.cleanHtmlStr( cItem.get('desc', '') )

    299. 

  299.             icon        =  self.host.cleanHtmlStr( cItem.get('icon', '') )

    300. 

  300.             

    301. 

  301.             hostItem = CDisplayListItem(name = title,

    302. 

  302.                                         description = description,

    303. 

  303.                                         type = type,

    304. 

  304.                                         urlItems = hostLinks,

    305. 

  305.                                         urlSeparateRequest = 1,

    306. 

  306.                                         iconimage = icon,

    307. 

  307.                                         possibleTypesOfSearch = possibleTypesOfSearch)

    308. 

  308.             hostList.append(hostItem)

    309. 

  309. 

    310. 

  310.         return hostList

    311. 

  311.     # end convertList

    312. 

  312. 

    313. 

  313.     def getSearchItemInx(self):

    314. 

  314.         try:

    315. 

  315.             list = self.host.getCurrList()

    316. 

  316.             for i in range( len(list) ):

    317. 

  317.                 if list[i]['category'] == 'search':

    318. 

  318.                     return i

    319. 

  319.         except:

    320. 

  320.             printDBG('getSearchItemInx EXCEPTION')

    321. 

  321.             return -1

    322. 

  322. 

    323. 

  323.     def setSearchPattern(self):

    324. 

  324.         try:

    325. 

  325.             list = self.host.getCurrList()

    326. 

  326.             if 'history' == list[self.currIndex]['name']:

    327. 

  327.                 pattern = list[self.currIndex]['title']

    328. 

  328.                 search_type = list[self.currIndex]['search_type']

    329. 

  329.                 self.host.history.addHistoryItem( pattern, search_type)

    330. 

  330.                 self.searchPattern = pattern

    331. 

  331.                 self.searchType = search_type

    332. 

  332.         except:

    333. 

  333.             printDBG('setSearchPattern EXCEPTION')

    334. 

  334.             self.searchPattern = ''

    335. 

  335.             self.searchType = ''

    336. 

  336.         return

    337. 

  337. 

    338. 

  338.