plugins/m3u_plugin.py

   1 __author__ = 'rvb'
   2 '''
   3 Local Playlist Plugin
   4 (based on ytv plugin by ValdikSS)
   5
   6 plsylist index
   7
   8 http://ip:port/index
   9
  10 m3u playlists
  11
  12 http://ip:port/m3u
  13 http://ip:port/m3u/
  14 http://ip:port/m3u/list-name
  15
  16 plain-tet channel names
  17
  18 http://ip:port/list
  19 http://ip:port/list/
  20 http://ip:port/list/list-name
  21
  22 forward to player
  23
  24 http://ip:port/play/channel-name
  25 http://ip:port/play/list-name/channel-name
  26 http://ip:port/play/{list-name}/(get|mp4|webm)/channel-name
  27
  28 '''
  29 import json
  30 import logging
  31 import urlparse
  32 import urllib
  33 from modules.PluginInterface import VPProxyPlugin
  34 from modules.PlaylistGenerator import PlaylistGenerator
  35 from modules.M3uParser import parseM3U
  36 import config.m3u as config
  37 import os
  38
  39 class M3u(VPProxyPlugin):
  40
  41     handlers = ('m3u', 'm3ut', 'm3uw', 'm3uo', 'list', 'play', 'index')
  42
  43     logger = logging.getLogger('plugin_m3u')
  44     playlist = None
  45
  46     def handle(self, connection):
  47
  48         logger = logging.getLogger('plugin_m3u')
  49         hostport = connection.headers['Host']
  50
  51         self.splitted_path=connection.path.split('/')
  52
  53         if self.splitted_path[1]=='m3u':
  54           prefix='get'
  55         elif self.splitted_path[1]=='m3uw':
  56           prefix='ogg'
  57         elif self.splitted_path[1]=='m3ut':
  58           prefix='mp4'
  59         elif self.splitted_path[1]=='m3uo':
  60           prefix='ogv'
  61         elif self.splitted_path[1] in ("list","play","index"):
  62           None
  63         else:
  64           connection.dieWithError(404)
  65           return
  66
  67         if len(self.splitted_path)>3 and self.splitted_path[1]!="play":
  68             connection.dieWithError()
  69             return
  70
  71         if self.splitted_path[1]=='index':
  72
  73             text=""
  74             for dir in os.walk(config.m3u_directory):
  75                 if dir[0]==config.m3u_directory:
  76                     for dirname in dir[2]:
  77                         if dirname.endswith('.m3u'):
  78                             text=text+'\n'+dirname
  79
  80             connection.send_response(200)
  81             connection.send_header('Content-Type', 'text/plain; charset=utf-8')
  82             connection.end_headers()
  83
  84             listing = text.encode('utf-8')
  85             connection.wfile.write(listing)
  86
  87             return
  88
  89         if len(self.splitted_path)<3 or (len(self.splitted_path)==3 and self.splitted_path[1]=="play"):
  90
  91            m3u_file=config.m3u_directory+'/'+config.m3u_default
  92
  93         else:
  94
  95            filename = self.splitted_path[2]
  96            if filename:
  97              m3u_file=config.m3u_directory+'/'+self.splitted_path[2]
  98            else:
  99              m3u_file=config.m3u_directory+'/'+config.m3u_default
 100
 101         try:
 102             playlist=parseM3U(m3u_file)
 103         except:
 104             connection.dieWithError(404)
 105             return
 106
 107         if self.splitted_path[1]=="list":
 108             connection.send_response(200)
 109             connection.send_header('Content-Type', 'text/plain; charset=utf-8')
 110             connection.end_headers()
 111         elif self.splitted_path[1]=="play":
 112             channel=self.splitted_path[len(self.splitted_path)-1]
 113             logger.debug('channel requestes= "%s"' % channel)
 114             if not channel:
 115               connection.dieWithError(404)
 116               return
 117             channel=urllib.unquote(channel).decode('utf-8')
 118             if len(self.splitted_path)<=4:
 119               prefix="get"
 120             elif len(self.splitted_path)==5:
 121               prefix=self.splitted_path[len(self.splitted_path)-2]
 122               if prefix not in ('get','mp4','webm'):
 123                 connection.dieWithError()
 124                 return
 125             else:
 126               connection.dieWithError()
 127               return
 128             url=None
 129             for record in playlist:
 130               if record.title:
 131                 if record.title.decode('utf-8').replace('/','')==channel:
 132                     url=record.path.decode('utf-8')
 133             if url:
 134                 redirect='/'+prefix+'/'+url
 135                 connection.send_response(302)
 136                 connection.send_header('Location', redirect)
 137                 connection.end_headers()
 138             else:
 139                 logger.debug('Nothing found!')
 140                 connection.dieWithError(404)
 141                 return
 142
 143         else:
 144             connection.send_response(200)
 145             connection.send_header('Content-Type', 'application/x-mpegurl')
 146             connection.end_headers()
 147
 148         try:
 149             playlist=parseM3U(m3u_file)
 150         except:
 151             connection.dieWithError(404)
 152             return
 153
 154         if not playlist:
 155             connection.dieWithError()
 156             return
 157
 158         exported = ""
 159
 160         if self.splitted_path[1]=="list":
 161
 162              for record in playlist:
 163                  if record.title:
 164                      exported = exported + "" + record.title.decode('utf-8').replace('/','') + "\n"
 165
 166         else:
 167
 168             playlistgen = PlaylistGenerator()
 169
 170             for record in playlist:
 171                 if record.title:
 172                     channel=dict()
 173                     channel['name']=record.title.decode('utf-8')
 174                     channel['url']=record.path.decode('utf-8')
 175                     try:
 176                       channel['tvg']=record.attrs['tvg-name'].decode('utf-8')
 177                     except:
 178                       None
 179                     try:
 180                       if record.attrs['group-title'] != 'None':
 181                         channel['group']=record.attrs['group-title'].decode('utf-8')
 182                     except:
 183                       None
 184                     try:
 185                       channel['country']=record.attrs['country'].decode('utf-8')
 186                     except:
 187                       None
 188                     try:
 189                       channel['logo']=record.attrs['tvg-logo'].decode('utf-8')
 190                     except:
 191                       None
 192                     playlistgen.addItem(channel)
 193
 194             exported = playlistgen.exportm3u(hostport,prefix)
 195
 196         if exported:
 197             exported = exported.encode('utf-8')
 198             connection.wfile.write(exported)
 199
 200     def getparam(self, key):
 201         if key in self.params:
 202             return self.params[key][0]
 203         else:
 204             return None