default.py

import sys
import xbmc, xbmcgui, xbmcplugin
import urllib2
import urllib
import CommonFunctions
common = CommonFunctions
from t0mm0.common.addon import Addon
from metahandler import metahandlers

addon = Addon('plugin.video.1channel', sys.argv)

from bs4 import BeautifulSoup
import soupselect; soupselect.monkeypatch()
import re
import feedparser
import json

#sys.path.append("./xbmc_env/bin/python2.5") 
#sys.path.append("./xbmc_env/lib/python2.5") 
#sys.path.append("./xbmc_env/lib/python2.5/site-packages") 

# plugin modes
MODE_FIRST = 10
MODE_SECOND = 20
MODE_SEARCH = 30
MODE_RESULT = 40
# parameter keys
PARAMETER_KEY_MODE = "mode"

# menu item names
FIRST_SUBMENU = "First Submenu"
SECOND_SUBMENU = "Second Submenu"

# plugin handle
handle = int(sys.argv[1])

#patterns
YOURUPLOAD = re.compile("content.*?(http:.*?\.mp4)")
SAPO = re.compile("url:\s+'(http:.*?\.mp4)'")
MP4UPLOAD = re.compile("(http:.*?:182/.*?\.mp4)")
UPLOADC = re.compile("(http:.*?:182/.*?\.mp4)")
ENGINE = re.compile("url:\s+'(http.*?mp4.*?)'")
VIDEOWEED = re.compile("advURL.*?(http.*?file.*?)\"")
VIDEONEST = re.compile("(http.*?mp4)")
UPLOAD2 = re.compile("video=(http.*?mp4.*?)\&")
RUTUBE = re.compile("file=(http.*?iflv)")

VIDEOEMBED = re.compile("div class=\"videoembed activeembed\".*?src=\"(http.*?)\"")

# utility functions
def parameters_string_to_dict(parameters):
    ''' Convert parameters encoded in a URL to a dict. '''
    paramDict = {}
    if parameters:
        paramPairs = parameters[1:].split("&")
        for paramsPair in paramPairs:
            paramSplits = paramsPair.split('=')
            if (len(paramSplits)) == 2:
                paramDict[paramSplits[0]] = paramSplits[1]
    return paramDict

def addDirectoryItem(name, isFolder=True, parameters={}, listitem = False):
    ''' Add a list item to the XBMC UI.'''
    if (listitem):
        li = listitem
    else:
        li = xbmcgui.ListItem(name)
    url = sys.argv[0] + '?' + urllib.urlencode(parameters)
    return xbmcplugin.addDirectoryItem(handle=int(sys.argv[1]), url=url, listitem=li, isFolder=isFolder)

def addLink(name,url):
    ok=True
    liz=xbmcgui.ListItem(name)
    liz.setInfo( type="video",  infoLabels = {
            'title' : name 
    })
    ok=xbmcplugin.addDirectoryItem(handle=int(sys.argv[1]),url=url,listitem=liz)
    return ok

def fetchUrl(url):
    req = urllib2.Request(url)
    req.add_header('User-Agent', 'Mozilla/5.0 (Windows; U; Windows NT 5.1; en-GB; rv:1.9.0.3) Gecko/2008092417 Firefox/3.0.3')
    response = urllib2.urlopen(req)
    html=response.read()
    response.close()
    return html

'''
XBMC plays mp4
'''
def get_video_url(provider_name, url):    
    html = fetchUrl(url)
    
    match = False
    video_url = ''
    if provider_name == 'rutube':
        match=RUTUBE.search(html)
    else:
        try:
       
           provider_page_url = VIDEOEMBED.search(html).group(1)
           html = fetchUrl(provider_page_url)
           
           if (provider_name == 'sapo'):
              match=SAPO.search(html)
           elif (provider_name == 'yourupload'):
              match=YOURUPLOAD.search(html)
           elif (provider_name == 'uploadc'):
              match=UPLOADC.search(html)
           elif (provider_name == 'upload2'):
              match=UPLOAD2.search(html)
           elif (provider_name == 'mp4upload'):
              match=MP4UPLOAD.search(html)
           #elif (provider_name == 'videoweed'):
           #  match=VIDEOWEED.search(html)
           elif (provider_name == 'engine'):
              match=ENGINE.search(html)
           elif (provider_name == 'videonest'):
              match=VIDEONEST.search(html)
           else:
             try:
                print "AD-TV: unknown provider - "+provider_page_url+"\n"+html
             except:
                print "AD-TV: unknown provider - "+provider_page_url+"\n"+repr(html)
             match=VIDEONEST.search(html)

        except Exception as inst:
           print "AD-TV: parsing problem - \n"+html, inst

    if(match):
        video_url = match.group(1)
        if (provider_name == 'engine'):
            video_url = urllib.unquote(video_url)

    return video_url
    
