o ˷eSW@sUdZddlZddlZddlmZddlZddlZddlm Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZdd lmZmZmZmZmZm Z m!Z!m"Z"ddl#m$Z$ddlm%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+dZ, dZ-dZ.dZ/e0e1d<dZ2e0e1d<dZ3e0e1d<e j4edZ5ee1d<dZ6e0e1d<Gd d!d!eZ7d"e8d#e!e8e9ffd$d%Z:d&e0d'e9d#e%fd(d)Z;d*ejj?fd/d0Z@d1eAd#e!e0fd2d3ZBdS)4aClient 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` N)AnonymousCredentials)retry) client_info)ClientWithProject)_helpers) __version__)types)DocumentSnapshotrender_field_path) BulkWriterBulkWriterOptions)AnyAsyncGenerator GeneratorIterableListOptionalTupleUnion)BaseCollectionReference)BaseDocumentReference)BaseTransaction)BaseWriteBatch) BaseQueryz (default)zgoogle-cloud-firestore-emulatorzCExactly one of ``last_update_time`` or ``exists`` must be provided.zGDocument {!r} appeared in response but was not present among references_BAD_DOC_TEMPLATEz'There is already an active transaction. _ACTIVE_TXNzThere is no active transaction. _INACTIVE_TXN)client_library_version _CLIENT_INFOFIRESTORE_EMULATOR_HOST_FIRESTORE_EMULATOR_HOSTcsDeZdZdZdZ dZdZdZddee df d:fdd Z de fddZ d d Z defd d Zed dZeddZdefddZdedefddZdedefddZdefddZd;deedefddZdeefddZ d;de eefd ed!de!fd"d#Z"e#d$edefd%d&Z$e#de e%j&e%j'ffd'd(Z(    d BaseClienta: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. For now, :attr:`DEFAULT_DATABASE` (the default value) is the only valid 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. )z.https://www.googleapis.com/auth/cloud-platformz)https://www.googleapis.com/auth/datastoreNreturncs|tt|_|jdur|durt}|durt}tt|j|||dd||_ |r6t |t kr6t j j|}||_||_dS)N)project credentialsclient_options_http)osgetenvr!_emulator_hostr_DEFAULT_EMULATOR_PROJECTsuperr"__init__ _client_infotypedictgoogleapi_corer& from_dict_client_options _database)selfr$r%databaserr& __class__\/var/www/ideatree/venv/lib/python3.10/site-packages/google/cloud/firestore_v1/base_client.pyr-ts(    zBaseClient.__init__cCsl|jdur3|jdur||}n|j|j|jddid}||j|d|_||j|jd|_|j |_ |jS)zLazy-loading getter GAPIC Firestore API. Returns: The GAPIC client with the credentials of the current client. Nzgrpc.keepalive_time_msi0u)r%options)hostchannel) transportr&) _firestore_api_internalr*_emulator_channelcreate_channel_target _credentialsitems _transportr4r.)r6r? client_class client_moduler>r:r:r;_firestore_api_helpers    z BaseClient._firestore_api_helpercCshd}|jdurt|jdddur|jj}dd|fg}dt|jvr,tjj|j|dStj|j|dS)a] Creates an insecure channel to communicate with the local emulator. If credentials are provided the token is extracted and added to the headers. This supports local testing of firestore rules if the credentials have been created from a signed custom token. :return: grpc.Channel or grpc.aio.Channel ownerNid_token AuthorizationzBearer GrpcAsyncIOTransport)r<) rDgetattrrKstr__name__grpcaioinsecure_channelr*)r6r?tokenr<r:r:r;rAs zBaseClient._emulator_channelcCs,|jdur|jS|jr|jjr|jjS|jS)zReturn the target (where the API is). Eg. "firestore.googleapis.com" Returns: str: The location of the API. N)r*r4 api_endpointDEFAULT_ENDPOINT)r6rGr:r:r;_target_helpers zBaseClient._target_helpercCs.|jdurtjjjd|j|jd}||_|jS)aThe database string corresponding to this client's project. This value is lazy-loaded and cached. Will be of the form ``projects/{project_id}/databases/{database_id}`` but ``database_id == '(default)'`` for the time being. Returns: str: The fully-qualified database string for the current project. (The default database is also in this string.) Nz'projects/{project}/databases/{database})r$r7)_database_string_internalr1r2 path_templateexpandr$r5)r6db_strr:r:r;_database_strings zBaseClient._database_stringcCs4|jdurt|j|_|jdur|jd|jS)zThe RPC metadata for this client's associated database. Returns: Sequence[Tuple(str, str)]: RPC metadata with resource prefix for the database associated with this client. N) authorizationz Bearer owner)_rpc_metadata_internalrmetadata_with_prefixr\r*appendr6r:r:r; _rpc_metadatas   zBaseClient._rpc_metadatacGtNNotImplementedError)r6collection_pathr:r:r; collectionzBaseClient.collection collection_idcCrcrdrer6rjr:r:r;collection_group rizBaseClient.collection_groupcCs"d|vr td|d||S)aChecks validity of collection_id and then uses subclasses collection implementation. 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: The created collection. /zInvalid collection_id z&. Collection IDs must not contain '/'.) ValueErrorrhrkr:r:r;_get_collection_references  z$BaseClient._get_collection_referencecGrcrdre)r6 document_pathr:r:r;document#rizBaseClient.documentr<cCs t||dS)aGet a BulkWriter instance from this client. Args: :class:`@google.cloud.firestore_v1.bulk_writer.BulkWriterOptions`: Optional control parameters for the :class:`@google.cloud.firestore_v1.bulk_writer.BulkWriter` returned. Returns: :class:`@google.cloud.firestore_v1.bulk_writer.BulkWriter`: A utility to efficiently create and save many `WriteBatch` instances to the server. )clientr<)r )r6r<r:r:r; bulk_writer&s zBaseClient.bulk_writercGsDt|}|jd}tj|}||r|t|d}|tjS)a2Standardize the format of path to tuple of path segments and strip the database string from path if present. Args: document_path (Tuple[str, ...]): Can either be * A single ``/``-delimited path to a document * A tuple of document path segments z /documents/N) _path_helperr\rDOCUMENT_PATH_DELIMITERjoin startswithlensplit)r6rppath base_path joined_pathr:r:r;_document_path_helper5s     z BaseClient._document_path_helper referencersr cCrcrdre)r6r~rsr:r:r;recursive_deleteEszBaseClient.recursive_delete field_namescGst|S)aCreate a **field path** from a list of nested field names. A **field path** is a ``.``-delimited concatenation of the field names. It is used to represent a nested field. For example, in the data .. code-block:: python data = { 'aa': { 'bb': { 'cc': 10, }, }, } the field path ``'aa.bb.cc'`` represents the data stored in ``data['aa']['bb']['cc']``. Args: field_names: The list of field names. Returns: str: The ``.``-delimited field path. r )rr:r:r; field_pathLszBaseClient.field_pathcKsXt|dkr tt|\}}|dkrt|S|dkr"t|Sd|}tt|)a,Create a write option for write operations. Write operations include :meth:`~google.cloud.DocumentReference.set`, :meth:`~google.cloud.DocumentReference.update` and :meth:`~google.cloud.DocumentReference.delete`. One of the following keyword arguments must be provided: * ``last_update_time`` (:class:`google.protobuf.timestamp_pb2. Timestamp`): A timestamp. When set, the target document must exist and have been last updated at that time. Protobuf ``update_time`` timestamps are typically returned from methods that perform write operations as part of a "write result" protobuf or directly. * ``exists`` (:class:`bool`): Indicates if the document being modified should already exist. Providing no argument would make the option have no effect (so it is not allowed). Providing multiple would be an apparent contradiction, since ``last_update_time`` assumes that the document **was** updated (it can't have been updated if it doesn't exist) and ``exists`` indicate that it is unknown if the document exists or not. Args: kwargs (Dict[str, Any]): The keyword arguments described above. Raises: TypeError: If anything other than exactly one argument is provided by the caller. Returns: :class:`~google.cloud.firestore_v1.client.WriteOption`: The option to be used to configure a write message. last_update_timeexistsz{!r} was provided)rx TypeError_BAD_OPTION_ERRpopitemrLastUpdateOption ExistsOptionformat)kwargsnamevalueextrar:r:r; write_optionis '     zBaseClient.write_option references field_paths transactionrtimeoutc Cs@t|\}}t|}|j||t|d} t||} | || fS)z,Shared setup for async/sync :meth:`get_all`.)r7 documentsmaskr)_reference_info _get_doc_maskr\rget_transaction_idmake_retry_timeout_kwargs) r6rrrrrdocument_paths reference_maprrequestrr:r:r; _prep_get_alls   zBaseClient._prep_get_allcCrcrdre)r6rrrrrr:r:r;get_alls zBaseClient.get_allcCs$dd|ji}t||}||fS)z0Shared setup for async/sync :meth:`collections`.parentz {}/documents)rr\rr)r6rrrrr:r:r;_prep_collectionss zBaseClient._prep_collectionscCrcrdre)r6rrr:r:r; collectionsszBaseClient.collectionscCrcrdrerar:r:r;batchrizBaseClient.batchcKrcrdre)r6rr:r:r;rrizBaseClient.transaction)r#Nrd)NNNN)NN)sb          (        6