U $FZhj6@sdZddlZddlZddlmZmZmZmZmZddl m Z m Z ddl m Z ddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZm Z m!Z!m"Z"GdddZ#GdddZ$dS)zDClasses for representing bundles for the Google Cloud Firestore API.N)BundledDocumentMetadata BundledQuery BundleElementBundleMetadata NamedQuery)_datetime_to_pb_timestampUTC)limit_type_of_query) AsyncQuery) BaseClient)DocumentSnapshot) BaseQuery)DocumentReference)_helpers) Timestamp) json_format)DictListOptionalUnionc@seZdZUdZdZeed<eddddZe ddd d Z ee dd d d Z e ee j dddZee e j ddddZeee j dddZee e j edddZee j efddddZeeedddZed d!d"Zeed#d$d%Zd&d'ZdS)(FirestoreBundleamA group of serialized documents and queries, suitable for longterm storage or query resumption. If any queries are added to this bundle, all associated documents will be loaded and stored in memory for serialization. Usage: .. code-block:: python from google.cloud.firestore import Client, _helpers from google.cloud.firestore_bundle import FirestoreBundle db = Client() bundle = FirestoreBundle('my-bundle') bundle.add_named_query('all-users', db.collection('users')._query()) bundle.add_named_query( 'top-ten-hamburgers', db.collection('hamburgers').limit(limit=10), ) serialized: str = bundle.build() # Store somewhere like a Google Cloud Storage bucket for retrieval by # a client SDK. Args: name (str): The Id of the bundle. BUNDLE_SCHEMA_VERSIONN)namereturncCs*||_i|_i|_tddd|_d|_dS)Nr)secondsZnanos)r documents named_queriesrlatest_read_time_deserialized_metadata)selfrr!k/home/aprabhat/apps/x.techxrdev.in/venv/lib/python3.8/site-packages/google/cloud/firestore_bundle/bundle.py__init__Ms zFirestoreBundle.__init__)snapshotrcCsg}|jj}|j|}|r$|jj}|dkp@t|j|j jdk}|rht |t ||j|j |dd|j|<| |j||S)aAdds a document to the bundle. Args: snapshot (DocumentSnapshot): The fully-loaded Firestore document to be preserved. Example: .. code-block:: python from google.cloud import firestore db = firestore.Client() collection_ref = db.collection(u'users') bundle = firestore.FirestoreBundle('my bundle') bundle.add_document(collection_ref.documents('some_id').get()) Returns: FirestoreBundle: self Nr)r read_timeexistsqueriesr$metadata) reference_document_pathrgetr)r'rcompare_timestampsr%r$_BundledDocumentrr&_update_last_read_time_reset_metadata)r r$Zoriginal_queriesZfull_document_pathZoriginal_documentZshould_use_snaphotr!r!r" add_documentTs4   zFirestoreBundle.add_document)rqueryrcCsbt|ts tdt|jd||jkr:td|d|j||d}||||||S)aAdds a query to the bundle, referenced by the provided name. Args: name (str): The name by which the provided query should be referenced. query (Query): Query of documents to be fully loaded and stored in the bundle for future access. Example: .. code-block:: python from google.cloud import firestore db = firestore.Client() collection_ref = db.collection(u'users') bundle = firestore.FirestoreBundle('my bundle') bundle.add_named_query('all the users', collection_ref._query()) Returns: FirestoreBundle: self Raises: ValueError: If anything other than a BaseQuery (e.g., a Collection) is supplied. If you have a Collection, call its `_query()` method to get what this method expects. ValueError: If the supplied name has already been added. z&Attempted to add named query of type: z. Expected BaseQuery.zQuery name conflict: z has already been added.) query_name) isinstancer ValueErrortype__name__r_save_documents_from_query_save_named_queryr0)r rr2 _read_timer!r!r"add_named_querys  zFirestoreBundle.add_named_query)r2r3rcCs|tjjjtd}t|trr?r@rrCr1rr,r*r+r)r'rDr%)r r$r3r:rFrGr!r!r"rBs  z$FirestoreBundle._process_async_query)rr$r%rcCs(t|t||jt|dt|dS)N)parentZstructured_queryZ limit_type)rZ bundled_queryr%)rr _to_protobuf_pbr rbuild_timestamp)r rr$r%r!r!r"rHsz"FirestoreBundle._build_named_query)r%rcCs2t|tr|nt|}t||jdkr.||_dS)Nr)r4rrrr-r)r r%Z_tsr!r!r"r/s z&FirestoreBundle._update_last_read_time)bundle_elementclientr6c Csddlm}t|dddkr"i|_|dkr4|j|_n|dkrN|j|j|jj<n|dkrh|j |j|j j<n|dkrt |j j}t t ||j d j|d t|j|j|d |j|j jj|j j|j jd }|||j|jj}|j|j jjD]}|jj|qntd |dS)z}Applies BundleElements to this FirestoreBundle instance as a part of deserializing a FirestoreBundle string. r)Document_doc_metadata_mapNr)Z namedQueryZdocumentMetadatadocument)mappingT)rN)datar&r*r% create_time update_timez"Unexpected type of BundleElement: )Z(google.cloud.firestore_v1.types.documentrOgetattrrPr)r named_queryrrdocument_metadatarZDocumentReferenceValuerQr Z decode_dictfieldsrZcollection_nameZ document_idr%rTrUr1rr,r*r+r'rDr5) r rMrNr6rOZ doc_ref_valuer$rGr3r!r!r"_add_bundle_elementsP      z#FirestoreBundle._add_bundle_element)rc Csd}|jD]}||t|d7}qd}|jD]>}||t|jd7}|d7}||t|jjd7}q6t|j pt |j t tj|t|ddd }|||S) aIterates over the bundle's stored documents and queries and produces a single length-prefixed json string suitable for long-term storage. Example: .. code-block:: python from google.cloud import firestore db = firestore.Client() collection_ref = db.collection(u'users') bundle = firestore.FirestoreBundle('my bundle') bundle.add_named_query('app-users', collection_ref._query()) serialized_bundle: str = bundle.build() # Now upload `serialized_bundle` to Google Cloud Storage, store it # in Memorystore, or any other storage solution. Returns: str: The length-prefixed string representation of this bundle' contents. )rWr)rXr)rQzutf-8)idrTversionZtotal_documents total_bytes)r))rvalues_compile_bundle_elementrrr)r$rJrKrrrrrLrrlenencode)r bufferrWZdocument_countrGr)r!r!r"build3s6    zFirestoreBundle.build)rMrcCs"tt|j}t||SN)jsondumpsrZ MessageToDictrKra)r rMZ serialized_ber!r!r"r`msz'FirestoreBundle._compile_bundle_elementcCs d|_dS)zeHydrating bundles stores cached data we must reset anytime new queries or documents are addedN)r)r r!r!r"r0qszFirestoreBundle._reset_metadata)r7 __module__ __qualname____doc__rint__annotations__strr#r r1r r;r>r8r9r rBrrHrrr/rr rZrdr`r0r!r!r!r"r-s<  8/      .:rc@s"eZdZdZeeddddZdS)r.zcConvenience class to hold both the metadata and the actual content of a document to be bundled.N)r$r)rcCs||_||_dSrer()r r$r)r!r!r"r#{sz_BundledDocument.__init__)r7rhrirjr rr#r!r!r!r"r.ws r.)%rjr>rfZ*google.cloud.firestore_bundle.types.bundlerrrrrZgoogle.cloud._helpersrrZ&google.cloud.firestore_bundle._helpersr Z%google.cloud.firestore_v1.async_queryr Z%google.cloud.firestore_v1.base_clientr Z'google.cloud.firestore_v1.base_documentr Z$google.cloud.firestore_v1.base_queryr Z"google.cloud.firestore_v1.documentrZgoogle.cloud.firestore_v1rZgoogle.protobuf.timestamp_pb2rZgoogle.protobufrtypingrrrrrr.r!r!r!r"s$         L