Rename database_setup directory to database.
This commit is contained in:
@@ -0,0 +1,18 @@
|
||||
from sqlmodel import Session, SQLModel, create_engine
|
||||
|
||||
from database.models import CategoryGroup, 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)
|
||||
|
||||
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()
|
||||
Reference in New Issue
Block a user