gmyth: gmyth-stream/server/0.2/lib/server.py@f5ef83bbe8b5 (annotated)

morphbr@565	1	#!/usr/bin/env python
morphbr@565	2
morphbr@565	3	__author__ = "Gustavo Sverzut Barbieri / Artur Duque de Souza"
morphbr@565	4	__author_email__ = "barbieri@gmail.com / artur.souza@indt.org.br"
morphbr@565	5	__license__ = "GPL"
morphbr@565	6	__version__ = "0.3"
morphbr@565	7
morphbr@565	8	import os
morphbr@565	9	import threading
morphbr@565	10	import SocketServer
morphbr@565	11	import BaseHTTPServer
morphbr@565	12	import socket
morphbr@565	13	import urlparse
morphbr@565	14	import cgi
morphbr@565	15	import lib.utils as utils
morphbr@565	16	import logging as log
morphbr@565	17
morphbr@565	18	__all__ = ("Transcoder", "RequestHandler", "Server", "serve_forever",
morphbr@565	19	"load_plugins_transcoders")
morphbr@565	20
morphbr@565	21	class Transcoder(object):
morphbr@565	22	log = log.getLogger("gmyth-stream.transcoder")
morphbr@565	23	priority = 0 # negative values have higher priorities
morphbr@565	24	name = None # to be used in requests
morphbr@565	25
morphbr@565	26	def __init__(self, params):
morphbr@565	27	self.params = params
morphbr@565	28	# __init__()
morphbr@565	29
morphbr@565	30
morphbr@565	31	def params_first(self, key, default=None):
morphbr@565	32	if default is None:
morphbr@565	33	return self.params[key][0]
morphbr@565	34	else:
morphbr@565	35	try:
morphbr@565	36	return self.params[key][0]
morphbr@565	37	except:
morphbr@565	38	return default
morphbr@565	39	# params_first()
morphbr@565	40
morphbr@565	41
morphbr@565	42	def get_mimetype(self):
morphbr@565	43	mux = self.params_first("mux", "mpg")
morphbr@565	44
morphbr@565	45	if mux == "mpeg":
morphbr@565	46	return "video/mpeg"
morphbr@565	47	elif mux == "avi":
morphbr@565	48	return "video/x-msvideo"
morphbr@565	49	else:
morphbr@565	50	return "application/octet-stream"
morphbr@565	51	# get_mimetype()
morphbr@565	52
morphbr@565	53
morphbr@565	54	def start(self, outfile):
morphbr@565	55	return True
morphbr@565	56	# start()
morphbr@565	57
morphbr@565	58
morphbr@565	59	def stop(self):
morphbr@565	60	return Tru
morphbr@565	61	# stop()
morphbr@565	62
morphbr@565	63
morphbr@565	64	def __str__(self):
morphbr@565	65	return '%s("%s", mux="%s", params=%s)' % \
morphbr@565	66	(self.__class__.__name__,
morphbr@565	67	self.params_first("uri", "None"),
morphbr@565	68	self.params_first("mux", "mpg"),
morphbr@565	69	self.params)
morphbr@565	70	# __str__()
morphbr@565	71	# Transcoder
morphbr@565	72
morphbr@565	73
morphbr@565	74
morphbr@565	75	class RequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):
morphbr@565	76	log = log.getLogger("gmyth-stream.request")
morphbr@565	77	def_transcoder = None
morphbr@565	78	transcoders = utils.PluginSet(Transcoder)
morphbr@565	79
morphbr@565	80	@classmethod
morphbr@565	81	def load_plugins_transcoders(cls, directory):
morphbr@565	82	cls.transcoders.load_from_directory(directory)
morphbr@565	83
morphbr@565	84	if cls.def_transcoder is None and cls.transcoders:
morphbr@565	85	cls.def_transcoder = cls.transcoders[0].name
morphbr@565	86	# load_plugins_transcoders()
morphbr@565	87
morphbr@565	88
morphbr@565	89	def do_dispatch(self, body):
morphbr@565	90	self.url = self.path
morphbr@565	91
morphbr@565	92	pieces = urlparse.urlparse(self.path)
morphbr@565	93	self.path = pieces[2]
morphbr@565	94	self.query = cgi.parse_qs(pieces[4])
morphbr@565	95
morphbr@565	96	if self.path == "/":
morphbr@565	97	self.serve_main(body)
morphbr@565	98	elif self.path == "/shutdown.do":
morphbr@565	99	self.serve_shutdown(body)
morphbr@565	100	elif self.path == "/stop-transcoder.do":
morphbr@565	101	self.serve_stop_transcoder(body)
morphbr@565	102	elif self.path == "/status.do":
morphbr@565	103	self.serve_status(body)
morphbr@565	104	elif self.path == "/play.do":
morphbr@565	105	self.serve_play(body)
morphbr@565	106	elif self.path == "/stream.do":
morphbr@565	107	self.serve_stream(body)
morphbr@565	108	else:
morphbr@565	109	self.send_error(404, "File not found")
morphbr@565	110	# do_dispatch()
morphbr@565	111
morphbr@565	112
morphbr@565	113	def do_GET(self):
morphbr@565	114	self.do_dispatch(True)
morphbr@565	115	# do_GET()
morphbr@565	116
morphbr@565	117
morphbr@565	118	def do_HEAD(self):
morphbr@565	119	self.do_dispatch(False)
morphbr@565	120	# do_HEAD()
morphbr@565	121
morphbr@565	122
morphbr@565	123	def _nav_items(self):
morphbr@565	124	self.wfile.write("""\
morphbr@565	125	<li><a href="/play.do">Play</a></li>
morphbr@565	126	<li><a href="/status.do">Status</a></li>
morphbr@565	127	<li><a href="/stop-transcoder.do">Stop transcoders</a></li>
morphbr@565	128	<li><a href="/shutdown.do">Shutdown Server</a></li>
morphbr@565	129	""")
morphbr@565	130	# _nav_items()
morphbr@565	131
morphbr@565	132
morphbr@565	133	def serve_main(self, body):
morphbr@565	134	self.send_response(200)
morphbr@565	135	self.send_header("Content-Type", "text/html")
morphbr@565	136	self.send_header('Connection', 'close')
morphbr@565	137	self.end_headers()
morphbr@565	138	if body:
morphbr@565	139	self.wfile.write("""\
morphbr@565	140	<html>
morphbr@565	141	<head><title>Catota Server</title></head>
morphbr@565	142	<body>
morphbr@565	143	<h1>Welcome to Catota Server</h1>
morphbr@565	144	<ul>
morphbr@565	145	""")
morphbr@565	146	self._nav_items()
morphbr@565	147	self.wfile.write("""\
morphbr@565	148	</ul>
morphbr@565	149	</body>
morphbr@565	150	</html>
morphbr@565	151	""")
morphbr@565	152	# serve_main()
morphbr@565	153
morphbr@565	154
morphbr@565	155	def serve_play(self, body):
morphbr@565	156	self.send_response(200)
morphbr@565	157	self.send_header("Content-Type", "text/html")
morphbr@565	158	self.send_header('Connection', 'close')
morphbr@565	159	self.end_headers()
morphbr@565	160	if body:
morphbr@565	161	self.wfile.write("""\
morphbr@565	162	<html>
morphbr@565	163	<head><title>Catota Server</title></head>
morphbr@565	164	<body>
morphbr@565	165	<h1>Play</h1>
morphbr@565	166	<form action="/stream.do" method="GET">
morphbr@565	167	<input type="text" name="type" value="file" />://<input type="text" name="location" value=""/>
morphbr@565	168	<input type="submit" />
morphbr@565	169	</form>
morphbr@565	170	<ul>
morphbr@565	171	""")
morphbr@565	172	self._nav_items()
morphbr@565	173	self.wfile.write("""\
morphbr@565	174	</ul>
morphbr@565	175	</body>
morphbr@565	176	</html>
morphbr@565	177	""")
morphbr@565	178	# serve_play()
morphbr@565	179
morphbr@565	180
morphbr@565	181	def serve_shutdown(self, body):
morphbr@565	182	self.send_response(200)
morphbr@565	183	self.send_header("Content-Type", "text/html")
morphbr@565	184	self.send_header('Connection', 'close')
morphbr@565	185	self.end_headers()
morphbr@565	186	if body:
morphbr@565	187	self.wfile.write("""\
morphbr@565	188	<html>
morphbr@565	189	<head><title>Catota Server Exited</title></head>
morphbr@565	190	<body>
morphbr@565	191	<h1>Catota is not running anymore</h1>
morphbr@565	192	</body>
morphbr@565	193	</html>
morphbr@565	194	""")
morphbr@565	195	self.server.server_close()
morphbr@565	196	# serve_shutdown()
morphbr@565	197
morphbr@565	198
morphbr@565	199	def serve_stop_all_transcoders(self, body):
morphbr@565	200	self.send_response(200)
morphbr@565	201	self.send_header("Content-Type", "text/html")
morphbr@565	202	self.send_header('Connection', 'close')
morphbr@565	203	self.end_headers()
morphbr@565	204	if body:
morphbr@565	205	self.server.stop_transcoders()
morphbr@565	206	self.wfile.write("""\
morphbr@565	207	<html>
morphbr@565	208	<head><title>Catota Server Stopped Transcoders</title></head>
morphbr@565	209	<body>
morphbr@565	210	<h1>Catota stopped running transcoders</h1>
morphbr@565	211	<ul>
morphbr@565	212	""")
morphbr@565	213	self._nav_items()
morphbr@565	214	self.wfile.write("""\
morphbr@565	215	</ul>
morphbr@565	216	</body>
morphbr@565	217	</html>
morphbr@565	218	""")
morphbr@565	219	# serve_stop_all_transcoders()
morphbr@565	220
morphbr@565	221
morphbr@565	222	def serve_stop_selected_transcoders(self, body, requests):
morphbr@565	223	self.send_response(200)
morphbr@565	224	self.send_header("Content-Type", "text/html")
morphbr@565	225	self.send_header('Connection', 'close')
morphbr@565	226	self.end_headers()
morphbr@565	227	if body:
morphbr@565	228	self.wfile.write("""\
morphbr@565	229	<html>
morphbr@565	230	<head><title>Catota Server Stopped Transcoders</title></head>
morphbr@565	231	<body>
morphbr@565	232	<h1>Catota stopped running transcoders:</h1>
morphbr@565	233	<ul>
morphbr@565	234	""")
morphbr@565	235	transcoders = self.server.get_transcoders()
morphbr@565	236
morphbr@565	237	for req in requests:
morphbr@565	238	try:
morphbr@565	239	host, port = req.split(":")
morphbr@565	240	except IndexError:
morphbr@565	241	continue
morphbr@565	242
morphbr@565	243	port = int(port)
morphbr@565	244	addr = (host, port)
morphbr@565	245
morphbr@565	246	for t, r in transcoders:
morphbr@565	247	if r.client_address == addr:
morphbr@565	248	t.stop()
morphbr@565	249	self.server.del_transcoders(self, t)
morphbr@565	250	self.wfile.write("""\
morphbr@565	251	<li>%s: %s:%s</li>
morphbr@565	252	""" % (t, addr[0], addr[1]))
morphbr@565	253	t.stop()
morphbr@565	254	break
morphbr@565	255	self.wfile.write("""\
morphbr@565	256	</ul>
morphbr@565	257	<ul>
morphbr@565	258	""")
morphbr@565	259	self._nav_items()
morphbr@565	260	self.wfile.write("""\
morphbr@565	261	</ul>
morphbr@565	262	</body>
morphbr@565	263	</html>
morphbr@565	264	""")
morphbr@565	265	# serve_stop_selected_transcoders()
morphbr@565	266
morphbr@565	267
morphbr@565	268	def serve_stop_transcoder(self, body):
morphbr@565	269	req = self.query.get("request", None)
morphbr@565	270	if req and "all" in req:
morphbr@565	271	self.serve_stop_all_transcoders(body)
morphbr@565	272	elif req:
morphbr@565	273	self.serve_stop_selected_transcoders(body, req)
morphbr@565	274	else:
morphbr@565	275	self.serve_status(body)
morphbr@565	276	# serve_stop_transcoder()
morphbr@565	277
morphbr@565	278
morphbr@565	279	def serve_status(self, body):
morphbr@565	280	self.send_response(200)
morphbr@565	281	self.send_header("Content-Type", "text/html")
morphbr@565	282	self.send_header('Connection', 'close')
morphbr@565	283	self.end_headers()
morphbr@565	284	if body:
morphbr@565	285	self.wfile.write("""\
morphbr@565	286	<html>
morphbr@565	287	<head><title>Catota Server Status</title></head>
morphbr@565	288	<body>
morphbr@565	289	<h1>Catota Status</h1>
morphbr@565	290	""")
morphbr@565	291	tl = self.server.get_transcoders()
morphbr@565	292	if not tl:
morphbr@565	293	self.wfile.write("<p>No running transcoder.</p>\n")
morphbr@565	294	else:
morphbr@565	295	self.wfile.write("<p>Running transcoders:</p>\n")
morphbr@565	296	self.wfile.write("""\
morphbr@565	297	<ul>
morphbr@565	298	<li><a href="/stop-transcoder.do?request=all">[STOP ALL]</a></li>
morphbr@565	299	""")
morphbr@565	300	for transcoder, request in tl:
morphbr@565	301	self.wfile.write("""\
morphbr@565	302	<li>%s: %s:%s <a href="/stop-transcoder.do?request=%s:%s">[STOP]</a></li>
morphbr@565	303	""" % (transcoder, request.client_address[0], request.client_address[1],
morphbr@565	304	request.client_address[0], request.client_address[1]))
morphbr@565	305
morphbr@565	306	self.wfile.write("""\
morphbr@565	307	</ul>
morphbr@565	308	<ul>
morphbr@565	309	""")
morphbr@565	310	self._nav_items()
morphbr@565	311	self.wfile.write("""\
morphbr@565	312	</ul>
morphbr@565	313	</body>
morphbr@565	314	</html>
morphbr@565	315	""")
morphbr@565	316	# serve_status()
morphbr@565	317
morphbr@565	318
morphbr@565	319	def _get_transcoder(self):
morphbr@565	320	# get transcoder option: mencoder is the default
morphbr@565	321	request_transcoders = self.query.get("transcoder", ["mencoder"])
morphbr@565	322
morphbr@565	323	for t in request_transcoders:
morphbr@565	324	transcoder = self.transcoders.get(t)
morphbr@565	325	if transcoder:
morphbr@565	326	return transcoder
morphbr@565	327
morphbr@565	328	if not transcoder:
morphbr@565	329	return self.transcoders[self.def_transcoder]
morphbr@565	330	# _get_transcoder()
morphbr@565	331
morphbr@565	332
morphbr@565	333	def serve_stream(self, body):
morphbr@565	334	transcoder = self._get_transcoder()
morphbr@565	335	try:
morphbr@565	336	obj = transcoder(self.query)
morphbr@565	337	except Exception, e:
morphbr@565	338	self.send_error(500, str(e))
morphbr@565	339	return
morphbr@565	340
morphbr@565	341	self.send_response(200)
morphbr@565	342	self.send_header("Content-Type", obj.get_mimetype())
morphbr@565	343	self.send_header('Connection', 'close')
morphbr@565	344	self.end_headers()
morphbr@565	345
morphbr@565	346	if body:
morphbr@565	347	self.server.add_transcoders(self, obj)
morphbr@565	348	obj.start(self.wfile)
morphbr@565	349	self.server.del_transcoders(self, obj)
morphbr@565	350	# serve_stream()
morphbr@565	351
morphbr@565	352
morphbr@565	353	def log_request(self, code='-', size='-'):
morphbr@565	354	self.log.info('"%s" %s %s', self.requestline, str(code), str(size))
morphbr@565	355	# log_request()
morphbr@565	356
morphbr@565	357
morphbr@565	358	def log_error(self, format, *args):
morphbr@565	359	self.log.error("%s: %s" % (self.address_string(), format % args))
morphbr@565	360	# log_error()
morphbr@565	361
morphbr@565	362
morphbr@565	363	def log_message(self, format, *args):
morphbr@565	364	self.log.info("%s: %s" % (self.address_string(), format % args))
morphbr@565	365	# log_message()
morphbr@565	366	# RequestHandler
morphbr@565	367
morphbr@565	368
morphbr@565	369
morphbr@565	370	class Server(SocketServer.ThreadingMixIn, BaseHTTPServer.HTTPServer):
morphbr@565	371	log = log.getLogger("gmyth-streamer.server")
morphbr@565	372	run = True
morphbr@565	373	_transcoders = {}
morphbr@565	374	_lock = threading.RLock()
morphbr@565	375
morphbr@565	376	def serve_forever(self):
morphbr@565	377	self.log.info("GMyth-Streamer serving HTTP on %s:%s" %
morphbr@565	378	self.socket.getsockname())
morphbr@565	379	try:
morphbr@565	380	while self.run:
morphbr@565	381	self.handle_request()
morphbr@565	382	except KeyboardInterrupt, e:
morphbr@565	383	pass
morphbr@565	384
morphbr@565	385	self.log.debug("Stopping all remaining transcoders...")
morphbr@565	386	self.stop_transcoders()
morphbr@565	387	self.log.debug("Transcoders stopped!")
morphbr@565	388	# serve_forever()
morphbr@565	389
morphbr@565	390
morphbr@565	391	def get_request(self):
morphbr@565	392	skt = self.socket
morphbr@565	393	old = skt.gettimeout()
morphbr@565	394	skt.settimeout(0.5)
morphbr@565	395	while self.run:
morphbr@565	396	try:
morphbr@565	397	r = skt.accept()
morphbr@565	398	skt.settimeout(old)
morphbr@565	399	return r
morphbr@565	400	except socket.timeout, e:
morphbr@565	401	pass
morphbr@565	402	raise socket.error("Not running")
morphbr@565	403	# get_request()
morphbr@565	404
morphbr@565	405
morphbr@565	406	def server_close(self):
morphbr@565	407	self.run = False
morphbr@565	408	self.stop_transcoders()
morphbr@565	409
morphbr@565	410	BaseHTTPServer.HTTPServer.server_close(self)
morphbr@565	411	# server_close()
morphbr@565	412
morphbr@565	413
morphbr@565	414	def stop_transcoders(self):
morphbr@565	415	self._lock.acquire()
morphbr@565	416	for transcoder, request in self._transcoders.iteritems():
morphbr@565	417	self.log.info("Stop transcoder: %s, client=%s" %
morphbr@565	418	(transcoder, request.client_address))
morphbr@565	419	transcoder.stop()
morphbr@565	420	self._lock.release()
morphbr@565	421	# stop_transcoders()
morphbr@565	422
morphbr@565	423
morphbr@565	424	def get_transcoders(self):
morphbr@565	425	self._lock.acquire()
morphbr@565	426	try:
morphbr@565	427	return self._transcoders.items()
morphbr@565	428	finally:
morphbr@565	429	self._lock.release()
morphbr@565	430	# get_transcoders()
morphbr@565	431
morphbr@565	432
morphbr@565	433	def add_transcoders(self, request, transcoder):
morphbr@565	434	self._lock.acquire()
morphbr@565	435	try:
morphbr@565	436	self._transcoders[transcoder] = request
morphbr@565	437	finally:
morphbr@565	438	self._lock.release()
morphbr@565	439	# add_transcoders()
morphbr@565	440
morphbr@565	441
morphbr@565	442	def del_transcoders(self, request, transcoder):
morphbr@565	443	self._lock.acquire()
morphbr@565	444	try:
morphbr@565	445	del self._transcoders[transcoder]
morphbr@565	446	finally:
morphbr@565	447	self._lock.release()
morphbr@565	448	# del_transcoders()
morphbr@565	449	# Server
morphbr@565	450
morphbr@565	451
morphbr@565	452
morphbr@565	453	def serve_forever(host="0.0.0.0", port=40000):
morphbr@565	454	addr = (host, port)
morphbr@565	455
morphbr@565	456	RequestHandler.protocol_version = "HTTP/1.0"
morphbr@565	457	httpd = Server(addr, RequestHandler)
morphbr@565	458	httpd.serve_forever()
morphbr@565	459	# serve_forever()
morphbr@565	460
morphbr@565	461
morphbr@565	462	def load_plugins_transcoders(directory):
morphbr@565	463	RequestHandler.load_plugins_transcoders(directory)
morphbr@565	464	# load_plugins_transcoders()

author	rosfran
	Wed Apr 18 23:13:26 2007 +0100 (2007-04-18)
branch	trunk
changeset 568	f5ef83bbe8b5
child 569	644a526d4086
permissions	-rw-r--r--