o
    ×èFh9  ã                   @   sè   d Z ddlmZ ddlmZ ddlmZmZmZm	Z	 ddl
mZ ddlmZ ddlmZ ddlmZ dd	lmZ dd
lmZ ddlmZ ddlmZ ddlmZmZmZm Z m!Z!m"Z"m#Z# ddl$m%Z% e#rjddl&m'Z' G dd„ deƒZ(dS )a¡  Client for interacting with the Google Cloud Firestore API.

This is the base from which all interactions with the API occur.

In the hierarchy of API concepts

* a :class:`~google.cloud.firestore_v1.client.Client` owns a
  :class:`~google.cloud.firestore_v1.collection.CollectionReference`
* a :class:`~google.cloud.firestore_v1.client.Client` owns a
  :class:`~google.cloud.firestore_v1.document.DocumentReference`
é    )Úgapic_v1)Úretry)Ú
BaseClientÚ_CLIENT_INFOÚ_parse_batch_getÚ_path_helper)ÚCollectionGroup©Ú
WriteBatch)ÚCollectionReference)ÚDocumentReference)Ú	FieldPath©ÚTransaction)Úclient)Úgrpc)ÚAnyÚ	GeneratorÚIterableÚListÚOptionalÚUnionÚTYPE_CHECKING)ÚDocumentSnapshot)Ú
BulkWriterc                       sn  e Zd ZdZdddedf	d-‡ fdd„Zedd„ ƒZedd	„ ƒZd
e	de
fdd„Zde	defdd„Zde	defdd„Zddejjdfdedee	 dedejdedeeedf fdd„Zejjdfdejdedeeedf fdd„Zdddœdee
ef ded  d!ee  de fd"d#„Z!dd$d%œdee
ef dd d!ee  d&ee  de f
d'd(„Z"de#fd)d*„Z$defd+d,„Z%‡  Z&S ).ÚClienta  Client for interacting with Google Cloud Firestore API.

    .. note::

        Since the Cloud Firestore API requires the gRPC transport, no
        ``_http`` argument is accepted by this class.

    Args:
        project (Optional[str]): The project which the client acts on behalf
            of. If not passed, falls back to the default inferred
            from the environment.
        credentials (Optional[~google.auth.credentials.Credentials]): The
            OAuth2 Credentials to use for this client. If not passed, falls
            back to the default inferred from the environment.
        database (Optional[str]): The database name that the client targets.
            If not passed, falls back to :attr:`DEFAULT_DATABASE`.
        client_info (Optional[google.api_core.gapic_v1.client_info.ClientInfo]):
            The client info used to send a user-agent string along with API
            requests. If ``None``, then default info will be used. Generally,
            you only need to set this if you're developing your own library
            or partner tool.
        client_options (Union[dict, google.api_core.client_options.ClientOptions]):
            Client options used to set user options on the client. API Endpoint
            should be set through client_options.
    NÚreturnc                    s   t t| ƒj|||||d� d S )N)ÚprojectÚcredentialsÚdatabaseÚclient_infoÚclient_options)Úsuperr   Ú__init__)Úselfr   r   r   r    r!   ©Ú	__class__© úY/var/www/html/loop/nvenv/lib/python3.10/site-packages/google/cloud/firestore_v1/client.pyr#   S   s   

