INNER CODE UNIT · Python
CfnTagLoader
awslabs/awsome-distributed-ai · .github/scripts/stage-template-publish.py:16
class CfnTagLoader(yaml.SafeLoader):
"""YAML loader that preserves structure while tolerating CloudFormation tags."""
def _construct_unknown(loader: CfnTagLoader, node: yaml.Node) -> Any:
if isinstance(node, yaml.ScalarNode):
return loader.construct_scalar(node)
if isinstance(node, yaml.SequenceNode):
return loader.construct_sequence(node)
if isinstance(node, yaml.MappingNode):
return loader.construct_mapping(node)
raise TypeError(f"Unsupported YAML node: {type(node).__name__}")
CfnTagLoader.add_multi_constructor("", lambda loader, _tag, node: _construct_unknown(loader, node))
S3_KEY_RE = re.compile(r"^[A-Za-z0-9!_.*'()/.-]+$")