imcompiling/agon
1
0
Fork 0
Code Issues 17 Pull Requests 4 Actions Packages Activity

Test file after connecting to prod database and testing relationships.

Browse Source
This commit is contained in:
imcompiling
2025-03-03 23:02:40 -05:00
parent f74b08b02e
commit 43f51f7f07
1 changed files with 15 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
database/database_test.py
+15 -15
View File
@@ -1,9 +1,9 @@
import os import os
from dotenv import load_dotenv 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() load_dotenv()
@@ -12,19 +12,19 @@ SQL_HOST = os.getenv("SQL_HOST", "")
SQL_PASS = os.getenv("SQL_PASS", "") SQL_PASS = os.getenv("SQL_PASS", "")
SQL_USER = os.getenv("SQL_USER", "") SQL_USER = os.getenv("SQL_USER", "")
url = "sqlite:///database.db" # url = "sqlite:///database.db"
# url = f"mysql+pymysql://{SQL_USER}:{SQL_PASS}@{SQL_HOST}/{SQL_DB}" url = f"mysql+pymysql://{SQL_USER}:{SQL_PASS}@{SQL_HOST}/{SQL_DB}"
engine = create_engine(url) engine = create_engine(url)
SQLModel.metadata.create_all(engine) # SQLModel.metadata.create_all(engine)
with Session(engine) as session: with Session(engine) as session:
guild = Guild(id=1, name="test guild", submission_channel=1, verification_channel=2) # guild = Guild(id=1, name="test guild", submission_channel=1, verification_channel=2)
session.add(guild) # session.add(guild)
group1 = CategoryGroup(name="group1", channel_id=1, guild_id=1, message_id=1) # 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) # group2 = CategoryGroup(name="group2", channel_id=2, guild_id=1, message_id=2)
session.add(group1) # session.add(group1)
session.add(group2) # session.add(group2)
session.commit() # session.commit()
# anotherone = Guild(id=2, name="test guild 2", submission_channel=1, verification_channel=2) # anotherone = Guild(id=2, name="test guild 2", submission_channel=1, verification_channel=2)
# session.add(anotherone) # session.add(anotherone)
@@ -34,7 +34,7 @@ with Session(engine) as session:
# session.add(group2) # session.add(group2)
# session.commit() # session.commit()
# stmt = select(Guild) stmt = select(Guild).where(Guild.name == "Dakiny")
# guild = session.scalars(stmt) guild = session.scalar(stmt)
# print(guild) print(guild.category_groups)