netinvent/npbackup
1
1
Fork 0
mirror of https://github.com/netinvent/npbackup.git synced 2025-09-07 05:24:47 +08:00
Code Issues Projects Releases Packages Wiki Activity

upgrade_server: Implement server disabling

Browse source
This commit is contained in:
deajan 2025-01-14 22:38:58 +01:00
parent ba0e6b831c
commit 0c6a7472bd
2 changed files with 20 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
upgrade_server/upgrade_server/api.py
View file

@ -119,6 +119,10 @@ async def current_version(
crud.store_host_info(config_dict["upgrades"]["statistics_file"], host_id=data)
except KeyError:
logger.error("No statistics file set.")
if not crud.is_enabled():
return CurrentVersion(version="0.00-disabled")
try:
result = crud.get_current_version()
if not result:
@ -188,8 +192,12 @@ async def upgrades(
except KeyError:
logger.error("No statistics file set.")
# TODO:
# This can be amended by adding specific rules for host identity or groups or installed
if not crud.is_enabled():
raise HTTPException(
status_code=503,
detail="Service is currently disabled for maintenance"
)
file = FileGet(platform=platform, arch=arch, build_type=build_type, auto_upgrade_host_identity=auto_upgrade_host_identity, installed_version=installed_version, group=group)
try:
result = crud.get_file(file)
@ -252,11 +260,18 @@ async def download(
"arch": arch.value,
"build_type": build_type.value,
}
try:
crud.store_host_info(config_dict["upgrades"]["statistics_file"], host_id=data)
except KeyError:
logger.error("No statistics file set.")
if not crud.is_enabled():
raise HTTPException(
status_code=503,
detail="Service is currently disabled for maintenance"
)
file = FileGet(platform=platform, arch=arch, build_type=build_type, auto_upgrade_host_identity=auto_upgrade_host_identity, installed_version=installed_version, group=group)
try:
result = crud.get_file(file, content=True)

3
upgrade_server/upgrade_server/crud.py
View file

@ -53,7 +53,8 @@ def sha256sum_data(data):
def is_enabled() -> bool:
return not os.path.isfile("DISABLED")
path = os.path.join(config_dict["upgrades"]["data_root"], "DISABLED")
return not os.path.isfile(path)
def store_host_info(destination: str, host_id: dict) -> None: