rotated text

2020-04-11 21:00:21 +02:00
parent 6a4acebef1
commit 6b0d4cdca0
5 changed files with 111 additions and 29 deletions
@@ -0,0 +1,123 @@
+import os
+import traceback
+import logging
+import discord
+import re
+import tempfile
+from datetime import datetime
+from dotenv import load_dotenv
+
+from meme_otron import img_factory as imgf
+from meme_otron import meme_db as db
+from meme_otron import utils
+from meme_otron import main
+
+VERSION = "1.0-dev"
+t0 = datetime.now()
+logging.basicConfig(format="[%(asctime)s][%(levelname)s][%(module)s] %(message)s", level=logging.INFO)
+
+imgf.load_fonts()
+db.load_memes()
+
+# Loading token
+load_dotenv()
+token = os.getenv('DISCORD_TOKEN')
+
+client = discord.Client()
+
+
+def debug(message, txt):
+    """
+    Print a log with the context of the current event
+
+    :param (discord.Message) message: message that triggered the event
+    :param (str) txt: text of the log
+    """
+    logging.info(f"{message.guild} > #{message.channel}: {txt}")
+
+
+@client.event
+async def on_ready():
+    """
+    Called when client is connected
+    """
+    # Change status
+    await client.change_presence(
+        activity=discord.Game(f"v{VERSION}"),
+        status=discord.Status.online
+    )
+    # Debug connected guilds
+    logging.info(f'{client.user} v{VERSION} has connected to Discord\nto the following guilds:')
+    for guild in client.guilds:
+        logging.info(f'- {guild.name}(id: {guild.id})')
+
+
+@client.event
+async def on_message(message):
+    """
+    Called when a message is sent to any channel on any guild
+
+    :param (discord.Message) message: message sent
+    """
+    # Ignore self messages
+    if message.author == client.user:
+        return
+    direct = message.channel.type == discord.ChannelType.private
+    if direct or client.user in message.mentions:
+        message.content = re.sub(r'<@[^>]+>', '', message.content).strip()
+        args = utils.parse_arguments(message.content)
+        debug(message, str(args))
+        if len(args) == 0 or args[0].lower().strip() == "help":
+            await message.channel.send(f"Hey {message.author.mention},\n"
+                                       f"You can generate a meme with the syntax\n"
+                                       f"```\n"
+                                       f"[template] \"text 1\" \"text 2\" ...\n"
+                                       f"```"
+                                       f"You can find a more detailed help and a list of templates at:\n"
+                                       f"<https://github.com/klemek/meme-otron/tree/master/discord>")
+            return
+        async with message.channel.typing():
+            left_wmark_text = None
+            if not direct and len(args) > 1:
+                f"By {message.author.display_name}"
+            img = main.compute(*args, left_wmark_text=left_wmark_text)
+            if img is None:
+                await message.channel.send(f"Template `{args[0]}` not found\n"
+                                           f"You can find a more detailed help and a list of templates at:\n"
+                                           f"<https://github.com/klemek/meme-otron/tree/master/discord>")
+                return
+            with tempfile.NamedTemporaryFile() as output:
+                img.save(output, format="JPEG")
+                response = None
+                if len(args) == 1:
+                    response = f"Template `{args[0]}`:"
+                elif not direct:
+                    response = f"A meme by {message.author.mention}:"
+                await message.channel.send(response,
+                                           file=discord.File(filename="meme.jpg", fp=output.name))
+            if not direct:
+                try:
+                    await message.delete()
+                except discord.Forbidden:
+                    pass
+                except discord.NotFound:
+                    pass
+
+
+# Launch client and rerun on errors
+while True:
+    try:
+        client.run(token)
+        break  # clean kill
+    except Exception as e:
+        t = datetime.now()
+        logging.error(f"Exception raised at {t:%Y-%m-%d %H:%M} : {repr(e)}")
+        fileName = f"error_{t:%Y-%m-%d_%H-%M-%S}.txt"
+        if os.path.exists(fileName):
+            logging.error("Two many errors, killing")
+            break
+        with open(fileName, 'w') as f:
+            f.write(f"Discord AI Dungeon 2 v{VERSION} started at {t0:%Y-%m-%d %H:%M}\r\n"
+                    f"Exception raised at {t:%Y-%m-%d %H:%M}\r\n"
+                    f"\r\n"
+                    f"{traceback.format_exc()}")