93 lines
2.3 KiB
Python
93 lines
2.3 KiB
Python
from configparser import ConfigParser
|
|
from typing import Any, List, Optional
|
|
|
|
from facefusion import state_manager
|
|
from facefusion.common_helper import cast_float, cast_int
|
|
|
|
CONFIG_PARSER = None
|
|
|
|
|
|
def get_config_parser() -> ConfigParser:
|
|
global CONFIG_PARSER
|
|
|
|
if CONFIG_PARSER is None:
|
|
CONFIG_PARSER = ConfigParser()
|
|
CONFIG_PARSER.read(state_manager.get_item('config_path'), encoding = 'utf-8')
|
|
return CONFIG_PARSER
|
|
|
|
|
|
def clear_config_parser() -> None:
|
|
global CONFIG_PARSER
|
|
|
|
CONFIG_PARSER = None
|
|
|
|
|
|
def get_str_value(key : str, fallback : Optional[str] = None) -> Optional[str]:
|
|
value = get_value_by_notation(key)
|
|
|
|
if value or fallback:
|
|
return str(value or fallback)
|
|
return None
|
|
|
|
|
|
def get_int_value(key : str, fallback : Optional[str] = None) -> Optional[int]:
|
|
value = get_value_by_notation(key)
|
|
|
|
if value or fallback:
|
|
return cast_int(value or fallback)
|
|
return None
|
|
|
|
|
|
def get_float_value(key : str, fallback : Optional[str] = None) -> Optional[float]:
|
|
value = get_value_by_notation(key)
|
|
|
|
if value or fallback:
|
|
return cast_float(value or fallback)
|
|
return None
|
|
|
|
|
|
def get_bool_value(key : str, fallback : Optional[str] = None) -> Optional[bool]:
|
|
value = get_value_by_notation(key)
|
|
|
|
if value == 'True' or fallback == 'True':
|
|
return True
|
|
if value == 'False' or fallback == 'False':
|
|
return False
|
|
return None
|
|
|
|
|
|
def get_str_list(key : str, fallback : Optional[str] = None) -> Optional[List[str]]:
|
|
value = get_value_by_notation(key)
|
|
|
|
if value or fallback:
|
|
return [ str(value) for value in (value or fallback).split(' ') ]
|
|
return None
|
|
|
|
|
|
def get_int_list(key : str, fallback : Optional[str] = None) -> Optional[List[int]]:
|
|
value = get_value_by_notation(key)
|
|
|
|
if value or fallback:
|
|
return [ cast_int(value) for value in (value or fallback).split(' ') ]
|
|
return None
|
|
|
|
|
|
def get_float_list(key : str, fallback : Optional[str] = None) -> Optional[List[float]]:
|
|
value = get_value_by_notation(key)
|
|
|
|
if value or fallback:
|
|
return [ cast_float(value) for value in (value or fallback).split(' ') ]
|
|
return None
|
|
|
|
|
|
def get_value_by_notation(key : str) -> Optional[Any]:
|
|
config_parser = get_config_parser()
|
|
|
|
if '.' in key:
|
|
section, name = key.split('.')
|
|
if section in config_parser and name in config_parser[section]:
|
|
return config_parser[section].get(name)
|
|
if key in config_parser:
|
|
return config_parser.get(key)
|
|
return None
|