135 lines
2.7 KiB
Python
135 lines
2.7 KiB
Python
import os
|
|
import discord
|
|
|
|
# Get bot token
|
|
|
|
token = ""
|
|
with open("token.cfg") as t:
|
|
token = t.read()
|
|
|
|
# Create bot client
|
|
|
|
client = discord.Client()
|
|
|
|
# IDs of bot admins
|
|
|
|
admins = []
|
|
|
|
# Default prefix
|
|
|
|
prefix = "r."
|
|
|
|
# Default color, to be used for embeds.
|
|
|
|
color = discord.Colour.blue()
|
|
|
|
# API functions
|
|
|
|
def register_command(command, function):
|
|
|
|
commands[command] = function
|
|
|
|
def register_admin_command(command, function):
|
|
|
|
admin_commands[command] = function
|
|
|
|
def get_commands():
|
|
return commands
|
|
|
|
def get_admin_commands():
|
|
return admin_commands
|
|
|
|
def get_bot_prefix():
|
|
return prefix
|
|
|
|
def get_embed_color():
|
|
return color
|
|
|
|
def command_exists(command):
|
|
|
|
if command in commands:
|
|
return True
|
|
|
|
if command in admin_commands:
|
|
return True
|
|
|
|
return False
|
|
|
|
def is_admin(id):
|
|
|
|
return id in admins
|
|
|
|
async def restart():
|
|
os.system('./restart.sh')
|
|
# Exit cleanly
|
|
exit()
|
|
|
|
# Initializing the command dictionaries
|
|
|
|
# Commands anyone can use!
|
|
commands = {}
|
|
|
|
# Commands only accessible to admins
|
|
admin_commands = {}
|
|
|
|
# Load commands
|
|
|
|
os.chdir('commands')
|
|
|
|
for filename in os.listdir():
|
|
|
|
# Make sure it isn't this source file. Needs to be abstracted in the future.
|
|
if filename != 'bot.py':
|
|
|
|
# Open the file.
|
|
with open(filename) as file:
|
|
|
|
# Run the file in the current scope.
|
|
exec(file.read())
|
|
|
|
# When message detected
|
|
|
|
@client.event
|
|
async def on_message(message):
|
|
|
|
# If it looks like a command...
|
|
if message.content.startswith(prefix):
|
|
|
|
for command in commands:
|
|
|
|
if message.content.startswith(prefix+command):
|
|
|
|
try:
|
|
args = message.content[len(message.content.split()[0])+1:]
|
|
await commands[command](message, args)
|
|
|
|
except Exception as e:
|
|
msg = "**Exception in command `" + command + "`:**\n" + str(e)
|
|
await message.channel.send(msg)
|
|
|
|
# Checking for admin commands as well! But only if the author is an admin.
|
|
if is_admin(str(message.author.id)):
|
|
|
|
for command in admin_commands:
|
|
|
|
if message.content.startswith(prefix+command):
|
|
|
|
try:
|
|
args = message.content[len(message.content.split()[0])+1:]
|
|
await commands[command](message, args)
|
|
|
|
except Exception as e:
|
|
msg = "**Exception in command `" + command + "`:**\n" + str(e)
|
|
await message.channel.send(msg)
|
|
|
|
|
|
# When logged in
|
|
|
|
@client.event
|
|
async def on_ready():
|
|
print("Logged in as\n" + client.user.name + "\n" + str(client.user.id) + "\n---")
|
|
|
|
# Log in
|
|
|
|
client.run(token)
|