#!/usr/bin/env python
#encoding:utf-8
 
#
# TVNamer.py
# Automatic TV episode namer.
# Uses data from www.thetvdb.com via tvdb_api
#
__author__ = "dbr/Ben"
__version__ = "0.1"
 
import os,sys,re
from optparse import OptionParser
from tvdb_api import tvdb
 
config={}
 
config['with_ep_name'] = '%(showname)s - [%(seasno)02dx%(epno)02d] - %(epname)s.%(ext)s'
config['without_ep_name'] = '%(showname)s - [%(seasno)02dx%(epno)02d].%(ext)s'
 
config['name_parse'] = [
    re.compile('''
    ^(
        [\w\. ]*
    )
    (?= \- |\.)?
    s(\d+)e(\d+)\D
    ''', re.IGNORECASE|re.VERBOSE ),
    
    re.compile('''
    ^(
        [\w\. ]*
    )
    (?= \- |\.)?
    .*?
    (\d+)x(\d+)\D
    ''', re.IGNORECASE|re.VERBOSE ),
    
    re.compile('''
    ^(
        [\w\. ]*
    )
    (?= \- )?
    \D+
    (\d+?)(\d{2})
    \D''', re.IGNORECASE|re.VERBOSE ),
]
 
config['valid_filename_chars'] = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!@£$%^&*()_+=-[]{}\"'.,<>`~? "
 
def findFiles(args):
    allfiles=[]
    for x in args:
        if os.path.isdir(x):
            for sf in os.listdir(x):
                newpath = os.path.join(x,sf)
                if os.path.isfile(newpath):
                    allfiles.append(newpath)
                #end if isfile
            #end for sf
        elif os.path.isfile(x):
            allfiles.append(x)
        #end if isdir
    #end for x
    return allfiles
#end findFiles
 
def processNames(names):
    allEps=[]
    for f in names:
        filepath,filename = os.path.split( f )
        filename,ext = os.path.splitext( filename )
        for r in config['name_parse']:
            match = r.match(filename)
            if match:
                showname,seasno,epno = match.groups()
                showname = showname.replace("."," ").strip()
                seasno,epno = int(seasno), int(epno)
                allEps.append({ 'showname':showname,
                                 'seasno':seasno,
                                 'epno':epno,
                                 'filepath':filepath,
                                 'filename':filename,
                                 'ext':ext
                             })
                break
        else:
            print "Invalid name %s"%(f)
        #end for r
    #end for f
    
    return allEps
#end processNames
 
def formatName(cfile):
    orig_ext = cfile['ext']
    cfile['ext'] = cfile['ext'].replace(".","",1)
    if cfile['epname']:
        n = config['with_ep_name'] % (cfile)
    else:
        n = config['without_ep_name'] % (cfile)
    #end if epname
    cfile['ext'] = orig_ext 
    return n
#end formatName
 
def cleanName(name):
    return ''.join( [c for c in name if c in config['valid_filename_chars']] )
        
 
def renameFile(oldfile,newfile,force=False):
    new_exists = os.access(newfile,os.F_OK)
    if new_exists:
        sys.stderr.write("New filename already exists.. ")
        if force:
            sys.stderr.write("overwriting\n")
            os.rename(oldfile,newfile)
        else:
            sys.stderr.write("skipping\n")
            return False
        #end if force
    else:
        os.rename(oldfile,newfile)
        return True
    #end if new_exists
    
 
def main():
    parser = OptionParser()
 
    parser.add_option("-d","--debug",action="store_true",default=False,dest="debug", help="Show debugging info")
    parser.add_option("-b","--batch",action="store_false",dest="interactive",help="Selects first search result, requires no human intervention once launched",default=False)
    parser.add_option("-i","--interactive",action="store_true",dest="interactive",help="Interactivly prompt for show",default=True)
    parser.add_option("-f","--force",action="store_true",default=False,dest="force",help="Skips checking if new filename already exists before renaming")
    
    opts,args = parser.parse_args()
    if len(args) == 0:
        parser.error("No filenames or directories supplied")
    #end if len(args)
    
    allFiles = findFiles(args)
    validFiles = processNames(allFiles)
    
    if len(validFiles) == 0:
        sys.stderr.write("No valid files found")
        sys.exit(1)
    
    print "Starting tvdb"
    t = tvdb(debug = opts.debug, interactive = opts.interactive)
    print "Done"
    print "Starting processing files"
    always=False
    for cfile in validFiles:
        epname = t[ cfile['showname'] ][ cfile['seasno'] ][ cfile['epno'] ]['name']
        if epname:
            cfile['epname'] = epname
        else:
            cfile['epname'] = None
            sys.stderr.write("Episode name not found for %s" % ( cfile['fullpath']) )
        #end if epname
 
        newname = formatName( cfile )
        newname = cleanName(newname)
        oldfile = os.path.join( cfile['filepath'],cfile['filename'] + cfile['ext'] )
        newfile = os.path.join( cfile['filepath'],newname )
        print "*"*20
        print "Old name: %s" % ( cfile['filename'] + cfile['ext'] )
        print "New name: %s" % ( newname )
        
        if always:
            rename_result = renameFile(oldfile,newname,force=opts.force)
            if rename_result:
                print "..auto-renaming"
            else:
                print "..not renamed"
            continue # next filename!
        #end if always
        
        print "Rename?"
        print "(y/n/a/q)",
        ans=raw_input()
        if ans[0] == "a":
            always=True
            rename_result = renameFile(oldfile,newname,force=opts.force)
        elif ans[0] == "q":
            print "Aborting"
            break
        elif ans[0] == "y":
            rename_result = renameFile(oldfile,newname,force=opts.force)
        elif ans[0] == "n":
            print "Skipping"
            continue
        #end if ans
        if rename_result:
            print "..renamed"
        else:
            print "..not renamed"
        #end if rename_result
    #end for cfile
#end main
 
if __name__ == "__main__":
    main()
 
def prompts():
    if always:
        print "Processing",x
    else:
        ans=raw_input()
        if ans=="a":
            always=True
            print "processing",x
        elif ans=="q":
            pass#break
        elif ans=="n":
            print "Skipping",x
            pass#continue
        elif ans=="y":
            print "processing",x
        #end if ans
    #end if always