diff --git a/app/routers/e_impexium.py b/app/routers/e_impexium.py index 5365979..a12b47b 100644 --- a/app/routers/e_impexium.py +++ b/app/routers/e_impexium.py @@ -94,9 +94,32 @@ async def event_import_reg( event_badge_data['event_id'] = event_id event_badge_data['external_id'] = external_id - if event_registrant.get('registrantTypeCode'): + if reg_type_code := event_registrant.get('registrantTypeCode'): event_badge_data['badge_type_code'] = event_registrant.get('registrantTypeCode') # Using this as the badge_type - event_badge_data['badge_type'] = event_registrant.get('registrantTypeName') + + ishlt_m = ('AHMB', 'IMBR', 'MBR') + ishlt_nm = ('ANHM', 'INMB', 'NMBR') + ishlt_student_m = ('SMBR') + ishlt_student_nm = ('SNMB') + ishlt_ex_all = ('EX') + ishlt_ex_booth = ('EX') + ishlt_guest = ('GUEST') + if reg_type_code in ishlt_m: + event_badge_data['badge_type'] = 'Member' + elif reg_type_code in ishlt_nm: + event_badge_data['badge_type'] = 'Non-Member' + elif reg_type_code in ishlt_student_m: + event_badge_data['badge_type'] = 'Student/Trainee Member' + elif reg_type_code in ishlt_student_m: + event_badge_data['badge_type'] = 'Student/Trainee Non-Member' + elif reg_type_code in ishlt_ex_all: + event_badge_data['badge_type'] = 'Exhibitor All Access' + elif reg_type_code in ishlt_ex_booth: + event_badge_data['badge_type'] = 'Exhibitor Booth Staff' + elif reg_type_code in ishlt_guest: + event_badge_data['badge_type'] = 'Guest' + else: + event_badge_data['badge_type'] = event_registrant.get('registrantTypeName') elif e_impexium_event_id == 'EX22_AM': event_badge_data['badge_type_code'] = 'EX' event_badge_data['badge_type'] = 'Exhibit Staff' diff --git a/app/routers/event_badge.py b/app/routers/event_badge.py index 59990b1..0f052f4 100644 --- a/app/routers/event_badge.py +++ b/app/routers/event_badge.py @@ -1,4 +1,4 @@ -import datetime +import datetime, time from fastapi import APIRouter, Body, Depends, Header, HTTPException, Query, Response, status from pydantic import BaseModel, EmailStr, Field from typing import Dict, List, Optional, Set, Union @@ -328,6 +328,8 @@ async def get_event_badge_obj( log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL log.debug(locals()) + # time.sleep(2) + if event_badge_id := redis_lookup_id_random(record_id_random=event_badge_id, table_name='event_badge'): pass else: return mk_resp(data=None, status_code=404, response=commons.response)