import os
import shutil
import MySQLdb as mdb
import glob
import json
from urllib import urlopen
from fuzzywuzzy import fuzz
from operator import itemgetter

PROCESSDIR="/srv/storage2/files/VideoProcessing/"
THOMAS="Thomas"
CHUGGINGTON="Chuggington"
MIKE="MikeTheKnight"
OCTONAUTS="Octonauts"
NIGHTGARDEN="InTheNightGarden"
RAARAA="RaaRaa"
INPUTDIR="Input"
SICKBEARDAPI="http://192.168.0.2:8081/api/3678177136222bf5002be209220ccb20/"

class TVShow:
    def __init__(self, episode, season, title, subtitle, description):
        self.episode = episode
        self.season = season
        self.title = title
        self.subtitle = subtitle
        self.description = description

def FindShowId(showName):
    jsonurl = urlopen(SICKBEARDAPI+"?cmd=shows")
    result = json.loads(jsonurl.read())

    shows = []
    for show in result['data']:
        shows.append((show, fuzz.partial_ratio(showName.lower(), result['data'][show]['show_name'].lower())))

    shows = sorted(shows, key=itemgetter(1), reverse=True)

    if shows[0][1] > 85:
        return shows[0][0]

def FindEpisode(showId, name=None, description=None):
    jsonurl = urlopen("{0}?cmd=show.seasons&tvdbid={1}".format(SICKBEARDAPI, showId))
    result = json.loads(jsonurl.read())
    
    for season in result['data']:
        for episode in result['data'][season]:
            if name is not None and name.lower() == result['data'][season][episode]['name'].lower():
                return (season, episode)
            elif description is not None:
                result = FindEpisodeByDescription(showId, season, episode, description)
                if result is not None:
                    return result
    
    return (0, 0)

def GetEpisodeName(subtitle, showName):
    if subtitle[:len(showName)].lower() == showName.lower():
        return subtitle[len(showName + ' and the '):]
    else:
        return subtitle

def FindEpisodeByDescription(showId, season, episode, description):
    jsonEpisodeUrl = urlopen("{0}?cmd=episode&tvdbid={1}&season={2}&episode={3}".format(SICKBEARDAPI, showId, season, episode))
    episodeResult = json.loads(jsonEpisodeUrl.read())

    if fuzz.ratio(episodeResult['data']['description'].lower(), description.lower()) > 85:
        return (season, episode)
    
    return None

def DetermineTargetFilename(directory, filename, inputFilename):
    dir = os.path.join(directory, inputFilename[:-4])
	
    if not os.path.exists(dir):
        os.makedirs(dir)
  
    return os.path.join(dir, filename)

def ProcessKnownEpisode(directory, filename, inputFilename):
    target = DetermineTargetFilename(directory, filename, inputFilename)
    shutil.move(inputFilename, target)
	
def ProcessUnknownEpisode(inputFilename):
	print "do this"

def RetrieveEpisodeData(inputFile):
    con = mdb.connect('localhost', 'script', 'script', 'mythconverg')
    
    with con:
        cur = con.cursor(mdb.cursors.DictCursor)
        cur.execute("select episode, season, title, subtitle, description from mythconverg.recorded where basename = '{0}'".format(inputFile))
        result = cur.fetchone()
        
        return TVShow(result['episode'], result['season'], result['title'], result['subtitle'], result['description'])
    
def FixEpisodeSeasonNumber(number):
    if number < 10:
        return "0{0}".format(number)
    else:
        return str(number)

def GetDirectory(title, season):
    directory = ""
    if title == "Thomas and Friends" or title == "Thomas the Tank Engine & Friends":
        directory = THOMAS
    elif title == "Chuggington":
        directory = CHUGGINGTON
    elif title == "Mike the Knight":
        directory = MIKE
    elif title == "Octonauts" or title == "The Octonauts":
        directory = OCTONAUTS
    elif title == "In the Night Garden":
        directory = NIGHTGARDEN
    elif title == "Raa Raa! The Noisy Lion":
        directory = RAARAA
    else:
        print "Didn't match"
    
    return os.path.join(PROCESSDIR, directory, INPUTDIR, season)
    
    
def ProcessEpisode(inputFile):
    show = RetrieveEpisodeData(inputFile)

    if show.title:
        if show.subtitle:
            show.subtitle = GetEpisodeName(show.subtitle, show.title)

        if (show.season == '0' or show.episode == '0'):
            showId = FindShowId(show.title)
            
            result = FindEpisode(showId, show.subtitle, show.description)
            show.season = result[0]
            show.episode = result[1]
            
        if show.season != "0" and show.episode != "0":
            show.season = FixEpisodeSeasonNumber(show.season)
            show.episode = FixEpisodeSeasonNumber(show.episode)
            
            seasonFolder = "Season {0}".format(show.season)
            season = "S{0}".format(show.season)
            episode = "E{0}".format(show.episode)
            renamedFile = "{0}{1} - {2} - SD TV_.mpg".format(season, episode, show.subtitle)
            
            directory = GetDirectory(show.title, seasonFolder)
            ProcessKnownEpisode(directory, renamedFile, os.path.basename(inputFile))
    else:
        print "no show name"

def GetFilesToProcess():
    return glob.glob("*.mpg")
        
for file in GetFilesToProcess():
    ProcessEpisode(file)