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

11
geeksbot.py
View File

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