#
# -*- coding: utf-8 -*-
#
#  Scarica la lista di URI per vedere i canali rai
#
 
import urllib2, re, httplib
 
class ChannelList():
 
    def __init__(self):
 
        ## Inizializzo un dizionario di canali
        self.channels = {}
 
        self.UA = 'Mozilla/5.0 (X11; U; Linux i686; it; rv:1.9.0.6) Gecko/2009011912 Firefox/3.0.11'
 
    def updateChannels(self):
 
        ## Qui dobbiamo un pochino imbrogliare..ehm ehm
        req = urllib2.Request("http://www.rai.tv/dl/RaiTV/videoWall/PublishingBlock-5566288c-3d21-48dc-b3e2-af7fbe3b2af8.xml",
                              None,
                              { 'User-Agent': self.UA })
        r = urllib2.urlopen(req)
        self.parseChannels(r.read())
 
    def parseChannels(self, xmlfile):
        """Fai il parse dei canali scaricati dalla rai"""
 
        for channel, url in re.findall("<videoUnit name=\"([^\"]*)\".*>.*\\n.*<url>(\S*)</url>", xmlfile):
            url = re.sub(r"&amp;", "&", url)
            self.channels[channel] = url
 
 
    def getChannel(self, key):
        """Ottiene il vero canale da cui guardare.. :)"""
 
        if not self.channels.has_key(key):
            return None
 
        host = re.search(r"^.*//([^/]*)/", self.channels[key]).group(1)
        # print " => host = %s" % host
        path = re.search(r"%s/(.*)$" % host, self.channels[key]).group(1)
        # print " => path = %s" % path
 
        req = urllib2.Request(self.channels[key],
                              None,
                              {'User-Agent': self.UA})
 
        r = urllib2.urlopen(req)
        mms = r.read()
        mms = re.search(r"HREF=\"(\S*)\"", mms).group(1)
        return mms