ûzClient.__init__c                 C   s   |   tjtjt¡S )zãLazy-loading getter GAPIC Firestore API.
        Returns:
            :class:`~google.cloud.gapic.firestore.v1`.firestore_client.FirestoreClient:
            The GAPIC client with the credentials of the current client.
        )Ú_firestore_api_helperÚfirestore_grpc_transportÚFirestoreGrpcTransportÚfirestore_clientÚFirestoreClient©r$   r'   r'   r(   Ú_firestore_apic   s
   ýzClient._firestore_apic                 C   s   |   tj¡S )z‘Return the target (where the API is).
        Eg. "firestore.googleapis.com"

        Returns:
            str: The location of the API.
        )Ú_target_helperr,   r-   r.   r'   r'   r(   Ú_targetp   s   zClient._targetÚcollection_pathc                 G   s   t t|ƒd| iŽS )a  Get a reference to a collection.

        For a top-level collection:

        .. code-block:: python

            >>> client.collection('top')

        For a sub-collection:

        .. code-block:: python

            >>> client.collection('mydocs/doc/subcol')
            >>> # is the same as
            >>> client.collection('mydocs', 'doc', 'subcol')

        Sub-collections can be nested deeper in a similar fashion.

        Args:
            collection_path: Can either be

                * A single ``/``-delimited path to a collection
                * A tuple of collection path segments

        Returns:
            :class:`~google.cloud.firestore_v1.collection.CollectionReference`:
            A reference to a collection in the Firestore database.
        r   )r   r   )r$   r2   r'   r'   r(   Ú
collectionz   s   zClient.collectionÚcollection_idc                 C   s   t |  |¡ƒS )a†  
        Creates and returns a new Query that includes all documents in the
        database that are contained in a collection or subcollection with the
        given collection_id.

        .. code-block:: python

            >>> query = client.collection_group('mygroup')

        Args:
            collection_id (str) Identifies the collections to query over.

                Every collection or subcollection with this ID as the last segment of its
                path will be included. Cannot contain a slash.

        Returns:
            :class:`~google.cloud.firestore_v1.query.CollectionGroup`:
            The created Query.
        )r   Ú_get_collection_reference)r$   r4   r'   r'   r(   Úcollection_group™   s   zClient.collection_groupÚdocument_pathc                 G   s   t | j|Ž d| iŽS )a}  Get a reference to a document in a collection.

        For a top-level document:

        .. code-block:: python

            >>> client.document('collek/shun')
            >>> # is the same as
            >>> client.document('collek', 'shun')

        For a document in a sub-collection:

        .. code-block:: python

            >>> client.document('mydocs/doc/subcol/child')
            >>> # is the same as
            >>> client.document('mydocs', 'doc', 'subcol', 'child')

        Documents in sub-collections can be nested deeper in a similar fashion.

        Args:
            document_path): Can either be

                * A single ``/``-delimited path to a document
                * A tuple of document path segments

        Returns:
            :class:`~google.cloud.firestore_v1.document.DocumentReference`:
            A reference to a document in a collection.
        r   )r   Ú_document_path_helper)r$   r7   r'   r'   r(   Údocument¯   s
   ÿÿzClient.documentÚ
referencesÚfield_pathsÚtransactionr   Útimeoutc                 c   sP   � |   |||||¡\}}}| jjd|| jdœ|¤Ž}	|	D ]	}
t|
|| ƒV  qdS )aš  Retrieve a batch of documents.

        .. note::

           Documents returned by this method are not guaranteed to be
           returned in the same order that they are given in ``references``.

        .. note::

           If multiple ``references`` refer to the same document, the server
           will only return one result.

        See :meth:`~google.cloud.firestore_v1.client.Client.field_path` for
        more information on **field paths**.

        If a ``transaction`` is used and it already has write operations
        added, this method cannot be used (i.e. read-after-write is not
        allowed).

        Args:
            references (List[.DocumentReference, ...]): Iterable of document
                references to be retrieved.
            field_paths (Optional[Iterable[str, ...]]): An iterable of field
                paths (``.``-delimited list of field names) to use as a
                projection of document fields in the returned results. If
                no value is provided, all fields will be returned.
            transaction (Optional[:class:`~google.cloud.firestore_v1.transaction.Transaction`]):
                An existing transaction that these ``references`` will be
                retrieved in.
            retry (google.api_core.retry.Retry): Designation of what errors, if any,
                should be retried.  Defaults to a system-specified policy.
            timeout (float): The timeout for this request.  Defaults to a
                system-specified value.

        Yields:
            .DocumentSnapshot: The next document snapshot that fulfills the
            query, or :data:`None` if the document does not exist.
        ©ÚrequestÚmetadataNr'   )Ú_prep_get_allr/   Úbatch_get_documentsÚ_rpc_metadatar   )r$   r:   r;   r<   r   r=   r?   Úreference_mapÚkwargsÚresponse_iteratorÚget_doc_responser'   r'   r(   Úget_allÒ   s   €.

ÿþýÿzClient.get_allc                 c   sF   � |   ||¡\}}| jjd|| jdœ|¤Ž}|D ]}|  |¡V  qdS )a  List top-level collections of the client's database.

        Args:
            retry (google.api_core.retry.Retry): Designation of what errors, if any,
                should be retried.  Defaults to a system-specified policy.
            timeout (float): The timeout for this request.  Defaults to a
                system-specified value.

        Returns:
            Sequence[:class:`~google.cloud.firestore_v1.collection.CollectionReference`]:
                iterator of subcollections of the current document.
        r>   Nr'   )Ú_prep_collectionsr/   Úlist_collection_idsrC   r3   )r$   r   r=   r?   rE   Úiteratorr4   r'   r'   r(   Úcollections  s   €þýÿzClient.collectionsiˆ  )Úbulk_writerÚ
