diff --git a/database/database_test.py b/database/database_test.py index 5ef69be..7e9218a 100644 --- a/database/database_test.py +++ b/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)