From 08cd3d0e1a10a942f794380d827001322c8ec42c Mon Sep 17 00:00:00 2001 From: sepehr Date: Fri, 10 Jan 2025 23:45:51 +0300 Subject: [PATCH] Reload server if db connection failed --- py/libs/db/db_sysconfig.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/py/libs/db/db_sysconfig.py b/py/libs/db/db_sysconfig.py index 055a1d9..20972b6 100644 --- a/py/libs/db/db_sysconfig.py +++ b/py/libs/db/db_sysconfig.py @@ -6,7 +6,7 @@ # Author: sepehr.ha@gmail.com from peewee import * - +from pathlib import Path from libs.db.db import User,BaseModel,get_object_or_404 import logging log = logging.getLogger("db_sysconfig") @@ -44,7 +44,13 @@ def get_scan_mode(): return get_object_or_404(Sysconfig, key="scan_mode") def get_sysconfig(key): - return get_object_or_404(Sysconfig, key=key).value + try: + return get_object_or_404(Sysconfig, key=key).value + except Exception as e: + log.error(e) + if "server closed the connection unexpectedly" in str(e): + Path('/app/reload').touch() + return None def get_firmware_latest(): return get_object_or_404(Sysconfig, key="latest_version")