dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd l mZdd lmZdd lmZdd lmZdddZe j*e j,j.Gdde j0Ze j*e j,j4GddeZdZdZdZdZee_ y)z+Bare Metal Solution network update command.)absolute_import)division)unicode_literals) BmsClient)IpRangeReservation)waiter)base) exceptions)flags) labels_util)log) resourcesa  Update a Bare Metal Solution network. This call returns immediately, but the update operation may take several minutes to complete. To check if the operation is complete, use the `describe` command for the network. aH To update an network called ``my-network'' in region ``us-central1'' with a new label ``key1=value1'', run: $ {command} my-network --region=us-central1 --update-labels=key1=value1 To clear all labels, run: $ {command} my-network --region=us-central1 --clear-labels ) DESCRIPTIONEXAMPLESc&eZdZdZedZdZy)Update%Update a Bare Metal Solution network.ctj|dtj|tj j |tj|dy)z Register flags for this command.T) positionalF)hiddenN)r AddNetworkArgToParserr AddUpdateLabelsFlagsr ASYNC_FLAG AddToParserAddNetworkIpReservationToParserparsers "lib/surface/bms/networks/update.pyArgsz Update.Args:sD 48$$V,OO' ))&?ct}|jjj}d}tj j |}|j|}|j|jjj|jj}t||}|j|||}|j r't#j$|j'd|S|j(rIt"j*j-dj/|j'|j0|St2j4j7|j0dd} t9j:|j<} t9j>| | dj/|j0} t#j$|j'd| S) N)network_resourcelabelsip_reservationsnetwork)kindz@Update request issued for: [{}] Check operation [{}] for status.z/baremetalsolution.projects.locations.operationsv2) collection api_versionz&Waiting for operation [{}] to complete) rCONCEPTSr%Parser DiffFromUpdateArgs GetNetworkApplymessagesNetwork LabelsValuer# GetOrNone_ApplyIpReservationsUpdates UpdateNetworkdoner UpdatedResourceNameasync_statusPrintformatnamerREGISTRYParseRelativeNamerCloudOperationPollerNoResourcesoperation_serviceWaitFor) selfargsclientr% labels_update labels_diff orig_resourcer$op_ref op_resourcepollerress rRunz Update.RunBs [Fmm##))+GM""11$7K%%g.M%%++'ik2$ FO  ! ! '")F{{ ',,.y9 m {{ jj**0&*MO m$$66 D7K 3 3  "F ..AHH#[[* +C Y7 Jr N)__name__ __module__ __qualname____doc__ staticmethodrrMr rrr6s-@@#r rc eZdZdZedZy) UpdateAlpharc.tj|yN)rrrs rrzUpdateAlpha.Argsls KKr N)rNrOrPrQrRrrSr rrUrUhs-r rUct|drgS|jDcgc]-}t|j|j|j /}}t|drA|j }t|tjd|t|d|d|dgzSt|drt||jSycc}w) aGApplies the changes in args to the reservations in existing_network. Returns None if no changes were to be applied. Args: args: The arguments passed to the command. existing_network: The existing network. Returns: List of IP range reservations after applying updates or None if there are no changes. clear_ip_range_reservationsadd_ip_range_reservationzadd-ip-range-reservation start-address end-addressnoteremove_ip_range_reservationN) _IsSpecified reservationsr startAddress endAddressr]rZ_ValidateAgainstSpecr IP_RESERVATION_SPEC_RemoveReservationr^)rDexisting_networkrLexisting_reservationsres_dicts rr4r4rs$56 I"...#))3>>388D. $23,,H5#<#<35 8O4h}6M#F+ -$  $56 3">> @@7s2Cct|tjd|d}|d}t|D]4\}}|j|k(s|j |k(s&|d|||dzdzcSt dj||)Nzremove-ip-range-reservationr[r\zPCannot find an IP range reservation with start-address [{}] and end-address [{}])rcr IP_RESERVATION_KEY_SPEC enumerate start_address end_address LookupErrorr<)r`remove_key_dictrmrnirLs rreres(E(E46!/2- .+,'fa M)coo.L "1  QUV 4 44( 117 8C2E FFr cb|jD]}||vstj||yrW)keysr MissingPropertyError)dict_to_validatespec flag_nameprops rrcrcs.iikd ##  + +It <<r c$|j|S)zAReturns true if an arg is defined and specified, false otherwise.)IsKnownAndSpecified)rDr=s rr_r_s ! !$ ''r N)!rQ __future__rrr%googlecloudsdk.api_lib.bms.bms_clientrrgooglecloudsdk.api_lib.utilrgooglecloudsdk.callioper googlecloudsdk.command_lib.bmsr r $googlecloudsdk.command_lib.util.argsr googlecloudsdk.corer r DETAILED_HELP ReleaseTracks ReleaseTrackGA UpdateCommandrALPHArUr4rercr_ detailed_helprSr rrs2&';D.(50<#)   .D%%(().T  .*.bD%%++,&-!@H F= (%r