mirror of
https://github.com/nikdoof/tvorganise.git
synced 2025-12-13 06:42:16 +00:00
Added script to automatically move properly named files into the correct location
This commit is contained in:
dbr
209
autoPathTv.py
Normal file
209
autoPathTv.py
Normal file
@@ -0,0 +1,209 @@
|
||||
#!/usr/bin/env python
|
||||
#encoding:utf-8
|
||||
"""
|
||||
autoPathTv.py
|
||||
Use at your own risk. "It works for me".
|
||||
|
||||
This parses "Show Name - [01x23] - Episode Name.avi"
|
||||
filenames and automatically copies them to
|
||||
/Volumes/aodDrive/video/tv/Show Name/season 1/
|
||||
|
||||
(obviously the path is changable, as is in the input
|
||||
format if you change the reges)
|
||||
"""
|
||||
import os, sys, re
|
||||
from optparse import OptionParser
|
||||
import shutil
|
||||
|
||||
|
||||
config = {}
|
||||
regex_config={}
|
||||
|
||||
##############################################
|
||||
# Path configs
|
||||
|
||||
# Where to move the files
|
||||
config['target_path'] = "/Volumes/aodDrive/video/tv/%(file_showname)s/season %(seasno)s/"
|
||||
|
||||
|
||||
##############################################
|
||||
# Regex configs
|
||||
|
||||
# Character class for valid episode/show names.
|
||||
# Example: [a-zA-Z0-9\-'\ ]
|
||||
regex_config['valid_in_names'] = "[\w\(\).,\[\]'\ \-?]"
|
||||
|
||||
config['name_parse'] = [
|
||||
re.compile("^(%(valid_in_names)s+) - \[(\d{2})x(\d{2})\] - (%(valid_in_names)s+)$" % (regex_config)),
|
||||
re.compile("^(%(valid_in_names)s+) - \[(\d{2})x(\d{2}-\d{2})\] - (%(valid_in_names)s+)$" % (regex_config)),
|
||||
re.compile("^(%(valid_in_names)s+) - \[(\d{2})x(Special\d{1,2})\] - (%(valid_in_names)s+)$" % (regex_config)),
|
||||
re.compile("^(%(valid_in_names)s+) - \[(\d{2})xExtra(\d{1,2})\] - (%(valid_in_names)s+)$" % (regex_config)),
|
||||
re.compile("^(%(valid_in_names)s+) - \[(\d{2})] - (%(valid_in_names)s+)$" % (regex_config)),
|
||||
]
|
||||
# end configs
|
||||
##############################################
|
||||
|
||||
def colour(text,colour="red"):
|
||||
nocolour=False
|
||||
if nocolour: # Colour no supported, return plain text
|
||||
return text
|
||||
#end if
|
||||
|
||||
c = {'red':'[31m',
|
||||
'green':'[32m',
|
||||
'blue':'[34m',
|
||||
}
|
||||
CLR=chr(27)+'[0m'
|
||||
if not colour in c.keys():
|
||||
raise ValueError("Invalid colour")
|
||||
else:
|
||||
return chr(27)+c[colour] + text + CLR
|
||||
#end if
|
||||
#end colour
|
||||
|
||||
def findFiles(args):
|
||||
"""
|
||||
Takes a list of files/folders, grabs files inside them. Does not recurse
|
||||
more than one level (if a folder is supplied, it will list files within)
|
||||
"""
|
||||
allfiles = []
|
||||
for cfile in args:
|
||||
if os.path.isdir(cfile):
|
||||
for sf in os.listdir(cfile):
|
||||
newpath = os.path.join(cfile, sf)
|
||||
if os.path.isfile(newpath):
|
||||
allfiles.append(newpath)
|
||||
#end if isfile
|
||||
#end for sf
|
||||
elif os.path.isfile(cfile):
|
||||
allfiles.append(cfile)
|
||||
#end if isdir
|
||||
#end for cfile
|
||||
return allfiles
|
||||
#end findFiles
|
||||
|
||||
def processNames(names, verbose=False):
|
||||
"""
|
||||
Takes list of names, runs them though the config['name_parse'] regexs
|
||||
"""
|
||||
allEps = []
|
||||
for f in names:
|
||||
filepath, filename = os.path.split( f )
|
||||
filename, ext = os.path.splitext( filename )
|
||||
|
||||
# Remove leading . from extension
|
||||
ext = ext.replace(".", "", 1)
|
||||
|
||||
for r in config['name_parse']:
|
||||
match = r.match(filename)
|
||||
if match:
|
||||
showname, seasno, epno, epname = match.groups()
|
||||
|
||||
#remove ._- characters from name (- removed only if next to end of line)
|
||||
showname = re.sub("[\._]|\-(?=$)", " ", showname).strip()
|
||||
|
||||
seasno, epno = int(seasno), int(epno)
|
||||
|
||||
if verbose:
|
||||
print "*"*20
|
||||
print "File:", filename
|
||||
print "Pattern:", r.pattern
|
||||
print "Showname:", showname
|
||||
print "Seas:", seasno
|
||||
print "Ep:", epno
|
||||
print "*"*20
|
||||
|
||||
allEps.append({ 'file_showname':showname,
|
||||
'seasno':seasno,
|
||||
'epno':epno,
|
||||
'filepath':filepath,
|
||||
'filename':filename,
|
||||
'ext':ext
|
||||
})
|
||||
break # Matched - to the next file!
|
||||
else:
|
||||
print "Invalid name: %s" % (f)
|
||||
#end for r
|
||||
#end for f
|
||||
|
||||
return allEps
|
||||
#end processNames
|
||||
|
||||
def confirm(question="Rename files?"):
|
||||
ans = None
|
||||
while ans not in ["q", "quit"]:
|
||||
print "y/n/q?",
|
||||
ans = raw_input()
|
||||
if ans.lower() in ["y", "yes"]:
|
||||
return True
|
||||
elif ans.lower() in ["n", "no"]:
|
||||
return False
|
||||
else:
|
||||
sys.exit(1)
|
||||
#end confirm
|
||||
|
||||
def make_path(path):
|
||||
try:
|
||||
os.makedirs(path)
|
||||
except OSError:
|
||||
print "Couldn't make path"
|
||||
#end make_path
|
||||
|
||||
def does_file_exist(path):
|
||||
try:
|
||||
os.stat(path)
|
||||
except OSError:
|
||||
file_exists = False
|
||||
else:
|
||||
file_exists = True
|
||||
return file_exists
|
||||
|
||||
|
||||
###########################
|
||||
|
||||
|
||||
def main():
|
||||
|
||||
parser = OptionParser(usage="%prog [options] <file or directories>")
|
||||
opts, args = parser.parse_args()
|
||||
|
||||
files = findFiles(args)
|
||||
files = processNames(files)
|
||||
|
||||
# Warn if no files are found, then exit
|
||||
if files.__len__() == 0:
|
||||
print colour('No files found','red')
|
||||
sys.exit(1)
|
||||
#end if files == 0
|
||||
|
||||
for name in files:
|
||||
oldfile = os.path.join(name['filepath'], name['filename']) + "." + name['ext']
|
||||
newpath = config['target_path'] % name
|
||||
newfile = os.path.join(newpath, name['filename']) + "." + name['ext']
|
||||
|
||||
print "Old path:", oldfile
|
||||
print "New path:", newfile
|
||||
ans=confirm()
|
||||
if ans:
|
||||
make_path(newpath)
|
||||
file_exists = does_file_exist(newfile)
|
||||
if file_exists:
|
||||
print "[!] File already exists, not copying"
|
||||
else:
|
||||
print "[*] Copying file"
|
||||
try:
|
||||
shutil.copy(oldfile, newpath)
|
||||
except Exception, errormsg:
|
||||
print "[!] Error copying file! %s" % (errormsg)
|
||||
else:
|
||||
print "[*] ..done"
|
||||
#end try
|
||||
#end if not file_exists
|
||||
else:
|
||||
print "Skipping file"
|
||||
#end if ans
|
||||
#end for name in files
|
||||
#end main
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user