Reload server if db connection failed

This commit is contained in:
sepehr 2025-01-10 23:45:51 +03:00
parent 4af68c2fb7
commit 08cd3d0e1a

View file

@ -6,7 +6,7 @@
# Author: sepehr.ha@gmail.com # Author: sepehr.ha@gmail.com
from peewee import * from peewee import *
from pathlib import Path
from libs.db.db import User,BaseModel,get_object_or_404 from libs.db.db import User,BaseModel,get_object_or_404
import logging import logging
log = logging.getLogger("db_sysconfig") log = logging.getLogger("db_sysconfig")
@ -44,7 +44,13 @@ def get_scan_mode():
return get_object_or_404(Sysconfig, key="scan_mode") return get_object_or_404(Sysconfig, key="scan_mode")
def get_sysconfig(key): 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(): def get_firmware_latest():
return get_object_or_404(Sysconfig, key="latest_version") return get_object_or_404(Sysconfig, key="latest_version")