Test file after connecting to prod database and testing relationships.

database/database_test.py

@@ -1,9 +1,9 @@
 import os
 
 from dotenv import load_dotenv
-from sqlmodel import Session, SQLModel, create_engine
+from sqlmodel import Session, create_engine, select
 
-from models import CategoryGroup, Guild
+from models import Guild
 
 load_dotenv()
 
@@ -12,19 +12,19 @@ SQL_HOST = os.getenv("SQL_HOST", "")
 SQL_PASS = os.getenv("SQL_PASS", "")
 SQL_USER = os.getenv("SQL_USER", "")
 
-url = "sqlite:///database.db"
-# url = f"mysql+pymysql://{SQL_USER}:{SQL_PASS}@{SQL_HOST}/{SQL_DB}"
+# url = "sqlite:///database.db"
+url = f"mysql+pymysql://{SQL_USER}:{SQL_PASS}@{SQL_HOST}/{SQL_DB}"
 engine = create_engine(url)
-SQLModel.metadata.create_all(engine)
+# SQLModel.metadata.create_all(engine)
 
 with Session(engine) as session:
-    guild = Guild(id=1, name="test guild", submission_channel=1, verification_channel=2)
-    session.add(guild)
-    group1 = CategoryGroup(name="group1", channel_id=1, guild_id=1, message_id=1)
-    group2 = CategoryGroup(name="group2", channel_id=2, guild_id=1, message_id=2)
-    session.add(group1)
-    session.add(group2)
-    session.commit()
+    # guild = Guild(id=1, name="test guild", submission_channel=1, verification_channel=2)
+    # session.add(guild)
+    # group1 = CategoryGroup(name="group1", channel_id=1, guild_id=1, message_id=1)
+    # group2 = CategoryGroup(name="group2", channel_id=2, guild_id=1, message_id=2)
+    # session.add(group1)
+    # session.add(group2)
+    # session.commit()
 
     # anotherone = Guild(id=2, name="test guild 2", submission_channel=1, verification_channel=2)
     # session.add(anotherone)
@@ -34,7 +34,7 @@ with Session(engine) as session:
     # session.add(group2)
     # session.commit()
 
-    # stmt = select(Guild)
-    # guild = session.scalars(stmt)
+    stmt = select(Guild).where(Guild.name == "Dakiny")
+    guild = session.scalar(stmt)
 
-#     print(guild)
+    print(guild.category_groups)