accounting/bench
2
0
Fork 0
mirror of https://github.com/frappe/bench.git synced 2025-01-24 15:38:25 +00:00
Code Issues Releases Activity

refactor: bench.utils.log

Browse Source 
Maintain `bench.LOG_BUFFER` for entire process when `bench.cli.fancy` is set
This commit is contained in:
Gavin D'souza 2021-11-18 00:36:13 +05:30
parent 747be65f92
commit ac07d8dc52
2 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
bench/cli.py
View File

@ -30,6 +30,7 @@ from bench.utils import (
from bench.utils.bench import get_env_cmd from bench.utils.bench import get_env_cmd
from_command_line = False from_command_line = False
bench.LOG_BUFFER = []
change_uid_msg = "You should not run this command as root" change_uid_msg = "You should not run this command as root"
src = os.path.dirname(__file__) src = os.path.dirname(__file__)

25
bench/utils/__init__.py
View File

@ -48,19 +48,32 @@ def is_frappe_app(directory: str) -> bool:
return bool(is_frappe_app) return bool(is_frappe_app)
def log(message, level=0): def log(message, level=0, no_log=False):
import bench
import bench.cli
levels = { levels = {
0: ("blue", "INFO"), # normal 0: ("blue", "INFO"), # normal
1: ("green", "SUCCESS"), # success 1: ("green", "SUCCESS"), # success
2: ("red", "ERROR"), # fail 2: ("red", "ERROR"), # fail
3: ("yellow", "WARN"), # warn/suggest 3: ("yellow", "WARN"), # warn/suggest
} }
loggers = {2: logger.error, 3: logger.warning}
color, prefix = levels.get(level, levels[0])
level_logger = loggers.get(level, logger.info)
level_logger(message) color, prefix = levels.get(level, levels[0])
click.secho(f"{prefix}: {message}", fg=color)
if bench.cli.from_command_line and bench.cli.fancy:
bench.LOG_BUFFER.append(
{"prefix": prefix, "message": message, "color": color,}
)
if no_log:
click.secho(message, fg=color)
else:
loggers = {2: logger.error, 3: logger.warning}
level_logger = loggers.get(level, logger.info)
level_logger(message)
click.secho(f"{prefix}: {message}", fg=color)
def check_latest_version(): def check_latest_version():