Spaces:
Sleeping
Sleeping
File size: 1,353 Bytes
b99ca3e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
from logging import Logger, basicConfig, getLogger
import facefusion.choices
from facefusion.common_helper import get_first, get_last
from facefusion.types import LogLevel
def init(log_level : LogLevel) -> None:
basicConfig(format = '%(message)s')
get_package_logger().setLevel(facefusion.choices.log_level_set.get(log_level))
def get_package_logger() -> Logger:
return getLogger('facefusion')
def debug(message : str, module_name : str) -> None:
get_package_logger().debug(create_message(message, module_name))
def info(message : str, module_name : str) -> None:
get_package_logger().info(create_message(message, module_name))
def warn(message : str, module_name : str) -> None:
get_package_logger().warning(create_message(message, module_name))
def error(message : str, module_name : str) -> None:
get_package_logger().error(create_message(message, module_name))
def create_message(message : str, module_name : str) -> str:
module_names = module_name.split('.')
first_module_name = get_first(module_names)
last_module_name = get_last(module_names)
if first_module_name and last_module_name:
return '[' + first_module_name.upper() + '.' + last_module_name.upper() + '] ' + message
return message
def enable() -> None:
get_package_logger().disabled = False
def disable() -> None:
get_package_logger().disabled = True
|