Trying to get asynchronous database connections working

pull/8/head
sgoudham 4 years ago
parent 5d67859ab4
commit 7e9a8aeaab

@ -157,17 +157,17 @@ async def _help(ctx, *, command: Optional[str] = None):
async def reload_db(ctx): async def reload_db(ctx):
pool = await db.connection2(db.loop) pool = await db.connection2(db.loop)
with db.connection() as conn: async with pool.acquire() as conn:
with closing(conn.cursor()) as cursor: async with conn.cursor() as cur:
# Define the insert statement that will insert the user's information # Define the insert statement that will insert the user's information
insert = "INSERT IGNORE INTO members (guildID, discordID) VALUES " + ", ".join( insert = "INSERT IGNORE INTO members (guildID, discordID) VALUES " + ", ".join(
map(lambda m: f"({ctx.guild.id}, {m.id})", ctx.guild.members)) map(lambda m: f"({ctx.guild.id}, {m.id})", ctx.guild.members))
try: # Execute the query try: # Execute the query
cursor.execute(insert) await cur.execute(insert)
except Exception as e: except Exception as e:
print(e) print(e)
print(cursor.rowcount, f"Record(s) inserted successfully into Members from {ctx.guild.name}") print(cur.rowcount, f"Record(s) inserted successfully into Members from {ctx.guild.name}")
@client.command(name="prefix", aliases=["Prefix"]) @client.command(name="prefix", aliases=["Prefix"])
@ -214,7 +214,7 @@ async def on_guild_join(guild):
print(cursor.rowcount, f"Record inserted successfully into Guilds from {guild.name}") print(cursor.rowcount, f"Record inserted successfully into Guilds from {guild.name}")
# Define the insert statement that will insert the user's information # Define the insert statement that will insert the user's information
insert = "INSERT INTO members (guildID, discordID) VALUES" + ", ".join( insert = "INSERT IGNORE INTO members (guildID, discordID) VALUES" + ", ".join(
map(lambda m: f"({guild.id}, {m.id})", guild.members)) map(lambda m: f"({guild.id}, {m.id})", guild.members))
with closing(conn.cursor()) as cursor: with closing(conn.cursor()) as cursor:
# Execute the query # Execute the query

Loading…
Cancel
Save