1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
"""Provide a model for a log message event."""
from __future__ import annotations
from dataclasses import dataclass, field
from typing import Literal, TypedDict
from ..const import CommandClass
class LogMessageContextDataType(TypedDict, total=False):
"""Represent a log message context data dict type."""
source: Literal["config", "serial", "controller", "driver"] # required
type: Literal["controller", "value", "node"]
nodeId: int
header: str
direction: Literal["inbound", "outbound", "none"]
change: Literal["added", "removed", "updated", "notification"]
internal: bool
endpoint: int
commandClass: int
property: int | str
propertyKey: int | str
@dataclass
class LogMessageContext:
"""Represent log message context information."""
data: LogMessageContextDataType = field(repr=False)
source: Literal["config", "serial", "controller", "driver"] = field(init=False)
type: Literal["controller", "value", "node"] | None = field(init=False)
node_id: int | None = field(init=False)
header: str | None = field(init=False)
direction: Literal["inbound", "outbound", "none"] | None = field(init=False)
change: Literal["added", "removed", "updated", "notification"] | None = field(
init=False
)
internal: bool | None = field(init=False)
endpoint: int | None = field(init=False)
command_class: CommandClass | None = field(init=False, default=None)
property_: int | str | None = field(init=False)
property_key: int | str | None = field(init=False)
def __post_init__(self) -> None:
"""Post initialize."""
self.source = self.data["source"]
self.type = self.data.get("type")
self.node_id = self.data.get("nodeId")
self.header = self.data.get("header")
self.direction = self.data.get("direction")
self.change = self.data.get("change")
self.internal = self.data.get("internal")
self.endpoint = self.data.get("endpoint")
if (command_class := self.data.get("commandClass")) is not None:
self.command_class = CommandClass(command_class)
self.property_ = self.data.get("property")
self.property_key = self.data.get("propertyKey")
class LogMessageDataType(TypedDict, total=False):
"""Represent a log message data dict type."""
source: Literal["driver"] # required
event: Literal["logging"] # required
message: str | list[str] # required
formattedMessage: str | list[str] # required
direction: str # required
level: str # required
context: LogMessageContextDataType # required
primaryTags: str
secondaryTags: str
secondaryTagPadding: int
multiline: bool
timestamp: str
label: str
def _process_message(message: str | list[str]) -> list[str]:
"""Process a message and always return a list."""
if isinstance(message, str):
return str(message).splitlines()
# We will assume each item in the array is on a separate line so we can
# remove trailing line breaks
return [message.rstrip("\n") for message in message]
@dataclass
class LogMessage:
"""Represent a log message."""
data: LogMessageDataType = field(repr=False)
message: list[str] = field(init=False)
formatted_message: list[str] = field(init=False)
direction: str = field(init=False)
level: str = field(init=False)
context: LogMessageContext = field(init=False)
primary_tags: str | None = field(init=False)
secondary_tags: str | None = field(init=False)
secondary_tag_padding: int | None = field(init=False)
multiline: bool | None = field(init=False)
timestamp: str | None = field(init=False)
label: str | None = field(init=False)
def __post_init__(self) -> None:
"""Post initialize."""
self.message = _process_message(self.data["message"])
self.formatted_message = _process_message(self.data["formattedMessage"])
self.direction = self.data["direction"]
self.level = self.data["level"]
self.context = LogMessageContext(self.data["context"])
self.primary_tags = self.data.get("primaryTags")
self.secondary_tags = self.data.get("secondaryTags")
self.secondary_tag_padding = self.data.get("secondaryTagPadding")
self.multiline = self.data.get("multiline")
self.timestamp = self.data.get("timestamp")
self.label = self.data.get("label")
|
