a !fF@sdZddlmZddlmZddlmZddlmZ ddlm Z ddlm Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZe jZdZdZejZdZdZe j Z!dZ"dZ#e j$Z%eZ&eZ'dZ(dZ)dZ*dZ+d,edj-Z.GdddeeZ/Gddde0Z1ddZ2dS)aParent client for calling the Google Cloud Bigtable API. This is the base from which all interactions with the API occur. In the hierarchy of API concepts * a :class:`Client` owns an :class:`.Instance` * a :class:`.Instance` owns a :class:`Table ` * a :class:`Table ` owns a :class:`ColumnFamily <.column_family.ColumnFamily>` * a :class:`Table ` owns a :class:`Row <.row.Row>` (and all the cells in the row) )get_distribution)implementations)bigtable_instance_admin_pb2)bigtable_table_admin_pb2) bigtable_pb2)operations_grpc_pb2)DEFAULT_SERVE_NODESInstance)_EXISTING_INSTANCE_LOCATION_ID)_ClientFactoryMixin)_ClientProjectMixin)get_credentialszbigtableadmin.googleapis.comizbigtable.googleapis.comz.https://www.googleapis.com/auth/bigtable.adminz-https://www.googleapis.com/auth/bigtable.dataz6https://www.googleapis.com/auth/bigtable.data.readonly zgcloud-python/{0}Zgcloudc@seZdZdZddddeefddZddZedd Z ed d Z ed d Z eddZ eddZ eddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Zedefd&d'Zd(d)ZdS)*ClientaClient for interacting with Google Cloud Bigtable API. .. note:: Since the Cloud Bigtable API requires the gRPC transport, no ``http`` argument is accepted by this class. :type project: :class:`str` or :func:`unicode ` :param project: (Optional) The ID of the project which owns the instances, tables and data. If not provided, will attempt to determine from the environment. :type credentials: :class:`OAuth2Credentials ` or :data:`NoneType ` :param credentials: (Optional) The OAuth2 Credentials to use for this client. If not provided, defaults to the Google Application Default Credentials. :type read_only: bool :param read_only: (Optional) Boolean indicating if the data scope should be for reading only (or for writing as well). Defaults to :data:`False`. :type admin: bool :param admin: (Optional) Boolean indicating if the client will be used to interact with the Instance Admin or Table Admin APIs. This requires the :const:`ADMIN_SCOPE`. Defaults to :data:`False`. :type user_agent: str :param user_agent: (Optional) The user agent to be used with API request. Defaults to :const:`DEFAULT_USER_AGENT`. :type timeout_seconds: int :param timeout_seconds: Number of seconds for request time-out. If not passed, defaults to :const:`DEFAULT_TIMEOUT_SECONDS`. :raises: :class:`ValueError ` if both ``read_only`` and ``admin`` are :data:`True` NFcCstj||d|durt}|r,|r,tdg}|r@|tn |t|rX|tt||_ z| |}Wnt yYn0||_ ||_ ||_d|_d|_d|_d|_dS)Nprojectz=A read-only client cannot also performadministrative actions.)r __init__r ValueErrorappendREAD_ONLY_SCOPE DATA_SCOPE ADMIN_SCOPEbool_admin create_scopedAttributeError _credentials user_agenttimeout_seconds_data_stub_internal_instance_stub_internal_operations_stub_internal_table_stub_internal)selfr credentialsZ read_onlyZadminrrscopesr'W/var/www/html/python-backend/venv/lib/python3.9/site-packages/gcloud/bigtable/client.pyrs.     zClient.__init__cCs4|j}||j}||j|t|jv|j|j|jS)a Make a copy of this client. Copies the local data stored as simple types but does not copy the current state of any open connections with the Cloud Bigtable API. :rtype: :class:`.Client` :returns: A copy of the current client. ) rrr& __class__rrrrr)r$r%Z copied_credsr'r'r(copys  z Client.copycCs|jS)zGetter for client's credentials. :rtype: :class:`OAuth2Credentials ` :returns: The credentials stored on the client. )rr$r'r'r(r%szClient.credentialscCs d|jS)aProject name to be used with Instance Admin API. .. note:: This property will not change if ``project`` does not, but the return value is not cached. The project name is of the form ``"projects/{project}"`` :rtype: str :returns: The project name to be used with the Cloud Bigtable Admin API RPC service. z projects/rr+r'r'r( project_nameszClient.project_namecCs|jdurtd|jS)aGetter for the gRPC stub used for the Data API. :rtype: :class:`grpc.beta._stub._AutoIntermediary` :returns: A gRPC stub object. :raises: :class:`ValueError ` if the current client has not been :meth:`start`-ed. NClient has not been started.)r rr+r'r'r( _data_stubs zClient._data_stubcCs&|jstd|jdur td|jS)aXGetter for the gRPC stub used for the Instance Admin API. :rtype: :class:`grpc.beta._stub._AutoIntermediary` :returns: A gRPC stub object. :raises: :class:`ValueError ` if the current client is not an admin client or if it has not been :meth:`start`-ed. Client is not an admin client.Nr-)rrr!r+r'r'r(_instance_stubs  zClient._instance_stubcCs&|jstd|jdur td|jS)aTGetter for the gRPC stub used for the Operations API. :rtype: :class:`grpc.beta._stub._AutoIntermediary` :returns: A gRPC stub object. :raises: :class:`ValueError ` if the current client is not an admin client or if it has not been :meth:`start`-ed. r/Nr-)rrr"r+r'r'r(_operations_stubs  zClient._operations_stubcCs&|jstd|jdur td|jS)aUGetter for the gRPC stub used for the Table Admin API. :rtype: :class:`grpc.beta._stub._AutoIntermediary` :returns: A gRPC stub object. :raises: :class:`ValueError ` if the current client is not an admin client or if it has not been :meth:`start`-ed. r/Nr-)rrr#r+r'r'r( _table_stub s  zClient._table_stubcCst|tttS)zCreates gRPC stub to make requests to the Data API. :rtype: :class:`grpc.beta._stub._AutoIntermediary` :returns: A gRPC stub object. ) _make_stubDATA_STUB_FACTORY_V2DATA_API_HOST_V2DATA_API_PORT_V2r+r'r'r(_make_data_stubszClient._make_data_stubcCst|tttS)zCreates gRPC stub to make requests to the Instance Admin API. :rtype: :class:`grpc.beta._stub._AutoIntermediary` :returns: A gRPC stub object. )r3INSTANCE_STUB_FACTORY_V2INSTANCE_ADMIN_HOST_V2INSTANCE_ADMIN_PORT_V2r+r'r'r(_make_instance_stub"szClient._make_instance_stubcCst|tttS)aCreates gRPC stub to make requests to the Operations API. These are for long-running operations of the Instance Admin API, hence the host and port matching. :rtype: :class:`grpc.beta._stub._AutoIntermediary` :returns: A gRPC stub object. )r3OPERATIONS_STUB_FACTORY_V2OPERATIONS_API_HOST_V2OPERATIONS_API_PORT_V2r+r'r'r(_make_operations_stub+s zClient._make_operations_stubcCst|tttS)zCreates gRPC stub to make requests to the Table Admin API. :rtype: :class:`grpc.beta._stub._AutoIntermediary` :returns: A gRPC stub object. )r3TABLE_STUB_FACTORY_V2TABLE_ADMIN_HOST_V2TABLE_ADMIN_PORT_V2r+r'r'r(_make_table_stub7szClient._make_table_stubcCs |jduS)zCheck if the client has been started. :rtype: bool :returns: Boolean indicating if the client has been started. N)r r+r'r'r( is_started@szClient.is_startedcCsf|r dS||_|j|jrb||_||_| |_ |j|j|j dS)zPrepare the client to make requests. Activates gRPC contexts for making requests to the Bigtable Service(s). N) rDr7r __enter__rr;r!r?r"rCr#r+r'r'r(startHs       z Client.startcCs ||S)z'Starts the client as a context manager.)rFr+r'r'r(rEaszClient.__enter__cCsn|s dS|jddd|jrR|jddd|jddd|jdddd|_d|_d|_d|_dS)z!Closes all the open gRPC clients.N)rDr __exit__rr!r"r#r+r'r'r(stopfsz Client.stopcCs |dS)z&Stops the client as a context manager.N)rH)r$exc_typeexc_valZexc_tr'r'r(rGxszClient.__exit__cCst|||||dS)aFactory to create a instance associated with this client. :type instance_id: str :param instance_id: The ID of the instance. :type location: string :param location: location name, in form ``projects//locations/``; used to set up the instance's cluster. :type display_name: str :param display_name: (Optional) The display name for the instance in the Cloud Console UI. (Must be between 4 and 30 characters.) If this value is not set in the constructor, will fall back to the instance ID. :type serve_nodes: int :param serve_nodes: (Optional) The number of nodes in the instance's cluster; used to set up the instance's cluster. :rtype: :class:`.Instance` :returns: an instance owned by this client. ) display_name serve_nodesr )r$Z instance_idlocationrKrLr'r'r(instance|szClient.instancecs<tjjd}j|j}fdd|jD}||jfS)aList instances owned by the project. :rtype: tuple :returns: A pair of results, the first is a list of :class:`.Instance` objects returned and the second is a list of strings (the failed locations in the request). )parentcsg|]}t|qSr')r Zfrom_pb).0Z instance_pbr+r'r( sz)Client.list_instances..)instance_admin_v2_pb2ZListInstancesRequestr,r0Z ListInstancesr instancesZfailed_locations)r$Z request_pbresponserSr'r+r(list_instancess zClient.list_instances)__name__ __module__ __qualname____doc__DEFAULT_USER_AGENTDEFAULT_TIMEOUT_SECONDSrr*propertyr%r,r.r0r1r2r7r;r?rCrDrFrErHrGr rrNrUr'r'r'r(r]s>* #           rc@s eZdZdZddZddZdS)_MetadataPluginzCallable class to transform metadata for gRPC requests. :type client: :class:`.client.Client` :param client: The client that owns the instance. Provides authorization and user agent. cCs|j|_|j|_dS)N)r%rr _user_agent)r$clientr'r'r(rsz_MetadataPlugin.__init__cCs0|jj}dd|fd|jfg}||ddS)z.Adds authorization header to request metadata. AuthorizationzBearer z User-agentN)rZget_access_token access_tokenr^)r$Zunused_contextcallbackraheadersr'r'r(__call__s   z_MetadataPlugin.__call__N)rVrWrXrYrrdr'r'r'r(r]sr]c CsFtddd}t|}tj|dd}t||}t|||}||S)aMakes a stub for an RPC service. Uses / depends on the beta implementation of gRPC. :type client: :class:`.client.Client` :param client: The client that owns the instance. Provides authorization and user agent. :type stub_factory: callable :param stub_factory: A factory which will create a gRPC stub for a given service. :type host: str :param host: The host for the service. :type port: int :param port: The port for the service. :rtype: :class:`grpc.beta._stub._AutoIntermediary` :returns: The stub object used to make gRPC requests to a given API. NZ google_creds)name)rZssl_channel_credentialsr]Zmetadata_call_credentialsZcomposite_channel_credentialsZsecure_channel) r_Z stub_factoryhostportZtransport_credsZcustom_metadata_pluginZ auth_credsZ channel_credsZchannelr'r'r(r3sr3N)3rY pkg_resourcesrZ grpc.betarZgcloud.bigtable._generatedrrRrZtable_admin_v2_pb2rZ data_v2_pb2rZoperations_grpc_v2_pb2Zgcloud.bigtable.clusterrZgcloud.bigtable.instancer r Z gcloud.clientr r Zgcloud.credentialsrZ#beta_create_BigtableTableAdmin_stubr@rArBZ&beta_create_BigtableInstanceAdmin_stubr8r9r:Zbeta_create_Bigtable_stubr4r5r6Zbeta_create_Operations_stubr<r=r>rrrr[formatversionrZrobjectr]r3r'r'r'r(sL            P