wip

htmx_patterns/models.py

@@ -0,0 +1,60 @@
+from datetime import datetime
+from typing import Optional
+
+from faker import Faker
+from polyfactory.factories.pydantic_factory import ModelFactory
+from pydantic import UUID4
+from sqlmodel import Field, SQLModel
+
+faker = Faker()
+
+
+class PersonBase(SQLModel, table=False):
+    id: UUID4
+    name: str
+    birthday: datetime
+    phone_number: str
+
+    def get_by_id(session, id):
+        return session.get(Person, id)
+
+    def save(self, session):
+        session.add(self)
+        session.commit()
+        session.refresh(self)
+        return self
+
+    def delete(self, session):
+        session.delete(self)
+        session.commit()
+        return self
+
+    def update(self, session):
+        session.merge(self)
+        session.commit()
+        session.refresh(self)
+        return self
+
+    def all(session):
+        return session.query(Person).all()
+
+    def paginate(session, page=1, page_size=10):
+        return session.query(Person).offset((page - 1) * page_size).limit(page_size)
+
+
+class Person(PersonBase, table=True):
+    id: Optional[int] = Field(default=None, primary_key=True)
+
+
+class PersonCreate(PersonBase):
+    pass
+
+
+class PersonRead(PersonBase):
+    id: int = Field(default=None, primary_key=True)
+
+
+class PersonFactory(ModelFactory):
+    name = faker.name
+    phone_number = faker.phone_number
+    __model__ = Person