74 lines
1.8 KiB
Python
74 lines
1.8 KiB
Python
|
# Python imports
|
||
|
|
import json
|
||
|
|
import asyncio
|
||
|
|
import time
|
||
|
|
|
||
|
|
# Lib imports
|
||
|
|
from kasa import Discover
|
||
|
|
|
||
|
|
# Application imports
|
||
|
|
|
||
|
|
|
||
|
|
class SmartDeviceManager:
|
||
|
|
def __init__(self):
|
||
|
|
self.BASE_IP = "192.168.0."
|
||
|
|
self.IPs = ["12", "13", "14", "15", "16", "17"]
|
||
|
|
self.devices = self.retrieve_devices()
|
||
|
|
|
||
|
|
|
||
|
|
def retrieve_devices(self):
|
||
|
|
devices = []
|
||
|
|
for ip in self.IPs:
|
||
|
|
device = self.get_device(self.BASE_IP + ip)
|
||
|
|
if device:
|
||
|
|
print("Device: {} is available...".format(device.alias))
|
||
|
|
print("{}".format(device.hw_info))
|
||
|
|
devices.append(device)
|
||
|
|
|
||
|
|
return devices
|
||
|
|
|
||
|
|
def set_device_on(self, device):
|
||
|
|
asyncio.run(device.turn_on())
|
||
|
|
|
||
|
|
def set_device_off(self, device):
|
||
|
|
asyncio.run(device.turn_off())
|
||
|
|
|
||
|
|
def set_led_on(self, device):
|
||
|
|
asyncio.run(device.set_led(True))
|
||
|
|
|
||
|
|
def set_led_off(self, device):
|
||
|
|
asyncio.run(device.set_led(False))
|
||
|
|
|
||
|
|
def set_device_alias(self, device, alias):
|
||
|
|
asyncio.run(device.set_alias(alias))
|
||
|
|
|
||
|
|
def update_info(self, device):
|
||
|
|
try:
|
||
|
|
asyncio.run(device.update())
|
||
|
|
except Exception as e:
|
||
|
|
pass
|
||
|
|
|
||
|
|
def get_device(self, addr):
|
||
|
|
try:
|
||
|
|
device = asyncio.run( Discover.discover_single(addr) )
|
||
|
|
self.update_info(device)
|
||
|
|
return device
|
||
|
|
except Exception as e:
|
||
|
|
print(repr(e))
|
||
|
|
return None
|
||
|
|
|
||
|
|
def pulsate(self, device, rate = 1):
|
||
|
|
state = "on"
|
||
|
|
while True:
|
||
|
|
if state == "on":
|
||
|
|
state = "off"
|
||
|
|
self.set_device_on(device)
|
||
|
|
else:
|
||
|
|
state = "on"
|
||
|
|
self.set_device_off(device)
|
||
|
|
|
||
|
|
time.sleep(rate)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
SmartDeviceManager()
|