mirror of
https://github.com/myned/modufur.git
synced 2024-11-01 13:02:38 +00:00
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
import toml
|
|
import hikari
|
|
|
|
|
|
# Hikari activity type
|
|
# https://www.hikari-py.dev/hikari/presences.html#hikari.presences.ActivityType
|
|
ACTIVITY = hikari.ActivityType.LISTENING
|
|
# Global command error response
|
|
ERROR = "```❗ An internal error has occurred. This has been reported to my master 🐺```"
|
|
# Local bot configuration
|
|
CONFIG = """\
|
|
guilds = [] # guild IDs to register commands, empty for global
|
|
master = 0 # guild ID to register owner commands
|
|
client = 0 # bot application ID
|
|
token = "" # bot token
|
|
activity = "" # bot status
|
|
saucenao = "" # saucenao token
|
|
e621 = "" # e621 token
|
|
|
|
"""
|
|
|
|
|
|
# Load or create config.toml
|
|
try:
|
|
config = toml.load("config.toml")
|
|
except FileNotFoundError:
|
|
with open("config.toml", "w") as f:
|
|
f.write(CONFIG)
|
|
print("config.toml created with default values. Restart when modified")
|
|
exit()
|
|
|
|
|
|
# Global command error response for owner
|
|
def error(event):
|
|
exception = event.exception.__cause__ or event.exception
|
|
|
|
return (
|
|
f"**`{event.context.command.name}` in {event.context.get_channel().mention if event.context.guild_id else 'DMs'}"
|
|
f"```❗ {type(exception).__name__}: {exception}```**"
|
|
)
|
|
|
|
|
|
# Write config to file
|
|
def dump():
|
|
with open("config.toml", "w") as file:
|
|
toml.dump(config, file)
|