@@ -374,8 +374,9 @@ def update_citizen_db():
374
374
375
375
citizen_id = request .values ['citizen_id' ]
376
376
vaccine_name = request .values ['vaccine_name' ]
377
+ option = request .values ['option' ]
377
378
378
- if not (citizen_id and vaccine_name ):
379
+ if not (citizen_id and vaccine_name and option ):
379
380
logger .error (REPORT_FEEDBACK ["missing_key" ])
380
381
return {"feedback" : REPORT_FEEDBACK ["missing_key" ]}
381
382
@@ -391,32 +392,29 @@ def update_citizen_db():
391
392
logger .error (REPORT_FEEDBACK ["invalid_vaccine" ])
392
393
return {"feedback" : REPORT_FEEDBACK ["invalid_vaccine" ]}
393
394
394
- try :
395
- option = request .values ['option' ]
396
- if (option == "walk-in" ):
397
- if is_reserved (citizen_id ):
398
- logger .error (REPORT_FEEDBACK ["has_reservation" ])
399
- return {"feedback" : REPORT_FEEDBACK ["has_reservation" ]}
400
-
401
- try :
402
- citizen = get_citizen (citizen_id )
403
- is_valid , feedback = validate_vaccine (citizen , vaccine_name )
404
-
405
- if not is_valid :
406
- logger .error ("{} - {}" .format (citizen_id ,
407
- feedback ['feedback' ]))
408
- return feedback
409
-
410
- citizen .vaccine_taken = [
411
- * (citizen .vaccine_taken ), vaccine_name
412
- ]
413
- db .session .commit ()
414
-
415
- except :
416
- db .session .rollback ()
417
- logger .error (REPORT_FEEDBACK ["other" ])
418
- return {"feedback" : REPORT_FEEDBACK ["other" ]}
419
- except :
395
+ if (option == "walk-in" ):
396
+ if is_reserved (citizen_id ):
397
+ logger .error (REPORT_FEEDBACK ["has_reservation" ])
398
+ return {"feedback" : REPORT_FEEDBACK ["has_reservation" ]}
399
+ try :
400
+ citizen = get_citizen (citizen_id )
401
+ is_valid , feedback = validate_vaccine (citizen , vaccine_name )
402
+ if not is_valid :
403
+ logger .error ("{} - {}" .format (citizen_id ,
404
+ feedback ['feedback' ]))
405
+ return feedback
406
+ citizen .vaccine_taken = [* (citizen .vaccine_taken ), vaccine_name ]
407
+ db .session .commit ()
408
+ except :
409
+ db .session .rollback ()
410
+ logger .error (REPORT_FEEDBACK ["other" ])
411
+ return {"feedback" : REPORT_FEEDBACK ["other" ]}
412
+
413
+ logger .info ("{} - updated citizen - vaccine name: {}" .format (
414
+ citizen_id , vaccine_name ))
415
+ return {"feedback" : REPORT_FEEDBACK ["success" ]}
416
+
417
+ elif (option == "reserve" ):
420
418
if not is_reserved (citizen_id ):
421
419
logger .error (REPORT_FEEDBACK ["not_reservation" ])
422
420
return {"feedback" : REPORT_FEEDBACK ["not_reservation" ]}
@@ -426,19 +424,22 @@ def update_citizen_db():
426
424
citizen_data .vaccine_taken = [
427
425
* (citizen_data .vaccine_taken ), vaccine_name
428
426
]
429
- reservation_data = get_unchecked_reservations (
430
- db , citizen_id ).filter (
431
- Reservation .vaccine_name == vaccine_name ).first ()
427
+ reservation_data = get_unchecked_reservations (citizen_id ).filter (
428
+ Reservation .vaccine_name == vaccine_name ).first ()
432
429
reservation_data .checked = True
433
430
db .session .commit ()
434
431
except :
435
432
db .session .rollback ()
436
433
logger .error (REPORT_FEEDBACK ["not_match_vaccine" ])
437
434
return {"feedback" : REPORT_FEEDBACK ["not_match_vaccine" ]}
438
435
439
- logger .info ("{} - updated citizen - vaccine name: {}" .format (
440
- citizen_id , vaccine_name ))
441
- return {"feedback" : REPORT_FEEDBACK ["success" ]}
436
+ logger .info ("{} - updated citizen - vaccine name: {}" .format (
437
+ citizen_id , vaccine_name ))
438
+ return {"feedback" : REPORT_FEEDBACK ["success" ]}
439
+
440
+ else :
441
+ logger .error (REPORT_FEEDBACK ["invalid_option" ])
442
+ return {"feedback" : REPORT_FEEDBACK ["invalid_option" ]}
442
443
443
444
@app .route ('/register_user' , methods = ['POST' ])
444
445
@cross_origin ()
0 commit comments