httpserver.py

   1 import time, os, BaseHTTPServer, SocketServer, socket, re
   2 from urllib import unquote_plus, quote, unquote
   3 from urlparse import urlparse
   4 from cgi import parse_qs
   5 from Cheetah.Template import Template
   6 from plugin import GetPlugin
   7 from xml.sax.saxutils import escape
   8
   9 SCRIPTDIR = os.path.dirname(__file__)
  10
  11 class TivoHTTPServer(SocketServer.ThreadingMixIn, BaseHTTPServer.HTTPServer):
  12     containers = {}
  13
  14     def __init__(self, server_address, RequestHandlerClass):
  15         BaseHTTPServer.HTTPServer.__init__(self, server_address,
  16                                            RequestHandlerClass)
  17         self.daemon_threads = True
  18
  19     def add_container(self, name, settings):
  20         if self.containers.has_key(name) or name == 'TiVoConnect':
  21             raise "Container Name in use"
  22         settings['content_type'] = GetPlugin(settings['type']).CONTENT_TYPE
  23         self.containers[name] = settings
  24
  25 class TivoHTTPHandler(BaseHTTPServer.BaseHTTPRequestHandler):
  26
  27     def address_string(self):
  28         host, port = self.client_address[:2]
  29         return host
  30
  31     def do_GET(self):
  32
  33         basepath = unquote_plus(self.path).split('/')[1]
  34
  35         ## Get File
  36         for name, container in self.server.containers.items():
  37             if basepath == name:
  38                 plugin = GetPlugin(container['type'])
  39                 plugin.send_file(self, container, name)
  40                 return
  41
  42         ## Not a file not a TiVo command fuck them
  43         if not self.path.startswith('/TiVoConnect'):
  44             self.infopage()
  45             return
  46
  47         o = urlparse("http://fake.host" + self.path)
  48         query = parse_qs(o[4])
  49
  50         mname = False
  51         if query.has_key('Command') and len(query['Command']) >= 1:
  52
  53             command = query['Command'][0]
  54
  55             # If we are looking at the root container
  56             if command == "QueryContainer" and \
  57                (not query.has_key('Container') or query['Container'][0] == '/'):
  58                 self.root_container()
  59                 return
  60
  61             if query.has_key('Container'):
  62                 # Dispatch to the container plugin
  63                 for name, container in self.server.containers.items():
  64                     if query['Container'][0].startswith(name):
  65                         plugin = GetPlugin(container['type'])
  66                         if hasattr(plugin,command):
  67                             method = getattr(plugin, command)
  68                             method(self, query)
  69                         else:
  70                             self.unsupported(query)
  71                         break
  72         else:
  73             self.unsupported(query)
  74
  75     def root_container(self):
  76          t = Template(file=os.path.join(SCRIPTDIR, 'templates',
  77                                         'root_container.tmpl'))
  78          t.containers = self.server.containers
  79          t.hostname = socket.gethostname()
  80          t.escape = escape
  81          self.send_response(200)
  82          self.end_headers()
  83          self.wfile.write(t)
  84
  85     def infopage(self):
  86         self.send_response(200)
  87         self.send_header('Content-type', 'text/html')
  88         self.end_headers()
  89         t = Template(file=os.path.join(SCRIPTDIR, 'templates',
  90                                        'info_page.tmpl'))
  91         self.wfile.write(t)
  92         self.end_headers()
  93
  94     def unsupported(self, query):
  95         self.send_response(404)
  96         self.send_header('Content-type', 'text/html')
  97         self.end_headers()
  98         t = Template(file=os.path.join(SCRIPTDIR, 'templates',
  99                                        'unsupported.tmpl'))
 100         t.query = query
 101         self.wfile.write(t)
 102
 103 if __name__ == '__main__':
 104     def start_server():
 105         httpd = TivoHTTPServer(('', 9032), TivoHTTPHandler)
 106         httpd.add_container('test', 'x-container/tivo-videos',
 107                             r'C:\Documents and Settings\Armooo\Desktop\pyTivo\test')
 108         httpd.serve_forever()
 109
 110     start_server()