From 120ce61261a6a0729b45089bba8bcc983575a665 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 20 Sep 2024 12:22:32 +0200 Subject: [PATCH] Shorten scope in the logger --- facefusion/logger.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/facefusion/logger.py b/facefusion/logger.py index ef29ae54..1ea484e0 100644 --- a/facefusion/logger.py +++ b/facefusion/logger.py @@ -2,7 +2,7 @@ from logging import Logger, basicConfig, getLogger from typing import Tuple from facefusion.choices import log_level_set -from facefusion.common_helper import get_last +from facefusion.common_helper import get_first, get_last from facefusion.typing import LogLevel, TableContents, TableHeaders @@ -32,10 +32,12 @@ def error(message : str, module_name : str) -> None: def create_message(message : str, module_name : str) -> str: - scope = get_last(module_name.split('.')) + scopes = module_name.split('.') + first_scope = get_first(scopes) + last_scope = get_last(scopes) - if scope: - return '[' + scope.upper() + '] ' + message + if first_scope and last_scope: + return '[' + first_scope.upper() + '.' + last_scope.upper() + '] ' + message return message