a !f H@sdZddlZddlmZddlmZddlmZddlm Z ddlm Z ddl mZdd l mZdd lmZd Zed Zed ZdZedZedZee jiZddZdddZddZGdddeZGdddeZ dS)z;User friendly container for Google Cloud Bigtable Instance.N)operations_pb2)_pb_timestamp_to_datetime) instance_pb2)bigtable_instance_admin_pb2)bigtable_table_admin_pb2Cluster)DEFAULT_SERVE_NODESTablezsee-existing-clusterzH^projects/(?P[^/]+)/instances/(?P[a-z][-a-z0-9]*)$z^operations/projects/([^/]+)/instances/([a-z][-a-z0-9]*)/locations/(?P[a-z][-a-z0-9]*)/operations/(?P\d+)$z$type.googleapis.com/google.bigtable.z admin.v2.CreateInstanceMetadatacCsbd|jj}tj||jtj|jdd}|j|j}|j d|j|_ |d|j |_ |j |_ |S)a Creates a protobuf request for a CreateInstance request. :type instance: :class:`Instance` :param instance: The instance to be created. :rtype: :class:`.messages_v2_pb2.CreateInstanceRequest` :returns: The CreateInstance request object containing the instance info. z projects/ display_name)parent instance_idinstancez /clusters/z /locations/)_clientprojectmessages_v2_pb2ZCreateInstanceRequestr data_v2_pb2Instancerclustersname_cluster_location_idlocation_cluster_serve_nodes serve_nodes)r parent_namemessageclusterr Y/var/www/html/python-backend/venv/lib/python3.9/site-packages/gcloud/bigtable/instance.py_prepare_create_request1s   r"cCs:|dur$||jkr$td||jft|j}||jS)a Convert a serialized "google.protobuf.Any" value to actual type. :type any_val: :class:`google.protobuf.any_pb2.Any` :param any_val: A serialized protobuf value container. :type expected_type: str :param expected_type: (Optional) The type URL we expect ``any_val`` to have. :rtype: object :returns: The de-serialized object. :raises: :class:`ValueError ` if the ``expected_type`` does not match the ``type_url`` on the input. NzExpected type: %s, Received: %s)Ztype_url ValueError _TYPE_URL_MAPZ FromStringvalue)Zany_val expected_typeZcontainer_classr r r!_parse_pb_any_to_nativeJs  r'cCsVt|j}|dur td|j|d}t|d}t|j}t|j }|||fS)aProcesses a create protobuf response. :type operation_pb: :class:`google.longrunning.operations_pb2.Operation` :param operation_pb: The long-running operation response from a Create/Update/Undelete instance request. :rtype: (int, str, datetime) :returns: (operation_id, location_id, operation_begin). :raises: :class:`ValueError ` if the operation name doesn't match the :data:`_OPERATION_NAME_RE` regex. NzFOperation name was not in the expected format after instance creation. location_id operation_id) _OPERATION_NAME_REmatchrr#groupintr'metadatarZ request_time) operation_pbr+r(r)Zrequest_metadataZoperation_beginr r r!_process_operation`s   r0c@s2eZdZdZd ddZddZddZd d ZdS) OperationaRepresentation of a Google API Long-Running Operation. In particular, these will be the result of operations on instances using the Cloud Bigtable API. :type op_type: str :param op_type: The type of operation being performed. Expect ``create``, ``update`` or ``undelete``. :type op_id: int :param op_id: The ID of the operation. :type begin: :class:`datetime.datetime` :param begin: The time when the operation was started. :type location_id: str :param location_id: ID of the location in which the operation is running :type instance: :class:`Instance` :param instance: The instance that created the operation. NcCs(||_||_||_||_||_d|_dSNF)op_typeop_idbeginr( _instance _complete)selfr3r4r5r(rr r r!__init__s zOperation.__init__cCsXt||jsdS|j|jkoV|j|jkoV|j|jkoV|j|jkoV|j|jkoV|j|jkSr2) isinstance __class__r3r4r5r(r6r7r8otherr r r!__eq__s       zOperation.__eq__cCs || SNr>r<r r r!__ne__szOperation.__ne__cCs`|jrtdd|jj|j|jf}tj|d}|jjj ||jjj }|j rXd|_dSdSdS)aCheck if the operation has finished. :rtype: bool :returns: A boolean indicating if the current operation has completed. :raises: :class:`ValueError ` if the operation has already completed. zThe operation has completed.z(operations/%s/locations/%s/operations/%drTFN) r7r#r6rr(r4rZGetOperationRequestrZ_operations_stubZ GetOperationtimeout_secondsdone)r8Zoperation_name request_pbr/r r r!finisheds   zOperation.finished)N)__name__ __module__ __qualname____doc__r9r>rArFr r r r!r1{s   r1c@seZdZdZedefddZddZeddZ d d Z e d d Z d dZ ddZddZddZddZddZd"ddZddZddZd d!ZdS)#raIRepresentation of a Google Cloud Bigtable Instance. We can use a :class:`Instance` to: * :meth:`reload` itself * :meth:`create` itself * :meth:`update` itself * :meth:`delete` itself * :meth:`undelete` itself .. note:: For now, we leave out the ``default_storage_type`` (an enum) which if not sent will end up as :data:`.data_v2_pb2.STORAGE_SSD`. :type instance_id: str :param instance_id: The ID of the instance. :type client: :class:`Client ` :param client: The client that owns the instance. Provides authorization and a project ID. :type location_id: str :param location_id: ID of the location in which the instance will be created. Required for instances which do not yet exist. :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. NcCs&||_|p ||_||_||_||_dSr?)rrrrr)r8rclientr(rrr r r!r9s  zInstance.__init__cCs|jstd|j|_dS)zpRefresh self from the server-provided protobuf. Helper for :meth:`from_pb` and :meth:`reload`. z/Instance protobuf does not contain display_nameN)rr#)r8 instance_pbr r r!_update_from_pbszInstance._update_from_pbcCs\t|j}|dur td|j|d|jkr8td|d}|||t}|||S)aCreates a instance instance from a protobuf. :type instance_pb: :class:`instance_pb2.Instance` :param instance_pb: A instance protobuf object. :type client: :class:`Client ` :param client: The client that owns the instance. :rtype: :class:`Instance` :returns: The instance parsed from the protobuf response. :raises: :class:`ValueError ` if the instance name does not match ``projects/{project}/instances/{instance_id}`` or if the parsed project ID does not match the project ID on the client. Nz6Instance protobuf name was not in the expected format.rzBProject ID on instance does not match the project ID on the clientr)_INSTANCE_NAME_REr+rr#r,r_EXISTING_INSTANCE_LOCATION_IDrM)clsrLrKr+rresultr r r!from_pbs    zInstance.from_pbcCs"|j}|j|j||j|jdS)zMake a copy of this instance. Copies the local data stored as simple types and copies the client attached to this instance. :rtype: :class:`.Instance` :returns: A copy of the current instance. r )rcopyr;rrr)r8Z new_clientr r r!rSs  z Instance.copycCs|jjd|jS)aPInstance name used in requests. .. note:: This property will not change if ``instance_id`` does not, but the return value is not cached. The instance name is of the form ``"projects/{project}/instances/{instance_id}"`` :rtype: str :returns: The instance name. z /instances/)r project_namerr8r r r!r(sz Instance.namecCs(t||jsdS|j|jko&|j|jkSr2)r:r;rrr<r r r!r>9s    zInstance.__eq__cCs || Sr?r@r<r r r!rAEszInstance.__ne__cCs0tj|jd}|jj||jj}||dS)z&Reload the metadata for this instance.rBN)rZGetInstanceRequestrr_instance_stubZ GetInstancerCrM)r8rErLr r r!reloadHs zInstance.reloadcCs<t|}|jj||jj}t|\}}}td||||dS)aECreate this instance. .. note:: Uses the ``project`` and ``instance_id`` on the current :class:`Instance` in addition to the ``display_name``. To change them before creating, reset the values via .. code:: python instance.display_name = 'New display name' instance.instance_id = 'i-changed-my-mind' before calling :meth:`create`. :rtype: :class:`Operation` :returns: The long-running operation corresponding to the create operation. create)r)r"rrVZCreateInstancerCr0r1)r8rEr/r4Zloc_idZop_beginr r r!rXSs zInstance.createcCs*tj|j|jd}|jj||jjdS)a)Update this instance. .. note:: Updates the ``display_name``. To change that value before updating, reset its values via .. code:: python instance.display_name = 'New display name' before calling :meth:`update`. )rrN)rrrrrrVZUpdateInstancerCr8rEr r r!updateoszInstance.updatecCs&tj|jd}|jj||jjdS)aWDelete this instance. Marks a instance and all of its tables for permanent deletion in 7 days. Immediately upon completion of the request: * Billing will cease for all of the instance's reserved resources. * The instance's ``delete_time`` field will be set 7 days in the future. Soon afterward: * All tables within the instance will become unavailable. Prior to the instance's ``delete_time``: * The instance can be recovered with a call to ``UndeleteInstance``. * All other attempts to modify or delete the instance will be rejected. At the instance's ``delete_time``: * The instance and **all of its tables** will immediately and irrevocably disappear from the API, and their data will be permanently deleted. rBN)rZDeleteInstanceRequestrrrVZDeleteInstancerCrYr r r!deleteszInstance.deletecCst|||dS)aFactory to create a cluster associated with this client. :type cluster_id: str :param cluster_id: The ID of the cluster. :type serve_nodes: int :param serve_nodes: (Optional) The number of nodes in the cluster. Defaults to 3. :rtype: :class:`.Cluster` :returns: The cluster owned by this client. )rr)r8Z cluster_idrr r r!rs zInstance.clustercsNtjjd}jj|jj}dd|jD}fdd|jD}||fS)aLists clusters in this instance. :rtype: tuple :returns: A pair of results, the first is a list of :class:`.Cluster` s returned and the second is a list of strings (the failed locations in the request). rcSsg|]}|qSr r ).0rr r r! sz*Instance.list_clusters..csg|]}t|qSr )rrR)r^Z cluster_pbrUr r!r_s) rZListClustersRequestrrrVZ ListClustersrCfailed_locationsr)r8rEZlist_clusters_responser`rr rUr! list_clusterss zInstance.list_clusterscCs t||S)zFactory to create a table associated with this instance. :type table_id: str :param table_id: The ID of the table. :rtype: :class:`Table ` :returns: The table owned by this instance. r )r8table_idr r r!tables zInstance.tablecCs~tj|jd}|jj||jj}g}|jD]L}|jd}|j|sVt d|jf|jt |d}| | |q,|S)aGList the tables in this instance. :rtype: list of :class:`Table ` :returns: The list of tables owned by the instance. :raises: :class:`ValueError ` if one of the returned tables has a name that is not of the expected format. r]z/tables/z$Table name %s not of expected formatN) table_messages_v2_pb2ZListTablesRequestrrZ _table_stubZ ListTablesrCZtables startswithr#lenappendrc)r8rEZ table_list_pbrQZtable_pbZ table_prefixrbr r r! list_tabless   zInstance.list_tables)r\)rGrHrIrJrOr r9rM classmethodrRrSpropertyrr>rArWrXrZr[rrarcrhr r r r!rs*'        r)N)!rJreZgoogle.longrunningrZgcloud._helpersrZgcloud.bigtable._generatedrrrrrrdZgcloud.bigtable.clusterrr Zgcloud.bigtable.tabler rOcompilerNr*Z_TYPE_URL_BASEZ_ADMIN_TYPE_URL_BASEZ_INSTANCE_CREATE_METADATAr r$r"r'r0objectr1rr r r r!s,           F