74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
import sys
|
||
|
|
from os import rename
|
||
|
|
from argparse import ArgumentParser
|
||
|
|
|
||
|
|
from normalisename import Normalisename
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
# parse arguments
|
||
|
|
parser = ArgumentParser(description='Eliminate funky stuff from filenames.')
|
||
|
|
parser.add_argument(
|
||
|
|
'--separator', '-s', type=str, nargs='?', default='_',
|
||
|
|
help='Set separator to use (to replace spaces with).'
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--allow', '-a', type=str, nargs='?', action='append',
|
||
|
|
help='Specify a special character to allow (it will not be removed).'
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--disallow', '-d', type=str, nargs='?', action='append',
|
||
|
|
help='Specify a special character to disallow (it will removed).'
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'--whitelist', '-w', type=str,
|
||
|
|
help='Overwrite default whitelist (format: whitespace separated string).'
|
||
|
|
)
|
||
|
|
|
||
|
|
mgroup = parser.add_mutually_exclusive_group()
|
||
|
|
mgroup.add_argument(
|
||
|
|
'--dryrun', '-n', action='store_true',
|
||
|
|
help='Do not touch anything, just tell me what would change!'
|
||
|
|
)
|
||
|
|
mgroup.add_argument(
|
||
|
|
'--funky', '-f', action='store_true',
|
||
|
|
help='List filenames with special characters'
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'files', type=str, nargs='+',
|
||
|
|
help='File(s) to normalise the name of (relative or absolute path).'
|
||
|
|
)
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
# verify arguments
|
||
|
|
if args.disallow:
|
||
|
|
if args.separator in args.disallow:
|
||
|
|
sys.exit('Disallowing your chosen separator makes no sense!')
|
||
|
|
|
||
|
|
# declare special characters that will not be removed (spaces are handled elsewhere)
|
||
|
|
whitelist = {' ', '.', '-'}
|
||
|
|
if args.whitelist:
|
||
|
|
whitelist = set(args.whitelist.split())
|
||
|
|
|
||
|
|
# modify whitelist based on arguments
|
||
|
|
whitelist.add(args.separator)
|
||
|
|
if args.allow:
|
||
|
|
whitelist = whitelist.union(set(args.allow))
|
||
|
|
if args.disallow:
|
||
|
|
whitelist = whitelist.difference(set(args.disallow))
|
||
|
|
|
||
|
|
# define operations
|
||
|
|
listchanges = lambda a, b: print('{}{} --> {}'.format(a, ' '*(60-len(a)), b))
|
||
|
|
listfunky = lambda a, b: print(a)
|
||
|
|
|
||
|
|
operation = rename
|
||
|
|
if args.dryrun:
|
||
|
|
operation = listchanges
|
||
|
|
elif args.funky:
|
||
|
|
operation = listfunky
|
||
|
|
|
||
|
|
# do what needs to be done
|
||
|
|
Normalisename(operation, args.separator, whitelist)(args.files)
|