U
|
ú{`H ã @ sf d Z ddlZddlZddlmZ ddlZddlmZ ddl m
|
Z
|
dZ G dd deZed d
|
d Z dS ) z9
|
@author: l1aoxingyu
|
@contact: sherlockliao01@gmail.com
|
é N)ÚAny)ÚCfgNode)Ú PathManagerZ_BASE_c sn e Zd ZdZedeedddZdeedddZ fd d
|
Z e
|
d fd d Z ee d fdd Z ZS )r aM
|
Our own extended version of :class:`yacs.config.CfgNode`.
|
It contains the following extra features:
|
1. The :meth:`merge_from_file` method supports the "_BASE_" key,
|
which allows the new CfgNode to inherit all the attributes from the
|
base configuration file.
|
2. Keys that start with "COMPUTED_" are treated as insertion-only
|
"computed" attributes. They can be inserted regardless of whether
|
the CfgNode is frozen or not.
|
3. With "allow_unsafe=True", it supports pyyaml tags that evaluate
|
expressions in config. See examples in
|
https://pyyaml.org/wiki/PyYAMLDocumentation#yaml-tags-and-python-types
|
Note that this may lead to arbitrary code execution: you must not
|
load a config file from untrusted sources before manually inspecting
|
the content of the file.
|
F)ÚfilenameÚ allow_unsafec s t | d¡v}zt |¡}W n` tjjk
|
r| |s6 t t¡}| d
|
| ¡¡ | ¡ t| d}t |¡}W 5 Q R X Y nX W 5 Q R X fdd t |kr
|
|t }| d¡r¼tj |¡}tt|jdddgsètj tj | ¡|¡}tj||d }|t = || |S |S )
|
a
|
Just like `yaml.load(open(filename))`, but inherit attributes from its
|
`_BASE_`.
|
Args:
|
filename (str): the file name of the current config. Will be used to
|
find the base config file.
|
allow_unsafe (bool): whether to allow loading the config file with
|
`yaml.unsafe_load`.
|
Returns:
|
(dict): the loaded yaml
|
ÚrzlLoading config {} with yaml.unsafe_load. Your machine may be at risk if the file contains malicious content.c s\ | ¡ D ]N\}}t|trN||krNt|| ts>td |¡ ||| q|||< qd S )Nz"Cannot inherit key '{}' from base!)ÚitemsÚ
|
isinstanceÚdictÚAssertionErrorÚformat)ÚaÚbÚkÚv©Úmerge_a_into_b© ú./config/config.pyr C s ÿþz3CfgNode.load_yaml_with_base.<locals>.merge_a_into_bú~ú/zhttps://zhttp://©r )r ÚopenÚyamlZ safe_loadÚ constructorZConstructorErrorÚloggingÚ getLoggerÚ__name__Úwarningr ÚcloseZ unsafe_loadÚBASE_KEYÚ
|
startswithÚosÚpathÚ
|
expanduserÚanyÚmapÚjoinÚdirnamer Úload_yaml_with_base)r r ÚfÚcfgÚloggerZ base_cfg_fileZbase_cfgr r r r) % sF
|
þÿ $
|
|
ÿ
|
ÿ ÿ
|
zCfgNode.load_yaml_with_base)Ú cfg_filenamer c C s t j||d}t| |}dS )zí
|
Merge configs from a given yaml file.
|
Args:
|
cfg_filename: the file name of the yaml config.
|
allow_unsafe: whether to allow loading the config file with
|
`yaml.unsafe_load`.
|
r N)r r) Útype)Úselfr- r Z
|
loaded_cfgr r r Úmerge_from_fileb s
|
ÿzCfgNode.merge_from_filec s" t |kstd t ¡t |¡S )zO
|
Args:
|
cfg_other (CfgNode): configs to merge from.
|
ú0The reserved key '{}' can only be used in files!)r r r ÚsuperÚmerge_from_other_cfg)r/ Z cfg_other©Ú __class__r r r3 q s
|
ÿþz CfgNode.merge_from_other_cfg)Úcfg_listc s4 t |ddd
}t|ks(td t¡t |¡S )zS
|
Args:
|
cfg_list (list): list of configs to merge from.
|
r Né r1 )Úsetr r r r2 Úmerge_from_list)r/ r6 Úkeysr4 r r r9 } s ÿþzCfgNode.merge_from_list)ÚnameÚvalc sT | d¡rB|| kr8| | }||kr&d S td |||¡|| |< nt ||¡ d S )NZ COMPUTED_zOComputed attributed '{}' already exists with a different value! old={}, new={}.)r! ÚKeyErrorr r2 Ú __setattr__)r/ r; r< Zold_valr4 r r r> s
|
þÿ
|
zCfgNode.__setattr__)F)F)r Ú
|
__module__Ú __qualname__Ú__doc__Ú staticmethodÚstrÚboolr) r0 r3 Úlistr9 r r> Ú __classcell__r r r4 r r s < r )Úreturnc C s ddl m} | ¡ S )z]
|
Get a copy of the default config.
|
Returns:
|
a fastreid CfgNode instance.
|
é ©Ú_C)ÚdefaultsrJ ÚclonerI r r r Úget_cfg s rM ) rA r r" Útypingr r Z yacs.configr Z_CfgNodeÚ utils.file_ior r rM r r r r Ú<module> s
|
