tdZddlmZddlmZddlmZddlZddlZddlmZddlm Z ddlm Z dd lm Z dd l m Z dd lmZdd lmZdd lmZddlZddZej,ddZ ddZ ddZy)z1Provider-neutral tools for manipulating metadata.)absolute_import)division)unicode_literalsN)errors) posix_util) symlink_util)user_request_args_factory)resource_reference)yaml)function_result_cache)filescd}|r|dj|z } tj|}t|tst|t r|S tj|S#tj $r1}|djtj|z }Yd}~Ud}~wwxYw#tj$r1}|djtj|z }Yd}~nd}~wwxYwtj|)z2Converts JSON or YAML stream to an in-memory dict.zFound invalid JSON/YAMLz in {}z YAML Error: {}Nz JSON Error: {})formatr load isinstancedictlistYAMLParseErrorsix text_typejsonloadsJSONDecodeErrorrInvalidUrlError)string source_pathcurrent_error_stringparsedes 3lib/googlecloudsdk/api_lib/storage/metadata_util.pyread_yaml_json_from_stringr!#s2HOOK88J YYv F&$:fd#; m$< J ::f    J077 a8HIIJ   J077 a8HIIJ 344s/6A(B/(B,;'B''B,/C3'C..C3)maxsizectjjtjj|}t j |}t ||S)a7Converts JSON or YAML file to an in-memory dict. Args: file_path (str): Path for the file to parse passed in by the user. Returns: parsed (dict): Parsed value from the provided file_path. Raises: InvalidUrlError: The provided file_path either failed to load or be parsed as a dict. )r)ospathrealpath expanduserr ReadFileContentsr!) file_pathexpanded_file_pathcontentss r cached_read_yaml_json_filer,8sFww''(:(:9(EF  # #$6 7( #H) DDct|tj}|xr|}|xr|}t|xs0|xs,|xs(|jxs|j xs |j S)z8Checks for the storage provider's custom metadata field.)rr FileObjectResourceboolcustom_fields_to_setcustom_fields_to_removecustom_fields_to_update) resource_argspreserve_posixpreserve_symlinksattributes_resource known_posix file_resourceshould_parse_file_posixshould_parse_symlinkss r has_updated_custom_fieldsr<Ks-@@-+<}+=  / / /  + +/  . . /  . .  r-c$|j}|syt|tjr|nd}|r|rt j dt ||j|j||sy|jxr|}|jxr|}i}|s|r<|r|} n!tj||j} tj|| |r tj||j|jt j"k(r|S|jr|j%|j|St'|fi|} |j(r|j(D] } | | vs| | = |j*r| j%|j*| S)aReturns a dictionary containing new custom metadata for an object. Assumes that the custom metadata setter, clear flag, and a group containing the update and flags are in a mutually exclusive group, meaning values can be provided for one of these three flags/groups. The preserve POSIX flag is not a member of this group, meaning it can be set with any of these flags. Args: existing_custom_fields (dict): Existing custom metadata provided by an API. request_config (request_config): May contain custom metadata fields that should be modified. attributes_resource (Resource|None): If present, used for parsing POSIX and symlink data from a resource for the --preserve-posix and/or --preserve_symlink flags. This value is ignored unless it is an instance of FileObjectResource. known_posix (PosixAttributes|None): Set as custom metadata on target, skipping re-parsing from system. Returns: Optional[dict] that should be the value of the storage provider's custom metadata field. `None` means that existing metadata should remain unchanged. Empty dictionary means it should be cleared. Raises: errors.Error: If incompatible existing_custom_fields were encountered. NzWExisting custom fields should not exist when updating custom fields using local source.)r7r8)r6)r4rr r/rErrorr<r5r6r"get_posix_attributes_from_resource1update_custom_metadata_dict_with_posix_attributesr3update_custom_metadata_dict_with_symlink_attributes is_symlinkr1r CLEARupdaterr2r3) existing_custom_fieldsrequest_configr7r8r4r9r:r;posix_metadataposix_attributes custom_fieldskeys r get_updated_custom_fieldsrKds@!..-   ');)N)N O    ,,   ###&&-   *99Km(::L}.+$#FF >+K+K@@(*DD 00''+D+J+JJ ''-<<= -@@-**44  # 5**>>? r-)N)NN)__doc__ __future__rrrrr$"googlecloudsdk.command_lib.storagerrrr ,googlecloudsdk.command_lib.storage.resourcesr googlecloudsdk.corer googlecloudsdk.core.cacher googlecloudsdk.core.utilr rr!lrur,r<rKr-r rUsx8&' 59;HK$;* 5*4(E)E, 8 cr-