a !f6@sdZddlZddlmZddlmZddlmZe dZ e dZ iZ dZ d d Zdd d Zd dZGdddeZGdddeZdS)z:User friendly container for Google Cloud Bigtable Cluster.N)operations_pb2) instance_pb2)bigtable_instance_admin_pb2zd^projects/(?P[^/]+)/instances/(?P[^/]+)/clusters/(?P[a-z][-a-z0-9]*)$zk^operations/projects/([^/]+)/instances/([^/]+)/clusters/([a-z][-a-z0-9]*)/operations/(?P\d+)$cCs tj|jj|jtj|jddS)aCreates a protobuf request for a CreateCluster request. :type cluster: :class:`Cluster` :param cluster: The cluster to be created. :rtype: :class:`.messages_v2_pb2.CreateClusterRequest` :returns: The CreateCluster request object containing the cluster info.  serve_nodes)parent cluster_idcluster)messages_v2_pb2ZCreateClusterRequest _instancenamer data_v2_pb2Clusterrr rX/var/www/html/python-backend/venv/lib/python3.9/site-packages/gcloud/bigtable/cluster.py_prepare_create_request+s rcCs:|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_classrrr_parse_pb_any_to_native=s  rcCs2t|j}|dur td|jt|d}|S)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 cluster request. :rtype: tuple :returns: integer ID of the operation (``operation_id``). :raises: :class:`ValueError ` if the operation name doesn't match the :data:`_OPERATION_NAME_RE` regex. NzKOperation name was not in the expected format after a cluster modification. operation_id)_OPERATION_NAME_REmatchr rintgroup) operation_pbrrrrr_process_operationSs rc@s2eZdZdZd ddZddZddZd d ZdS) OperationaRepresentation of a Google API Long-Running Operation. In particular, these will be the result of operations on clusters 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 cluster: :class:`Cluster` :param cluster: The cluster that created the operation. NcCs||_||_||_d|_dSNF)op_typeop_id_cluster _complete)selfr"r#r rrr__init__zszOperation.__init__cCs@t||jsdS|j|jko>|j|jko>|j|jko>|j|jkSr!) isinstance __class__r"r#r$r%r&otherrrr__eq__s     zOperation.__eq__cCs || SNr,r*rrr__ne__szOperation.__ne__cCsd|jrtdd|jjd|jf}tj|d}|jjj}|j ||j }|j r\d|_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/z/operations/%dr TFN) r%rr$r r#rZGetOperationRequestr _clientZ_operations_stubZ GetOperationtimeout_secondsdone)r&Zoperation_name request_pbclientrrrrfinisheds    zOperation.finished)N)__name__ __module__ __qualname____doc__r'r,r/r6rrrrr is  r c@steZdZdZefddZddZeddZdd Z e d d Z d d Z ddZ ddZddZddZddZdS)raRepresentation of a Google Cloud Bigtable Cluster. We can use a :class:`Cluster` 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 cluster_id: str :param cluster_id: The ID of the cluster. :type instance: :class:`.instance.Instance` :param instance: The instance where the cluster resides. :type serve_nodes: int :param serve_nodes: (Optional) The number of nodes in the cluster. Defaults to :data:`DEFAULT_SERVE_NODES`. cCs||_||_||_d|_dSr-)r r rlocation)r&r instancerrrrr'szCluster.__init__cCs"|jstd|j|_|j|_dS)zpRefresh self from the server-provided protobuf. Helper for :meth:`from_pb` and :meth:`reload`. z-Cluster protobuf does not contain serve_nodesN)rrr;)r& cluster_pbrrr_update_from_pbszCluster._update_from_pbcCspt|j}|dur td|j|d|jjkr:td|d|jkrRtd||d|}|||S)aCreates a cluster instance from a protobuf. :type cluster_pb: :class:`instance_pb2.Cluster` :param cluster_pb: A cluster protobuf object. :type instance: :class:`.instance.Instance>` :param instance: The instance that owns the cluster. :rtype: :class:`Cluster` :returns: The cluster parsed from the protobuf response. :raises: :class:`ValueError ` if the cluster name does not match ``projects/{project}/instances/{instance}/clusters/{cluster_id}`` or if the parsed project ID does not match the project ID on the client. Nz5Cluster protobuf name was not in the expected format.projectzAProject ID on cluster does not match the project ID on the clientr<zCInstance ID on cluster does not match the instance ID on the clientr ) _CLUSTER_NAME_RErr rrr1r?Z instance_idr>)clsr=r<rresultrrrfrom_pbs  zCluster.from_pbcCs|j}|j|j||jdS)zMake a copy of this cluster. Copies the local data stored as simple types and copies the client attached to this instance. :rtype: :class:`.Cluster` :returns: A copy of the current cluster. r)r copyr)r r)r&Z new_instancerrrrDs  z Cluster.copycCs|jjd|jS)aoCluster name used in requests. .. note:: This property will not change if ``_instance`` and ``cluster_id`` do not, but the return value is not cached. The cluster name is of the form ``"projects/{project}/instances/{instance}/clusters/{cluster_id}"`` :rtype: str :returns: The cluster name. z /clusters/)r r r )r&rrrr sz Cluster.namecCs(t||jsdS|j|jko&|j|jkSr!)r(r)r r r*rrrr,s    zCluster.__eq__cCs || Sr-r.r*rrrr/szCluster.__ne__cCs4tj|jd}|jjj||jjj}||dS)z%Reload the metadata for this cluster.r0N) r ZGetClusterRequestr r r1_instance_stubZ GetClusterr2r>)r&r4r=rrrreload s   zCluster.reloadcCs6t|}|jjj||jjj}t|}td||dS)a:Create this cluster. .. note:: Uses the ``project``, ``instance`` and ``cluster_id`` on the current :class:`Cluster` in addition to the ``serve_nodes``. To change them before creating, reset the values via .. code:: python cluster.serve_nodes = 8 cluster.cluster_id = 'i-changed-my-mind' before calling :meth:`create`. :rtype: :class:`Operation` :returns: The long-running operation corresponding to the create operation. creater)rr r1rEZ CreateClusterr2rr r&r4rr#rrrrG+s   zCluster.createcCs@tj|j|jd}|jjj||jjj}t |}t d||dS)aUpdate this cluster. .. note:: Updates the ``serve_nodes``. If you'd like to change them before updating, reset the values via .. code:: python cluster.serve_nodes = 8 before calling :meth:`update`. :rtype: :class:`Operation` :returns: The long-running operation corresponding to the update operation. )r rupdater) rrr rr r1rEZ UpdateClusterr2rr rHrrrrIGs  zCluster.updatecCs*tj|jd}|jjj||jjjdS)a:Delete this cluster. Marks a cluster and all of its tables for permanent deletion in 7 days. Immediately upon completion of the request: * Billing will cease for all of the cluster's reserved resources. * The cluster's ``delete_time`` field will be set 7 days in the future. Soon afterward: * All tables within the cluster will become unavailable. Prior to the cluster's ``delete_time``: * The cluster can be recovered with a call to ``UndeleteCluster``. * All other attempts to modify or delete the cluster will be rejected. At the cluster's ``delete_time``: * The cluster and **all of its tables** will immediately and irrevocably disappear from the API, and their data will be permanently deleted. r0N)r ZDeleteClusterRequestr r r1rEZ DeleteClusterr2)r&r4rrrdeleteds  zCluster.deleteN)r7r8r9r:DEFAULT_SERVE_NODESr'r> classmethodrCrDpropertyr r,r/rFrGrIrJrrrrrs   !    r)N)r:reZgoogle.longrunningrZgcloud.bigtable._generatedrrrr compiler@rrrKrrrobjectr rrrrrs      <