wayfinderak/Geeksbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Switching to fetchval

Browse Source
This commit is contained in:
Dusty.P 2018-04-20 19:49:01 -08:00
parent 0654bd51eb
commit 9ef06824a7
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
geeksbot.py
View File

@ -69,7 +69,7 @@ class Geeksbot(commands.Bot):
@staticmethod
async def get_custom_prefix(bot_inst, message):
return bot_inst.db_con.fetchval('select prefix from guild_config where guild_id = $1',
return await bot_inst.db_con.fetchval('select prefix from guild_config where guild_id = $1',
message.guild.id) or bot_inst.default_prefix
async def load_ext(self, ctx, mod=None):
@ -122,14 +122,14 @@ async def unload(ctx, mod):
async def on_message(ctx):
if not ctx.author.bot:
if ctx.guild:
if int(bot.db_con.fetchval("select channel_lockdown from guild_config where guild_id = $1",
if int(await bot.db_con.fetchval("select channel_lockdown from guild_config where guild_id = $1",
ctx.guild.id)):
if ctx.channel.id in json.loads(bot.db_con.fetchval("select allowed_channels from guild_config "
if ctx.channel.id in json.loads(await bot.db_con.fetchval("select allowed_channels from guild_config "
"where guild_id = $1",
ctx.guild.id)):
await bot.process_commands(ctx)
elif ctx.channel.id == 418452585683484680:
prefix = bot.db_con.fetchval('select prefix from guild_config where guild_id = $1', ctx.guild.id)
prefix = await bot.db_con.fetchval('select prefix from guild_config where guild_id = $1', ctx.guild.id)
prefix = prefix[0] if prefix else bot.default_prefix
ctx.content = f'{prefix}{ctx.content}'
await bot.process_commands(ctx)