Skip to content

Commit 7bc3567

Browse files
authored
Merge pull request #16 from CM000n/bugfix/fix_tls_handshake_error
Fix tls handshake
2 parents 8347f76 + ae0338d commit 7bc3567

File tree

3 files changed

+48
-26
lines changed

3 files changed

+48
-26
lines changed

custom_components/qss/io.py

+26-4
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,28 @@
1212
_LOGGER = logging.getLogger(__name__)
1313

1414

15-
def _insert_row(host: str, port: int, auth: tuple, event: Event) -> None:
16-
tls = bool(auth)
17-
with Sender(host, port, auth=auth, tls=tls) as sender:
15+
def _insert_row_with_auth(host: str, port: int, auth: tuple, event: Event) -> None:
16+
with Sender(host, port, auth=auth, tls=True) as sender:
17+
entity_id = event.data["entity_id"]
18+
state = event.data.get("new_state")
19+
attrs = dict(state.attributes)
20+
sender.row(
21+
"qss",
22+
symbols={
23+
"entity_id": entity_id,
24+
},
25+
columns={
26+
"state": state.state,
27+
"attributes": dumps(attrs, sort_keys=True, default=str),
28+
},
29+
at=event.time_fired,
30+
)
31+
32+
sender.flush()
33+
34+
35+
def _insert_row_without_auth(host: str, port: int, event: Event) -> None:
36+
with Sender(host, port) as sender:
1837
entity_id = event.data["entity_id"]
1938
state = event.data.get("new_state")
2039
attrs = dict(state.attributes)
@@ -40,7 +59,10 @@ def _insert_row(host: str, port: int, auth: tuple, event: Event) -> None:
4059
)
4160
def _retry_data_insertion(host: str, port: int, auth: tuple, event: Event) -> None:
4261
"""Use a retry for inserting event data into QuestDB."""
43-
_insert_row(host, port, auth, event)
62+
if all(auth):
63+
_insert_row_with_auth(host, port, auth, event)
64+
else:
65+
_insert_row_without_auth(host, port, event)
4466

4567

4668
def insert_event_data_into_questdb(host: str, port: int, auth: tuple, event: Event, queue: Queue) -> None:

poetry.lock

+21-21
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

pyproject.toml

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[tool.poetry]
22
name = "qss"
3-
version = "v0.0.7"
3+
version = "v0.0.8"
44
description = "QuestDB State Storage (QSS) for Home Assistant"
55
license = "MIT"
66
readme = "README.md"

0 commit comments

Comments
 (0)