From 73a783cd6b2b9f02d7071947e70da7eb5bc7d2b8 Mon Sep 17 00:00:00 2001 From: Josh Date: Fri, 15 Jan 2021 20:34:05 +1000 Subject: [PATCH] Strip both - and _ from newcog class names --- discord/__main__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/discord/__main__.py b/discord/__main__.py index 47504b7f9..02803cf07 100644 --- a/discord/__main__.py +++ b/discord/__main__.py @@ -250,8 +250,9 @@ def newcog(parser, args): name = args.class_name else: name = str(directory.stem) - if '-' in name: - name = name.replace('-', ' ').title().replace(' ', '') + if '-' in name or '_' in name: + translation = str.maketrans('-_', ' ') + name = name.translate(translation).title().replace(' ', '') else: name = name.title()