o ViZ@sVdZddlZddlZddlmZmZmZGdddZdad dee defdd Z dS) z+ Configuration handling for Perth library. N)DictAnyOptionalc @seZdZdZdddddddd dd d Zd d eefd dZd eddfddZd eddfddZ d dedede de fddZ dedede ddfddZ dede ee ffddZdefddZdS)!Configz Configuration management for Perth. Handles loading, saving, and accessing configuration settings for the library. perthT)default_watermarkerverbosecpuimplicitN)devicerun_name models_diriD)default_sample_rate normalize)generalraudio config_pathc Csi|_|jD] \}}||j|<qtjtjtjtjt ddd|jdd<|r=tj |r?| |dSdSdS)z Initialize configuration with default values and optional user config. Args: config_path: Path to a YAML configuration file to load r perth_net pretrainedr N) _config _defaultsitemscopyospathjoindirnameabspath__file__existsload)selfrsectionvaluesr$@/home/ubuntu/.local/lib/python3.10/site-packages/perth/config.py__init__&szConfig.__init__returnc Csz>t|d }t|}Wdn1swY|r9|D]\}}||jvr3|j||q!||j|<q!WdSWdSty[}ztd|d|WYd}~dSd}~ww)z Load configuration from a YAML file. Args: config_path: Path to a YAML configuration file rNz$Warning: Could not load config from z: )openyaml safe_loadrrupdate Exceptionprint)r!rf user_configr"r#er$r$r%r <s    "z Config.loadcCs`tjtjtj|ddt|d}tj|j|ddWddS1s)wYdS)z Save current configuration to a YAML file. Args: config_path: Path to save the configuration to T)exist_okwFdefault_flow_styleN) rmakedirsrrrr)r*dumpr)r!rr/r$r$r%saveQs "z Config.saver"keydefaultcCs*||jvr||j|vr|j||S|S)a Get a configuration value. Args: section: Configuration section key: Configuration key default: Default value to return if key is not found Returns: Configuration value or default r)r!r"r9r:r$r$r%get\s z Config.getvaluecCs&||jvr i|j|<||j||<dS)z Set a configuration value. Args: section: Configuration section key: Configuration key value: Value to set Nr;)r!r"r9r=r$r$r%setls z Config.setcCs|j|iS)z Get an entire configuration section. Args: section: Configuration section name Returns: Dictionary of configuration values for the section )rr<r)r!r"r$r$r% get_sectionys zConfig.get_sectioncCstj|jddS)z4Return a string representation of the configuration.Fr4)r*r7r)r!r$r$r%__str__szConfig.__str__N)__name__ __module__ __qualname____doc__rrstrr&r r8rr<r>rr?r@r$r$r$r%r s(    rrr'cCs&tdur t|atS|rt|tS)z Get the global configuration instance. Args: config_path: Optional path to a configuration file to load Returns: Config instance N)_config_instancerr )rr$r$r% get_configs   rHrA) rErr*typingrrrrrGrFrHr$r$r$r%s