jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 03:07:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use NamedTuple for repetier API methods (#56941)

Browse Source
This commit is contained in:
Marc Mueller 2021-10-04 17:41:51 +02:00 committed by GitHub
parent eb9b9c57a4
commit 2f9943fe7a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 50 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

90
homeassistant/components/repetier/__init__.py
View File

@ -37,28 +37,38 @@ UPDATE_SIGNAL = "repetier_update_signal"
TEMP_DATA = {"tempset": "temp_set", "tempread": "state", "output": "output"}
API_PRINTER_METHODS = {
"bed_temperature": {
"offline": {"heatedbeds": None, "state": "off"},
"state": {"heatedbeds": "temp_data"},
"temp_data": TEMP_DATA,
"attribute": "heatedbeds",
},
"extruder_temperature": {
"offline": {"extruder": None, "state": "off"},
"state": {"extruder": "temp_data"},
"temp_data": TEMP_DATA,
"attribute": "extruder",
},
"chamber_temperature": {
"offline": {"heatedchambers": None, "state": "off"},
"state": {"heatedchambers": "temp_data"},
"temp_data": TEMP_DATA,
"attribute": "heatedchambers",
},
"current_state": {
"offline": {"state": None},
"state": {
@dataclass
class APIMethods:
"""API methods for properties."""
offline: dict[str, str | None]
state: dict[str, str]
temp_data: dict[str, str] | None = None
attribute: str | None = None
API_PRINTER_METHODS: dict[str, APIMethods] = {
"bed_temperature": APIMethods(
offline={"heatedbeds": None, "state": "off"},
state={"heatedbeds": "temp_data"},
temp_data=TEMP_DATA,
attribute="heatedbeds",
),
"extruder_temperature": APIMethods(
offline={"extruder": None, "state": "off"},
state={"extruder": "temp_data"},
temp_data=TEMP_DATA,
attribute="extruder",
),
"chamber_temperature": APIMethods(
offline={"heatedchambers": None, "state": "off"},
state={"heatedchambers": "temp_data"},
temp_data=TEMP_DATA,
attribute="heatedchambers",
),
"current_state": APIMethods(
offline={"state": None},
state={
"state": "state",
"activeextruder": "active_extruder",
"hasxhome": "x_homed",
@ -67,10 +77,10 @@ API_PRINTER_METHODS = {
"firmware": "firmware",
"firmwareurl": "firmware_url",
},
},
"current_job": {
"offline": {"job": None, "state": "off"},
"state": {
),
"current_job": APIMethods(
offline={"job": None, "state": "off"},
state={
"done": "state",
"job": "job_name",
"jobid": "job_id",
@ -84,25 +94,25 @@ API_PRINTER_METHODS = {
"y": "y",
"z": "z",
},
},
"job_end": {
"offline": {"job": None, "state": "off", "start": None, "printtime": None},
"state": {
),
"job_end": APIMethods(
offline={"job": None, "state": "off", "start": None, "printtime": None},
state={
"job": "job_name",
"start": "start",
"printtime": "print_time",
"printedtimecomp": "from_start",
},
},
"job_start": {
"offline": {
),
"job_start": APIMethods(
offline={
"job": None,
"state": "off",
"start": None,
"printedtimecomp": None,
},
"state": {"job": "job_name", "start": "start", "printedtimecomp": "from_start"},
},
state={"job": "job_name", "start": "start", "printedtimecomp": "from_start"},
),
}
@ -251,17 +261,17 @@ class PrinterAPI:
"""Get data from the state cache."""
printer = self.printers[printer_id]
methods = API_PRINTER_METHODS[sensor_type]
for prop, offline in methods["offline"].items():
for prop, offline in methods.offline.items():
state = getattr(printer, prop)
if state == offline:
# if state matches offline, sensor is offline
return None
data = {}
for prop, attr in methods["state"].items():
for prop, attr in methods.state.items():
prop_data = getattr(printer, prop)
if attr == "temp_data":
temp_methods = methods["temp_data"]
temp_methods = methods.temp_data or {}
for temp_prop, temp_attr in temp_methods.items():
data[temp_attr] = getattr(prop_data[temp_id], temp_prop)
else:
@ -290,8 +300,8 @@ class PrinterAPI:
continue
methods = API_PRINTER_METHODS[sensor_type]
if "temp_data" in methods["state"].values():
prop_data = getattr(printer, methods["attribute"])
if "temp_data" in methods.state.values():
prop_data = getattr(printer, methods.attribute or "")
if prop_data is None:
continue
for idx, _ in enumerate(prop_data):