Added relationship between receipt and item

Mythridor · Aug 19, 2024 · 7b84096 · 7b84096
1 parent 40e4ade
commit 7b84096
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 26 deletions.
diff --git a/spendizer/sql/model/__init__.py → spendizer/model/__init__.py b/spendizer/sql/model/__init__.py → spendizer/model/__init__.py
diff --git a/spendizer/model/main.py b/spendizer/model/main.py
@@ -0,0 +1,28 @@
+import datetime
+import uuid
+from sqlmodel import Field, SQLModel, Relationship
+
+
+class Receipt(SQLModel, table=True):
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    date: datetime.datetime = Field(nullable=False)
+    market_id: uuid.UUID = Field(foreign_key='marketplace.id', nullable=False)
+    total: float = Field(nullable=False)
+    items: list['Item'] = Relationship(
+        back_populates='receipt', nullable=False
+    )
+
+
+class Item(SQLModel, table=True):
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    name: str = Field(nullable=False)
+    receipt_id: uuid.UUID | None = Field(
+        default=None, foreign_key='receipt.id'
+    )
+    unit_price: float = Field(nullable=False)
+    receipt: Receipt = Relationship(back_populates='items', nullable=False)
+
+
+class Marketplace(SQLModel, table=True):
+    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
+    name: str = Field(nullable=False)
diff --git a/spendizer/sql/model/main.py b/spendizer/sql/model/main.py