def _create_person(self, pers):
from sormas import Sex
date = pers.reporting_date
birthdate_yyyy = None if type(date) is float else int(date.split('-')[0]) - pers.age
lat = 0 if np.isnan(pers.latitude) else pers.latitude # Fixme 0 condition
lon = 0 if np.isnan(pers.longitude) else pers.longitude # Fixme 0 condition
location_dto = gen_location_dto(
lat, lon,
region_ref(self.regions['Voreingestellte Bundesländer']['uuid']),
self.districts['Voreingestellter Landkreis']['uuid']
)
person = gen_person_dto(
first_name=pers.first_name,
last_name=pers.family_name,
# todo other values like diverse
sex=Sex.MALE if pers.sex == 'm' else Sex.FEMALE,
birthdate_yyyy=birthdate_yyyy,
address=location_dto,
present_condition=PresentCondition.DEAD if pers.died else PresentCondition.ALIVE
)
# todo use Person class?
return person
def pre_populate_infected(self, n=5):
for _ in range(n):
person = gen_person_dto()
case = Infected(person, Disease.CORONAVIRUS)
self.today.infected.append(case)
def pre_populate_susceptible(self, n=5):
for _ in range(n):
p = gen_person_dto()
self.today.susceptible.append(p)