DdZddlmZddlmZddlmZddlZddlZddlZddlZddl m Z ddl m Z ddl m Z dd l mZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlZej<ddZdZ dZ!dZ"dZ#Gdde$Z%dZ&dZ'd)dZ(d*dZ)dZ*dZ+ d+d Z,d!Z- d,d"Z.d-d#Z/d$Z0Gd%d&e$Z1Gd'd(e1Z2y).+Common utility functions for sql instances.)absolute_import)division)unicode_literalsN) list_pager)api_util) constants) exceptions)apis)config)execution_utils)log) properties) console_io)encoding)filessqlv1beta4z#Failed to start the Cloud SQL ProxyMYSQLPOSTGRES SQLSERVERc"eZdZdZdZdZdZy)DatabaseInstancePresentationzKRepresents a DatabaseInstance message that is modified for user visibility.c|jD]}|jdk(ra|jrC|jjtj j jk(rd|_a|j|_st||j}|t|tr|s|jdvr+t||jtj|t||j|y)NstateSTOPPED)currentDiskSize maxDiskSize) all_fieldsnamesettingsactivationPolicymessagesSettingsActivationPolicyValueValuesEnumNEVERrgetattr isinstancelistsetattrsix text_type)selforigfieldvalues +lib/googlecloudsdk/api_lib/sql/instances.py__init__z%DatabaseInstancePresentation.__init__5s" w  ==T]];;x?P?P?p?p?v?vv $*zz$*ejj)  j&=e ZZ= = D%**cmmE&: ; D%**e ,#cVt|tr|j|jk(Sy)zAOverrides the default implementation by checking attribute dicts.F)r(r__dict__r-others r1__eq__z#DatabaseInstancePresentation.__eq__Ds#%56 ]]enn ,, r3c&|j| S)z@Overrides the default implementation (only needed for Python 2).)r8r6s r1__ne__z#DatabaseInstancePresentation.__ne__Js{{5! !!r3N)__name__ __module__ __qualname____doc__r2r8r:r3r1rr2sS - "r3rcL|jd}dj|ddS)z>Parses and returns the region string from the gce_zone string.-N)splitjoin)gce_zonezone_componentss r1GetRegionFromZonerGOs&NN3'/ /#2& ''r3ctjj|xr$tj|tjS)zBChecks if cloud_sql_proxy_path binary is present in cloud sdk bin.)ospathexistsaccessX_OK)cloud_sql_proxy_paths r1_IsCloudSqlProxyPresentInSdkBinrOVs0 ''..- . 3 ))("'' 24r3c0tjj}|r-tjj ||}t |r|Stj|}|rtjd|d|d|Stjd)zDetermines the path to the Cloud SQL Proxy binary. Args: binary_name: Name of Cloud SQL Proxy binary to look for in path. (v1: 'cloud_sql_proxy', v2: 'cloud-sql-proxy') Returns: str: The path to the Cloud SQL Proxy binary. zUsing z found at []zbA Cloud SQL Proxy SDK root could not be found, or access is denied.Please check your installation.) r Paths sdk_bin_pathrIrJrDrO file_utilsFindExecutableOnPathrdebugsql_exceptionsSqlProxyNotFound) binary_namerSrN proxy_paths r1_GetCloudSqlProxyPathr[\s,,,77<< kB&';< !!..{;*II{m;zl!<=   ) ) * ++r3cdjt}|rdjt|}tj|)Nz{}.z{}: {})format_BASE_CLOUD_SQL_PROXY_ERRORrWCloudSqlProxyError) error_msgmessages r1_RaiseProxyErrorrb{s6 LL4 5'oo99EG))'22r3c\tj|jjS)z=Reads and returns the next line from the proxy stderr stream.)rDecodestderrreadline proxy_processs r1_ReadLineFromStderrri --668 99r3c\tj|jjS)z=Reads and returns the next line from the proxy stdout stream.)rrdstdoutrfrgs r1_ReadLineFromStdoutrmrjr3c|d}d}|jt|}|rxtjj |t j |vrtdj|n|rd|vr|S|st j|vr|St|}|rx||k\r td||z }tj||j|r5 t|}|sn&tjj |d|vr|S4|jj}|rtj|}|j!} g} | D]&}|j#drn| j%|(dj'| j)} | r t| yt|yty) aWait for the proxy to be ready for connections, then return proxy_process. Args: proxy_process: The Process corresponding to the Cloud SQL Proxy (v2). port: int, the port that the proxy was started on. seconds_to_timeout: Seconds to wait before timing out. run_connection_test: if true, waits for connection test success message. Returns: The Process object corresponding to the Cloud SQL Proxy (v2). r皙?NePort already in use. Exit the process running on port {} or try connecting again on a different port.zConnection test passed Timed out.zUsage: )pollrmrstatuswriter PROXY_ADDRESS_IN_USE_ERRORrbr]"PROXY_V2_READY_FOR_CONNECTIONS_MSGtimesleeprereadrrd splitlines startswithappendrDstrip) rhportseconds_to_timeoutrun_connection_testtotal_wait_secondsseconds_to_sleeplineerr decoded_err err_lineserror_only_linesr`s r1_WaitForProxyV2ToStartrs$ } -D  jjt  - - 5 44:F4L  #;t#C!::dB  /d $//|$**JJ 3 $:   /d   jjt !T )  !!##//#&K&&(I  " d#  *+113Iy!{#r3cd}d}|jt|}|rntjj |t j |vrtdj|nt j|vr|St|}|rn||k\r td||z }tj||jty)a_Wait for the proxy to be ready for connections, then return proxy_process. Args: proxy_process: Process, the process corresponding to the Cloud SQL Proxy. port: int, the port that the proxy was started on. seconds_to_timeout: Seconds to wait before timing out. Returns: The Process object corresponding to the Cloud SQL Proxy. rroNrprq) rsrirrtrur rvrbr]PROXY_READY_FOR_CONNECTIONS_MSGrxry)rhrrrrrs r1_WaitForProxyToStartrs$ } -D  jjt  - - 5 44:F4L B  4 4 <  /d //|$**JJ ' $,r3c td} |jdt|g} |r| jd|r| jd|r| jd|r| jd|r| jd| r| d | gz } | r| jd |r| d |gz } n\tj j jjd } | dtjj| gz } tj| g| }tjj!dj#dj%| t'j(|t&j*t&j*t&j*}t7|||| S#t,$rG}|j.t.j0k(r$t3j4dj#| d}~wwxYw)amStarts the Cloud SQL Proxy (v2) for instance on the given port. Args: instance: The instance to start the proxy for. port: The port to bind the proxy to. seconds_to_timeout: Seconds to wait before timing out. impersonate_service_account: Service account to impersonate. auto_iam_authn: Whether to use IAM DB Authentication. private_ip: Whether to use private IP. psc: Whether to use PSC. auto_ip: Whether to use auto IP detection. debug_logs: Whether to enable verbose logs in proxy. sqladmin_api_endpoint: Cloud SQL Admin API endpoint for Cloud SQL Proxy. run_connection_test: Whether to run connection test. Returns: The Process object corresponding to the Cloud SQL Proxy. Raises: CloudSqlProxyError: An error starting the Cloud SQL Proxy. SqlProxyNotFound: An error finding a Cloud SQL Proxy installation. zcloud-sql-proxy)rYz--portz --private-ipz--pscz --auto-ipz --debug-logsz--auto-iam-authnz--sqladmin-api-endpointz--run-connection-testz--impersonate-service-accountTrequiredz--credentials-filez#Starting Cloud SQL Proxy: [{args}]  argsrlstdinre'Failed to start Cloud SQL Proxy. Please make sure it is available in the PATH [{}]. Learn more about installing the Cloud SQL Proxy here: https://cloud.google.com/sql/docs/mysql/connect-auth-proxy#install. If you would like to report this issue, please run the following command: gcloud feedbackN)r[connectionNamestrr}rVALUEScoreaccountGetr rRLegacyCredentialsAdcPathr ArgsForExecutableToolrrtrur]rD subprocessPopenPIPEEnvironmentErrorerrnoENOENTrWr_r)instancerrimpersonate_service_accountauto_iam_authn private_ippscauto_ip debug_logssqladmin_api_endpointr command_pathrr proxy_argsrhes r1StartCloudSqlProxyV2rsB'3DE,  ! !8SY 7$KKKK KK KKKK"# &(= >>DKK'(  ,.I JJD$$,,00$0?G //8 D44\IDI***,33*9M3N $$oo M( T-/B    ww%,,  - - %&,VL%9     sAF,, G<5AG77G<ct}ddj|j|g}tjj j jd}|dtjj|gz }tj|g|}tjjdjdj!| t#j$|t"j&t"j&t"j& }t3|||S#t($rG}|j*t*j,k(r$t/j0d j|d }~wwxYw) aStarts the Cloud SQL Proxy (v1) for instance on the given port. Args: instance: The instance to start the proxy for. port: The port to bind the proxy to. seconds_to_timeout: Seconds to wait before timing out. Returns: The Process object corresponding to the Cloud SQL Proxy. Raises: CloudSqlProxyError: An error starting the Cloud SQL Proxy. SqlProxyNotFound: An error finding a Cloud SQL Proxy installation. z -instancesz {}=tcp:{}Trz-credential_filez$Starting Cloud SQL Proxy: [{args}]] rrrrN)r[r]rrrrrrr rRrr rrrtrurDrrrrrrrWr_r) rrrrrrrrhrs r1StartCloudSqlProxyrWs>'(,  **8+B+BDI J$    " " * * . . . =' v||~FFwO PP$44\IDI***-44#((::N4O $$oo M* mT3E FF   ww%,,  - - & '-f\&:     sAD** E:3AE55E:c\|j|jjjk(S)zDReturns a boolean indicating if the database instance is second gen.) backendTypeDatabaseInstanceBackendTypeValueValuesEnum SECOND_GEN) sql_messagesrs r1 IsInstanceV2rs%   !>!>!Y!Y!d!d ddr3creZdZdZed dZedZedZedZedZ edZ y) _BaseInstancesrNc tjtj}|j}|j}t j jjjd}i}|||d<d}||n||d<tj|j|j|fi| fd}|S)aGets SQL instances in a given project. Modifies current state of an individual instance to 'STOPPED' if activationPolicy is 'NEVER'. Args: limit: int, The maximum number of records to yield. None if all available records should be yielded. batch_size: int, The number of items to retrieve per request. Returns: List of yielded DatabaseInstancePresentation instances. Trlimiti batch_size)projectc36KD]}t|ywN)r)resultyieldeds r1 YieldInstancesWithAModifiedStatezM_BaseInstances.GetDatabaseInstances..YieldInstancesWithAModifiedStates&*622s)r SqlClientAPI_VERSION_DEFAULT sql_clientrrrrrrr YieldFromList instancesSqlInstancesListRequest) rrclientrr project_idparamsdefault_batch_sizerrs @r1GetDatabaseInstancesz#_BaseInstances.GetDatabaseInstancess    < < =F""J&&L""''//33T3BJ F fWo , 2D <&&,,Z,@LDJLG3 , --r3c4tjdddy)NzWhen adding a new IP address to authorized networks, make sure to also include any IP addresses that have already been authorized. Otherwise, they will be overwritten and de-authorized.Tradefault cancel_on_norPromptContinuer?r3r1*PrintAndConfirmAuthorizedNetworksOverwritez9_BaseInstances.PrintAndConfirmAuthorizedNetworksOverwritesM r3c4tjdddy)NzThis request will trigger a simulated maintenance event and will not change the maintenance version on the instance. Downtime will occur on the instance.FTrrr?r3r1(PrintAndConfirmSimulatedMaintenanceEventz7_BaseInstances.PrintAndConfirmSimulatedMaintenanceEvents& r3c@|jjtS)z>Returns a boolean indicating if the database version is MySQL.)r r|_MYSQL_DATABASE_VERSION_PREFIXdatabase_versions r1IsMysqlDatabaseVersionz%_BaseInstances.IsMysqlDatabaseVersions  + +,J KKr3c@|jjtS)zAReturns a boolean indicating if the database version is Postgres.)r r|!_POSTGRES_DATABASE_VERSION_PREFIXrs r1IsPostgresDatabaseVersionz(_BaseInstances.IsPostgresDatabaseVersions  + +,M NNr3c@|jjtS)zCReturns a boolean indicating if the database version is SQL Server.)r r|"_SQLSERVER_DATABASE_VERSION_PREFIXrs r1IsSqlServerDatabaseVersionz)_BaseInstances.IsSqlServerDatabaseVersions  + +,N OOr3)NN) r;r<r=r> staticmethodrrrrrrr?r3r1rrs3%.%.NLLOOPPr3rc0eZdZdZedZedZy)InstancesV1Beta4z3Common utility functions for sql instances V1Beta4.cH|j|_|j|_yr)rrr )instance_resource instance_refs r1SetProjectAndInstanceFromRefz-InstancesV1Beta4.SetProjectAndInstanceFromRefs , 4 4)22r3c||_yr)backupConfiguration)r! backup_configs r1AddBackupConfigToSettingsz*InstancesV1Beta4.AddBackupConfigToSettingss #0H r3N)r;r<r=r>rrrr?r3r1rrs+;3311r3r)cloud_sql_proxyr)F) NFFFFFNF)r)3r> __future__rrrrrIrrxapitools.base.pyrgooglecloudsdk.api_lib.sqlrr r rWgooglecloudsdk.api_lib.utilr googlecloudsdk.corer r rrgooglecloudsdk.core.consolergooglecloudsdk.core.utilrrrTr+GetMessagesModuler#r^rrrobjectrrGrOr[rbrirmrrrrrrrr?r3r1rs2&'  '/0C,&/#*2-8  "4 ! !% 3C!($.!%0""6":(4 +>3: : BGFR$R-/59(-$)"!&$)/3-2Xv1Ghe JPVJPZ 1~ 1r3