Skip to content

Commit

Permalink
feat: Add new id and created_at claim fields
Browse files Browse the repository at this point in the history
  • Loading branch information
clemlesne committed Jan 16, 2024
1 parent 28d2d79 commit 83c104a
Showing 1 changed file with 19 additions and 5 deletions.
24 changes: 19 additions & 5 deletions models/claim.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,22 @@
from pydantic import BaseModel
from typing import Optional
from datetime import datetime
from pydantic import BaseModel, Field
from typing import Optional, Set
import random
import string


class ClaimModel(BaseModel):
additional_documentation: Optional[str] = None
claim_explanation: Optional[str] = None
# Immutable fields
id: str = Field(
default_factory=(
lambda: "".join(
random.choice(string.ascii_lowercase + string.digits) for _ in range(6)
)
),
frozen=True,
)
created_at: datetime = Field(default_factory=datetime.utcnow, frozen=True)
# Editable fields
extra_details: Optional[str] = None
incident_date_time: Optional[str] = None
incident_description: Optional[str] = None
Expand All @@ -27,4 +38,7 @@ class ClaimModel(BaseModel):

@staticmethod
def editable_fields() -> Set[str]:
return ClaimModel.model_json_schema()["properties"].keys()
return ClaimModel.model_json_schema()["properties"].keys() - [
"id",
"created_at",
]

0 comments on commit 83c104a

Please sign in to comment.