pymeds/pyvr.py

#!/usr/bin/env python
# Copyright 2008 John-Mark Gurney <jmg@funktaht.com>
'''PVR Interface'''

__version__ = '$Change: 1109 $'
# $Id: //depot/python/pymeds/main/shoutcast.py#13 $

from DIDLLite import Container, Item, VideoItem, Resource
from FSStorage import registerklassfun, FSObject

import os.path
import time
from twisted.internet import reactor
from twisted.python import log
import twisted.web
import urlparse
import urllib2

def getPage(url, contextFactory=None, *args, **kwargs):
	"""Download a web page as a string.

	Download a page. Return the HTTPClientFactory, which will
	callback with a page (as a string) or errback with a
	description of the error.

	See HTTPClientFactory to see what extra args can be passed.
	"""
	from twisted.web.client import _parse, HTTPClientFactory

	scheme, host, port, path = _parse(url)
	factory = HTTPClientFactory(url, *args, **kwargs)
	if scheme == 'https':
		from twisted.internet import ssl
		if contextFactory is None:
			contextFactory = ssl.ClientContextFactory()
		reactor.connectSSL(host, port, factory, contextFactory)
	else:
		reactor.connectTCP(host, port, factory)
	return factory

class PYVRShow(VideoItem):
	def __init__(self, *args, **kwargs):
		baseurl = kwargs['url']
		self.info = kwargs['info']
		del kwargs['info'], kwargs['url']

		VideoItem.__init__(self, *args, **kwargs)

		url = self.info['link']
		sc = urlparse.urlparse(url)[0]
		if not sc:
			# need to combine w/ base url
			url = urlparse.urljoin(baseurl, url)
		self.res = Resource(url,
		    'http-get:*:%s:*' % self.info['mimetype'])
		self.res.duration = self.info['duration']

	def doUpdate(self):
		pass

import xml.sax
import xml.sax.handler
from xml.sax.saxutils import unescape

class RecordingXML(xml.sax.handler.ContentHandler):
	dataels = ('title', 'subtitle', 'duration', 'mimetype', 'link',
	    'delete', )

	def __init__(self):
		self.shows = {}
		self.data = None

	def characters(self, chars):
		if self.data is not None:
			self.data.append(chars)

	def startElement(self, name, attrs):
		if name in self.dataels:
			self.data = []
			self.curel = name
		elif name == 'record':
			self.currec = {}

	def endElement(self, name):
		if name in self.dataels:
			data = unescape(''.join(self.data))
			self.currec[self.curel] = data
		elif name == 'record':
			rec = self.currec
			try:
				self.shows[rec['title']].append(rec)
			except KeyError:
				self.shows[rec['title']] = [ rec ]

		self.data = None

def recxmltoobj(page):
	obj = RecordingXML()
	xml.sax.parseString(page, obj)

	return obj.shows

class PYVRShows(Container):
	def __init__(self, *args, **kwargs):
		self.pyvr = kwargs['pyvr']
		del kwargs['pyvr']
		self.show = kwargs['show']
		del kwargs['show']

		Container.__init__(self, *args, **kwargs)

		self.shows = {}
		self.lastmodified = None

	def checkUpdate(self):
		self.pyvr.checkUpdate()
		if self.pyvr.lastmodified != self.lastmodified:
			self.doUpdate()

	@staticmethod
	def getunique(eps, ep):
		i = 1
		while True:
			title = '%s Copy %d' % (ep['subtitle'], i)
			if not eps.has_key(title):
				return title
			i += 1

	@staticmethod
	def eplisttodict(eps):
		ret = {}
		for pos, i in enumerate(eps):
			title = i['subtitle']
			if ret.has_key(title):
				print 'WARNING: dup:', `i`, `ret[title]`
				title = PYVRShows.getunique(ret, i)
			i['pos'] = pos
			ret[title] = i

		return ret

	def genChildren(self):
		return self.eplisttodict(self.pyvr.shows[self.show])

	def createObject(self, i, arg):
		return PYVRShow, i, (), { 'url': self.pyvr.url, 'info': arg }

	def sort(self):
		Container.sort(self, lambda x, y: cmp(x.info['pos'],
		    y.info['pos']))

	def doUpdate(self):
		Container.doUpdate(self)

		self.lastmodified = self.pyvr.lastmodified

class PYVR(FSObject, Container):
	def __init__(self, *args, **kwargs):
		self.url = kwargs.pop('url')

		FSObject.__init__(self, kwargs.pop('path'))
		Container.__init__(self, *args, **kwargs)

		#self.pend = None
		self.lastmodified = None
		self.newobjs = None
		self.objs = {}
		self.lastcheck = 0

	def checkUpdate(self):
		#if self.pend is not None:
		#	raise self.pend

		if time.time() - self.lastcheck < 10:
			return

		# Check to see if any changes have been made
		self.runCheck()

	def runCheck(self):
		while True:
			try:
				self.page = urllib2.urlopen(self.url)
				break
			except urllib2.HTTPError:
				time.sleep(.1)
		#self.page = getPage(self.url, method='HEAD')
		#self.page.deferred.addErrback(self.errCheck).addCallback(
		#    self.doCheck)
		#self.pend = self.page.deferred
		return self.doCheck(self.page)

	def errCheck(self, x):
		print 'errCheck:', `x`
		self.runCheck()

	def doCheck(self, x):
		#print 'doCheck:', self.page.status
		#if self.page.status != '200':
		#	print 'foo'
		#	return reactor.callLater(.01, self.runCheck)

		self.lastcheck = time.time()
		slm = self.page.info()['last-modified']
		if slm == self.lastmodified:
			# Page the same, don't do anything
			#self.pend = None
			return

		return self.parsePage(self.page)
		#self.page = getPage(self.url)
		#self.page.deferred.addCallback(self.parsePage)
		#self.pend = self.page.deferred

		#return self.pend

	def parsePage(self, page):
		slm = self.page.info()['last-modified']
		self.lastmodified = slm
		del self.page
		#self.pend = None

		self.newobjs = recxmltoobj(page.read())
		self.doUpdate()

	def genChildren(self):
		return self.newobjs.iterkeys()

	def createObject(self, i):
		return PYVRShows, i, (), { 'show': i, 'pyvr': self }

	def doUpdate(self):
		if self.newobjs is None:
			import traceback
			traceback.print_stack(file=log.logfile)
			raise ValueError('did not get shows')

		self.shows = self.newobjs

		Container.doUpdate(self)

		self.newobjs = None

def detectpyvrfile(path, fobj):
	bn = os.path.basename(path)
	if bn == 'PYVR':
		return PYVR, { 'url': fobj.readline().strip(), 'path': path }
	return None, None

registerklassfun(detectpyvrfile)