monster/yeelight-mqtt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Lightbulbstate class extracted

Browse Source
This commit is contained in:
Your Name
2017-01-11 12:43:17 +03:00
parent 4d70d5d1a6
commit 910ed70e03
2 changed files with 66 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lightbulbstate.py
+64
View File
@@ -0,0 +1,64 @@
import pyyeelight
import os
import time
import threading
class LightBulbState:
bright = 0
color_temperature = 0
status = "off"
rgb = 0
ip = ""
name = ""
yeelight = None # yeelight object
def __init__(self, ip, yeelightObj):
self.yeelight = yeelightObj
self.name = yeelightObj.__name__
self.ip = ip
def update_properties(self, force = False):
if (force):
self.yeelight.refresh_property()
prop = self.yeelight.get_all_properties()
# print(prop)
self.bright = prop["bright"]
self.color_temperature = prop["ct"]
self.status = prop["power"]
self.rgb = prop["rgb"]
def hash(self):
return str(self.bright) + ":" + str(self.color_temperature) + ":" + str(self.status) + ":" + str(self.rgb)
def is_int(self, x):
try:
tmp = int(x)
return True
except Exception as e:
return False
def process_command(self, param, value):
try:
if (param == 'status'):
if (value == "on"):
print("Turning on bulb", self.name)
self.yeelight.turn_on()
if (value == "off"):
print("Turning off bulb", self.name)
self.yeelight.turn_off()
if (param == 'bright' and self.is_int(value)):
print("Setting brightness of bulb", self.name, 'to', value)
self.yeelight.set_brightness(int(value))
if (param == 'ct' and self.is_int(value)):
print("Setting temperature of bulb", self.name, 'to', value)
self.yeelight.set_color_temperature(int(value))
if (param == 'rgb' and self.is_int(value)):
intval = int(value)
Blue = intval & 255
Green = (intval >> 8) & 255
Red = (intval >> 16) & 255
print("Setting rgb of bulb", self.name, 'to', value)
self.yeelight.set_rgb_color(Red, Green, Blue)
except Exception as e:
print ('Error while set value of bulb', self.name , ' error:', e)
src/main.py
+2 -61
View File
@@ -3,72 +3,13 @@ import paho.mqtt.client as mqtt
import os
import time
import threading
class LightBulbState:
bright = 0
color_temperature = 0
status = "off"
rgb = 0
ip = ""
name = ""
yeelight = None # yeelight object
def __init__(self, ip, yeelightObj):
self.yeelight = yeelightObj
self.name = yeelightObj.__name__
self.ip = ip
def update_properties(self, force = False):
if (force):
self.yeelight.refresh_property()
prop = self.yeelight.get_all_properties()
# print(prop)
self.bright = prop["bright"]
self.color_temperature = prop["ct"]
self.status = prop["power"]
self.rgb = prop["rgb"]
def hash(self):
return str(self.bright) + ":" + str(self.color_temperature) + ":" + str(self.status) + ":" + str(self.rgb)
def is_int(self, x):
try:
tmp = int(x)
return True
except Exception as e:
return False
def process_command(self, param, value):
try:
if (param == 'status'):
if (value == "on"):
print("Turning on bulb", self.name)
self.yeelight.turn_on()
if (value == "off"):
print("Turning off bulb", self.name)
self.yeelight.turn_off()
if (param == 'bright' and self.is_int(value)):
print("Setting brightness of bulb", self.name, 'to', value)
self.yeelight.set_brightness(int(value))
if (param == 'ct' and self.is_int(value)):
print("Setting temperature of bulb", self.name, 'to', value)
self.yeelight.set_color_temperature(int(value))
if (param == 'rgb' and self.is_int(value)):
intval = int(value)
Blue = intval & 255
Green = (intval >> 8) & 255
Red = (intval >> 16) & 255
print("Setting rgb of bulb", self.name, 'to', value)
self.yeelight.set_rgb_color(Red, Green, Blue)
except Exception as e:
print ('Error while set value of bulb', self.name , ' error:', e)
from lightbulbstate import LightBulbState
MQTT_SERVER = os.getenv('MQTT_SERVER', "")
MQTT_PORT = os.getenv('MQTT_PORT', 1883)
MQTT_USER = os.getenv('MQTT_USER', "")
MQTT_PASS = os.getenv('MQTT_PASS', "")
QUERY_TIME = 3
QUERY_TIME = os.getenv('QUERY_TIME', 3)
ips = dict({
'192.168.1.51':'main-mono',