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