Skip to content

Commit d191752

Browse files
committed
Handle database connection failures gracefully
1 parent ed217dc commit d191752

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

source/database_handler.py

+5-2
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from influxdb_client.client.write_api import SYNCHRONOUS
77
from logger import LoggerMixin
88
from time_handler import TimeHandler
9+
from urllib3.exceptions import NewConnectionError
910

1011

1112
@dataclasses.dataclass
@@ -43,10 +44,12 @@ def write_to_database(
4344
for field_to_insert in fields_to_insert:
4445
point = point.field(field_to_insert.name, field_to_insert.value)
4546
point = point.time(timestamp)
46-
# point = Point("sun_forecast").field("pv_estimate", 20.52).time(TimeHandler.get_time())
4747

4848
self.log.trace(f"Writing to database: {point}")
49-
self.write_api.write(bucket=self.bucket, record=point)
49+
try:
50+
self.write_api.write(bucket=self.bucket, record=point)
51+
except NewConnectionError as e:
52+
self.log.warning(f"Connection to database failed (ignoring): {str(e)}")
5053

5154
def close_connection(self) -> None:
5255
self.write_api.close()

0 commit comments

Comments
 (0)