mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
unified logger
This commit is contained in:
+13
-12
@@ -5,11 +5,12 @@ from fastapi import Request, Depends
|
||||
from fastapi.exceptions import HTTPException
|
||||
from fastapi.responses import FileResponse
|
||||
from modules import shared
|
||||
from modules import logger
|
||||
from modules.api import models, helpers
|
||||
|
||||
|
||||
def post_shutdown():
|
||||
shared.log.info('Shutdown request received')
|
||||
logger.log.info('Shutdown request received')
|
||||
import sys
|
||||
sys.exit(0)
|
||||
|
||||
@@ -21,7 +22,7 @@ def get_js(request: Request):
|
||||
if ext not in ['js', 'css', 'map', 'html', 'wasm', 'ttf', 'mjs', 'json']:
|
||||
raise HTTPException(status_code=400, detail=f"invalid file extension: {ext}")
|
||||
if not os.path.exists(file):
|
||||
shared.log.error(f"API: file not found: {file}")
|
||||
logger.log.error(f"API: file not found: {file}")
|
||||
raise HTTPException(status_code=404, detail=f"file not found: {file}")
|
||||
if ext in ['js', 'mjs']:
|
||||
media_type = 'application/javascript'
|
||||
@@ -50,12 +51,12 @@ def get_motd():
|
||||
res = requests.get('https://vladmandic.github.io/sdnext/motd', timeout=3)
|
||||
if res.status_code == 200:
|
||||
msg = (res.text or '').strip()
|
||||
shared.log.info(f'MOTD: {msg if len(msg) > 0 else "N/A"}')
|
||||
logger.log.info(f'MOTD: {msg if len(msg) > 0 else "N/A"}')
|
||||
motd += res.text
|
||||
else:
|
||||
shared.log.error(f'MOTD: {res.status_code}')
|
||||
logger.log.error(f'MOTD: {res.status_code}')
|
||||
except Exception as err:
|
||||
shared.log.error(f'MOTD: {err}')
|
||||
logger.log.error(f'MOTD: {err}')
|
||||
return motd
|
||||
|
||||
def get_version():
|
||||
@@ -67,18 +68,18 @@ def get_platform():
|
||||
return { **installer_get_platform(), **loader_get_packages() }
|
||||
|
||||
def get_log(req: models.ReqGetLog = Depends()):
|
||||
lines = shared.log.buffer[:req.lines] if req.lines > 0 else shared.log.buffer.copy()
|
||||
lines = logger.log.buffer[:req.lines] if req.lines > 0 else logger.log.buffer.copy()
|
||||
if req.clear:
|
||||
shared.log.buffer.clear()
|
||||
logger.log.buffer.clear()
|
||||
return lines
|
||||
|
||||
def post_log(req: models.ReqPostLog):
|
||||
if req.message is not None:
|
||||
shared.log.info(f'UI: {req.message}')
|
||||
logger.log.info(f'UI: {req.message}')
|
||||
if req.debug is not None:
|
||||
shared.log.debug(f'UI: {req.debug}')
|
||||
logger.log.debug(f'UI: {req.debug}')
|
||||
if req.error is not None:
|
||||
shared.log.error(f'UI: {req.error}')
|
||||
logger.log.error(f'UI: {req.error}')
|
||||
return {}
|
||||
|
||||
|
||||
@@ -132,8 +133,8 @@ def get_progress(req: models.ReqProgress = Depends()):
|
||||
progress = min((current / total) if current > 0 and total > 0 else 0, 1)
|
||||
time_since_start = time.time() - shared.state.time_start
|
||||
eta_relative = (time_since_start / progress) - time_since_start if progress > 0 else 0
|
||||
# shared.log.critical(f'get_progress: batch {batch_x}/{batch_y} step {step_x}/{step_y} current {current}/{total} time={time_since_start} eta={eta_relative}')
|
||||
# shared.log.critical(shared.state)
|
||||
# logger.log.critical(f'get_progress: batch {batch_x}/{batch_y} step {step_x}/{step_y} current {current}/{total} time={time_since_start} eta={eta_relative}')
|
||||
# logger.log.critical(shared.state)
|
||||
res = models.ResProgress(id=shared.state.id, progress=round(progress, 2), eta_relative=round(eta_relative, 2), current_image=current_image, textinfo=shared.state.textinfo, state=shared.state.dict(), )
|
||||
return res
|
||||
|
||||
|
||||
Reference in New Issue
Block a user