Re-use types here too

This commit is contained in:
Luke Murphy 2021-02-27 11:10:45 +01:00
parent a38515ab10
commit 86c8495e88
No known key found for this signature in database
GPG Key ID: 5E2EF5A63E3718CC

View File

@ -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'<img src="{ media_path }">'
elif message.url.lower().endswith(".pdf"):
elif message.url.lower().endswith(FILE_TYPES):
media_post = f'<iframe src="{ media_path }" width="800" height="1000"></iframe>'
elif message.url.lower().endswith(
(".mp3", ".ogg", ".oga", ".mogg", ".wav", ".m4a", ".webm")
):
elif message.url.lower().endswith(AUDIO_TYPES):
media_post = f'<audio controls src="{ media_path }"></audio>'
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'<video controls src="{ media_path }"></video>'
else:
media_post = None