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

Switching to asyncpg

Browse Source
This commit is contained in:
Dusty.P 2018-04-20 21:02:28 -08:00
parent 13d4dc564c
commit 9cc1004f38
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
geeksbot.py
View File

@ -57,7 +57,7 @@ class Geeksbot(commands.Bot):
self.infected = {}
self.TOKEN = self.bot_secrets['token']
del self.bot_secrets['token']
self.db_con = self.connect_db()
self.db_con = None
del self.bot_secrets['db_con']
self.default_prefix = 'g~'
self.voice_chans = {}
@ -65,7 +65,7 @@ class Geeksbot(commands.Bot):
self.gcs_service = build('customsearch', 'v1', developerKey=self.bot_secrets['google_search_key'])
async def connect_db(self):
return await asyncpg.connect(host={self.bot_secrets['db_con']['host']},
self.db_con = await asyncpg.connect(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']})
@ -144,6 +144,7 @@ async def on_message(ctx):
@bot.event
async def on_ready():
bot.connect_db()
bot.recent_msgs = {}
for guild in bot.guilds:
bot.recent_msgs[guild.id] = deque(maxlen=50)