Checking if connection can be created through main

pull/8/head
sgoudham 4 years ago
parent 09e3bc0913
commit 8f473da1a7

@ -86,16 +86,6 @@ class Bot(commands.Bot):
self.modmail_cache = {} self.modmail_cache = {}
self.member_cache = MyCoolCache(2) self.member_cache = MyCoolCache(2)
async def create_connection():
"""Setting up connection using asyncpg"""
self.db = await asyncpg.create_pool(
host=host,
port=int(port),
user=user,
password=password,
database=db)
async def startup_cache_log(): async def startup_cache_log():
"""Store the guilds/modmail systems in cache from the database on startup""" """Store the guilds/modmail systems in cache from the database on startup"""
@ -133,7 +123,7 @@ class Bot(commands.Bot):
await pool.release(conn) await pool.release(conn)
# Establish Database Connection # Establish Database Connection
self.loop.run_until_complete(create_connection()) # self.loop.run_until_complete(create_connection())
# Load Information Into Cache # Load Information Into Cache
self.loop.run_until_complete(startup_cache_log()) self.loop.run_until_complete(startup_cache_log())
@ -710,3 +700,14 @@ class Bot(commands.Bot):
self.run(API_TOKEN) self.run(API_TOKEN)
except discord.errors.LoginFailure as e: except discord.errors.LoginFailure as e:
print(e, "Login unsuccessful.") print(e, "Login unsuccessful.")
async def create_connection(self):
"""Setting up connection using asyncpg"""
self.db = await asyncpg.create_pool(
host=host,
port=int(port),
user=user,
password=password,
database=db,
loop=self.loop)

@ -38,5 +38,7 @@ async def test(ctx):
print(client.member_cache.cache) print(client.member_cache.cache)
client.create_connection()
# Run the bot # Run the bot
client.execute() client.execute()

Loading…
Cancel
Save