#ndZddlmZddlmZddlmZddlmZddlmZdZ dZ d Z d Z d Z d Zd Zy)z*Create request hook for snapshot settings.)absolute_import)division)unicode_literals)apis) exceptionsc|j |St|jdk7rtjdd|S)z8Validates that only a single location name is specified.storage_location_namesz5only a single location name is permitted at this time)r lenrInvalidArgumentException_argsreqs Llib/googlecloudsdk/api_lib/compute/snapshot_settings/modify_request_hooks.pyvalidate_single_locationrsM   ( *  4 & &'1,  - - ?  *c(t|fd}|S)z4Adds locations to the request if they are specified.c |js|Sjj|jDcgc]}t|c}}t |||j j _|Scc}w)N)additionalProperties)r 'SnapshotSettingsStorageLocationSettingsLocationsValue_wrap_location_name_ensure_location_fieldsnapshotSettingsstorageLocation locations)rrrlocation locations_msgmessagess r_maybe_add_locationsz1maybe_add_locations.._maybe_add_locations)s  & & j 88GG!% ; ;" ;H$Hh7 ;" H 3)5BC((2 J"sA7)_get_message_module)versionr!r s @rmaybe_add_locationsr$%s  )($ rcp|jjj||j|S)z9Wraps a location name into the appropriate proto message.)name)keyvalue)rrAdditionalProperty@SnapshotSettingsStorageLocationSettingsStorageLocationPreference)rr s rrr>sC  9 9 H H [ [   U U V  \ rcVtjdtjd|S)z/Returns the message module for the Compute API.compute)rGetMessagesModuleResolveVersion)r#s rr"r"Is)   $$Y8 rc|js|j|_|jjs |j|j_yy)z'Ensures that the location field is set.N)rSnapshotSettingsrr)rr s rrrQsL  #446C   - -88:( .rc|jr/djt|jdgz|_|S|jr:djd|jj ddgzD|_|S)aAdjusts the update mask for storage locations. If storage location policy is specified, then the update mask is adjusted so that the whole storage location structure is replaced. If a storage location name is specified, then the update mask is specified so that other storage location names are clearead. Args: _: this is ignored args: the parsed CLI args. req: the request message, partially populated. Returns: the request message with modified update mask. ,rc3$K|]}|r| yw)N).0masks r z6adjust_storage_location_update_mask..xs , ,szstorageLocation.locations)storage_location_policyjoin"_remove_all_storage_location_masks updateMaskr splitr s r#adjust_storage_location_update_maskr=]s" !!XX*3>>:>O=PPCN * ""XX ,,S1*+, CN *rcp|jdDcgc]}|r|jds|c}Scc}w)Nr2r)r< startswith)r6s rr:r:s@**S/ !$ doo&78 !  s3N)__doc__ __future__rrrgooglecloudsdk.api_lib.utilrgooglecloudsdk.callioperrr$rr"rr=r:r4rrrDs=1&',.  2 # Lr