a ù!f)ã@s8dZddlmZddlmZGdd„deƒZdd„ZdS) zûCreate / interact with a batch of updates / deletes. Batches provide the ability to execute multiple operations in a single request to the Cloud Datastore API. See https://cloud.google.com/datastore/docs/concepts/entities#Datastore_Batch_operations é)Úhelpers)Ú datastore_pb2c@s¼eZdZdZdZdZdZdZdZdd„Z d d „Z e d d „ƒZ e d d„ƒZ e dd„ƒZdd„Zdd„Zdd„Ze dd„ƒZdd„Zdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%d&„Zd'd(„ZdS))ÚBatchacAn abstraction representing a collected group of updates / deletes. Used to build up a bulk mutuation. For example, the following snippet of code will put the two ``save`` operations and the ``delete`` operation into the same mutation, and send them to the server in a single API request:: >>> from gcloud import datastore >>> client = datastore.Client() >>> batch = client.batch() >>> batch.put(entity1) >>> batch.put(entity2) >>> batch.delete(key3) >>> batch.commit() You can also use a batch as a context manager, in which case :meth:`commit` will be called automatically if its block exits without raising an exception:: >>> with batch: ... batch.put(entity1) ... batch.put(entity2) ... batch.delete(key3) By default, no updates will be sent if the block exits with an error:: >>> with batch: ... do_some_work(batch) ... raise Exception() # rolls back :type client: :class:`gcloud.datastore.client.Client` :param client: The client used to connect to datastore. NrééécCs"||_t ¡|_g|_|j|_dS©N)Ú_clientÚ_datastore_pb2Z CommitRequestÚ_commit_requestÚ_partial_key_entitiesÚ_INITIALÚ_status)ÚselfÚclient©rúW/var/www/html/python-backend/venv/lib/python3.9/site-packages/gcloud/datastore/batch.pyÚ__init__Ns zBatch.__init__cCs|jjS)z0Return the topmost batch / transaction, or None.)r Z current_batch©rrrrÚcurrentTsz Batch.currentcCs|jjS)z‘Getter for project in which the batch will run. :rtype: :class:`str` :returns: The project in which the batch will run. )r ÚprojectrrrrrXsz Batch.projectcCs|jjS)z•Getter for namespace in which the batch will run. :rtype: :class:`str` :returns: The namespace in which the batch will run. )r Ú namespacerrrrraszBatch.namespacecCs|jjS)z¾Getter for connection over which the batch will run. :rtype: :class:`gcloud.datastore.connection.Connection` :returns: The connection over which the batch will run. )r Ú connectionrrrrrjszBatch.connectioncCs|j ¡}|jS)zùAdds a new mutation for an entity with a partial key. :rtype: :class:`gcloud.datastore._generated.entity_pb2.Entity` :returns: The newly created entity protobuf that will be updated and sent with a commit. )Ú mutationsÚaddÚinsert©rZ new_mutationrrrÚ_add_partial_key_entity_pbss z Batch._add_partial_key_entity_pbcCs|j ¡}|jS)zûAdds a new mutation for an entity with a completed key. :rtype: :class:`gcloud.datastore._generated.entity_pb2.Entity` :returns: The newly created entity protobuf that will be updated and sent with a commit. )rrZupsertrrrrÚ_add_complete_key_entity_pb}s z!Batch._add_complete_key_entity_pbcCs|j ¡}|jS)zëAdds a new mutation for a key to be deleted. :rtype: :class:`gcloud.datastore._generated.entity_pb2.Key` :returns: The newly created key protobuf that will be deleted when sent with a commit. )rrÚdeleterrrrÚ_add_delete_key_pbŠs zBatch._add_delete_key_pbcCs|jjS)a.Getter for the changes accumulated by this batch. Every batch is committed with a single commit request containing all the work to be done as mutations. Inside a batch, calling :meth:`put` with an entity, or :meth:`delete` with a key, builds up the request by adding a new mutation. This getter returns the protobuf that has been built-up so far. :rtype: iterable :returns: The list of :class:`._generated.datastore_pb2.Mutation` protobufs to be sent in the commit request. )r rrrrrr”szBatch.mutationscCs\|jdurtdƒ‚|j|jjkr(tdƒ‚|jjrF| ¡}|j |¡n| ¡}t||ƒdS)a@Remember an entity's state to be saved during :meth:`commit`. .. note:: Any existing properties for the entity will be replaced by those currently set on this instance. Already-stored properties which do not correspond to keys set on this instance will be removed from the datastore. .. note:: Property values which are "text" ('unicode' in Python2, 'str' in Python3) map to 'string_value' in the datastore; values which are "bytes" ('str' in Python2, 'bytes' in Python3) map to 'blob_value'. When an entity has a partial key, calling :meth:`commit` sends it as an ``insert`` mutation and the key is completed. On return, the key for the ``entity`` passed in is updated to match the key ID assigned by the server. :type entity: :class:`gcloud.datastore.entity.Entity` :param entity: the entity to be saved. :raises: ValueError if entity has no key assigned, or if the key's ``project`` does not match ours. NzEntity must have a keyú&Key must be from same project as batch) ÚkeyÚ ValueErrorrÚ is_partialrr ÚappendrÚ_assign_entity_to_pb)rÚentityÚ entity_pbrrrÚput¤s z Batch.putcCs<|jrtdƒ‚|j|jkr"tdƒ‚| ¡}| ¡ |¡dS)aRemember a key to be deleted during :meth:`commit`. :type key: :class:`gcloud.datastore.key.Key` :param key: the key to be deleted. :raises: ValueError if key is not complete, or if the key's ``project`` does not match ours. zKey must be completer!N)r$r#rZ to_protobufr ÚCopyFrom)rr"Zkey_pbrrrrËs  z Batch.deletecCs |j|jkrtdƒ‚|j|_dS)aUBegins a batch. This method is called automatically when entering a with statement, however it can be called explicitly if you don't want to use a context manager. Overridden by :class:`gcloud.datastore.transaction.Transaction`. :raises: :class:`ValueError` if the batch has already begun. z!Batch already started previously.N)rr r#Ú _IN_PROGRESSrrrrÚbeginÝs z Batch.begincCsN|j |j|j|j¡\}}t||jƒD]"\}}|jdj}|j   |¡|_ q&dS)zFCommits the batch. This is called by :meth:`commit`. éÿÿÿÿN) rÚcommitrr Ú_idÚzipr ÚpathÚidr"Z completed_key)rÚ_Z updated_keysZ new_key_pbr'Znew_idrrrÚ_commitìs ÿÿ  z Batch._commitcCs$z| ¡W|j|_n |j|_0dS)zÀCommits the batch. This is called automatically upon exiting a with statement, however it can be called explicitly if you don't want to use a context manager. N)r4Ú _FINISHEDrrrrrr.üs z Batch.commitcCs |j|_dS)z«Rolls back the current batch. Marks the batch as aborted (can't be used again). Overridden by :class:`gcloud.datastore.transaction.Transaction`. N)Ú_ABORTEDrrrrrÚrollbackszBatch.rollbackcCs|j |¡| ¡|Sr)r Z _push_batchr,rrrrÚ __enter__s zBatch.__enter__cCs:z(|dur| ¡n| ¡W|j ¡n |j ¡0dSr)r.r7r Z _pop_batch)rÚexc_typeÚexc_valÚexc_tbrrrÚ__exit__s   zBatch.__exit__)Ú__name__Ú __module__Ú __qualname__Ú__doc__r/r r+r6r5rrÚpropertyrrrrrr rr)rr,r4r.r7r8r<rrrrrs6#       '  rcCs&t |¡}|j |j¡| |¡dS)aXCopy ``entity`` into ``entity_pb``. Helper method for ``Batch.put``. :type entity_pb: :class:`gcloud.datastore._generated.entity_pb2.Entity` :param entity_pb: The entity owned by a mutation. :type entity: :class:`gcloud.datastore.entity.Entity` :param entity: The entity being updated within the batch / transaction. N)rZentity_to_protobufr"r*)r(r'Zbare_entity_pbrrrr& s r&N) r@Zgcloud.datastorerZgcloud.datastore._generatedrr Úobjectrr&rrrrÚs