DRMTalks
/
devine
mirror of https://github.com/devine-dl/devine.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

Replace logging FileHandler with console.save_text

This commit is contained in:
rlaphoenix 2023-02-25 11:38:02 +00:00
parent 34a2a8e4e6
commit 77c16f557c
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
devine/core/__main__.py
View File

@ -1,3 +1,4 @@
import atexit
import logging import logging
from datetime import datetime from datetime import datetime
from pathlib import Path from pathlib import Path
@ -11,8 +12,11 @@ from devine.core import __version__
from devine.core.commands import Commands from devine.core.commands import Commands
from devine.core.config import config from devine.core.config import config
from devine.core.constants import LOG_FORMAT, LOG_FORMATTER, context_settings from devine.core.constants import LOG_FORMAT, LOG_FORMATTER, context_settings
from devine.core.console import console
from devine.core.utilities import rotate_log_file from devine.core.utilities import rotate_log_file
LOGGING_PATH = None
@click.command(cls=Commands, invoke_without_command=True, context_settings=context_settings) @click.command(cls=Commands, invoke_without_command=True, context_settings=context_settings)
@click.option("-v", "--version", is_flag=True, default=False, help="Print version information.") @click.option("-v", "--version", is_flag=True, default=False, help="Print version information.")
@ -26,10 +30,10 @@ def main(version: bool, debug: bool, log_path: Path) -> None:
coloredlogs.install(level=log.level, fmt=LOG_FORMAT, style="{") coloredlogs.install(level=log.level, fmt=LOG_FORMAT, style="{")
if log_path: if log_path:
global LOGGING_PATH
console.record = True
new_log_path = rotate_log_file(log_path) new_log_path = rotate_log_file(log_path)
fh = logging.FileHandler(new_log_path, encoding="utf8") LOGGING_PATH = new_log_path
fh.setFormatter(LOG_FORMATTER)
log.addHandler(fh)
urllib3.disable_warnings(InsecureRequestWarning) urllib3.disable_warnings(InsecureRequestWarning)
@ -40,5 +44,12 @@ def main(version: bool, debug: bool, log_path: Path) -> None:
return return
@atexit.register
def save_log():
if console.record and LOGGING_PATH:
# TODO: Currently semi-bust. Everything that refreshes gets duplicated.
console.save_text(LOGGING_PATH)
if __name__ == "__main__": if __name__ == "__main__":
main() main()