def get_video_providers(url):
    html = fetchUrl(url)
    soup = BeautifulSoup(html)
    generic_video_items = soup.findSelect('div.generic-video-item')
    result = []
    for item in generic_video_items:
        ssoup = BeautifulSoup(str(item.contents))
        video_page = ssoup.findSelect('div.thumb')
        if (not video_page[0].a):
            provider_url = url
            temp = BeautifulSoup(str(video_page[0].contents))
            span = temp.findAll('span')
            provider_name = span[1].string
        else:
            provider_name = video_page[0].a.center.span.string
            provider_url = video_page[0].a['href']

        data = {'provider_name': provider_name, 
            'provider_url': provider_url}

        result.append(data)

    return result

# UI builder functions
def show_root_menu():
    ''' Show the plugin root menu. '''
    '''addDirectoryItem(name=FIRST_SUBMENU, parameters={ PARAMETER_KEY_MODE: MODE_FIRST }, isFolder=True)
    addDirectoryItem(name=SECOND_SUBMENU, parameters={ PARAMETER_KEY_MODE: MODE_SECOND }, isFolder=True)
    '''
    
    addDirectoryItem(' - Search - ', isFolder=True, parameters ={'mode': MODE_SEARCH})

    #d = feedparser.parse('http://www.animedreaming.tv/rss.xml')
    #for i in d.entries:
    entries = get_latest_episodes()
    for i in entries:
        name = i["title"]
        link = i["link"]
        meta = create_meta('tvshow', re.sub('episode\s\d+','', name), False, '')
        listitem = xbmcgui.ListItem(name, iconImage=meta['cover_url'], thumbnailImage=meta['cover_url'])
        listitem.setInfo('video', meta)
        fanart = ''
        try: fanart = meta['backdrop_url']
        except: pass
        listitem.setProperty('fanart_image', fanart)
        addDirectoryItem(name, isFolder=True, parameters ={'link': link, 'mode': MODE_FIRST}, listitem=listitem)
        #video_url = get_video_url(link) 
        #if (video_url):
        #    addLink(name, video_url)
    xbmcplugin.endOfDirectory(handle=handle, succeeded=True)

def get_latest_episodes():
    json_episodes = fetchUrl("""http://query.yahooapis.com/v1/public/yql?q=use%0A'https%3A%2F%2Fraw.github.com%2Fneofreko%2Fcss2xpath%2Fmaster%2Fdata.html.cssselect.xml'%20as%20data.html.cssselect%3B%0Aselect%20*%20from%20data.html.cssselect%20where%20url%3D%22http%3A%2F%2Fwww.animedreaming.tv%22%20and%20css%3D'%23latest_holder%20.episode_div%20a'&format=json&callback=""")
    json_titles = fetchUrl("""http://query.yahooapis.com/v1/public/yql?q=use%0A'https%3A%2F%2Fraw.github.com%2Fneofreko%2Fcss2xpath%2Fmaster%2Fdata.html.cssselect.xml'%20as%20data.html.cssselect%3B%0Aselect%20*%20from%20data.html.cssselect%20where%20url%3D%22http%3A%2F%2Fwww.animedreaming.tv%22%20and%20css%3D'%23latest_holder%20.latest_title%20a'&format=json&callback=""")
    print json_episodes
    print json_titles
    episodes = json.loads(json_episodes)
    titles = json.loads(json_titles)
    result = []
    i=0
    titles_result = titles["query"]["results"]["results"]["a"]
    episodes_result = episodes["query"]["results"]["results"]["a"]
    print episodes_result
    episode_regex = re.compile("episode-(\d+)")
    for link in titles_result:
        title = link["content"]
        episode_url = "http://www.animedreaming.tv/%s" % episodes_result[i]["href"]
        episode_number = episode_regex.search(episode_url)
        if (episode_number):
            title = "%s episode %s" % (title, episode_number.group(1))
        i=i+1
        obj = {'link': episode_url, 'title': title}
        result.append(obj)

    return result
    
