From 88bf77d6dd8247db81acc183dedc351d068dad03 Mon Sep 17 00:00:00 2001 From: Nama Date: Wed, 7 Apr 2021 23:09:58 +0200 Subject: [PATCH] Fix crash, if Artemis is closed while the lib is connected --- artemisremotecontrol/__init__.py | 6 +++--- scripts/tasmota.py | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/artemisremotecontrol/__init__.py b/artemisremotecontrol/__init__.py index 66fb252..98febc5 100644 --- a/artemisremotecontrol/__init__.py +++ b/artemisremotecontrol/__init__.py @@ -1,5 +1,5 @@ from requests import get, post -from requests.exceptions import ConnectionError +from requests.exceptions import ConnectionError, ChunkedEncodingError import logging from os import environ from json import loads, dumps @@ -13,7 +13,7 @@ def _getapiurl(ep): try: response = get(url_endpoints) - except ConnectionError: + except (ConnectionError, ChunkedEncodingError): logging.error('Can\'t connect to Artemis') return None if response.status_code != 200: @@ -51,7 +51,7 @@ def setleds(deviceids: list, ledids: list, color: str): logging.debug(data) try: response = post(url, data=str(data)) - except ConnectionError: + except (ConnectionError, ChunkedEncodingError): logging.error('Can\'t connect to Artemis to set color') return if response.status_code != 200: diff --git a/scripts/tasmota.py b/scripts/tasmota.py index 5a9360e..8d05435 100644 --- a/scripts/tasmota.py +++ b/scripts/tasmota.py @@ -5,7 +5,7 @@ from time import sleep from threading import Thread from requests import get -from requests.exceptions import ConnectionError +from requests.exceptions import ConnectionError, ChunkedEncodingError from artemisremotecontrol.config import Config from artemisremotecontrol import getleds, setleds @@ -26,7 +26,7 @@ def loop(): try: response = get(url, timeout=1) status = response.status_code - except ConnectionError: + except (ConnectionError, ChunkedEncodingError): status = False if status != 200: # send "Disconnected" state