nikdoof/tvorganise
1
0
Fork 0
mirror of https://github.com/nikdoof/tvorganise.git synced 2025-12-13 06:42:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Removed checkTvEps

Browse Source
This commit is contained in:
dbr
2008-10-20 19:09:50 +10:30
parent 40cf5b8d4f
commit 2f6c3d0e5d
1 changed files with 0 additions and 218 deletions
Download Patch File Download Diff File Expand all files Collapse all files

218
checkTvEps.py
View File

@@ -1,218 +0,0 @@
#!/usr/bin/env python
#encoding:utf-8
import os,re,sys
###################################
# Colourized output helper
###################################
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
###################################
# Valid-name regex.
###################################
# Should return 4 groups:
# Series name.
# Season number.
# Episode number#
# Episode name. (can be empty)
#
# Valid episode names:
# Show name - [01x01-02] - The Episode Name (Part 1)
# Show name - [01x23] - The Episode Name (Part 1)
# Show name - [01x23] - The Episode Name
# Show name - [04x01]
tv_ep_name = [
re.compile("([-\w ]+) - \[(\d{2})x(\d{2}|\d{2}-\d{2})\](?= - ([\w\(\) ]+))?"),
]
###################################
# Regex to match valid, but not-to-be-processed files (dot-files, folder.jpg artwork)
###################################
decrappify = [
re.compile("(?=^[.]{1}.*)"),
re.compile("folder.jpg"),
]
# Location to process
loc = "." # Runs from the current path
###################################
# Find all valid files
###################################
#TODO: Use os.path.walk() so no external dependancies!
#TODO: Array of valid extensions (not just .avi)
import path
d = path.path(loc)
allfiles=[]
for f in d.walkfiles("*"):
allfiles.append( str(f) )
#end for f
files = [x for x in allfiles if os.path.isfile(x)] # only get files, not folders
# Strip out dotfiles/folder.jpg
for current_file in allfiles:
current_file_path,current_file_name = os.path.split(current_file)
for cur_decrap in decrappify:
if cur_decrap.match(current_file_name):
files.remove(current_file)
#end for file
#files = [os.path.join(loc,x) for x in files] # append path to file name
# Warn if no files are found, then exit
if files.__len__() == 0:
print colour('No files found','red')
sys.exit(0)
###################################
# Validate filenames
###################################
valid = []
invalid = []
for cur in files:
for cur_checker in tv_ep_name:
check = cur_checker.findall(cur)
if check:
# Valid file name
valid.append(check)
break # Found valid episode, skip to the next one
#end if
else:
# No match found (Invalid name)
invalid.append(cur)
#end for cur_checker
#end for
###################################
# Show invalid names
###################################
if invalid.__len__() > 0:
print colour('WARNING','red')
print '[!] Invalid file-names found:'
print '\n'.join(invalid)
print "-" * 28
print
###################################
# Show valid names
###################################
if valid.__len__() > 0:
print '[?] Valid file-names found:'
allepisodes = {}
# Make a dict of shows, each show contains an array of episode-info.
# Example:
# { 'showname': [ [1,24,'Title'], ],
# 'another': [ [4,12,'More'], [4,13,'Stuff'], ]
# }
# This could(/should) change into something more managable.
for x in valid:
showname,season,epno,title = x[0] # Put regex-groups into named vars
if allepisodes.has_key(showname): # Does the show exist in allepisodes?
allepisodes[showname].append([season,epno,title]) # yes, append episode
else:
allepisodes[showname] = [[season,epno,title]] # no, init episodes array
#end for x
# Now we have a nice dict of all shows, output them as:
# Showname
# Season 01
# Episodes 01, 02, 03
# Season 01
# Episodes 01, 02
for showname,alleps in allepisodes.items():
print colour(showname,'green')
# Make a dict for the current show,
# an index for each season, containing an array of episodes.
# Example:
# { '02' : [20,21,22,23],
# '03' : [01,02,03]
# }
# This is for ease-of-output. It'd be possible to work
# directly of the allepisodes dict, but a lot messier..
curshow={}
for curep in alleps:
season,epno,title = curep
try:
season = int(season)
except ValueError:
print "Error: Season number not integer. Check naming or tv_ep_name regex."
#end try
try:
epno = int(epno)
except:
# deal with episode numbered like - [01x01-02]
# TODO: Currently very hackish, improve the handling of [01x01-02] names
split_two_eps = epno.split("-")
if len(split_two_eps) == 2:
p1,p2 = split_two_eps
epno,ep2 = int(p1),int(p2)
# Add the second part, then add the first as usual. Messy, but works
if curshow.has_key(season):
curshow[season].append(ep2)
else:
curshow[season]=[ep2]
#end if curshow
else:
print "Error: Episode number not integer, or two integers (01-02)"
#end try
if curshow.has_key(season):
curshow[season].append(epno)
else:
curshow[season]=[epno]
#end for curep
for seas,eps in curshow.items():
# Check for missing episodes (by looking for gaps in sequence)
mi,ma=int(min(eps)),int(max(eps))
missing_eps=[]
for x in range(1,int(ma)+1):
try:
if not x in [int(y) for y in eps]:
missing_eps.append(x)
except ValueError:
pass # start value is 01-02 probably.
#end try
#end for x
# if we are missing an episode, "season 01" title becomes red.
if missing_eps.__len__() == 0:
cur_colour = "blue"
else:
cur_colour = "red"
# Print season title in above colour.
print "\t",colour("Season " + str(seas),colour=cur_colour)
# And list episodes
print "\t\t", "Episodes:", ", ".join([str(e) for e in eps])
# And if there is missing episodes, list them.
if missing_eps.__len__() != 0:
print "\t\t","Missing:", ", ".join([str(m) for m in missing_eps])
#end for seas,eps