lydia-discord-bot/bot.py

64 lines
1.7 KiB
Python

"""https://discord.com/oauth2/authorize?client_id=605758711071506432&permissions=67119104&scope=bot"""
import discord
from discord.ext import commands
import os, sqlite3
def get_prefix(client, message):
try:
conn = sqlite3.connect('db.sqlite3')
curr = conn.cursor()
curr.execute(f'SELECT * FROM guilds WHERE guild_id="{message.guild.id}"')
query = curr.fetchone()
if query == None:
curr.execute(f"INSERT INTO guilds VALUES ('{message.guild.id}','$')")
conn.commit()
prefix = "$"
else:
prefix = query[1]
conn.close()
return prefix
except AttributeError:
conn.close()
return "$"
client = commands.Bot(command_prefix = get_prefix)
@client.event
async def on_ready():
await client.change_presence(activity=discord.Game(name="$help"))
@client.event
async def on_guild_join(guild):
conn = sqlite3.connect('db.sqlite3')
curr = conn.cursor()
curr.execute(f"INSERT INTO guilds VALUES ('{guild.id}','$')")
conn.commit()
conn.close()
@client.command()
@commands.is_owner()
async def load(ctx, extension):
client.load_extension(f'cogs.{extension}')
@client.command()
@commands.is_owner()
async def unload(ctx, extension):
client.unload_extension(f'cogs.{extension}')
@client.command()
@commands.is_owner()
async def reload(ctx, extension):
client.unload_extension(f'cogs.{extension}')
client.load_extension(f'cogs.{extension}')
for filename in os.listdir("./cogs"):
if filename.endswith('.py'):
client.load_extension(f'cogs.{filename[:-3]}')
client.run('NjA1NzU4NzExMDcxNTA2NDMy.XUBKww.sI5-DZzdCCXdgW44uEACqGzaVpc')