Shoko Series Scanner.py

import re, os, os.path, json
import Media, VideoFiles, Stack, Utils

import urllib2, urllib

Prefs = {
    'Hostname': '127.0.0.1',
    'Port': 8111,
    'Username': 'Default',
    'Password': '',
    'IncludeOther': True
}

API_KEY = ''

def log(methodName, message, *args):
    '''
        Create a log message given the message and arguments
    '''
    logMsg = message
    # Replace the arguments in the string
    if args:
        logMsg = message % args
        
    logMsg = methodName + ' :: ' + logMsg
    print logMsg


def HttpPost(url, postdata):
    myheaders = {'Content-Type': 'application/json'}
    
    req = urllib2.Request('http://%s:%s/%s' % (Prefs['Hostname'], Prefs['Port'], url), headers=myheaders)
    return json.load(urllib2.urlopen(req, postdata))


def HttpReq(url, authenticate=True):
    log('HttpReq' ,"Requesting: %s", url)
    api_string = ''
    if authenticate:
        api_string = '&apikey=%s' % GetApiKey()

    myheaders = {'Accept': 'application/json'}
    
    req = urllib2.Request('http://%s:%s/%s%s' % (Prefs['Hostname'], Prefs['Port'], url, api_string), headers=myheaders)
    return json.load(urllib2.urlopen(req))


def GetApiKey():
    global API_KEY

    if not API_KEY:
        data = '{"user":"%s", "pass":"%s", "device":"%s"}' % (
            Prefs['Username'], Prefs['Password'] if Prefs['Password'] != None else '', 'Shoko Series Scanner For Plex')
        resp = HttpPost('api/auth', data)['apikey']
        log('GetApiKey', "Got API KEY: %s", resp)
        API_KEY = resp
        return resp

    return API_KEY


def Scan(path, files, mediaList, subdirs, language=None, root=None):
    log('Scan', 'path: %s', path)
    log('Scan', 'files: %s', files)
    log('Scan', 'mediaList: %s', mediaList)
    log('Scan', 'subdirs: %s', subdirs)
    log('Scan', 'language: %s', language)
    log('Scan', 'root: %s', root)
    
    # Scan for video files.
    VideoFiles.Scan(path, files, mediaList, subdirs, root)
    
    for idx, file in enumerate(files):
        log('Scan', 'file: %s', file)
        # http://127.0.0.1:8111/api/ep/getbyfilename?apikey=d422dfd2-bdc3-4219-b3bb-08b85aa65579&filename=%5Bjoseole99%5D%20Clannad%20-%2001%20(1280x720%20Blu-ray%20H264)%20%5B8E128DF5%5D.mkv

        episode_data = HttpReq("api/ep/getbyfilename?filename=%s" % (urllib.quote(os.path.basename(file))))
        if len(episode_data) == 0: break

        series_data = HttpReq("api/serie/fromep?id=%d&nocast=1&notag=1" % episode_data['id'])
        showTitle = str(series_data['name']) #no idea why I need to do this.
        log('Scan', 'show title: %s', showTitle)

        seasonNumber = 0
        if episode_data['season'] == None:
            if episode_data['eptype'] == 'Episode': seasonNumber = 1
            if episode_data['eptype'] == 'Credits': seasonNumber = -1 #season -1 for OP/ED
        else:
            seasonNumber = episode_data['season'].split('x')[0]

        if seasonNumber <= 0 and Prefs['IncludeOther'] == False: break #Ignore this by choice.
            

        log('Scan', 'season number: %s', seasonNumber)

        seasonYear = episode_data['year']
        log('Scan', 'season year: %s', seasonYear)

        episodeNumber = int(episode_data['epnumber'])
        if episode_data['eptype'] != 'Episode':
            episodeNumber = str("%s%d" % (episode_data['eptype'][0], episode_data['epnumber']))

        log('Scan', 'episode number: %s', episodeNumber)

        episodeTitle = str(episode_data['name'])
        log('Scan', 'episode title: %s', episodeTitle)

        vid = Media.Episode(showTitle, int(seasonNumber), episodeNumber , episodeTitle, int(seasonYear))
        log('Scan', 'vid: %s', vid)
        vid.parts.append(file)
        mediaList.append(vid)
    
    log('Scan', 'stack media')
    Stack.Scan(path, files, mediaList, subdirs)
    log('Scan', 'media list %s', mediaList)