smarthome.py

from device import Device, Light, Thermostat, DoorLock
from controller import Controller, LightingController, TemperatureController


class SmartHome:
    def __init__(self):
        self.devices = []
        self.controllers = []

    def add_device(self, device: Device):
        self.devices.append(device)

    def add_controller(self, controller: Controller):
        self.controllers.append(controller)

    def get_device_status(self, device_id):
        for device in self.devices:
            if device.device_id == device_id and isinstance(device, Device):
                return device.status

    def get_controller_devices(self, controller_id):
        for controller in self.controllers:
            if controller.controller_id == controller_id and isinstance(controller, Controller):
                return [device.name for device in controller.devices]


smart_home = SmartHome()

# Adding devices
light1 = Light("L001", "Living Room Light", brightness=50)
light2 = Light("L002", "Bedroom Room Light", brightness=50)
thermostat1 = Thermostat("T001", "Hallway Thermostat", current_temp=19)
doorlock1 = DoorLock("D001", "Front Door Lock")

light1.turn_on()
light2.turn_on()
thermostat1.turn_on()

smart_home.add_device(light1)
smart_home.add_device(light2)
smart_home.add_device(thermostat1)
smart_home.add_device(doorlock1)

# Adding controllers
temp_controller = TemperatureController("C001", "Main Temperature Controller")
lighting_controller = LightingController("C002", "Main Lighting Controller")

smart_home.add_controller(temp_controller)
smart_home.add_controller(lighting_controller)

# Assign devices to controllers
temp_controller.add_device(thermostat1)
lighting_controller.add_device(light1)
lighting_controller.add_device(light2)

# Control devices
temp_controller.maintain_temperature()
temp_controller.get_current_temperature(thermostat1)
lighting_controller.adjust_lighting(75, "warm white")
print(doorlock1.turn_on())
print(light2.turn_on())
print(lighting_controller.control_device('L002', 'on'))
print(lighting_controller.control_device('L002', 'on'))

print(smart_home.get_device_status("L001"))  # True
print(smart_home.get_device_status("T001"))  # True
print(smart_home.get_controller_devices("C002"))  # ["Living Room Light"]