def medical_assist(city: City) -> None:
"""
Calculate medical assistances.
Lets healthy medical workers assist others and increase their recovery probability.
:param city: the City object
:precondition: city is a valid instance of City
"""
patients_per_doctor = 3
patient_prob_recovery_increase = 0.05
citizens = city.get_citizens()
# LIST COMPREHENSION
needs_assistance = [
citizen for citizen in citizens if citizen.is_infected()
]
for citizen in citizens:
if citizen.get_role() == 'medical' and not citizen.is_infected():
# RANGE FUNCTION
for i in range(patients_per_doctor):
if len(needs_assistance) == 0:
return
patient = needs_assistance.pop()
new_prob_recovery = patient.get_prob_recovery(
) + patient_prob_recovery_increase
patient.update_prob_recovery(min(100, new_prob_recovery))
patient.set_medical_assist()
def calculate_hp(city: City) -> None:
"""
Calculate HP.
Decays the HP of infected citizens by a fixed amount.
:param city: the City object
:precondition: city is a valid instance of City
"""
for citizen in city.get_citizens():
if citizen.is_infected() and not citizen.is_recovered():
new_hp = citizen.get_hp() - city.get_daily_decay()
citizen.update_hp(max(new_hp, 0))
if citizen.get_hp() == 0:
citizen.set_deceased()
def calculate_ppe(city: City) -> None:
"""
Calculate personal protective equipment.
Consumes one PPE per medical worker and protects them from infection.
:param city: the City object
:precondition: city is a valid instance of City
"""
num_ppe = city.get_num_ppe()
for citizen in city.get_citizens():
if citizen.get_role() == 'medical':
if num_ppe > 0:
num_ppe -= 1
citizen.update_prob_infected(0.0)
elif citizen.get_prob_infected() == 0:
citizen.update_prob_infected(round(uniform(0.05, 0.12), 2))
city.set_num_ppe(num_ppe)