klemek/meme-otron
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

image data from message attachment

Browse Source
This commit is contained in:
klemek
2020-04-29 12:31:48 +02:00
parent f83ae349d6
commit f99bfff3ed
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
discord_bot/__main__.py
+9 -3
View File
@@ -116,7 +116,12 @@ async def on_message(message: discord.Message):
if len(args) > 1 and message.author.display_name is not None: if len(args) > 1 and message.author.display_name is not None:
left_wmark_text = f"By {message.author.display_name}" left_wmark_text = f"By {message.author.display_name}"
logging.info(args[0]) logging.info(args[0])
img, errors = meme_otron.compute(*args, left_wmark_text=left_wmark_text)
input_data = None
if len(message.attachments) > 0:
input_data = await message.attachments[0].read()
img, errors = meme_otron.compute(*args, left_wmark_text=left_wmark_text, input_data=input_data)
if len(errors) > 0: if len(errors) > 0:
response = ":warning:" response = ":warning:"
for err in errors: for err in errors:
@@ -131,8 +136,9 @@ async def on_message(message: discord.Message):
with tempfile.NamedTemporaryFile(delete=False) as output: with tempfile.NamedTemporaryFile(delete=False) as output:
img.save(output, format="JPEG") img.save(output, format="JPEG")
response = None response = None
if len(args) == 1: meme_id = utils.sanitize_input(args[0])
meme = meme_db.get_meme(utils.sanitize_input(args[0])) if len(args) == 1 and meme_id not in ["image", "text"]:
meme = meme_db.get_meme(meme_id)
response = f"Template `{meme.id}`:" response = f"Template `{meme.id}`:"
if len(meme.aliases) > 0: if len(meme.aliases) > 0:
response += f"\n- Aliases: `{'`, `'.join(meme.aliases)}`" response += f"\n- Aliases: `{'`, `'.join(meme.aliases)}`"