simulate_two.py

import random
from recap import recap, do_nothing
from simul_methods import stud_is_sick, teach_is_sick
import csv

#Odds of a person catching the virus outside of school (per day)
DAILY_EXT_ODDS_CHILD = 0.00007866
DAILY_EXT_ODDS_ADULT = 0.0001361

#Used to determine secondary exposure within classroom (0.035 * 0.6)
HEAVY_EXPOSURE_ODDS = 0.014

def simulate_two(schools, days_in_sim, num_sims):
  print("Beginning simulation of Scenario 2")
  sim_ct = 1
  days = days_in_sim
  with open('sickness_sim_2.csv', mode = 'w') as sick_file:
    sick_writer = csv.writer(sick_file, delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL)
    while(sim_ct < num_sims+1):
      days = days_in_sim
      day_of_the_week = 1
      testing_flag = False
      while(days > 0):
        exposed_students = 0
        exposed_teachers = 0
        for school in schools:
          #For healthy students, simulate whether they are a new primary case. For sick students, add to exposure list and decrement sickness counters
          for stud in school.students: 
            if(isinstance(stud, int)):
              do_nothing
            elif(stud.is_sick):
              stud_is_sick(stud, school, testing_flag)
            else:
              if(random.random() < DAILY_EXT_ODDS_CHILD):
                stud.become_sick("s")
                sick_writer.writerow([days, 'Student', 'Primary', stud.severity, school.school_ID, sim_ct])
      #         print("Student",stud.ID, "is now sick")

          #For healthy teachers, simulate whether they are a new primary case. For sick teachers, add to exposure list and decrement sickness counters
          for teach in school.teachers:
            if(isinstance(teach, int)):
              do_nothing()
            elif(teach.is_sick):
              teach_is_sick(stud,school,testing_flag)
            else:
              if(random.random() < DAILY_EXT_ODDS_ADULT):
                teach.become_sick("t")
                sick_writer.writerow([days, 'Teacher', 'Primary', teach.severity, school.school_ID, sim_ct])
      #         print("Teacher",teach.ID, "is now sick")

          #For each classroom with an infected person, expose all others and determine if anyone becomes a secondary case.
          if(day_of_the_week < 6):
            for stud in school.students:
              if(isinstance(stud, int)):
                do_nothing()
              else:
                if(stud.is_sick):
                  do_nothing()
                else:
                  if(stud.classroom in school.exposed_classrooms):
                    exposed_students+=1
        #           print("Student", stud.ID, "has been exposed")
                    if(random.random() < HEAVY_EXPOSURE_ODDS):
                      stud.become_sick("s")
                      sick_writer.writerow([days, 'Student', 'Secondary', stud.severity, school.school_ID, sim_ct])
                      if(testing_flag and stud.classroom in school.symptomatic_classrooms):
                        stud.in_school = False
          #            print("Student", stud.ID, "is now sick")
            for teach in school.teachers:
              if(isinstance(teach, int)):
                do_nothing()
              else:
                if(teach.is_sick):
                  do_nothing
                else:
                  if(teach.classroom in school.exposed_classrooms):
                    exposed_teachers+=1
            #       print("Teacher", teach.ID, "has been exposed")
                    if(random.random()< HEAVY_EXPOSURE_ODDS):
                      teach.become_sick("t")
                      sick_writer.writerow([days, 'Teacher', 'Secondary', teach.severity, school.school_ID, sim_ct])
                      if(testing_flag and teach.classroom in school.symptomatic_classrooms):
                        teach.in_school = False
          school.symptomatic_classrooms = []
          school.exposed_classrooms = []
        sick_writer.writerow([days, exposed_students, exposed_teachers, sim_ct])
        day_of_the_week +=1      
        day_of_the_week +=1
        if(day_of_the_week == 7):
          day_of_the_week = 1
        elif(day_of_the_week > 7):
          print("ERROR: exceeded max days of the week")
        days -= 1
        testing_flag = False
      print("Simulation", sim_ct, "complete.")
      sim_ct += 1

  # recap(schools)