wip

templates/model/learn_sql_model/models/{{modelname.lower()}}.py.jinja

@@ -0,0 +1,99 @@
+from typing import Optional
+
+from fastapi import HTTPException
+from pydantic import BaseModel
+from sqlmodel import Field, Relationship, SQLModel, Session, select
+
+from learn_sql_model.config import Config
+from learn_sql_model.models.pet import Pet
+
+
+class {{modelname}}Base(SQLModel, table=False):
+
+
+class {{modelname}}({{modelname.lower()}}Base, table=True):
+    id: Optional[int] = Field(default=None, primary_key=True)
+
+
+class {{modelname}}Create({{modelname.lower()}}Base):
+    ...
+
+    def post(self, config: Config) -> {{modelname}}:
+        config.init()
+        with Session(config.database.engine) as session:
+            db_{{modelname.lower()}} = {{modelname}}.from_orm(self)
+            session.add(db_{{modelname.lower()}})
+            session.commit()
+            session.refresh(db_{{modelname.lower()}})
+            return db_{{modelname.lower()}}
+
+
+class {{modelname}}Read({{modelname.lower()}}Base):
+    id: int
+
+    @classmethod
+    def get(
+        cls,
+        config: Config,
+        id: int,
+    ) -> {{modelname}}:
+
+        with config.database.session as session:
+            {{modelname.lower()}} = session.get({{modelname}}, id)
+            if not {{modelname.lower()}}:
+                raise HTTPException(status_code=404, detail="{{modelname}} not found")
+        return {{modelname.lower()}}
+
+    @classmethod
+    def list(
+        self,
+        config: Config,
+        where=None,
+        offset=0,
+        limit=None,
+    ) -> {{modelname}}:
+
+        with config.database.session as session:
+            statement = select({{modelname}})
+            if where != "None":
+                from sqlmodel import text
+
+                statement = statement.where(text(where))
+            statement = statement.offset(offset).limit(limit)
+            {{modelname.lower()}}es = session.exec(statement).all()
+        return {{modelname.lower()}}es
+
+
+class {{modelname}}Update(SQLModel):
+    # id is required to get the {{modelname.lower()}}
+    id: int
+
+    # all other fields, must match the model, but with Optional default None
+
+    def update(self, config: Config) -> {{modelname}}:
+        with Session(config.database.engine) as session:
+            db_{{modelname.lower()}} = session.get({{modelname}}, self.id)
+            if not db_{{modelname.lower()}}:
+                raise HTTPException(status_code=404, detail="{{modelname}} not found")
+            {{modelname.lower()}}_data = self.dict(exclude_unset=True)
+            for key, value in {{modelname.lower()}}_data.items():
+                if value is not None:
+                    setattr(db_{{modelname.lower()}}, key, value)
+            session.add(db_{{modelname.lower()}})
+            session.commit()
+            session.refresh(db_{{modelname.lower()}})
+            return db_{{modelname.lower()}}
+
+
+class {{modelname}}Delete(BaseModel):
+    id: int
+
+    def delete(self, config: Config) -> {{modelname}}:
+        config.init()
+        with Session(config.database.engine) as session:
+            {{modelname.lower()}} = session.get({{modelname}}, self.id)
+            if not {{modelname.lower()}}:
+                raise HTTPException(status_code=404, detail="{{modelname}} not found")
+            session.delete({{modelname.lower()}})
+            session.commit()
+            return {"ok": True}