def show_first_submenu():
    ''' Show first submenu. '''
    params = parameters_string_to_dict(sys.argv[2])
    link = params.get('link', "")
    providers = get_video_providers(urllib.unquote(link))
    for provider in providers:
        print "processing %s from %s" % (provider['provider_name'], provider['provider_url'])
        video_url = get_video_url(provider['provider_name'], provider['provider_url'])
        print video_url
        if (video_url):
            addLink("Play from %s" % provider['provider_name'], video_url)
        #addDirectoryItem(provider['provider_name'], isFolder=True, parameters ={'link': provider['provider_url'], 'mode': MODE_SECOND})
    #addLink()
    xbmcplugin.endOfDirectory(handle=handle, succeeded=True)

def show_second_submenu():
    ''' Show second submenu. '''
    params = parameters_string_to_dict(sys.argv[2])
    link = params.get('link', "")
    video_url = get_video_url(urllib.unquote(link)) 
    if (video_url):
        addLink("Play from %s" % name, video_url)
    xbmcplugin.endOfDirectory(handle=handle, succeeded=True)

def getSearch():
    query = common.getUserInput('Search', '')
    if not query:
        return False
    html = fetchUrl('http://www.animedreaming.tv/search.php?searchquery='+urllib.quote_plus(query))
    for link in re.compile('<li.*?href="(http://www.animedreaming.tv/.*?)">(.*?)</a>').findall(html):
        title = re.sub('<b>|</b>','*',link[1])
        meta = create_meta('tvshow', title, False, '')
        listitem = xbmcgui.ListItem(meta['title'], iconImage=meta['cover_url'], thumbnailImage=meta['cover_url'])
        listitem.setInfo('video', meta)
        fanart = ''
        try: fanart = meta['backdrop_url']
        except: pass
        listitem.setProperty('fanart_image', fanart)
        addDirectoryItem(title, isFolder=True, parameters ={'link': link[0], 'mode': MODE_RESULT}, listitem=listitem)

    xbmcplugin.endOfDirectory(handle=handle, succeeded=True)

def parseResults():
    params = parameters_string_to_dict(sys.argv[2])
    link = params.get('link', "")
    html = fetchUrl(urllib.unquote(link))
    for link in re.compile('<li.*?href="(http://www.animedreaming.tv/.*?)".*?>(.*?)&nbsp;<').findall(html):
        addDirectoryItem(link[1], isFolder=True, parameters ={'link': link[0], 'mode': MODE_FIRST})
    xbmcplugin.endOfDirectory(handle=handle, succeeded=True)

def create_meta(video_type, title, year, thumb):
    metaget=metahandlers.MetaData()
    try:    year = int(year)
    except: year = 0
    year = str(year)
    meta = {'title':title, 'year':year, 'imdb_id':'', 'overlay':''}
    meta['imdb_id'] = ''
    #if META_ON:
    try:
        if video_type == 'tvshow':
            meta = metaget.get_meta(video_type, title)
            if not (meta['imdb_id'] or meta['tvdb_id']):
                meta = metaget.get_meta(video_type, title, year=year)
            alt_id = meta['tvdb_id']

            ###Temporary work around. t0mm0.common isn't happy with the episode key being a str
            meta['episode'] = int(meta['episode'])
            meta['rating'] = float(meta['rating'])

        else: #movie
            meta = metaget.get_meta(video_type, title, year=year)
            alt_id = meta['tmdb_id']

        '''
        if video_type == 'tvshow' and not USE_POSTERS:
            meta['cover_url'] = meta['banner_url']
        if POSTERS_FALLBACK and meta['cover_url'] in ('/images/noposter.jpg',''):
            meta['cover_url'] = thumb
        img = meta['cover_url']
        '''
    # except: addon.log('Error assigning meta data for %s %s %s' %(video_type, title, year))
    except: print 'Error assigning meta data for %s %s %s' %(video_type, title, year)
    return meta

# parameter values
params = parameters_string_to_dict(sys.argv[2])
mode = int(params.get(PARAMETER_KEY_MODE, "0"))
print "##########################################################"
print("Mode: %s" % mode)
print "##########################################################"

# Depending on the mode, call the appropriate function to build the UI.
if not sys.argv[2]:
    # new start
    ok = show_root_menu()
elif mode == MODE_FIRST:
    ok = show_first_submenu()
elif mode == MODE_SECOND:
    ok = show_second_submenu()
elif mode == MODE_SEARCH:
    ok = getSearch()
elif mode == MODE_RESULT:
    ok = parseResults()