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

Fix db con

Browse Source
This commit is contained in:
Dusty.P 2018-05-13 23:41:03 -08:00
parent 17d527adc3
commit c54743acf4
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
bot.py
View File

@ -74,7 +74,14 @@ class Submitter(commands.Bot):
self.infected = {}
self.TOKEN = self.bot_secrets['token']
del self.bot_secrets['token']
self.db_con = asyncio.get_event_loop().run_until_complete(self.connect_db())
async def connect_db():
return await asyncpg.create_pool(host=self.bot_secrets['db_con']['host'],
database=self.bot_secrets['db_con']['db_name'],
user=self.bot_secrets['db_con']['user'],
password=self.bot_secrets['db_con']['password'],
loop=self.loop)
self.db_con = asyncio.get_event_loop().run_until_complete(connect_db())
self.default_prefix = '!'
self.tpe = futures.ThreadPoolExecutor()
self.embed_color = discord.Colour.from_rgb(49, 107, 111)
@ -87,13 +94,6 @@ class Submitter(commands.Bot):
'o': '🇴',
}
async def connect_db():
return await asyncpg.create_pool(host=self.bot_secrets['db_con']['host'],
database=self.bot_secrets['db_con']['db_name'],
user=self.bot_secrets['db_con']['user'],
password=self.bot_secrets['db_con']['password'],
loop=self.loop)
@staticmethod
async def get_custom_prefix(bot_inst, message):
return await bot_inst.db_con.fetchval('select prefix from guild_config where guild_id = $1',