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