iddlmZddlmZddlmZmZddlZ dZGddZ Gdd Z Gd d ejZ y) ) annotations)StringIO)DictIteratorN__parsing_context__c2eZdZdZddZeddZddZy)ParsingContextz_Container class for file slice information used to populate model metadata for certain objects.c<||_||_||_||_y)zInitializer for the ParsingContext class. The contents are represented internally as a yaml.Node in order to allow for lazy serialization to string representations. N) start_lineend_linefilename _content_node)selfr r r content_nodes l/var/www/html/strategist-ai/venv/lib/python3.12/site-packages/dbt_semantic_interfaces/parsing/yaml_loader.py__init__zParsingContext.__init__s! %    )cBtj|jS)zSerialized contents associated with the file slice represented by this ParsingContext object. This should only be called when a string representation of the contents are needed. )node)yaml serializerrs rcontentzParsingContext.contents ~~4#5#566rc:d|jd|jS)Nzline: z , filename: )r r rs r__str__zParsingContext.__str__#s( T]]ODDrN) r intr rr strrz yaml.NodereturnNone)rr)__name__ __module__ __qualname____doc__rpropertyrrrrr r s$i *77Err c4eZdZdZeddZeddZy)YamlConfigLoaderzMHelper class for loading YAML config strings into an iterator of YAML output.c#Kt|5}||_tj|tD]}| dddy#1swYyxYww)a6Wraps the yaml.load_all method and returns the resulting iterator with parsing context added to output. This replaces any calls to yaml.load_all(loader=SafeLineLoaderWithAddedContext), which internally adds ParsingContext info. Note PyYAML reads the name property from the input stream IF that input stream is a file object, otherwise it replaces it with a constant. Therefore, we use as StringIO instance to pass the contents into PyYAML and set the value of the name property on the file object to the name parameter here. ) initial_value)streamLoaderN)rnamerload_allSafeLineLoaderWithAddedContext)r,contentsr*documents rload_all_with_contextz&YamlConfigLoader.load_all_with_context*sMH - FK MM@^_     s A,A AA AcJ|jdxs|jdS)zAChecks if YAML file name ends with one of the supported suffixes.z.yamlz.yml)endswith)r s ris_valid_yaml_file_endingz*YamlConfigLoader.is_valid_yaml_file_ending8s%  )FX->->v-FFrN)r,rr/rrr)r rrbool)r r!r"r# staticmethodr1r4r%rrr'r''s-W  GGrr'c&eZdZdZddfd ZxZS)r.zAdds special field __parsing_context__ to all mappings. Credit: https://stackoverflow.com/questions/13319067/parsing-yaml-return-with-line-number ctt| ||}t|jj dz|j j |jj||t<|S)aOverride of the construct_mapping method in the PyYAML SafeLoader class. This override exists in order to populate the parsing context object with file location and raw YAML content information, which will be used to populate the Metadata model construct for nodes which request it. The file identifier (node.start_mark.name) is derived from the name property on the file-like object passed in as the stream parameter of the load_all call. The line numbers are part of the start and end mark properties. The content_node is a PyYAML node. We do not serialize it inside the loader because it's a full serialization pass on whatever contents this particular mapping node might contain, which could be the entire model collection. Rather, we store the node and serialize it on-demand later. Note: PyYAML uses metaclasses quite heavily so construct_mapping is in fact defined in the SafeConstructor class. )deep) superr.construct_mappingr start_marklineend_markr,PARSING_CONTEXT_KEY)rrr9mapping __class__s rr<z0SafeLineLoaderWithAddedContext.construct_mappingHsb 6OPT[_O`'5 OO 1 $dmm&8&8$//:N:NPT( #$r)F)rzyaml.MappingNoder9r5rr)r r!r"r#r< __classcell__)rBs@rr.r.>srr.) __future__riortypingrrrr@r r' SafeLoaderr.r%rrrHsE"! T+EE4GG.T__r