From 84fa00fa165ec3c3e760cd8da5e432a2642fdd43 Mon Sep 17 00:00:00 2001 From: Scott Idem Date: Mon, 14 Mar 2022 15:55:40 -0400 Subject: [PATCH] Work on badge template and related --- app/models/event_badge_models.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/app/models/event_badge_models.py b/app/models/event_badge_models.py index d91c770..7359ad5 100644 --- a/app/models/event_badge_models.py +++ b/app/models/event_badge_models.py @@ -34,6 +34,9 @@ class Event_Badge_Base(BaseModel): event_id_random_only: Optional[str] event_id_only: Optional[int] + event_badge_template_id_random: Optional[str] + event_badge_template_id: Optional[int] + event_person_id_random: Optional[str] event_person_id: Optional[int] @@ -143,6 +146,13 @@ class Event_Badge_Base(BaseModel): return redis_lookup_id_random(record_id_random=id_random, table_name='event') return None + @validator('event_badge_template_id', always=True) + def event_badge_template_id_lookup(cls, v, values, **kwargs): + if isinstance(v, int) and v > 0: return v + elif id_random := values.get('event_badge_template_id_random'): + return redis_lookup_id_random(record_id_random=id_random, table_name='event_badge_template') + return None + @validator('event_person_id', always=True) def event_person_id_lookup(cls, v, values, **kwargs): if isinstance(v, int) and v > 0: return v