feat: add Proof.type field

alembic/versions/20231125_1726_cce1da5c6733_add_proof_type_field.py

@@ -0,0 +1,30 @@
+"""Add Proof.type field
+
+Revision ID: cce1da5c6733
+Revises: 012466c0013e
+Create Date: 2023-11-25 17:26:49.022693
+
+"""
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "cce1da5c6733"
+down_revision: Union[str, None] = "012466c0013e"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column("proofs", sa.Column("type", sa.String(length=255), nullable=True))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column("proofs", "type")
+    # ### end Alembic commands ###

app/enums.py

@@ -11,3 +11,9 @@ class LocationOSMEnum(Enum):
     NODE = "NODE"
     WAY = "WAY"
     RELATION = "RELATION"
+
+
+class ProofTypeEnum(Enum):
+    PRICE_TAG = "PRICE_TAG"
+    RECEIPT = "RECEIPT"
+    GDPR_REQUEST = "GDPR_REQUEST"

app/models.py

@@ -16,7 +16,7 @@
 from sqlalchemy_utils.types.choice import ChoiceType
 
 from app.db import Base
-from app.enums import CurrencyEnum, LocationOSMEnum
+from app.enums import CurrencyEnum, LocationOSMEnum, ProofTypeEnum
 
 force_auto_coercion()
 
@@ -77,6 +77,8 @@ class Proof(Base):
     file_path = Column(String, nullable=False)
     mimetype = Column(String, index=True)
 
+    type = Column(ChoiceType(ProofTypeEnum))
+
     prices: Mapped[list["Price"]] = relationship(back_populates="proof")
 
     owner = Column(String, index=True)

app/schemas.py

@@ -12,7 +12,7 @@
     model_validator,
 )
 
-from app.enums import CurrencyEnum, LocationOSMEnum
+from app.enums import CurrencyEnum, LocationOSMEnum, ProofTypeEnum
 from app.models import Price
 
 
@@ -181,6 +181,7 @@ class ProofCreate(BaseModel):
 
     file_path: str
     mimetype: str
+    type: ProofTypeEnum
 
 
 class ProofBase(ProofCreate):