init

learn_sql_model/config.py

@@ -0,0 +1,40 @@
+from typing import TYPE_CHECKING
+
+from pydantic import BaseSettings
+from sqlmodel import SQLModel, Session, create_engine
+
+from learn_sql_model.models import Hero, Pet
+from learn_sql_model.standard_config import load
+
+models = [Hero, Pet]
+
+if TYPE_CHECKING:
+    from sqlalchemy import Engine
+
+
+class Config(BaseSettings):
+    database_url: str = "sqlite:///database.db"
+
+    class Config:
+        env_prefix = "LEARN_SQL_MODEL_"
+
+    @property
+    def engine(self) -> "Engine":
+        return create_engine(self.database_url)
+
+    @property
+    def session(self) -> "Session":
+        return Session(self.engine)
+
+    def create_db_and_tables(self) -> None:
+        SQLModel.metadata.create_all(self.engine)
+
+    # def create_endpoints(self) -> None:
+    # for model in models:
+    # app.post("/heroes/")(Hero.post_local)
+    # app.get("/heroes/")(Hero.read_heroes)
+
+
+raw_config = load("learn_sql_model")
+config = Config(**raw_config)
+config.create_db_and_tables()