dZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl mZddl mZdd l mZdd l mZd Zd Zd ZdZdZdZdZefdZefdZGddej6ZGddeZGddeZGddeZGddeZGddeZ Gd d!e!Z"Gd"d#e!Z#Gd$d%e#Z$Gd&d'e$Z%efd(Z&y))*z=Useful commands for interacting with the Cloud Filestore API.) list_pager)utils)apis)waiter)util) exceptions)log) properties) resourcesfilev1 v1p1alpha1v1beta1z!file.projects.locations.instanceszfile.projects.locationsz"file.projects.locations.operationsc6tjt|S)zImport and return the appropriate Cloud Filestore client. Args: version: str, the version of the API desired. Returns: Cloud Filestore client for the appropriate release track. )rGetClientInstanceAPI_NAMEversions 8lib/googlecloudsdk/api_lib/filestore/filestore_client.py GetClientr&s   ' 22c6tjt|S)zs rr@zFilestoreClient.messages`s == ! !!rNcJ|jj|}|jjj |}|j D]&}t jdj|(tj|jj|d|dS)awMake API calls to List active Cloud Filestore instances. Args: location_ref: The parsed location of the listed Filestore instances. limit: The number of Cloud Filestore instances to limit the results to. This limit is passed to the server and the server does the limiting. Returns: Generator that yields the Cloud Filestore instances. )parentzLocation {} may be unreachable. instancespageSizefieldlimitbatch_size_attribute) r@)FileProjectsLocationsInstancesListRequestr=projects_locations_instancesList unreachabler warningr8r YieldFromList)r9 location_refrGrequestresponselocations r ListInstanceszFilestoreClient.ListInstancesdsmmEEFG{{77<>rc||jj|j|}|j||S)z-Deletes an existing Cloud Filestore instance.)rVforcer@+FileProjectsLocationsInstancesDeleteRequestrX_DeleteInstance)r9rZasync_rnrPs rDeleteInstancezFilestoreClient.DeleteInstances?mmGG  & & (HG    00rcz|jj|j}|j||S)z,Delete an existing Cloud Filestore instance.rUro)r9rZrrrPs rDeleteInstanceAlphaz#FilestoreClient.DeleteInstanceAlphas=mmGG  & & (HG    00rc|jjj|}|r|Stjj |j t}|j|S)N collection) r=rJDeleter REGISTRYParseRelativeNamerVOPERATIONS_COLLECTIONWaitForOperation)r9rrrP delete_op operation_refs rrqzFilestoreClient._DeleteInstances] 88??HI &&88#89M   //rc|jj|j}|jjj |S)zGets description of a long-running operation. Args: operation_ref: the operation reference. Returns: messages.GoogleLongrunningOperation, the operation. rU)r@)FileProjectsLocationsOperationsGetRequestrXr=projects_locations_operationsrYr9rrPs r GetOperationzFilestoreClient.GetOperationsGmmEE  ' ' )FG ;; 4 4 8 8 AArctjtj|jj|dj |j S)aWaits on the long-running operation until the done field is True. Args: operation_ref: the operation reference. Raises: waiter.OperationError: if the operation contains an error. Returns: the 'response' field of the Operation. zWaiting for [{0}] to finish)rWaitForCloudOperationPollerNoResourcesr=rr8Name)r9rs rr}z FilestoreClient.WaitForOperationsL >>.. KK 5 5  %,,]-?-?-AB  rc|jj|j}|jjj |S)zCancels a long-running operation. Args: operation_ref: the operation reference. Returns: Empty response message. rU)r@,FileProjectsLocationsOperationsCancelRequestrXr=rCancelrs rCancelOperationzFilestoreClient.CancelOperationsGmmHH  ' ' )IG ;; 4 4 ; ;G DDrcd|jj|jj|j |}|j j j|}|r|Stjj|jt}|j|S)z"Create a Cloud Filestore instance.)rB instanceIdinstancerw)r@+FileProjectsLocationsInstancesCreateRequestParentrXrr=rJCreater rzr{rVr|r})r9rZrrconfigrP create_oprs rCreateInstancezFilestoreClient.CreateInstancesmmGG""$113$$&HG  88??HI &&88#89M   //rc|jj|}|jjj |S)NrU)r@FileProjectsLocationsGetRequestr=projects_locationsrY)r9rOrPs r GetLocationzFilestoreClient.GetLocations4mm;;;NG ;; ) ) - -g 66rc|jj|j}tj|j j |d|dS)NrU locationsrDrE)r@ FileProjectsLocationsListRequestrXrrNr=r)r9 project_refrGrPs r ListLocationszFilestoreClient.ListLocationssUmm<<  % % '=G  # # &&'  rc|jj|}tj|jj |d|dS)ayMake API calls to List active Cloud Filestore operations. Args: operation_ref: The parsed location of the listed Filestore instances. limit: The number of Cloud Filestore instances to limit the results to. This limit is passed to the server and the server does the limiting. Returns: Generator that yields the Cloud Filestore instances. rU operationsrDrE)r@*FileProjectsLocationsOperationsListRequestrrNr=r)r9rrGrPs rListOperationszFilestoreClient.ListOperationssOmmFF GG  # # 11'  rc|jj}||_|r||_|r||_|r||_| r|j j|| | r|j j|| |r|j j|||r|j j||||_ | r| |_ |r||_ | r| |d<|j j||| |rg|_|jj!}|j#d|_d|vr |d|_|j#dd}|j j)|||j#d}|r|j j+|||jj-||||_|||_|S)aParses the command line arguments for Create into a config. Args: tier: The tier. protocol: The protocol values are NFS_V3 (default) or NFS_V4_1. description: The description of the instance. file_share: The config for the file share. network: The network for the instance. performance: The performance configuration for the instance. labels: The parsed labels value. tags: The parsed tags value. zone: The parsed zone of the instance. nfs_export_options: The nfs export options for the file share. kms_key_name: The kms key for instance encryption. managed_ad: The Managed Active Directory settings of the instance. ldap: The LDAPS configuration of the instance. source_instance: The replication source of the instance. deletion_protection_enabled: bool, whether to enable deletion protection. deletion_protection_reason: The reason for enabling deletion protection. backend_type: The backend type of the instance (Compute or Filestore). Returns: The configuration that will be used as the request body for creating a Cloud Filestore instance. nfs_export_optionsrVzreserved-ip-rangez connect-modeDIRECT_PEERINGzpsc-endpoint-project)r@Instancetiertagsprotocol backendTyper2ParseManagedADIntoInstanceParseLdapIntoInstanceParseSourceInstanceIntoInstanceParsePerformanceIntoInstancelabels kmsKeyName descriptionParseFileShareIntoInstancenetworks NetworkConfiggetnetworkreservedIpRangeParseConnectModeParsePscEndpointProjectappenddeletionProtectionEnableddeletionProtectionReason)r9rrr file_sharer performancerrzoner kms_key_name managed_adldapsource_instancedeletion_protection_enableddeletion_protection_reason backend_typernetwork_config connect_modepsc_endpoint_projects rParseFilestoreConfigz$FilestoreClient.ParseFilestoreConfigsZ}}%%'HHM hm"h)h mm..xD  mm))(D9 mm33HoN mm00;GHO(h(h);j%&MM,,Xz4Hh}}224n&{{62n  ')01D)E&[[1ABl mm$$^\B%[[)?@  -- . ~.".+Fh(!-*Dh' Orc T|jj||||||||| | | |  } | S)a Parses updates into an instance config. Args: instance_config: The Instance message to update. description: str, a new description, if any. labels: LabelsValue message, the new labels value, if any. file_share: dict representing a new file share config, if any. performance: The performance configuration for the instance. managed_ad: The Managed Active Directory settings of the instance. disconnect_managed_ad: Disconnect from Managed Active Directory. ldap: The LDAP configuration of the instance. disconnect_ldap: Disconnect from LDAP. clear_nfs_export_options: bool, whether to clear the NFS export options. deletion_protection_enabled: bool, whether to enable deletion protection. deletion_protection_reason: The reason for enabling deletion protection. Raises: InvalidCapacityError, if an invalid capacity value is provided. InvalidNameError, if an invalid file share name is provided. Returns: The instance message. ) rrrrrdisconnect_managed_adrdisconnect_ldapclear_nfs_export_optionsrr)r2ParseUpdatedInstanceConfig)r9instance_configrrrrrrrrrrrrs rrz*FilestoreClient.ParseUpdatedInstanceConfigsIN}}773 '!9$?#=8 H Orc|jj|||}|r|Stjj |j t }|j|S)aPUpdates an instance. Args: instance_ref: the reference to the instance. instance_config: Instance message, the updated instance. update_mask: str, a comma-separated list of updated fields. async_: bool, if False, wait for the operation to complete. Returns: an Operation or Instance message. rw)r2UpdateInstancer rzr{rVr|r})r9rZr update_maskrr update_oprs rrzFilestoreClient.UpdateInstancesb ,,o{I &&88#89M   //rcd}d}d}d}d}g}|gS|D]}|jjj|jd|} |jjj|jd|} |jdd|k(r%|jd|} |jd |} nd} d} |j|jd g| | | | } |j | |S) aSCreates an NfsExportOptions message. Args: messages: The messages module. nfs_export_options: list, containing NfsExportOptions dictionary. Returns: File share message populated with values, filled with defaults. In case no nfs export options are provided we rely on the API to apply a default. READ_WRITE ROOT_SQUASHNO_ROOT_SQUASHN access-mode squash-modeanon_uidanon_gid ip-ranges)ipRangesanonUidanonGid accessMode squashMode)NfsExportOptionsAccessModeValueValuesEnumlookup_by_namerSquashModeValueValuesEnumr)r@r read_write root_squashno_root_squash anonimous_uid anonimous_gidnfs_export_configsnfs_export_option access_mode squash_moderrnfs_export_configs rMakeNFSExportOptionsMsgz'FilestoreClient.MakeNFSExportOptionsMsgs)JK%NMM! i/  # # = = L L##M:>   # # = = L L##M>B    }d 3{ B$((]C$((]C"33$((b9  4 12102 rc d}d}d}d}d}g}|gS|D]C}|jjj|jd|} |jjj|jd|} |jdd|k(r%|jd|} |jd |} nd} d} g} |jd g}|D]6}| j |jj j|8|j|jd g| | | | | |jd d }|j |F|S)aCreates an MakeNFSExportOptionsMsgBeta message. This function is a copy MakeNFSExportOptionsMsg with addition of handling the SecurityFlavors for NFSv41. Args: messages: The messages module. nfs_export_options: list, containing NfsExportOptions dictionary. Returns: File share message populated with values, filled with defaults. In case no nfs export options are provided we rely on the API to apply a default. rrrrNrrrrzsecurity-flavorsrr)rrrrrsecurityFlavorsr)rrrrrr'SecurityFlavorsValueListEntryValuesEnum)r@rrrrrrrrrrrrsecurity_flavors_listflavorsflavorrs rMakeNFSExportOptionsMsgBetaz+FilestoreClient.MakeNFSExportOptionsMsgBetas"JK%NMM! i/  # # = = L L##M:>   # # = = L L##M>B    }d 3{ B$((]C$((]C !%%&8"=g&$$  % % M M \ \   #33$((b9  /#'' 484 12G0H rc|yd|vr1|j|j|jdSd|vr1|j|j|jdSt d) aQCreates a PerformanceConfig message. Args: messages: The messages module. performance_config: A dictionary containing performance configuration, if any. Returns: The PerformanceConfig message. Raises: InvalidArgumentError: If performance_config argument constraints are violated. Nzmax-iops)maxIops) fixedIopszmax-iops-per-tb) maxIopsPerTb) iopsPerTbzNInvalid performance configuration. Must be one of max-iops or max-iops-per-gb.)PerformanceConfig FixedIOPSr IOPSPerTBr$)r@performance_configs rMakePerformanceConfigMsgz(FilestoreClient.MakePerformanceConfigMsg?s ! ''  ' '&&(,,Z8' ( 0 0  ' '&&-112CD' ( !  rr<)F)NNNNNNNNNNNNNNNNN NNNNNNNNFNN)rrr r!r5r:propertyr=r@rSr[rcrgrlrsrurqrr}rrrrrrrr staticmethodrrrr"rrr.r.Os..+K     " "4AAK?110 B( E 07 0     "&!%%i\  $"&!%5n0,..`==~""rr.cVeZdZdZdZ d dZdZ d dZdZdZ d Z y) r1z$Adapter for the alpha filestore API.cXtt|_tt|_yNr)rr0r=rr@r>s rr:zAlphaFilestoreAdapter.__init__hs$56DK(9:DMrNc(|jg|_|rd}d}d}|jDcgc]#}|j|jdk7s"|%c}|_d|vrytjj j jd}|jdxs|}tjj|||jd}d|vrutjj j jd}|jd}tjj|||jd}d||fvr td | | td tj|j |jd g} |j j#|jdt%j&|jd ||| } |jj)| yycc}w)a)Parse specified file share configs into an instance message. Args: instance: The Filestore instance object. file_share: File share configuration. instance_zone: The instance zone. Raises: InvalidArgumentError: If file_share argument constraints are violated. NrVzsource-snapshotTrequiredzsource-snapshot-region source-backupsource-backup-regionzEAt most one of ['source-snapshot', 'source-backup'] can be specified.OIf 'source-backup' is specified, 'source-backup-region' must also be specified.nfs-export-optionscapacity)rV capacityGbsourceSnapshot sourceBackupnfsExportOptions) fileSharesrVrr VALUEScoreprojectrY snapshot_utilSNAPSHOT_NAME_TEMPLATEr8 backup_utilBACKUP_NAME_TEMPLATEr$r.rr@FileShareConfigr BytesToGbr) r9rr instance_zonesource_snapshot source_backuprRfsrrfile_share_configs rrz0AlphaFilestoreAdapter.ParseFileShareIntoInstancels"homh ***bgg9O.O"*h j (##((0044d4C>>":;L}'>>EE Xz~~.?@  J &##((0044d4C>>"89#88?? Xz~~o>  o}5 5"     "x'7"   +BB --(> 9  228&!&& v(>3  :rc|jj||j|}|jjj |}|S)z6Send a Patch request for the Cloud Filestore instance.)rrV updateMask)r@*FileProjectsLocationsInstancesPatchRequestrXr=rJPatch)r9rZrrupdate_requestrs rrz$AlphaFilestoreAdapter.UpdateInstancesP]]MM  & & (NN  88>>~NI rc |jjjj|}||_y#t $rt dj|wxYw)z-Parse and match the supplied connection mode.zZ[{}] is not a valid connect-mode. Must be one of DIRECT_PEERING or PRIVATE_SERVICE_ACCESS.N)r@rConnectModeValueValuesEnumr connectModeKeyErrorr$r8)r9rkeyvalues rrz&AlphaFilestoreAdapter.ParseConnectModesc ) -- % % @ @ O O  $)n    $$*F3K s /9$Ar<r) rrr r!r:rr(rr,rrr"rrr1r1esS,; 15:4x  $"&!%JX4)rr1cVeZdZdZfdZdZdZdZdZdZ dZ d d Z d Z xZ S) r4z#Adapter for the beta filestore API.c~tt| tt|_t t|_yr )superr4r:rr3r=rr@r9 __class__s rr:zBetaFilestoreAdapter.__init__1s+ .0$45DK(89DMrcd|vrytjjjj d}|j d}| t dtjj|||j dS)Nr Tr rr) r rrrrYrr$rrr8)r9rrrRs r_ParseSourceBackupFromFilesharez4BetaFilestoreAdapter._ParseSourceBackupFromFileshare6sj( $$,,00$0?G~~45H     + + 2 2:>>/: rc|jd}| td|jd}| td|jj|jj |||_yzParses managed-ad configs into an instance message. Args: instance: The filestore instance struct. managed_ad: The --managed-ad flag value. Raises: InvalidArgumentError: If managed_ad argument constraints are violated. domainNz,Domain parameter is missing in --managed-ad.computerz.Computer parameter is missing in --managed-ad.)rGrH)r+rr$r@DirectoryServicesConfigManagedActiveDirectoryConfigr.r9rrrGrHs rrz/BetaFilestoreAdapter.ParseManagedADIntoInstanceD^^H %F ~ !O PP~~j)H : "&!F!F#}}IIH J "G"Hrc^|jd}| td|jd}| td|jd}|jd}|jd}|jj |jj ||||  |_y zParses ldap configs into an instance message. Args: instance: The filestore instance struct. ldap: The ldap cli parameters Raises: InvalidArgumentError: If ldap argument constraints are violated. rGNz(Domain parameter is missing in `--ldap`.serversz)Servers parameter is missing in `--ldap`.,zusers-ouz groups-ou)rGrPusersOugroupsOu)rrr$splitr@rJ LdapConfigr.r9rrrGrPusersougroupsous rrz*BetaFilestoreAdapter.ParseLdapIntoInstance]XXh F ~ !K LLhhy!G !L MMmmC Ghhz"Gxx $H"&!F!F ]] % % & "G"HrcF|jj||_y)z(Parse and match the supplied PSC config.)endpointProjectN)r@ PscConfig pscConfig)r9rrs rrz,BetaFilestoreAdapter.ParsePscEndpointProject{s!#}}66, 7 Nrc|jjjjd}g}|j |jj ||jj|||_y)zParses source_instance into a replication config and into an instance message. Args: instance: The filestore instance struct. source_instance: The source_instance cli param. STANDBY) peerInstance)rolereplicasN)r@ ReplicationRoleValueValuesEnumrr ReplicaConfig replication)r9rrrbrcs rrz4BetaFilestoreAdapter.ParseSourceInstanceIntoInstancesm == $ $ 8 8 G G DH OODMM//_/MN==44 H5HrcNtj|j||_y)zParses performance into performance config message and assigns to instance. Args: instance: The filestore instance struct. performance: The performance cli param. N)r.rr@performanceConfig)r9rrs rrz1BetaFilestoreAdapter.ParsePerformanceIntoInstances "1!I!I {"Hrc~|jg|_|rEd}d}|jDcgc]#}|j|jdk7s"|%c}|_d|vrRtjj j jd}|jd}| td|j|}tj|j|jdg}|jj|jdtj|jd || } |jj!| yycc}w) >"89  $$  :::Fm*FF --(rsD'0,.LP*#*)  :0<% 3'3 2J  2D5D;5;BuBHeH77SfSlF)FF)RW0WtX+Xt&4r