Just trying different things

pull/8/head
sgoudham 4 years ago
parent f9d1fbcd2b
commit 272095e1fc

@ -12,11 +12,11 @@
# GNU General Public License for more details. # GNU General Public License for more details.
# You should have received a copy of the GNU General Public License # You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>. # along with this program. If not, see <https://www.gnu.org/licenses/>.
import asyncio
import datetime import datetime
import logging import logging
import os import os
import random import random
import threading
import aiohttp import aiohttp
import asyncpg as asyncpg import asyncpg as asyncpg
@ -49,6 +49,8 @@ API_TOKEN = config('DISCORD_TOKEN')
class Bot(commands.Bot): class Bot(commands.Bot):
threadLock = threading.Lock()
def __init__(self, **options): def __init__(self, **options):
async def get_prefix(bot, message): async def get_prefix(bot, message):
@ -88,7 +90,7 @@ class Bot(commands.Bot):
async def create_connection(): async def create_connection():
"""Setting up connection using asyncpg""" """Setting up connection using asyncpg"""
with self.threadLock:
self.db = await asyncpg.create_pool( self.db = await asyncpg.create_pool(
host=host, host=host,
port=int(port), port=int(port),
@ -97,8 +99,6 @@ class Bot(commands.Bot):
database=db, database=db,
loop=self.loop) loop=self.loop)
asyncio.run(create_connection())
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"""
@ -136,7 +136,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())

Loading…
Cancel
Save