countdown-bot

commit 6a5dfb367dde0d98e314ae5358684f90bc360d56
parent d5235c149c83811e3f26e44e790969cc5f309bf8
Author: Asher Morgan <59518073+ashermorgan@users.noreply.github.com>
Date:   Sun, 21 Apr 2024 15:32:48 -0700

Connect bot to postgresql db

Diffstat:
M
countdown_bot/__main__.py
 | 
4
+++-
M
countdown_bot/bot.py
 | 
3
++-
M
requirements.txt
 | 
1
+

3 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/countdown_bot/__main__.py b/countdown_bot/__main__.py
@@ -2,6 +2,7 @@
 from dotenv import load_dotenv
 import logging
 import os
+import psycopg
 
 # Import modules
 from .bot import CountdownBot
@@ -21,7 +22,8 @@ logging.basicConfig(
 
 # Connect to database
 databaseSessionMaker = getSessionMaker(os.environ.get("DATABASE"))
+db_connection = psycopg.connect(os.environ.get("DATABASE2"))
 
 # Run bot
-bot = CountdownBot(databaseSessionMaker, [os.environ.get("PREFIX", "!")])
+bot = CountdownBot(databaseSessionMaker, [os.environ.get("PREFIX", "!")], db_connection)
 bot.run(os.environ.get("TOKEN"))
diff --git a/countdown_bot/bot.py b/countdown_bot/bot.py
@@ -12,10 +12,11 @@ from .models import EmptyCountdownError
 
 
 class CountdownBot(commands.Bot):
-    def __init__(self, databaseSessionMaker, prefixes):
+    def __init__(self, databaseSessionMaker, prefixes, db_connection):
         # Set properties
         self.databaseSessionMaker = databaseSessionMaker
         self.prefixes = prefixes
+        self.db_connection = db_connection
         self.logger = logging.getLogger(__name__)
 
         # Get intents
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,5 @@
 discord
 matplotlib
+psycopg2
 python-dotenv
 sqlalchemy