mirror of
https://github.com/nikdoof/tvorganise.git
synced 2025-12-13 06:42:16 +00:00
Further PEP8 work.
This commit is contained in:
272
tvorganise.py
272
tvorganise.py
@@ -19,172 +19,174 @@ import logging
|
||||
|
||||
class TvOrganiser():
|
||||
|
||||
_config = {}
|
||||
_config = {}
|
||||
|
||||
def __init__(self):
|
||||
pass
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
@property
|
||||
def _logger(self):
|
||||
if not hasattr(self, "__logger"):
|
||||
self.__logger = logging.getLogger(self.__class__.__name__)
|
||||
self.__logger.addHandler(logging.StreamHandler())
|
||||
return self.__logger
|
||||
@property
|
||||
def _logger(self):
|
||||
if not hasattr(self, "__logger"):
|
||||
self.__logger = logging.getLogger(self.__class__.__name__)
|
||||
self.__logger.addHandler(logging.StreamHandler())
|
||||
return self.__logger
|
||||
|
||||
def _getConfig(self, file):
|
||||
def _getConfig(self, file):
|
||||
|
||||
config = {}
|
||||
config = {}
|
||||
|
||||
configpsr = ConfigParser.RawConfigParser()
|
||||
configpsr.read('tvorganise.cfg')
|
||||
configpsr = ConfigParser.RawConfigParser()
|
||||
configpsr.read('tvorganise.cfg')
|
||||
|
||||
if configpsr.has_section('main'):
|
||||
for k, v in configpsr.items('main'):
|
||||
config[k] = v
|
||||
if configpsr.has_section('main'):
|
||||
for k, v in configpsr.items('main'):
|
||||
config[k] = v
|
||||
|
||||
if configpsr.has_section('regex'):
|
||||
if configpsr.has_section('regex'):
|
||||
|
||||
regex_config = {}
|
||||
regex = []
|
||||
regex_config = {}
|
||||
regex = []
|
||||
|
||||
# Load in subs before reading in the regex
|
||||
for k, v in configpsr.items('regex'):
|
||||
if k[:5] != 'regex':
|
||||
regex_config[k] = v
|
||||
# Load in subs before reading in the regex
|
||||
for k, v in configpsr.items('regex'):
|
||||
if k[:5] != 'regex':
|
||||
regex_config[k] = v
|
||||
|
||||
for k, v in configpsr.items('regex'):
|
||||
if k[:5] == 'regex':
|
||||
regex.append(re.compile(v % regex_config))
|
||||
for k, v in configpsr.items('regex'):
|
||||
if k[:5] == 'regex':
|
||||
regex.append(re.compile(v % regex_config))
|
||||
|
||||
config['regex'] = regex
|
||||
config['regex'] = regex
|
||||
|
||||
self._config = config
|
||||
return config
|
||||
self._config = config
|
||||
return config
|
||||
|
||||
def _findFiles(self, 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)
|
||||
elif os.path.isfile(cfile):
|
||||
allfiles.append(cfile)
|
||||
return allfiles
|
||||
def _findFiles(self, 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)
|
||||
elif os.path.isfile(cfile):
|
||||
allfiles.append(cfile)
|
||||
return allfiles
|
||||
|
||||
def processNames(self, names, verbose=False):
|
||||
"""
|
||||
Takes list of names, runs them though the regexs
|
||||
"""
|
||||
allEps = []
|
||||
for f in names:
|
||||
filepath, filename = os.path.split(f)
|
||||
filename, ext = os.path.splitext(filename)
|
||||
def processNames(self, names, verbose=False):
|
||||
"""
|
||||
Takes list of names, runs them though the 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)
|
||||
# Remove leading . from extension
|
||||
ext = ext.replace(".", "", 1)
|
||||
|
||||
for r in config['regex']:
|
||||
match = r.match(filename)
|
||||
if match:
|
||||
showname, seasno, epno, epname = match.groups()
|
||||
for r in config['regex']:
|
||||
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()
|
||||
#remove ._- characters from name
|
||||
showname = re.sub("[\._]|\-(?=$)", " ", showname).strip()
|
||||
|
||||
seasno, epno = int(seasno), int(epno)
|
||||
seasno, epno = int(seasno), int(epno)
|
||||
|
||||
self._logger.debug("File:", filename)
|
||||
self._logger.debug("Pattern:", r.pattern)
|
||||
self._logger.debug("Showname:", showname)
|
||||
self._logger.debug("Seas:", seasno)
|
||||
self._logger.debug("Ep:", epno)
|
||||
self._logger.debug("File:", filename)
|
||||
self._logger.debug("Pattern:", r.pattern)
|
||||
self._logger.debug("Showname:", showname)
|
||||
self._logger.debug("Seas:", seasno)
|
||||
self._logger.debug("Ep:", epno)
|
||||
|
||||
allEps.append({'file_showname': showname,
|
||||
'seasno': seasno,
|
||||
'epno': epno,
|
||||
'filepath': filepath,
|
||||
'filename': filename,
|
||||
'ext': ext})
|
||||
break # Matched - to the next file!
|
||||
else:
|
||||
self._logger.warning("Invalid name: %s" % (f))
|
||||
allEps.append({'file_showname': showname,
|
||||
'seasno': seasno,
|
||||
'epno': epno,
|
||||
'filepath': filepath,
|
||||
'filename': filename,
|
||||
'ext': ext})
|
||||
break # Matched - to the next file!
|
||||
else:
|
||||
self._logger.warning("Invalid name: %s" % (f))
|
||||
|
||||
return allEps
|
||||
return allEps
|
||||
|
||||
def _same_partition(f1, f2):
|
||||
return os.stat(f1).st_dev == os.stat(f2).st_dev
|
||||
def _same_partition(f1, f2):
|
||||
return os.stat(f1).st_dev == os.stat(f2).st_dev
|
||||
|
||||
###########################
|
||||
###########################
|
||||
|
||||
def main(self):
|
||||
parser = OptionParser(usage="%prog [options] <file or directories>")
|
||||
parser.add_option("-a", "--always", dest = "always",
|
||||
action = "store_true", default = False,
|
||||
help = "Do not ask for confirmation before copying")
|
||||
parser.add_option("-q", "--quiet", dest = "quiet",
|
||||
action = "store_true", default = False,
|
||||
help = "Silence output")
|
||||
parser.add_option("-c", "--config", dest = "config",
|
||||
action = "store", default = "tvorganise.cfg",
|
||||
help = "Use a custom configuration file")
|
||||
parser.add_option("-v", "", dest = "verbose",
|
||||
action = "store_true", default = False,
|
||||
help = "Verbose output")
|
||||
def main(self):
|
||||
parser = OptionParser(usage="%prog [options] <file or directories>")
|
||||
parser.add_option("-a", "--always", dest="always",
|
||||
action="store_true", default=False,
|
||||
help="Do not ask for confirmation before copying")
|
||||
parser.add_option("-q", "--quiet", dest="quiet",
|
||||
action="store_true", default=False,
|
||||
help="Silence output")
|
||||
parser.add_option("-c", "--config", dest="config",
|
||||
action="store", default="tvorganise.cfg",
|
||||
help="Use a custom configuration file")
|
||||
parser.add_option("-v", "", dest="verbose",
|
||||
action="store_true", default=False,
|
||||
help="Verbose output")
|
||||
|
||||
opts, args = parser.parse_args()
|
||||
opts, args = parser.parse_args()
|
||||
|
||||
if os.path.exists(opts.config):
|
||||
config = self._getConfig(opts.config)
|
||||
else:
|
||||
self._logger.error('Unable to find configuration file!')
|
||||
sys.exit(1)
|
||||
if os.path.exists(opts.config):
|
||||
config = self._getConfig(opts.config)
|
||||
else:
|
||||
self._logger.error('Unable to find configuration file!')
|
||||
sys.exit(1)
|
||||
|
||||
files = self._findFiles(args)
|
||||
files = self.processNames(files, opts.verbose)
|
||||
files = self._findFiles(args)
|
||||
files = self.processNames(files, opts.verbose)
|
||||
|
||||
# Warn if no files are found, then exit
|
||||
if len(files) == 0:
|
||||
self._logger.error('No files found')
|
||||
sys.exit(0)
|
||||
# Warn if no files are found, then exit
|
||||
if len(files) == 0:
|
||||
self._logger.error('No files found')
|
||||
sys.exit(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']
|
||||
for name in files:
|
||||
filename = "%s.%s" % (name['filename'], name['ext'])
|
||||
oldfile = os.path.join(name['filepath'], filename)
|
||||
newpath = config['target_path'] % name
|
||||
newfile = os.path.join(newpath, filename)
|
||||
|
||||
self._logger.info("Old path:", oldfile)
|
||||
self._logger.info("New path:", newfile)
|
||||
self._logger.info("Old path:", oldfile)
|
||||
self._logger.info("New path:", newfile)
|
||||
|
||||
if opts.always:
|
||||
if not os.path.exists(newpath):
|
||||
os.mkdirs(newpath)
|
||||
if os.path.exists(newfile):
|
||||
self._logger.warning("[!] File already exists, not copying")
|
||||
else:
|
||||
if self._same_partition(oldfile, newpath):
|
||||
self._logger.info("[*] Moving file")
|
||||
try:
|
||||
os.rename(oldfile, newfile)
|
||||
except Exception, errormsg:
|
||||
self._logger.error("[!] Error moving file! %s" % (errormsg))
|
||||
else:
|
||||
self._logger.info("[*] Copying file")
|
||||
try:
|
||||
shutil.copy(oldfile, newfile)
|
||||
except Exception, errormsg:
|
||||
self._logger.error("[!] Error copying file! %s" % (errormsg))
|
||||
else:
|
||||
self._logger.info("[*] ..done")
|
||||
else:
|
||||
self._logger.warning("Skipping file")
|
||||
if opts.always:
|
||||
if not os.path.exists(newpath):
|
||||
os.mkdirs(newpath)
|
||||
if os.path.exists(newfile):
|
||||
self._logger.warning("[!] File already exists, not copying")
|
||||
else:
|
||||
if self._same_partition(oldfile, newpath):
|
||||
self._logger.info("[*] Moving file")
|
||||
try:
|
||||
os.rename(oldfile, newfile)
|
||||
except Exception, errormsg:
|
||||
self._logger.error("[!] Error moving file! %s" % (errormsg))
|
||||
else:
|
||||
self._logger.info("[*] Copying file")
|
||||
try:
|
||||
shutil.copy(oldfile, newfile)
|
||||
except Exception, errormsg:
|
||||
self._logger.error("[!] Error copying file! %s" % (errormsg))
|
||||
else:
|
||||
self._logger.info("[*] ..done")
|
||||
else:
|
||||
self._logger.warning("Skipping file")
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
t = TvOrganiser()
|
||||
t.main()
|
||||
t = TvOrganiser()
|
||||
t.main()
|
||||
|
||||
Reference in New Issue
Block a user