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

Just more testing.

Browse Source
This commit is contained in:
imcompiling
2025-03-03 21:53:19 -05:00
parent 0ad425b982
commit ce3ed3c001
1 changed files with 20 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
database/database_test.py
+20 -9
View File
@@ -1,18 +1,29 @@
from sqlmodel import Session, SQLModel, create_engine
from sqlmodel import Session, SQLModel, create_engine, select
from database.models import CategoryGroup, Guild
from models import Guild
url = "sqlite:///database.db"
engine = create_engine(url)
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)
# 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()
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)
# anotherone = Guild(id=2, name="test guild 2", submission_channel=1, verification_channel=2)
# session.add(anotherone)
# group1 = CategoryGroup(name="group1", channel_id=1, guild_id=2, message_id=1)
# group2 = CategoryGroup(name="group2", channel_id=2, guild_id=2, message_id=2)
# session.add(group1)
# session.add(group2)
# session.commit()
session.commit()
stmt = select(Guild).where(Guild.id == 1)
guild = session.scalar(stmt)
print(guild.category_groups)