Add initial version
This commit is contained in:
commit
2568658ac3
151
vu1.py
Normal file
151
vu1.py
Normal file
|
@ -0,0 +1,151 @@
|
|||
from os.path import join as joinpath
|
||||
import json
|
||||
import colorsys
|
||||
import requests
|
||||
|
||||
API_BASE = "api/v0/dial"
|
||||
|
||||
class DialServer():
|
||||
def __init__(self, url, key):
|
||||
self.url = url
|
||||
self._base = joinpath(self.url, API_BASE)
|
||||
self.key = key
|
||||
self._dials = []
|
||||
pass
|
||||
|
||||
@property
|
||||
def dials(self):
|
||||
dial_data = self._req("list")
|
||||
for d in dial_data:
|
||||
dial = Dial(self, d["uid"])
|
||||
if not dial in self._dials:
|
||||
self._dials.append(dial)
|
||||
return self._dials
|
||||
|
||||
@property
|
||||
def color(self):
|
||||
return None
|
||||
|
||||
@color.setter
|
||||
def color(self, color):
|
||||
for dial in self._dials:
|
||||
dial.color = color
|
||||
|
||||
def _req(self, endpoint, params={}, post=False):
|
||||
data_dict_base = {
|
||||
"key": self.key
|
||||
}
|
||||
data_dict = {**data_dict_base, **params}
|
||||
|
||||
req = requests.post(joinpath(self._base, endpoint), data=data_dict_base, files=data_dict) if post else requests.get(joinpath(self._base, endpoint), data=data_dict)
|
||||
|
||||
try:
|
||||
result = req.json()
|
||||
if result["status"] == "ok":
|
||||
return result["data"]
|
||||
else:
|
||||
raise Exception
|
||||
except KeyError as e:
|
||||
raise e
|
||||
|
||||
class Dial():
|
||||
def __init__(self, server, uid):
|
||||
self._server = server
|
||||
self.uid = uid
|
||||
pass
|
||||
|
||||
def __eq__(self, other):
|
||||
try:
|
||||
return self.uid.lower() == other.uid.lower() #if other is another dial instance
|
||||
except AttributeError:
|
||||
return self.uid.lower() == other.lower() #fall back to uid string comparision
|
||||
|
||||
def __repr__(self):
|
||||
return f"{super().__repr__()} - {self.name}, {self.uid}"
|
||||
|
||||
def _get_status(self):
|
||||
s = self._server._req(f"{self.uid}/status")
|
||||
self._index = int(s["index"])
|
||||
self._name = s["dial_name"]
|
||||
self._value = int(s["value"])
|
||||
self._fw_hash = s["fw_hash"]
|
||||
self._fw_version = s["fw_version"]
|
||||
self._hw_version = s["hw_version"]
|
||||
self._protocol_version = s["protocol_version"]
|
||||
self._color = Color(*s["backlight"].values())
|
||||
self._image = s["image_file"]
|
||||
|
||||
@property
|
||||
def index(self):
|
||||
self._get_status()
|
||||
return self._index
|
||||
|
||||
@property
|
||||
def name(self):
|
||||
self._get_status()
|
||||
return self._name
|
||||
|
||||
@name.setter
|
||||
def name(self, new_name):
|
||||
self._server._req(f"{self.uid}/name", {"name": new_name})
|
||||
|
||||
@property
|
||||
def value(self):
|
||||
self._get_status()
|
||||
return self._value
|
||||
|
||||
@value.setter
|
||||
def value(self, new_value):
|
||||
self._server._req(f"{self.uid}/set", {"value": max(0, min(100, int(new_value)))})
|
||||
|
||||
@property
|
||||
def fw_hash(self):
|
||||
self._get_status()
|
||||
return self._fw_hash
|
||||
|
||||
@property
|
||||
def fw_version(self):
|
||||
self._get_status()
|
||||
return self._fw_version
|
||||
|
||||
@property
|
||||
def hw_version(self):
|
||||
self._get_status()
|
||||
return self._hw_version
|
||||
|
||||
@property
|
||||
def protocol_version(self):
|
||||
self._get_status()
|
||||
return self._protocol_version
|
||||
|
||||
@property
|
||||
def color(self):
|
||||
self._get_status()
|
||||
return self._color
|
||||
|
||||
@color.setter
|
||||
def color(self, new_color):
|
||||
self._server._req(f"{self.uid}/backlight", dict(new_color))
|
||||
|
||||
@property
|
||||
def image(self):
|
||||
self._get_status()
|
||||
return self._image
|
||||
|
||||
@image.setter
|
||||
def image(self, new_image):
|
||||
file = {'imgfile': (open(new_image, 'rb'))}
|
||||
self._server._req(f"{self.uid}/image/set", file, post=True)
|
||||
|
||||
class Color():
|
||||
def __init__(self, red, green, blue):
|
||||
self.red = max(0, min(100, red))
|
||||
self.green = max(0, min(100, green))
|
||||
self.blue = max(0, min(100, blue))
|
||||
|
||||
def hsv(self):
|
||||
return colorsys.rgb_to_hsv(self.red, self.green, self.blue)
|
||||
|
||||
def __iter__(self):
|
||||
for attr in ["red", "green", "blue"]:
|
||||
yield (attr, self.__dict__[attr])
|
Loading…
Reference in a new issue