SurgeryRoom.py

from Location import Location
from patient import Patient
import threading

class SurgeryRoom(threading.Thread):
    def __init__(self, id):
        self.id = id
        self.is_used = False
        self.surgery_stopped = True
        self.doctors_number = 0
        self.patient: Patient = None
        self.doctors = []
        self.lock = threading.Lock()

    def take_room(self, patient, doctors):
        self.lock.acquire()
        try:
            self.surgery_stopped = False
            self.doctors = doctors
            self.patient = patient
            self.doctors_number = len(self.doctors)
            self.is_used = True

            for doctor in self.doctors:
                doctor.surgery_room = self
        finally:
            self.lock.release()

    def stop_surgery(self):
        self.surgery_stopped = True

    def start_surgery(self):
        self.surgery_stopped = False

    def complete_surgery(self):        
        self.lock.acquire()
        try:
            self.is_used = False
            self.patient = None
            self.doctors = []
            self.patient = None
        except:
            pass
        finally:            
            self.lock.release()