Blunt tried 💀

Reviewed-on: #2

bot.py

@@ -1,8 +1,9 @@
 import simplematrixbotlib as botLibrary
 import json
 import fetch
+import time
 
-version = "0.2.0"
+version = "0.3.0"
 
 print("Imag/quotes bot version " + version)
 print("Loading config…")
@@ -22,6 +23,7 @@ else:
 
 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.")
@@ -33,29 +35,27 @@ async def help_message(room, message):
             and match.command("help")):
         return
     if debugMode == False:
-        message = (f"""
-        Help
+        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
+        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.
-        """)
+NOTE: Debug mode is on. Output will be more verbose.""")
 
 
     await bot.api.send_text_message(room.room_id, message)
@@ -71,41 +71,51 @@ async def make_choice(room, message):
         temp = False
     else:
         id = match.args()
-        id = str(id[1])
+        try:
+            id = int(id[1])
+        except ValueError:
+            youTriedMessage = ("Blunt tried :skull:")
+            await bot.api.send_text_message(room.room_id, youTriedMessage)
+
         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 = str(data['created'])
+            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}
+        message1 = (f"""Created: {created}
 Description: {description}
 Image ID: {imageid}
 OCR (Tesseract): {ocr}
-                    Rating: {score}
-                    """)
+Rating: {score}""")
     else:
         created, description, imageid, ocr, score = formatData(quoteData)
-        message1 = (f"""
-                    Created: {created}
+        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.)
-        """)
+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)
@@ -122,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)
 
@@ -136,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

@@ -9,6 +9,7 @@
     "server": "https://quotes.example.com"
   },
   "bot": {
-    "debugMode": true
+    "debugMode": true,
+    "admin": "@alice:example.com"
   }
 }

fetch.py

@@ -30,6 +30,12 @@ def fetchQuoteData(quoteId):
         print(quoteData.json())
     else:
         print(f"An error occured while saving the file: {quoteData.status_code}")
-        return "This file probably doesn't exist."
+        return None
+
+def fetchTopQuotes(instance):
+    raise NotImplementedError
+
+def fetchBottomQuotes(instance):
+    raise NotImplementedError
 
 fetchQuoteData(178)