Strip both - and _ from newcog class names

This commit is contained in:
Josh 2021-01-15 20:34:05 +10:00 committed by GitHub
parent 69bdc3a184
commit 73a783cd6b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -250,8 +250,9 @@ def newcog(parser, args):
name = args.class_name name = args.class_name
else: else:
name = str(directory.stem) name = str(directory.stem)
if '-' in name: if '-' in name or '_' in name:
name = name.replace('-', ' ').title().replace(' ', '') translation = str.maketrans('-_', ' ')
name = name.translate(translation).title().replace(' ', '')
else: else:
name = name.title() name = name.title()