diff --git a/src/memes.py b/src/memes.py index 8c5f45c..e0413b0 100644 --- a/src/memes.py +++ b/src/memes.py @@ -197,6 +197,18 @@ def getRandomMeme(): recentMemes.insertMeme(choice) return choice +# Build an underscored name from a string +def nameBuilder(command): + uploaded_file_name = "" + for word in command: + uploaded_file_name += word + "_" + + # remove the last _ from the name + uploaded_file_name = uploaded_file_name.rstrip(uploaded_file_name[-1]) + + # append the file extension + return uploaded_file_name + ############################################################################### # Calldict methods @@ -282,18 +294,11 @@ async def uploadMeme(command, message, client): file_names = [] for file in message.attachments: - # if we want to override the name uploaded_file_name = file.filename if len(command) > 1: - uploaded_file_name = "" - for word in command[1:]: - uploaded_file_name += word + "_" - - # remove the last _ from the name - uploaded_file_name = uploaded_file_name.rstrip(uploaded_file_name[-1]) - - # append the file extension + uploaded_file_name = nameBuilder(command[1:]) + # add the extension name, ext = os.path.splitext(file.filename) uploaded_file_name += ext