Compare commits

..

No commits in common. "main" and "0.2.0" have entirely different histories.
main ... 0.2.0

3 changed files with 56 additions and 79 deletions

76
bot.py
View file

@ -1,9 +1,8 @@
import simplematrixbotlib as botLibrary import simplematrixbotlib as botLibrary
import json import json
import fetch import fetch
import time
version = "0.3.0" version = "0.2.0"
print("Imag/quotes bot version " + version) print("Imag/quotes bot version " + version)
print("Loading config…") print("Loading config…")
@ -23,7 +22,6 @@ else:
print("Logging in…") print("Logging in…")
credentials = botLibrary.Creds(config['matrix']['server'], config['matrix']['userid'], config['matrix']['password']) credentials = botLibrary.Creds(config['matrix']['server'], config['matrix']['userid'], config['matrix']['password'])
admin = config['bot']['admin']
bot = botLibrary.Bot(credentials) bot = botLibrary.Bot(credentials)
PREFIX = "!q" PREFIX = "!q"
print("Logged in.") print("Logged in.")
@ -35,27 +33,29 @@ async def help_message(room, message):
and match.command("help")): and match.command("help")):
return return
if debugMode == False: 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. A work-in-progress Python rewrite of the original quotes bot, aiming for a mostly complete remake.
{PREFIX} help - show this message {PREFIX} help - show this message
{PREFIX} get - fetch a image from the defined instance {PREFIX} get - fetch a image from the defined instance
{PREFIX} source - send a link to the source code {PREFIX} source - send a link to the source code
{PREFIX} version - show the bot version {PREFIX} version - show the bot version
{PREFIX} die - kills the bot (only admins can do this!)""") """)
else: else:
message = (f"""Help message = (f"""
Help
============================ ============================
A work-in-progress Python rewrite of the original quotes bot, aiming for a mostly complete remake. A work-in-progress Python rewrite of the original quotes bot, aiming for a mostly complete remake.
{PREFIX} help - show this message {PREFIX} help - show this message
{PREFIX} get - fetch a image from the defined instance {PREFIX} get - fetch a image from the defined instance
{PREFIX} source - send a link to the source code {PREFIX} source - send a link to the source code
{PREFIX} version - show the bot version {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) await bot.api.send_text_message(room.room_id, message)
@ -71,51 +71,41 @@ async def make_choice(room, message):
temp = False temp = False
else: else:
id = match.args() id = match.args()
try: id = str(id[1])
id = int(id[1])
except ValueError:
youTriedMessage = ("Blunt tried :skull:")
await bot.api.send_text_message(room.room_id, youTriedMessage)
quoteImage = fetch.fetchQuote(id) quoteImage = fetch.fetchQuote(id)
quoteData = fetch.fetchQuoteData(id) quoteData = fetch.fetchQuoteData(id)
def formatData(data): def formatData(data):
if data is not None:
timestamp = time.strftime("%Y-%m-%d %H:%M:%S", time.gmtime(data['created']))
dataToFormat = data dataToFormat = data
created = timestamp created = str(data['created'])
description = str(data['desc']) description = str(data['desc'])
imageid = str(data['iid']) imageid = str(data['iid'])
ocr = str(data['ocr']) ocr = str(data['ocr'])
score = str(data['score']) score = str(data['score'])
return created, description, imageid, ocr, 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: if debugMode == False:
created, description, imageid, ocr, score = formatData(quoteData) created, description, imageid, ocr, score = formatData(quoteData)
message1 = (f"""Created: {created} message1 = (f"""
Created: {created}
Description: {description} Description: {description}
Image ID: {imageid} Image ID: {imageid}
OCR (Tesseract): {ocr} OCR (Tesseract): {ocr}
Rating: {score}""") Rating: {score}
""")
else: else:
created, description, imageid, ocr, score = formatData(quoteData) created, description, imageid, ocr, score = formatData(quoteData)
message1 = (f"""Created: {created} message1 = (f"""
Created: {created}
Description: {description} Description: {description}
Image ID: {imageid} Image ID: {imageid}
OCR (Tesseract): OCR (Tesseract): {ocr}
> {ocr} Rating: {score}
Rating: {score}""") """)
message2 = (f"""Command recived (DEBUG). message2 = (f"""
(We're supposed to fetch quote number {id} now.)""") Command recived (DEBUG).
(We're supposed to fetch quote number {id} now.)
""")
if debugMode == True: if debugMode == True:
await bot.api.send_text_message(room.room_id, message2) await bot.api.send_text_message(room.room_id, message2)
@ -132,8 +122,10 @@ async def help_message(room, message):
and match.command("source")): and match.command("source")):
return return
message = (f"""quotes-bot-python by Nyx Tutt (@n:everypizza.im) message = (f"""
https://git.everypizza.im/n/quotes-bot-python/""") 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) await bot.api.send_text_message(room.room_id, message)
@ -144,18 +136,10 @@ async def help_message(room, message):
and match.command("version")): and match.command("version")):
return 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) 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() bot.run()

View file

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

View file

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