chunk_sizeÚ	referencerM   r   rN   c                C   s    |du r|   ¡ }| j|||d�S )aU  Deletes documents and their subcollections, regardless of collection
        name.

        Passing a CollectionReference leads to each document in the collection
        getting deleted, as well as all of their descendents.

        Passing a DocumentReference deletes that one document and all of its
        descendents.

        Args:
            reference (Union[
                :class:`@google.cloud.firestore_v1.collection.CollectionReference`,
                :class:`@google.cloud.firestore_v1.document.DocumentReference`,
            ])
                The reference to be deleted.

            bulk_writer (Optional[:class:`@google.cloud.firestore_v1.bulk_writer.BulkWriter`])
                The BulkWriter used to delete all matching documents. Supply this
                if you want to override the default throttling behavior.

        N)rN   )rM   Ú_recursive_delete)r$   rO   rM   rN   r'   r'   r(   Úrecursive_delete)  s   ýzClient.recursive_deleter   ©rN   ÚdepthrS   c          	      C   sº   d}t |tƒr'| ¡  t ¡ g¡ |¡D ]}|D ]}|d7 }| |j¡ qqn,t |t	ƒrJ| 
¡ D ]}|| j||||d d�7 }q0|d7 }| |¡ n	td|jj› �ƒ‚|dkr[| ¡  |S )z'Recursion helper for `recursive_delete.r   é   rR   zUnexpected type for reference: )Ú
isinstancer   Ú	recursiveÚselectr   Údocument_idÚ	_chunkifyÚdeleterO   r   rL   rP   Ú	TypeErrorr&   Ú__name__Úclose)	r$   rO   rM   rN   rS   Únum_deletedÚchunkÚdoc_snapÚcol_refr'   r'   r(   rP   N  s6   

ýþú


üÿzClient._recursive_deletec                 C   s   t | ƒS )zýGet a batch instance from this client.

        Returns:
            :class:`~google.cloud.firestore_v1.batch.WriteBatch`:
            A "write" batch to be used for accumulating document changes and
            sending the changes all at once.
        r	   r.   r'   r'   r(   Úbatch|  s   zClient.batchc                 K   s   t | fi |¤ŽS )aD  Get a transaction that uses this client.

        See :class:`~google.cloud.firestore_v1.transaction.Transaction` for
        more information on transactions and the constructor arguments.

        Args:
            kwargs (Dict[str, Any]): The keyword arguments (other than
                ``client``) to pass along to the
                :class:`~google.cloud.firestore_v1.transaction.Transaction`
                constructor.

        Returns:
            :class:`~google.cloud.firestore_v1.transaction.Transaction`:
            A transaction attached to this client.
        r   )r$   rE   r'   r'   r(   r<   †  s   zClient.transaction)r   N)'r\   Ú
__module__Ú__qualname__Ú__doc__r   r#   Úpropertyr/   r1   Ústrr   r3   r   r6   r   r9   r   ÚmethodÚDEFAULTÚlistr   r   ÚretriesÚRetryÚfloatr   r   r   rH   rL   r   r   ÚintrQ   rP   r
   rb   r<   Ú__classcell__r'   r'   r%   r(   r   8   sŠ    úù

	&úþýüûú
ù=ýþý
ü û
þüû
ú*ú
þýûú
ù.
r   N))re   Úgoogle.api_corer   r   rk   Ú%google.cloud.firestore_v1.base_clientr   r   r   r   Úgoogle.cloud.firestore_v1.queryr   Úgoogle.cloud.firestore_v1.batchr
   Ú$google.cloud.firestore_v1.collectionr   Ú"google.cloud.firestore_v1.documentr   Ú$google.cloud.firestore_v1.field_pathr   Ú%google.cloud.firestore_v1.transactionr   Ú,google.cloud.firestore_v1.services.firestorer   r,   Ú7google.cloud.firestore_v1.services.firestore.transportsr   r*   Útypingr   r   r   r   r   r   r   Ú'google.cloud.firestore_v1.base_documentr   Ú%google.cloud.firestore_v1.bulk_writerr   r   r'   r'   r'   r(   Ú<module>   s"   $