Hospital-Simulation/main.py at main · mkaro91/Hospital-Simulation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
from hospital import Hospital
from department import Department
from doctor import Doctor
from nurse import Nurse

from enums import Severity, Shift

hospital = Hospital("City General")

icu = Department("ICU", capacity=3, accepted_severities=[Severity.CRITICAL, Severity.FATAL])
icu.add_doctor(Doctor("Dr. Patel", "Cardiology", Shift.DAY))
icu.add_doctor(Doctor("Dr. Nguyen", "Neurology", Shift.DAY))
icu.add_doctor(Doctor("Dr. Okafor", "Cardiology", Shift.NIGHT))
icu.add_doctor(Doctor("Dr. Chen", "Neurology", Shift.NIGHT))

icu.add_nurse(Nurse("Amy", "ICU", Shift.DAY))
icu.add_nurse(Nurse("Ben", "ICU", Shift.NIGHT))


er = Department("ER", capacity=5, accepted_severities=[Severity.SERIOUS, Severity.MODERATE])
er.add_doctor(Doctor("Dr. Lee", "Surgery",  Shift.DAY))
er.add_doctor(Doctor("Dr. Kim", "Emergency", Shift.DAY))
er.add_doctor(Doctor("Dr. Santos", "Surgery", Shift.NIGHT))
er.add_doctor(Doctor("Dr. Park", "Emergency", Shift.NIGHT))

er.add_nurse(Nurse("Clara", "ER", Shift.DAY))
er.add_nurse(Nurse("Dan", "ER", Shift.NIGHT))

general = Department("General Ward", capacity=10, accepted_severities=[Severity.MINOR, Severity.MODERATE])
general.add_doctor(Doctor("Dr. Nash", "General Practice", Shift.DAY))
general.add_doctor(Doctor("Dr. Cruz", "General Practice", Shift.NIGHT))

general.add_nurse(Nurse("Eve", "General Ward", Shift.DAY))
general.add_nurse(Nurse("Frank", "General Ward", Shift.NIGHT))

hospital.add_department(icu)
hospital.add_department(er)
hospital.add_department(general)

hospital.clock = 6
hospital.run(24)