mrtoth/normalisename
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add option to append special characters to whitelist

Browse Source
This commit is contained in:
Kjistóf 2017-05-28 19:55:21 +02:00 committed by Kristóf Tóth
parent 0da93de25a
commit e20f442016
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
normalisename.py
View File

@ -5,18 +5,26 @@ from os.path import basename, dirname
from os.path import join as joinpath from os.path import join as joinpath
from argparse import ArgumentParser from argparse import ArgumentParser
parser = ArgumentParser(description='Eliminate funky stuff from filenames.') parser = ArgumentParser(description='Eliminate funky stuff from filenames.')
parser.add_argument('--separator', '-s', type=str, nargs='?', default='_', parser.add_argument('--separator', '-s', type=str, nargs='?', default='_',
help='Set separator to use (to replace spaces with).') 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('--whitelist', '-w', type=str,
help='Overwrite default whitelist (format: whitespace separated string).')
parser.add_argument('files', type=str, nargs='+', parser.add_argument('files', type=str, nargs='+',
help='File(s) to normalise the name of (relative or absolute path).') help='File(s) to normalise the name of (relative or absolute path).')
args = parser.parse_args() args = parser.parse_args()
# special characters that will not be removed # special characters that will not be removed
whitelist = {' ', '.', '-'} whitelist = {' ', '.', '-'}
if args.whitelist: whitelist = set(args.whitelist.split())
whitelist.add(args.separator) whitelist.add(args.separator)
if args.allow: whitelist = whitelist.union(set(args.allow))
for path in args.files: for path in args.files:
dir = dirname(path) dir = dirname(path)