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

Improve type hints in trace (#78366)

Browse Source
This commit is contained in:
epenet 2022-09-13 15:09:38 +02:00 committed by GitHub
parent 458ddb6f4b
commit 69b59c9d59
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
homeassistant/components/trace/__init__.py
View File

@ -133,7 +133,9 @@ async def async_list_traces(hass, wanted_domain, wanted_key):
return traces return traces
def async_store_trace(hass, trace, stored_traces): def async_store_trace(
hass: HomeAssistant, trace: ActionTrace, stored_traces: int
) -> None:
"""Store a trace if its key is valid.""" """Store a trace if its key is valid."""
if key := trace.key: if key := trace.key:
traces = hass.data[DATA_TRACE] traces = hass.data[DATA_TRACE]
@ -144,7 +146,7 @@ def async_store_trace(hass, trace, stored_traces):
traces[key][trace.run_id] = trace traces[key][trace.run_id] = trace
def _async_store_restored_trace(hass, trace): def _async_store_restored_trace(hass: HomeAssistant, trace: RestoredTrace) -> None:
"""Store a restored trace and move it to the end of the LimitedSizeDict.""" """Store a restored trace and move it to the end of the LimitedSizeDict."""
key = trace.key key = trace.key
traces = hass.data[DATA_TRACE] traces = hass.data[DATA_TRACE]
@ -154,7 +156,7 @@ def _async_store_restored_trace(hass, trace):
traces[key].move_to_end(trace.run_id, last=False) traces[key].move_to_end(trace.run_id, last=False)
async def async_restore_traces(hass): async def async_restore_traces(hass: HomeAssistant) -> None:
"""Restore saved traces.""" """Restore saved traces."""
if DATA_TRACES_RESTORED in hass.data: if DATA_TRACES_RESTORED in hass.data:
return return
@ -216,15 +218,15 @@ class ActionTrace(BaseTrace):
def __init__( def __init__(
self, self,
item_id: str, item_id: str | None,
config: dict[str, Any], config: dict[str, Any] | None,
blueprint_inputs: dict[str, Any], blueprint_inputs: dict[str, Any] | None,
context: Context, context: Context,
) -> None: ) -> None:
"""Container for script trace.""" """Container for script trace."""
self._trace: dict[str, deque[TraceElement]] | None = None self._trace: dict[str, deque[TraceElement]] | None = None
self._config: dict[str, Any] = config self._config = config
self._blueprint_inputs: dict[str, Any] = blueprint_inputs self._blueprint_inputs = blueprint_inputs
self.context: Context = context self.context: Context = context
self._error: Exception | None = None self._error: Exception | None = None
self._state: str = "running" self._state: str = "running"
@ -239,7 +241,7 @@ class ActionTrace(BaseTrace):
trace_set_child_id(self.key, self.run_id) trace_set_child_id(self.key, self.run_id)
trace_id_set((self.key, self.run_id)) trace_id_set((self.key, self.run_id))
def set_trace(self, trace: dict[str, deque[TraceElement]]) -> None: def set_trace(self, trace: dict[str, deque[TraceElement]] | None) -> None:
"""Set action trace.""" """Set action trace."""
self._trace = trace self._trace = trace