dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd l m Z dd l mZdd l mZdd l mZddlmZddlmZddlmZddlmZddlmZdZdZdZdZe j8j:fdZdZe j@e j8j:e jBGdde jDZ#e j@e j8jHGdde#Z%y)zFImplementation of objects update command for updating object settings.)absolute_import)division)unicode_literals) cloud_api) arg_parsers)base)encryption_util)errors)flags)name_expansion)stdin_iterator)user_request_args_factory) task_executor)task_graph_executor) task_status)patch_object_task)rewrite_object_taskz Update Cloud Storage objects. a Update a Google Cloud Storage object's custom-metadata: $ {command} gs://bucket/my-object --custom-metadata=key1=value1,key2=value2 One can use [wildcards](https://cloud.google.com/storage/docs/wildcards) to update multiple objects in a single command. for instance to update all objects to have a custom-metadata key: $ {command} gs://bucket/** --custom-metadata=key1=value1,key2=value2 Rewrite all JPEG images to the NEARLINE storage class: $ {command} gs://bucket/*.jpg --storage-class=NEARLINE You can also provide a precondition on an object's metageneration in order to avoid potential race conditions: $ {command} gs://bucket/*.jpg --storage-class=NEARLINE --if-metageneration-match=123456789  c#\K|jxs|jxs |j}|rtj}nt j }tj|tjj}tj|}t|ddxs t|dd}t|ddxs:t|ddxs+t|ddxst|ddxs t|d d}|s|rtjj}n9|s|rtjj }ntjj"}|j$r/|j&s#|j(s|st+j,d |j.rt0j2j4} nt0j2j6} t1j8||t0j:j<t?j@|| D]} || jB| yw) z0Yields PatchObjectTask's or RewriteObjectTask's.) metadata_type retain_untilNretention_modecustom_contextscustom_contexts_fileremove_custom_contextsupdate_custom_contextsclear_custom_contextsz;--all_versions flag is only allowed for ACL modifier flags.) fields_scopeinclude_buckets object_staterecursion_requested)user_request_args)"encryption_keyclear_encryption_key storage_classrRewriteObjectTaskrPatchObjectTaskr'get_user_request_args_from_command_args MetadataTypeOBJECTadds_or_removes_aclsgetattrr FieldsScopeFULLNO_ACLSHORT all_versionspredefined_aclacl_filer Error recursiver RecursionSettingYESNONameExpansionIterator BucketSetting NO_WITH_ERRORr get_object_state_from_flagsresource) urlsargsrequires_rewrite task_typer"r+updates_retentionupdates_custom_contextsrrecursion_settingname_expansion_results %lib/surface/storage/objects/update.py_get_task_iteratorrGAs LT66L$:L:L#55I!11I GG 7DDKKM3GGdND9W d>d%t,6 -t 46 / 66 / 66 . 5 -((--L3((//L((..L  T]].B ,,E GG ^^&77;;&77::-CC $22@@44T:+   &&:K  sH*H,c|jddd|jddd|j}tj|tj||jd t j d |jd d d dd|jddd|jdt j dtj|tj|tj|dtj|tj|d|tj|dtj|dy)zRegister flags for this command. Args: parser (argparse.ArgumentParser): The parser to add the arguments to. release_track (ReleaseTrack): The release track to add flags for. Returns: objects update flag group url*z$Specifies URLs of objects to update.)nargshelpz--all-versions store_truez-Perform the operation on all object versions.)actionrLz--event-based-holdz3Enables or disables an event-based hold on objects.z-Rz-rz --recursivezZRecursively update objects under any buckets or directories that match the URL expression.z-sz--storage-classzfSpecify the storage class of the object. Using this flag triggers a rewrite of underlying object data.)rLz--temporary-holdz0Enables or disables a temporary hold on objects.T) allow_patch)rO release_track) is_updatezRead the list of objects to update from stdin. No need to enter a source argument if this flag is present. Example: "storage objects update -I --content-type=new-type") help_textN) add_argument add_groupr add_acl_modifier_flagsadd_preserve_acl_flagrStoreTrueFalseActionadd_additional_headers_flagadd_continue_on_error_flagadd_encryption_flagsadd_precondition_flagsadd_object_metadata_flagsadd_per_object_retention_flagsadd_read_paths_from_stdin_flag)parserrPacl_flags_groups rF_add_common_argsrazsq  3CE   : $$&//o.  - - @B     # $   ./    - - =? ##F+""6*V6v&!! $m&&v>&&  Ac~y)zRegister flags for the alpha version of this command. Args: parser (argparse.ArgumentParser): The parser to add the arguments to. Returns: objects update flag group Nr_s rF_add_alpha_argsrfs rbc0eZdZdZeedZedZdZ y)UpdateUpdate Cloud Storage objects. DESCRIPTIONEXAMPLESct|yN)rares rFArgsz Update.Argss Vrbc tj||js|jd|_t j |j |j}t||}tjj}tj|d|tjtj j"d|j$|_y)NT)increment_type manifest_path)parallelizabletask_status_queueprogress_manager_argscontinue_on_error)r initialize_key_storer2 preserve_aclr get_urls_iterablerIread_paths_from_stdinrGrmultiprocessing_contextQueuer execute_tasksrProgressManagerArgs IncrementTypeINTEGERrv exit_code)selfr?r> task_iteratorrts rFRunz Update.Runs((.   4#4#4#<d  + + $,, D'tT2M+CCIIK"00+)==&44<< 00DNrbN) __name__ __module__ __qualname____doc___COMMAND_DESCRIPTION _GA_EXAMPLES detailed_help staticmethodrorrdrbrFrhrhs/&*- rbrhc0eZdZdZeeezdZedZ y) UpdateAlpharirjcbt|tjjt |yrn)rar ReleaseTrackALPHArfres rFrozUpdateAlpha.Argss VT..445FrbN) rrrrrr_ALPHA_EXAMPLESrrrordrbrFrrs-%*0- rbrN)&r __future__rrrgooglecloudsdk.api_lib.storagergooglecloudsdk.callioperr"googlecloudsdk.command_lib.storager r r r r r(googlecloudsdk.command_lib.storage.tasksrrr0googlecloudsdk.command_lib.storage.tasks.objectsrrrrrrGrGArarf ReleaseTracksUniverseCompatibleCommandrhrrrdrbrFrsM&'4/(>54==HBH@NP *6r,0+<+<+?+?;|  D%%(() T\\ * FD%%++, & - rb