Set up basics of SQLAlchemy

database/__init__.py

@@ -0,0 +1,2 @@
+from .db import engine, session, Base
+from .models import *

database/db.py

@@ -0,0 +1,16 @@
+from settings import DB_HOST, DB_NAME, DB_PASSWORD, DB_USERNAME, DB_DIALECT, DB_DRIVER
+from sqlalchemy import create_engine
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+from urllib.parse import quote_plus
+
+# Encode password
+_encoded_pw = quote_plus(DB_PASSWORD)
+
+engine = create_engine(
+    # Format: dialect+driver://username:password@host:port/database
+    f"{DB_DIALECT}{'+' if DB_DRIVER else ''}{DB_DRIVER}://{DB_USERNAME}:{_encoded_pw}@{DB_HOST}/{DB_NAME}"
+)
+session = sessionmaker(bind=engine)()
+
+Base = declarative_base()

database/models.py

@@ -0,0 +1,16 @@
+from database import Base
+from sqlalchemy import Column, String, Integer
+from typing import List
+
+
+class TestTable(Base):
+
+    __tablename__ = "test_table"
+
+    id = Column(Integer, autoincrement=True, primary_key=True)
+    name = Column(String, nullable=False)
+
+
+# A list of all models in this file, this is used to
+# create all tables later on
+all_models: List[Base] = [TestTable]

database/utils.py

@@ -0,0 +1,11 @@
+from database import all_models, engine
+
+
+def create_all_tables():
+    """
+    Create all tables in case they don't exist yet
+    so the user doesn't have to do this manually
+    when a new table is added
+    """
+    for model in all_models:
+        model.__table__.create(engine, checkfirst=True)

requirements.txt

@@ -1,3 +1,4 @@
+sqlalchemy==1.4.18
 python-dotenv==0.14.0
 beautifulsoup4==4.9.1
 discord.py==1.7.3

settings.py

@@ -15,25 +15,27 @@ def _to_bool(value: str) -> bool:
 
 
 # Sandbox or live
-SANDBOX = _to_bool(os.getenv("SANDBOX", "true"))
+SANDBOX: bool = _to_bool(os.getenv("SANDBOX", "true"))
 
 # Tokens & API keys
-URBANDICTIONARY = os.getenv("URBANDICTIONARY", "")
+URBANDICTIONARY: str = os.getenv("URBANDICTIONARY", "")
-IMGFLIP_NAME = os.getenv("IMGFLIPNAME", "")
+IMGFLIP_NAME: str = os.getenv("IMGFLIPNAME", "")
-IMGFLIP_PASSWORD = os.getenv("IMGFLIPPASSWORD", "")
+IMGFLIP_PASSWORD: str = os.getenv("IMGFLIPPASSWORD", "")
-UFORA_TOKEN = os.getenv("UFORA_TOKEN", "")
+UFORA_TOKEN: str = os.getenv("UFORA_TOKEN", "")
 
 # Database credentials
-DB_USERNAME = os.getenv("DBUSERNAME", "")
+DB_USERNAME: str = os.getenv("DBUSERNAME", "postgres")
-DB_PASSWORD = os.getenv("DBPASSWORD", "")
+DB_PASSWORD: str = os.getenv("DBPASSWORD", "")
-DB_HOST = os.getenv("DBHOST", "")
+DB_HOST: str = os.getenv("DBHOST", "localhost")
-DB_NAME = os.getenv("DBNAME", "")
+DB_NAME: str = os.getenv("DBNAME", "")
+DB_DIALECT: str = os.getenv("DBDIALECT", "postgresql")
+DB_DRIVER: str = os.getenv("DBDRIVER", "")
 
 # Discord-related
-TOKEN = os.getenv("TOKEN", "")
+TOKEN: str = os.getenv("TOKEN", "")
-HOST_IPC = _to_bool(os.getenv("HOSTIPC", "false"))
+HOST_IPC: bool = _to_bool(os.getenv("HOSTIPC", "false"))
-READY_MESSAGE = os.getenv("READYMESSAGE", "I'M READY I'M READY I'M READY I'M READY")  # Yes, this is a Spongebob reference
+READY_MESSAGE: str = os.getenv("READYMESSAGE", "I'M READY I'M READY I'M READY I'M READY")  # Yes, this is a Spongebob reference
-STATUS_MESSAGE = os.getenv("STATUSMESSAGE", "with your Didier Dinks.")
+STATUS_MESSAGE: str = os.getenv("STATUSMESSAGE", "with your Didier Dinks.")
 
 # Guilds to test slash commands in
 # Ex: 123,456,789

startup/didier.py

@@ -1,3 +1,4 @@
+from database.utils import create_all_tables
 from data.snipe import Snipe
 from discord.ext import commands, ipc
 from dislash import InteractionClient
@@ -38,6 +39,9 @@ class Didier(commands.Bot):
         # Load all extensions
         self.init_extensions()
 
+        # Initialize database tables
+        create_all_tables()
+
         # Check missing files
         check_all()