|
|
|
@ -12,11 +12,11 @@
|
|
|
|
|
# GNU General Public License for more details.
|
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
|
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
|
import asyncio
|
|
|
|
|
import datetime
|
|
|
|
|
import logging
|
|
|
|
|
import os
|
|
|
|
|
import random
|
|
|
|
|
import threading
|
|
|
|
|
|
|
|
|
|
import aiohttp
|
|
|
|
|
import asyncpg as asyncpg
|
|
|
|
@ -49,6 +49,8 @@ API_TOKEN = config('DISCORD_TOKEN')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Bot(commands.Bot):
|
|
|
|
|
threadLock = threading.Lock()
|
|
|
|
|
|
|
|
|
|
def __init__(self, **options):
|
|
|
|
|
|
|
|
|
|
async def get_prefix(bot, message):
|
|
|
|
@ -88,7 +90,7 @@ class Bot(commands.Bot):
|
|
|
|
|
|
|
|
|
|
async def create_connection():
|
|
|
|
|
"""Setting up connection using asyncpg"""
|
|
|
|
|
|
|
|
|
|
with self.threadLock:
|
|
|
|
|
self.db = await asyncpg.create_pool(
|
|
|
|
|
host=host,
|
|
|
|
|
port=int(port),
|
|
|
|
@ -97,8 +99,6 @@ class Bot(commands.Bot):
|
|
|
|
|
database=db,
|
|
|
|
|
loop=self.loop)
|
|
|
|
|
|
|
|
|
|
asyncio.run(create_connection())
|
|
|
|
|
|
|
|
|
|
async def startup_cache_log():
|
|
|
|
|
"""Store the guilds/modmail systems in cache from the database on startup"""
|
|
|
|
|
|
|
|
|
@ -136,7 +136,7 @@ class Bot(commands.Bot):
|
|
|
|
|
await pool.release(conn)
|
|
|
|
|
|
|
|
|
|
# Establish Database Connection
|
|
|
|
|
# self.loop.run_until_complete(create_connection())
|
|
|
|
|
self.loop.run_until_complete(create_connection())
|
|
|
|
|
# Load Information Into Cache
|
|
|
|
|
self.loop.run_until_complete(startup_cache_log())
|
|
|
|
|
|
|
|
|
|