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
8 from xml
.sax
.saxutils
import escape
9 from debug
import debug_write
, fn_attr
11 SCRIPTDIR
= os
.path
.dirname(__file__
)
13 hack83
= config
.getHack83()
15 class TivoHTTPServer(SocketServer
.ThreadingMixIn
, BaseHTTPServer
.HTTPServer
):
18 def __init__(self
, server_address
, RequestHandlerClass
):
19 BaseHTTPServer
.HTTPServer
.__init
__(self
, server_address
,
21 self
.daemon_threads
= True
23 def add_container(self
, name
, settings
):
24 if self
.containers
.has_key(name
) or name
== 'TiVoConnect':
25 raise "Container Name in use"
27 settings
['content_type'] = GetPlugin(settings
['type']).CONTENT_TYPE
28 self
.containers
[name
] = settings
30 print 'Unable to add container', name
32 class TivoHTTPHandler(BaseHTTPServer
.BaseHTTPRequestHandler
):
34 def address_string(self
):
35 host
, port
= self
.client_address
[:2]
40 basepath
= unquote_plus(self
.path
).split('/')[1]
43 for name
, container
in self
.server
.containers
.items():
45 plugin
= GetPlugin(container
['type'])
46 plugin
.send_file(self
, container
, name
)
49 ## Not a file not a TiVo command fuck them
50 if not self
.path
.startswith('/TiVoConnect'):
54 o
= urlparse("http://fake.host" + self
.path
)
55 query
= parse_qs(o
[4])
58 if query
.has_key('Command') and len(query
['Command']) >= 1:
60 command
= query
['Command'][0]
62 # If we are looking at the root container
63 if command
== "QueryContainer" and \
64 (not query
.has_key('Container') or query
['Container'][0] == '/'):
68 if query
.has_key('Container'):
69 # Dispatch to the container plugin
70 basepath
= query
['Container'][0].split('/')[0]
71 for name
, container
in self
.server
.containers
.items():
73 plugin
= GetPlugin(container
['type'])
74 if hasattr(plugin
, command
):
75 method
= getattr(plugin
, command
)
81 # If we made it here it means we couldn't match the request to
83 self
.unsupported(query
)
85 def root_container(self
):
86 tsn
= self
.headers
.getheader('TiVo_TCD_ID', '')
87 tsnshares
= config
.getShares(tsn
)
89 for section
, settings
in tsnshares
:
91 settings
['content_type'] = \
92 GetPlugin(settings
['type']).CONTENT_TYPE
93 tsncontainers
[section
] = settings
94 except Exception, msg
:
95 print section
, '-', msg
96 t
= Template(file=os
.path
.join(SCRIPTDIR
, 'templates',
97 'root_container.tmpl'))
98 t
.containers
= tsncontainers
99 t
.hostname
= socket
.gethostname()
101 self
.send_response(200)
106 self
.send_response(200)
107 self
.send_header('Content-type', 'text/html')
109 t
= Template(file=os
.path
.join(SCRIPTDIR
, 'templates',
114 def unsupported(self
, query
):
115 if hack83
and 'Command' in query
and 'Filter' in query
:
116 debug_write(__name__
, fn_attr(), ['Unsupported request,',
117 'checking to see if it is video.'])
118 command
= query
['Command'][0]
119 plugin
= GetPlugin('video')
120 if ''.join(query
['Filter']).find('video') >= 0 and \
121 hasattr(plugin
, command
):
122 debug_write(__name__
, fn_attr(), ['Unsupported request,',
124 'send to video plugin for it to sort out.'])
125 method
= getattr(plugin
, command
)
129 self
.send_response(404)
130 self
.send_header('Content-type', 'text/html')
132 t
= Template(file=os
.path
.join(SCRIPTDIR
, 'templates',
137 if __name__
== '__main__':
139 httpd
= TivoHTTPServer(('', 9032), TivoHTTPHandler
)
140 httpd
.add_container('test', 'x-container/tivo-videos',
141 r
'C:\Documents and Settings\Armooo\Desktop\pyTivo\test')
142 httpd
.serve_forever()