From 60dae1d381aefcf78e6fe2a76eb97e86fa835f44 Mon Sep 17 00:00:00 2001
From: Anton Starikov <ant.starikov@gmail.com>
Date: Thu, 12 Dec 2024 18:23:50 +0100
Subject: [PATCH] Change formatting to make linter happy, update of project
 file

---
 custom_components/qss/event_handling.py |  4 ++--
 custom_components/qss/io.py             | 16 +++-------------
 pyproject.toml                          |  8 ++++----
 3 files changed, 9 insertions(+), 19 deletions(-)

diff --git a/custom_components/qss/event_handling.py b/custom_components/qss/event_handling.py
index 0ab8e36..ed186bb 100644
--- a/custom_components/qss/event_handling.py
+++ b/custom_components/qss/event_handling.py
@@ -11,8 +11,8 @@ def put_event_to_queue(event: Event, entity_filter: Callable[[str], bool], queue
     entity_id = event.data.get(ATTR_ENTITY_ID)
     state = event.data.get("new_state")
     if state is not None:
-      if all([entity_id, state, state.state != STATE_UNKNOWN, entity_filter(entity_id)]):
-        queue.put(event)
+        if all([entity_id, state, state.state != STATE_UNKNOWN, entity_filter(entity_id)]):
+            queue.put(event)
 
 
 def get_event_from_queue(queue: Queue) -> Event:
diff --git a/custom_components/qss/io.py b/custom_components/qss/io.py
index 4cad701..ad697f7 100644
--- a/custom_components/qss/io.py
+++ b/custom_components/qss/io.py
@@ -4,7 +4,7 @@
 from queue import Queue
 
 from homeassistant.core import Event
-from questdb.ingress import IngressError, Sender, Protocol
+from questdb.ingress import IngressError, Protocol, Sender
 from tenacity import retry, retry_if_exception_type, stop_after_attempt, wait_fixed
 
 from .const import RETRY_ATTEMPTS, RETRY_WAIT_SECONDS
@@ -13,14 +13,7 @@
 
 
 def _insert_row_with_auth(host: str, port: int, auth: tuple, event: Event) -> None:
-    with Sender(
-               Protocol.Tcps,
-               host,
-               port,
-               username=auth[0],
-               token=auth[1],
-               token_x=auth[2],
-               token_y=auth[3]) as sender:
+    with Sender(Protocol.Tcps, host, port, username=auth[0], token=auth[1], token_x=auth[2], token_y=auth[3]) as sender:
         entity_id = event.data["entity_id"]
         state = event.data.get("new_state")
         attrs = dict(state.attributes)
@@ -40,10 +33,7 @@ def _insert_row_with_auth(host: str, port: int, auth: tuple, event: Event) -> No
 
 
 def _insert_row_without_auth(host: str, port: int, event: Event) -> None:
-    with Sender(
-               Protocol.Tcp,
-               host,
-               port) as sender:
+    with Sender(Protocol.Tcp, host, port) as sender:
         entity_id = event.data["entity_id"]
         state = event.data.get("new_state")
         attrs = dict(state.attributes)
diff --git a/pyproject.toml b/pyproject.toml
index 9a208d5..63d07f3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "qss"
-version = "v0.0.10"
+version = "v0.0.11"
 description = "QuestDB State Storage (QSS) for Home Assistant"
 authors = ["CM000n"]
 keywords = [
@@ -17,14 +17,14 @@ readme = "README.md"
 
 [tool.poetry.dependencies]
 python = "^3.11"
-homeassistant = "^2023.12"
-questdb = "^1.2.0"
+homeassistant = "^2024.12"
+questdb = "^2.0.3"
 tenacity = "^8.0.0"
 
 
 [tool.poetry.dev-dependencies]
 voluptuous-stubs = "^0.1"
-homeassistant-stubs = "^2023.12"
+homeassistant-stubs = "^2024.12"
 pre-commit = "^2.21.0"
 pre-commit-hooks = "^4.5.0"
 codespell = "^2.0.0"