DB Connection is now created within the same event loop as client

Referencing pool connection from client/self.bot
pull/8/head
sgoudham 4 years ago
parent ec34fe0dd3
commit 55e87b3fc3

@ -5,8 +5,6 @@ import random
from discord import Member, Embed, Colour from discord import Member, Embed, Colour
from discord.ext.commands import BucketType, command, cooldown, bot_has_permissions, Cog from discord.ext.commands import BucketType, command, cooldown, bot_has_permissions, Cog
import db
from db import connection
from settings import colour_list from settings import colour_list
@ -62,7 +60,7 @@ class Relationship(Cog):
# Getting the guild of the user # Getting the guild of the user
guild = ctx.author.guild guild = ctx.author.guild
# Setup pool # Setup pool
pool = await connection(db.loop) pool = self.bot.db
# Setup pool connection and cursor # Setup pool connection and cursor
async with pool.acquire() as conn: async with pool.acquire() as conn:
@ -118,7 +116,7 @@ class Relationship(Cog):
if msg.content.lower() in ['y', 'yes', 'yea']: if msg.content.lower() in ['y', 'yes', 'yea']:
# Setup pool # Setup pool
pool = await connection(db.loop) pool = self.bot.db
# Setup pool connection and cursor # Setup pool connection and cursor
async with pool.acquire() as conn: async with pool.acquire() as conn:
@ -162,7 +160,7 @@ class Relationship(Cog):
# Getting the guild of the user # Getting the guild of the user
guild = ctx.author.guild guild = ctx.author.guild
# Setup pool # Setup pool
pool = await connection(db.loop) pool = self.bot.db
# Setup pool connection and cursor # Setup pool connection and cursor
async with pool.acquire() as conn: async with pool.acquire() as conn:
@ -210,7 +208,7 @@ class Relationship(Cog):
# if the person says yes # if the person says yes
if msg.content.lower() in ['y', 'yes', 'yea']: if msg.content.lower() in ['y', 'yes', 'yea']:
# Setup pool # Setup pool
pool = await connection(db.loop) pool = self.bot.db
# Setup pool connection and cursor # Setup pool connection and cursor
async with pool.acquire() as conn: async with pool.acquire() as conn:
@ -265,7 +263,7 @@ class Relationship(Cog):
guild = member.guild guild = member.guild
# Setup pool # Setup pool
pool = await connection(db.loop) pool = self.bot.db
# Setup pool connection and cursor # Setup pool connection and cursor
async with pool.acquire() as conn: async with pool.acquire() as conn:

Loading…
Cancel
Save