Blunt tried 💀

Reviewed-on: #2

bot.py

@@ -1,8 +1,9 @@
 import simplematrixbotlib as botLibrary
 import json
 import fetch
+import time
 
-version = "0.1.0"
+version = "0.3.0"
 
 print("Imag/quotes bot version " + version)
 print("Loading config…")
@@ -15,8 +16,14 @@ except FileNotFoundError:
 except:
     print("Something bad and unforseen happened, please report the bug to @n:everypizza.im")
 
+if config['bot']['debugMode'] == True:
+    debugMode = True
+else:
+    debugMode = False
+
 print("Logging in…")
 credentials = botLibrary.Creds(config['matrix']['server'], config['matrix']['userid'], config['matrix']['password'])
+admin = config['bot']['admin']
 bot = botLibrary.Bot(credentials)
 PREFIX = "!q"
 print("Logged in.")
@@ -27,16 +34,29 @@ async def help_message(room, message):
     if not (match.is_not_from_this_bot() and match.prefix()
             and match.command("help")):
         return
+    if debugMode == False:
+        message = (f"""Help
+============================
+A work-in-progress Python rewrite of the original quotes bot, aiming for a mostly complete remake.
+{PREFIX} help - show this message
+{PREFIX} get - fetch a image from the defined instance
+{PREFIX} source - send a link to the source code
+{PREFIX} version - show the bot version
+{PREFIX} die - kills the bot (only admins can do this!)""")
+    else:
+        message = (f"""Help
+============================
+A work-in-progress Python rewrite of the original quotes bot, aiming for a mostly complete remake.
+{PREFIX} help - show this message
+{PREFIX} get - fetch a image from the defined instance
+{PREFIX} source - send a link to the source code
+{PREFIX} version - show the bot version
+{PREFIX} die - kills the bot (only admins can do this!)
+
+---
+
+NOTE: Debug mode is on. Output will be more verbose.""")
 
-    message = (f"""
-    Help
-    ============================
-    A work-in-progress Python rewrite of the original quotes bot, aiming for a mostly complete remake.
-        {PREFIX} help - show this message
-        {PREFIX} get - fetch a image from the defined instance
-        {PREFIX} source - send a link to the source code
-        {PREFIX} version - show the bot version
-    """)
 
     await bot.api.send_text_message(room.room_id, message)
 
@@ -51,14 +71,57 @@ async def make_choice(room, message):
         temp = False
     else:
         id = match.args()
-        id = str(id[1])
-        quoteImage = fetch.fetchQuote(id)
-    message = (f"""
-    Command recived (DEBUG).
-    (We're supposed to fetch quote number {id} now.)
-    """)
+        try:
+            id = int(id[1])
+        except ValueError:
+            youTriedMessage = ("Blunt tried :skull:")
+            await bot.api.send_text_message(room.room_id, youTriedMessage)
 
-    await bot.api.send_text_message(room.room_id, message)
+        quoteImage = fetch.fetchQuote(id)
+        quoteData = fetch.fetchQuoteData(id)
+
+    def formatData(data):
+        if data is not None:
+            timestamp = time.strftime("%Y-%m-%d %H:%M:%S", time.gmtime(data['created']))
+            dataToFormat = data
+            created = timestamp
+            description = str(data['desc'])
+            imageid = str(data['iid'])
+            ocr = str(data['ocr'])
+            score = str(data['score'])
+            return created, description, imageid, ocr, score
+        else:
+            dataToFormat = data
+            created = "Error"
+            description = "Error"
+            imageid = "Error"
+            ocr = "Error"
+            score = "Error"
+            return created, description, imageid, ocr, score
+
+    if debugMode == False:
+        created, description, imageid, ocr, score = formatData(quoteData)
+        message1 = (f"""Created: {created}
+Description: {description}
+Image ID: {imageid}
+OCR (Tesseract): {ocr}
+Rating: {score}""")
+    else:
+        created, description, imageid, ocr, score = formatData(quoteData)
+        message1 = (f"""Created: {created}
+Description: {description}
+Image ID: {imageid}
+OCR (Tesseract):
+> {ocr}
+Rating: {score}""")
+        message2 = (f"""Command recived (DEBUG).
+(We're supposed to fetch quote number {id} now.)""")
+
+    if debugMode == True:
+        await bot.api.send_text_message(room.room_id, message2)
+        await bot.api.send_text_message(room.room_id, message1)
+    else:
+        await bot.api.send_text_message(room.room_id, message1)
     imageMessage = (quoteImage)
     await bot.api.send_image_message(room.room_id, imageMessage) # https://quotes.everypizza.im/image/178
 
@@ -69,10 +132,8 @@ async def help_message(room, message):
             and match.command("source")):
         return
 
-    message = (f"""
-    quotes-bot-python by Nyx Tutt (@n:everypizza.im)
-    https://git.everypizza.im/n/quotes-bot-python/
-    """)
+    message = (f"""quotes-bot-python by Nyx Tutt (@n:everypizza.im)
+https://git.everypizza.im/n/quotes-bot-python/""")
 
     await bot.api.send_text_message(room.room_id, message)
 
@@ -83,10 +144,18 @@ async def help_message(room, message):
             and match.command("version")):
         return
 
-    message = (f"""
-    quotes-bot-python version {version}
-    """)
+    message = (f"""quotes-bot-python version {version}""")
 
     await bot.api.send_text_message(room.room_id, message)
 
+@bot.listener.on_message_event
+async def help_message(room, message):
+    match = botLibrary.MessageMatch(room, message, bot, PREFIX)
+    if match.is_not_from_this_bot() and match.prefix() and match.command("die"):
+        if match.is_from_userid(admin):
+            await bot.api.send_text_message(room.room_id, "Ok")
+            exit(0)
+        else:
+            await bot.api.send_text_message(room.room_id, "You don't have permissions for that.")
+
 bot.run()

config.sample.json

@@ -7,5 +7,9 @@
   "imag":{
     "_comment": "Don't put a slash at an end of this. That took quite a bit of running around in circles to fix.",
     "server": "https://quotes.example.com"
+  },
+  "bot": {
+    "debugMode": true,
+    "admin": "@alice:example.com"
   }
 }

fetch.py

@@ -17,4 +17,25 @@ def fetchQuote(quoteId):
         return filename
     else:
         print(f"An error occured while saving the file: {quote.status_code}")
+        return "error.png"
+
+def fetchQuoteData(quoteId):
+    processedId = str(config['imag']['server'] + "/api/image/" + str(quoteId))
+    print(processedId)
+    quoteData = requests.get(processedId)
+    code = 200
+    if quoteData.status_code == 200:
+        return quoteData.json()
+        # print(str(json.dumps(quoteData)))
+        print(quoteData.json())
+    else:
+        print(f"An error occured while saving the file: {quoteData.status_code}")
         return None
+
+def fetchTopQuotes(instance):
+    raise NotImplementedError
+
+def fetchBottomQuotes(instance):
+    raise NotImplementedError
+
+fetchQuoteData(178)