This commit is contained in:
rooba 2022-11-27 18:43:36 -08:00
commit 6933b03b4a
5 changed files with 555 additions and 0 deletions

6
.gitignore vendored Normal file
View File

@ -0,0 +1,6 @@
**/app.py
**/phlyght.*
**/phlyght.*
**/poetry.lock
**/__pycache__
!test.py

3
phlyght/__init__.py Normal file
View File

@ -0,0 +1,3 @@
__all__ = ("Router", "route", "RouterMeta", "SubRouter", "HughApi")
from .api import Router, route, RouterMeta, SubRouter, HughApi

514
phlyght/api.py Normal file
View File

@ -0,0 +1,514 @@
__all__ = ("Router", "route", "RouterMeta", "SubRouter", "HughApi")
from inspect import signature
from re import compile
from typing import Any, Optional
from httpx import AsyncClient
from httpx._urls import URL as _URL
from yarl import URL as UR
STR_FMT_RE = compile(r"""(?=(\{([^:]+)(?::([^}]+))?\}))\1""")
URL_TYPES = {"str": str, "int": int}
def get_url_args(url):
kwds = {}
match = STR_FMT_RE.finditer(url)
for m in match:
name = m.group(2)
if len(m.groups()) >= 4:
type_ = URL_TYPES[m.group(3)]
else:
type_ = str
kwds[name] = type_
return kwds
class URL(_URL):
def __truediv__(self, other):
return URL(str(UR(f"{self}") / other.lstrip("/")))
def __floordiv__(self, other):
return URL(str(UR(f"{self}") / other.lstrip("/")))
def route(method, endpoint) -> Any:
def wrapped(fn):
async def sub_wrap(
self: SubRouter,
base_uri=None,
content: Optional[bytes] = None,
data: Optional[dict[str, str]] = None,
params: Optional[dict[str, str]] = None,
**kwargs,
):
data = data or {}
for param_name, param in signature(fn).parameters.items():
if param_name == "self":
continue
if param_name in fn.__annotations__:
anno = fn.__annotations__[param_name]
if anno._name == "Optional":
if hasattr(anno.__args__[0], "_name"):
type_ = str
else:
type_ = anno.__args__[0]
else:
type_ = fn.__annotations__[param_name]
else:
type_ = str
if param.default is param.empty:
if param_name not in kwargs and param_name not in data:
raise TypeError(
f"Missing required argument {param_name} for {fn.__name__}"
)
if param_name in kwargs:
data[param_name] = type_(kwargs.pop(param_name))
else:
if kwargs.get(param_name, None):
data[param_name] = type_(kwargs.pop(param_name))
url_args = get_url_args(endpoint)
for k, v in url_args.items():
if k not in kwargs and k not in params and k not in data:
raise ValueError(f"Missing required argument {k}")
if k in kwargs:
url_args[k] = v(kwargs.pop(k))
elif k in params:
url_args[k] = v(params.pop(k)) # type: ignore
elif k in data:
url_args[k] = v(data.pop(k))
try:
if kwargs:
data = data | kwargs if data else kwargs
except Exception:
...
if url_args:
new_endpoint = URL(f"{self._api_path}") / endpoint.format(**url_args)
else:
new_endpoint = URL(f"{self._api_path}") / endpoint
return await self._client.request(
method,
new_endpoint,
content=content,
data=data,
params=params,
)
return sub_wrap
return wrapped
class RouterMeta(type):
@classmethod
def __prepare__(cls, name, bases, **kwargs):
if bases:
return kwargs | bases[0].__dict__
return kwargs
def __new__(cls, name, bases, kwds, **kwargs):
cells = {}
_root = kwargs.get("root", "")
_base = kwds.get("BASE_URI", "")
_api = f"{_root}{_base}"
def set_key(v, base_uri=None):
def wrap(self, **_kwds):
return v(self, base_uri=base_uri, **_kwds)
return wrap
if any(map(lambda x: not x.startswith("__"), kwds.keys())):
funcs = list(
filter(
lambda k: not k[0].startswith("__") and callable(k[1]), kwds.items()
)
)
for k, v in funcs:
if hasattr(v, "__closure__"):
# val = v.__closure__[0].cell_contents
cells[k] = set_key(v, base_uri=_api)
kwds["handlers"] = cells
for base in bases:
if hasattr(base, "handlers"):
kwds["handlers"] |= base.handlers
kwds.update(**kwds["handlers"])
return super().__new__(cls, cls.__name__, bases, kwds, **kwargs)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
class SubRouter(metaclass=RouterMeta):
BASE_URI: str
_api_path: str
_client: AsyncClient
def __new__(cls, hue_api_key: str):
if not hasattr(cls, "handlers"):
cls.handlers = {}
for base in cls.__bases__:
if hasattr(base, "handlers"):
cls.handlers |= getattr(base, "handlers")
return super().__new__(cls)
def __init__(self, hue_api_key: str):
self._hue_api_key = hue_api_key
self._headers = {
"User-Agent": "Python/HueClient",
"hue-application-key": self._hue_api_key,
}
def __init_subclass__(cls, *args, **kwargs) -> None:
super().__init_subclass__()
if kwargs.get("root"):
cls._api_path = f'{kwargs.get("root")}{cls.BASE_URI}'
class HughApi(SubRouter):
BASE_URI = "/clip/v2"
@route("GET", "/resource/light")
async def get_lights(self, friendly_name: Optional[str] = None):
...
@route("GET", "/resource/light/{light_id}")
async def get_light(self, light_id: int):
...
@route("PUT", "/resource/light/{light_id}")
async def set_light(self, light_id: int, **kwargs):
...
@route("GET", "/resource/scene")
async def get_scenes(self):
...
@route("POST", "/resource/scene")
async def create_scene(self, **kwargs):
...
@route("GET", "/resource/scene/{scene_id}")
async def get_scene(self, scene_id: int):
...
@route("PUT", "/resource/scene/{scene_id}")
async def set_scene(self, scene_id: int, **kwargs):
...
@route("DELETE", "/resource/scene/{scene_id}")
async def delete_scene(self, scene_id: int):
...
@route("GET", "/resource/room")
async def get_rooms(self):
...
@route("POST", "/resource/room")
async def create_room(self, **kwargs):
...
@route("GET", "/resource/room/{room_id}")
async def get_room(self, room_id: int):
...
@route("PUT", "/resource/room/{room_id}")
async def set_room(self, room_id: int, **kwargs):
...
@route("DELETE", "/resource/room/{room_id}")
async def delete_room(self, room_id: int):
...
@route("GET", "/resource/zone")
async def get_zones(self):
...
@route("POST", "/resource/zone")
async def create_zone(self, **kwargs):
...
@route("GET", "/resource/zone/{zone_id}")
async def get_zone(self, zone_id: int):
...
@route("PUT", "/resource/zone/{zone_id}")
async def set_zone(self, zone_id: int, **kwargs):
...
@route("DELETE", "/resource/zone/{zone_id}")
async def delete_zone(self, zone_id: int):
...
@route("GET", "/resource/bridge_home")
async def get_bridge_homes(self):
...
@route("GET", "/resource/bridge_home/{bridge_home_id}")
async def get_bridge_home(self, bridge_home_id: int):
...
@route("PUT", "/resource/bridge_home/{bridge_home_id}")
async def set_bridge_home(self, bridge_home_id: int, **kwargs):
...
@route("GET", "/resource/grouped_light")
async def get_grouped_lights(self):
...
@route("GET", "/resource/grouped_light/{grouped_light_id}")
async def get_grouped_light(self, grouped_light_id: int):
...
@route("PUT", "/resource/grouped_light/{grouped_light_id}")
async def set_grouped_light(self, grouped_light_id: int, **kwargs):
...
@route("GET", "/resource/device")
async def get_devices(self):
...
@route("GET", "/resource/device/{device_id}")
async def get_device(self, device_id: int):
...
@route("PUT", "/resource/device/{device_id}")
async def set_device(self, device_id: int, **kwargs):
...
@route("GET", "/resource/bridges")
async def get_bridges(self):
...
@route("GET", "/resource/bridges/{bridge_id}")
async def get_bridge(self, bridge_id: int):
...
@route("PUT", "/resource/bridges/{bridge_id}")
async def set_bridge(self, bridge_id: int, **kwargs):
...
@route("GET", "/resource/device_power")
async def get_device_powers(self):
...
@route("GET", "/resource/device_power/{device_power_id}")
async def get_device_power(self, device_power_id: int):
...
@route("PUT", "/resource/device_power/{device_power_id}")
async def set_device_power(self, device_power_id: int, **kwargs):
...
@route("GET", "/resource/zigbee_connectivity")
async def get_zigbee_connectivities(self):
...
@route("GET", "/resource/zigbee_connectivity/{zigbee_connectivity_id}")
async def get_zigbee_connectivity(self, zigbee_connectivity_id: int):
...
@route("PUT", "/resource/zigbee_connectivity/{zigbee_connectivity_id}")
async def set_zigbee_connectivity(self, zigbee_connectivity_id: int, **kwargs):
...
@route("GET", "/resource/zgb_connectivity")
async def get_zgb_connectivities(self):
...
@route("GET", "/resource/zgb_connectivity/{zgb_connectivity_id}")
async def get_zgb_connectivity(self, zgb_connectivity_id: int):
...
@route("PUT", "/resource/zgb_connectivity/{zgb_connectivity_id}")
async def set_zgb_connectivity(self, zgb_connectivity_id: int, **kwargs):
...
@route("GET", "/resource/motion")
async def get_motions(self):
...
@route("GET", "/resource/motion/{motion_id}")
async def get_motion(self, motion_id: int):
...
@route("PUT", "/resource/motion/{motion_id}")
async def set_motion(self, motion_id: int, **kwargs):
...
@route("GET", "/resource/temperature")
async def get_temperatures(self):
...
@route("GET", "/resource/temperature/{temperature_id}")
async def get_temperature(self, temperature_id: int):
...
@route("PUT", "/resource/temperature/{temperature_id}")
async def set_temperature(self, temperature_id: int, **kwargs):
...
@route("GET", "/resource/light_level")
async def get_light_levels(self):
...
@route("GET", "/resource/light_level/{light_level_id}")
async def get_light_level(self, light_level_id: int):
...
@route("PUT", "/resource/light_level/{light_level_id}")
async def set_light_level(self, light_level_id: int, **kwargs):
...
@route("GET", "/resource/button")
async def get_buttons(self):
...
@route("GET", "/resource/button/{button_id}")
async def get_button(self, button_id: int):
...
@route("PUT", "/resource/button/{button_id}")
async def set_button(self, button_id: int, **kwargs):
...
@route("GET", "/resource/behavior_script")
async def get_behavior_scripts(self):
...
@route("GET", "/resource/behavior_script/{behavior_script_id}")
async def get_behavior_script(self, behavior_script_id: int):
...
@route("GET", "/resource/behavior_instance")
async def get_behavior_instances(self):
...
@route("POST", "/resource/behavior_instance")
async def create_behavior_instance(self, **kwargs):
...
@route("GET", "/resource/behavior_instance/{behavior_instance_id}")
async def get_behavior_instance(self, behavior_instance_id: int):
...
@route("PUT", "/resource/behavior_instance/{behavior_instance_id}")
async def set_behavior_instance(self, behavior_instance_id: int, **kwargs):
...
@route("DELETE", "/resource/behavior_instance/{behavior_instance_id}")
async def delete_behavior_instance(self, behavior_instance_id: int):
...
@route("GET", "/resource/geofence_client")
async def get_geofence_clients(self):
...
@route("POST", "/resource/geofence_client")
async def create_geofence_client(self, **kwargs):
...
@route("GET", "/resource/geofence_client/{geofence_client_id}")
async def get_geofence_client(self, geofence_client_id: int):
...
@route("PUT", "/resource/geofence_client/{geofence_client_id}")
async def set_geofence_client(self, geofence_client_id: int, **kwargs):
...
@route("DELETE", "/resource/geofence_client/{geofence_client_id}")
async def delete_geofence_client(self, geofence_client_id: int):
...
@route("GET", "/resource/geolocation")
async def get_geolocations(self):
...
@route("GET", "/resource/geolocation/{geolocation_id}")
async def get_geolocation(self, geolocation_id: int):
...
@route("PUT", "/resource/geolocation/{geolocation_id}")
async def set_geolocation(self, geolocation_id: int, **kwargs):
...
@route("GET", "/resource/entertainment_configuration")
async def get_entertainment_configurations(self):
...
@route("POST", "/resource/entertainment_configuration")
async def create_entertainment_configuration(self, **kwargs):
...
@route(
"GET", "/resource/entertainment_configuration/{entertainment_configuration_id}"
)
async def get_entertainment_configuration(
self, entertainment_configuration_id: int
):
...
@route(
"PUT", "/resource/entertainment_configuration/{entertainment_configuration_id}"
)
async def set_entertainment_configuration(
self, entertainment_configuration_id: int, **kwargs
):
...
@route(
"DELETE",
"/resource/entertainment_configuration/{entertainment_configuration_id}",
)
async def delete_entertainment_configuration(
self, entertainment_configuration_id: int
):
...
@route("GET", "/resource/entertainment")
async def get_entertainments(self):
...
@route("GET", "/resource/entertainment/{entertainment_id}")
async def get_entertainment(self, entertainment_id: int):
...
@route("PUT", "/resource/entertainment/{entertainment_id}")
async def set_entertainment(self, entertainment_id: int, **kwargs):
...
@route("GET", "/resource/homekit")
async def get_homekits(self):
...
@route("GET", "/resource/homekit/{homekit_id}")
async def get_homekit(self, homekit_id: int):
...
@route("PUT", "/resource/homekit/{homekit_id}")
async def set_homekit(self, homekit_id: int, **kwargs):
...
@route("GET", "/resource")
async def get_resources(self):
...
class Router(HughApi, root="https://192.168.69.104"):
def __init__(self, hue_api_key: str):
super().__init__(hue_api_key)
self._client = AsyncClient(headers=self._headers, verify=False)

19
pyproject.toml Normal file
View File

@ -0,0 +1,19 @@
[tool.poetry]
name = "lights"
version = "0.0.1"
description = "Lights"
authors = ["Ra <ra@tcp.direct>"]
[tool.poetry.dependencies]
python = ">=3.11,<4.0.0"
rich = ">=12.6.0"
aioredis = ">=2.0.1"
httpx = "^0.23.1"
yarl = "^1.8.1"
[tool.poetry.dev-dependencies]
black = ">=22.10.0"
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"

13
test.py Normal file
View File

@ -0,0 +1,13 @@
from asyncio import run
from rich import print
from phlyght.api import Router
async def main():
router = Router("ur api key")
lights = await router.get_lights()
print(lights.json())
run(main())