diff --git a/LogBot/logbot.py b/LogBot/logbot.py index fd87ece..ecd1d03 100644 --- a/LogBot/logbot.py +++ b/LogBot/logbot.py @@ -92,33 +92,13 @@ def download(self, message, logger): # define media_post media_path = os.path.join(media_type, filename) - if message.url.lower().endswith( - (".jpg", "jpeg", "png", ".gif", ".bmp", ".svg", "eps") - ): + if message.url.lower().endswith(IMAGE_TYPES): media_post = f'' - elif message.url.lower().endswith(".pdf"): + elif message.url.lower().endswith(FILE_TYPES): media_post = f'' - elif message.url.lower().endswith( - (".mp3", ".ogg", ".oga", ".mogg", ".wav", ".m4a", ".webm") - ): + elif message.url.lower().endswith(AUDIO_TYPES): media_post = f'' - elif message.url.lower().endswith( - ( - ".mp4", - ".webm", - ".flv", - ".vob", - ".avi", - ".mov", - ".qt", - ".mpg", - ".mpeg", - ".mp4", - ".m2v", - ".mpe", - ".3gp", - ) - ): + elif message.url.lower().endswith(VIDEO_TYPES): media_post = f'' else: media_post = None