diff --git a/src/teilchensammler_cli/database.py b/src/teilchensammler_cli/database.py new file mode 100644 index 0000000..fe3906a --- /dev/null +++ b/src/teilchensammler_cli/database.py @@ -0,0 +1,8 @@ +from sqlmodel import SQLModel, create_engine + +sqlite_url = "sqlite:///:memory:" +engine = create_engine(sqlite_url, echo=True) + + +def create_db_and_tables(): + SQLModel.metadata.create_all(engine) diff --git a/src/teilchensammler_cli/models.py b/src/teilchensammler_cli/models.py index 397733d..13e7dc2 100644 --- a/src/teilchensammler_cli/models.py +++ b/src/teilchensammler_cli/models.py @@ -3,8 +3,6 @@ import uuid from sqlmodel import ( Field, SQLModel, - create_engine, - text, ) @@ -21,11 +19,5 @@ class Teilchen(TeilchenInput, table=True): number: int = Field(default=1) -sqlite_url = "sqlite:///:memory:" -engine = create_engine(sqlite_url, echo=True) -def create_db_and_tables(): - SQLModel.metadata.create_all(engine) - with engine.connect() as connection: - _ = connection.execute(text("PRAGMA foreign_keys=ON")) # for sqlite only