pymeds/pymediaserv

#!/usr/bin/env python

# Licensed under the MIT license
# http://opensource.org/licenses/mit-license.php

# Copyright 2005, Tim Potter <tpot@samba.org>

from DIDLLite import TextItem, AudioItem, VideoItem, Resource
import os
import os.path
import random
import socket
import string
import sys
import traceback
from twisted.python import log
from twisted.internet import reactor

def generateuuid():
	if False:
		return 'uuid:asdflkjewoifjslkdfj'
	return ''.join([ 'uuid:'] + map(lambda x: random.choice(string.letters), xrange(20)))

listenAddr = sys.argv[1]
if len(sys.argv) > 2:
	listenPort = int(sys.argv[2])
	if listenPort < 1024 or listenPort > 65535:
		raise ValueError, 'port out of range'
else:
	listenPort = random.randint(10000, 65000)

log.startLogging(sys.stdout)

# Create SSDP server

from SSDP import SSDPServer, SSDP_PORT, SSDP_ADDR

s = SSDPServer()

port = reactor.listenMulticast(SSDP_PORT, s)
port.joinGroup(SSDP_ADDR)
port.setLoopbackMode(0)		# don't get our own sends

uuid = generateuuid()
urlbase = 'http://%s:%d/' % (listenAddr, listenPort)

# Create SOAP server

from twisted.web import server, resource, static
from ContentDirectory import ContentDirectoryServer
from ConnectionManager import ConnectionManagerServer

class WebServer(resource.Resource):
    def __init__(self):
        resource.Resource.__init__(self)

class RootDevice(static.Data):
    def __init__(self):
	r = {
		'hostname': socket.gethostname(),
		'uuid': uuid,
		'urlbase': urlbase,
	}
	d = file('root-device.xml').read() % r
        static.Data.__init__(self, d, 'text/xml')

root = WebServer()
cds = ContentDirectoryServer('My Media Server')
root.putChild('ContentDirectory', cds)
cds = cds.control
root.putChild('ConnectionManager', ConnectionManagerServer())
root.putChild('root-device.xml', RootDevice())


# Area of server to serve media files from

from MediaServer import MediaServer

medianode = static.File('media')
medianode.contentTypes.update( {
	'.wmv':	'video/x-ms-wmv',
	'.ts':	'video/mp2t',
	#'.ts':	'video/mpeg',	# we may want this instead of mp2t
	'.mp4':	'video/mp4',
})
root.putChild('media', medianode)

# Set up media files
allmedia = cds.addContainer('0', 'All Media')
log.msg('allmedia: %s' % repr(allmedia))
for i in os.listdir('media'):
	fpath = os.path.join('media', i)
	try:
		if not os.path.isfile(fpath):
			log.msg('skipping: %s' % fpath)
			continue
		fn, ext = os.path.splitext(i)
		mt = medianode.contentTypes[ext]
		ty = mt.split('/')[0]
		if ty == 'video':
			klass = VideoItem
		elif ty == 'audio':
			klass = AudioItem
		elif ty == 'text':
			klass = TextItem
		else:
			raise KeyError, 'no item for mt: %s' % mt

		item = cds.addItem(allmedia, klass, fn)
		cds[item].res = Resource('%smedia/%s' % (urlbase, i), 'http-get:*:%s:*' % mt)
		cds[item].res.size = os.path.getsize(fpath)
	except KeyError:
            traceback.print_exc(file=log.logfile)

site = server.Site(root)
reactor.listenTCP(listenPort, site)

# we need to do this after the children are there, since we send notifies
s.register('%s::upnp:rootdevice' % uuid,
           'upnp:rootdevice',
           urlbase + 'root-device.xml')

s.register(uuid,
           uuid,
           urlbase + 'root-device.xml')

s.register('%s::urn:schemas-upnp-org:device:MediaServer:1' % uuid,
           'urn:schemas-upnp-org:device:MediaServer:1',
           urlbase + 'root-device.xml')

s.register('%s::urn:schemas-upnp-org:service:ConnectionManager:1' % uuid,
           'urn:schemas-upnp-org:device:ConnectionManager:1',
           urlbase + 'root-device.xml')

s.register('%s::urn:schemas-upnp-org:service:ContentDirectory:1' % uuid,
           'urn:schemas-upnp-org:device:ContentDirectory:1',
           urlbase + 'root-device.xml')

# Main loop

reactor.run()