a !fN?@sdZddlZddlZddlmZddlmZddlZddlm Z ddlm Z ddl m Z ddlmZdd lmZd Zd"d d ZddZddZddZd#ddZddZddZddZddZddZGd d!d!eZdS)$zrHelper functions for dealing with Cloud Datastore's Protobuf API. The non-private functions are part of the API. N) struct_pb2) latlng_pb2)_datetime_to_pb_timestamp)_pb_timestamp_to_datetime) entity_pb2)Entity)Key)entity_from_protobufkey_from_protobufFcCsbd}|rRt|jjdkrdSdd|jjD}t|}t|dkrL|}q^|}n |jr^|j}|S)atGet the meaning from a protobuf value. :type value_pb: :class:`gcloud.datastore._generated.entity_pb2.Value` :param value_pb: The protobuf value to be checked for an associated meaning. :type is_list: bool :param is_list: Boolean indicating if the ``value_pb`` contains a list value. :rtype: int :returns: The meaning for the ``value_pb`` if one is set, else :data:`None`. For a list value, if there are disagreeing means it just returns a list of meanings. If all the list meanings agree, it just condenses them. NrcSsg|] }t|qS) _get_meaning).0 sub_value_pbr r Y/var/www/html/python-backend/venv/lib/python3.9/site-packages/gcloud/datastore/helpers.py >sz _get_meaning..)len array_valuevaluessetpopmeaning)value_pbis_listrZ all_meaningsZunique_meaningsr r rr $s  r cCs |j|S)aAdd (by name) a new ``Value`` protobuf to an entity protobuf. :type entity_pb: :class:`gcloud.datastore._generated.entity_pb2.Entity` :param entity_pb: An entity protobuf to add a new property to. :type name: string :param name: The name of the new property. :rtype: :class:`gcloud.datastore._generated.entity_pb2.Value` :returns: The new ``Value`` protobuf that was added to the entity. ) propertiesZ get_or_create) entity_pbnamer r r _new_value_pbMs rcCs t|jS)amIterator of name, ``Value`` tuples from entity properties. :type entity_pb: :class:`gcloud.datastore._generated.entity_pb2.Entity` :param entity_pb: An entity protobuf to add a new property to. :rtype: :class:`generator` :returns: An iterator that yields tuples of a name and ``Value`` corresponding to properties on the entity. )six iteritemsr)rr r r_property_tuples\s r c Csd}|drt|j}i}i}g}t|D]\}}t|}|||<t|t}t||d} | durn| |f||<|rtdd|j j D} t | dkrt d| r||q,|jr,||q,t||d} | || j|| S) awFactory method for creating an entity based on a protobuf. The protobuf should be one returned from the Cloud Datastore Protobuf API. :type pb: :class:`gcloud.datastore._generated.entity_pb2.Entity` :param pb: The Protobuf representing the entity. :rtype: :class:`gcloud.datastore.entity.Entity` :returns: The entity derived from the protobuf. Nkeyrcss|] }|jVqdSN)exclude_from_indexes)r rr r r sz'entity_from_protobuf..rzVFor an array_value, subvalues must either all be indexed or all excluded from indexes.)r!r$)ZHasFieldr r!r _get_value_from_value_pb isinstancelistr rrrr ValueErrorrappendr$rupdate _meanings) pbr!Z entity_propsZentity_meaningsr$Z prop_namervaluerrZexclude_valuesentityr r rr is6            r c Csz||jvrdS|j|\}}||ur(dS|rpt|ts@t|}tj|jj |}|D]\}} | durV| |_ qVn||_ dS)abAdd meaning information (from an entity) to a protobuf. :type entity: :class:`gcloud.datastore.entity.Entity` :param entity: The entity to be turned into a protobuf. :type name: string :param name: The name of the property. :type value: object :param value: The current value stored as property ``name``. :type value_pb: :class:`gcloud.datastore._generated.entity_pb2.Value` :param value_pb: The protobuf value to add meaning / meanings to. :type is_list: bool :param is_list: (Optional) Boolean indicating if the ``value`` is a list value. N) r,r'r( itertoolsrepeatrmovesziprrr) r/rr.rrrZ orig_valueZval_iterrZ sub_meaningr r r_set_pb_meaning_from_entitys      r4cCst}|jdur(|j}|j||D]r\}}t|t}|rTt|dkrTq0t ||}t ||||j vr|s|d|_ |j j D] }d|_ qt|||||dq0|S)aConverts an entity into a protobuf. :type entity: :class:`gcloud.datastore.entity.Entity` :param entity: The entity to be turned into a protobuf. :rtype: :class:`gcloud.datastore._generated.entity_pb2.Entity` :returns: The protobuf representing the entity. NrTr") _entity_pb2rr! to_protobufCopyFromitemsr'r(rr_set_protobuf_valuer$rrr4)r/rZkey_pbrr.Z value_is_listrZ sub_valuer r rentity_to_protobufs&          r:cCsxg}|jD]4}||j|jr,||j|jr ||jq d}|jjrT|jj}d}|jjrh|jj}t|||dS)aVFactory method for creating a key based on a protobuf. The protobuf should be one returned from the Cloud Datastore Protobuf API. :type pb: :class:`gcloud.datastore._generated.entity_pb2.Key` :param pb: The Protobuf representing the key. :rtype: :class:`gcloud.datastore.key.Key` :returns: a new `Key` instance N) namespaceproject) pathr*kindidrZ partition_idZ project_idZ namespace_idr)r-Z path_argselementr<r;r r rr s    r cCs(t|tjrd}t|}nt|tr6d|}}nt|trLd|}}nt|trbd|}}nt|tjrzd|}}nt|tj rd|}}nt|t t frd|}}npt|t rd|}}nZt|t rd |}}nDt|trd |}}n(|d ur d tj}}ntd t||d|fS)aGiven a value, return the protobuf attribute name and proper value. The Protobuf API uses different attribute names based on value types rather than inferring the type. This function simply determines the proper attribute name based on the type of the value provided and returns the attribute name as well as a properly formatted value. Certain value types need to be coerced into a different type (such as a `datetime.datetime` into an integer timestamp, or a `gcloud.datastore.key.Key` into a Protobuf representation. This function handles that for you. .. note:: 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'. For example: >>> _pb_attr_value(1234) ('integer_value', 1234) >>> _pb_attr_value('my_string') ('string_value', 'my_string') :type val: `datetime.datetime`, :class:`gcloud.datastore.key.Key`, bool, float, integer, string :param val: The value to be scrutinized. :rtype: tuple :returns: A tuple of the attribute name and proper value type. timestampr!booleandoubleintegerstringZblobr/arrayZ geo_pointNnullzUnknown protobuf attr type %s_value)r'datetimerrr6boolfloatr integer_types text_typebytesstrrr(GeoPointrZ NULL_VALUEr)type)valrr.r r r_pb_attr_values2!                  rScCs|d}|dkrt|j}n|dkr2t|j}n|dkrB|j}n|dkrR|j}n|dkrb|j}n|dkrr|j}nr|dkr|j }nb|d krt |j }nN|d krd d |j j D}n2|d krt|jj|jj}n|dkrd}ntd|S)aGiven a protobuf for a Value, get the correct value. The Cloud Datastore Protobuf API returns a Property Protobuf which has one value set and the rest blank. This function retrieves the the one value provided. Some work is done to coerce the return value into a more useful type (particularly in the case of a timestamp value, or a key value). :type value_pb: :class:`gcloud.datastore._generated.entity_pb2.Value` :param value_pb: The Value Protobuf. :rtype: object :returns: The value provided by the Protobuf. :raises: :class:`ValueError ` if no value type has been set. value_typetimestamp_value key_value boolean_value double_value integer_value string_value blob_value entity_valuercSsg|] }t|qSr )r&)r r.r r rrysz,_get_value_from_value_pb..geo_point_valueZ null_valueNz)Value protobuf did not have any value set)Z WhichOneofrrUr rVrWrXrYrZr[r r\rrrPr]latitude longituder))rrTresultr r rr&Ls:    r&cCst|\}}|dkr"|j|n|dkr8|j|nn|dkrVt|}|j|nP|dkr|jj}|D]}|}t ||qjn"|dkr|j |n t |||dS)aAssign 'val' to the correct subfield of 'value_pb'. The Protobuf API uses different attribute names based on value types rather than inferring the type. Some value types (entities, keys, lists) cannot be directly assigned; this function handles them correctly. :type value_pb: :class:`gcloud.datastore._generated.entity_pb2.Value` :param value_pb: The value protobuf to which the value is being assigned. :type val: :class:`datetime.datetime`, boolean, float, integer, string, :class:`gcloud.datastore.key.Key`, :class:`gcloud.datastore.entity.Entity` :param val: The value to be assigned. rVrUr\rr]N) rSrVr7rUr:r\rraddr9r]setattr)rrRattrrZl_pbitemZi_pbr r rr9s  r9c@s0eZdZdZddZddZddZdd Zd S) rPzSimple container for a geo point value. :type latitude: float :param latitude: Latitude of a point. :type longitude: float :param longitude: Longitude of a point. cCs||_||_dSr#r^r_)selfr^r_r r r__init__szGeoPoint.__init__cCstj|j|jdS)zConvert the current object to protobuf. :rtype: :class:`google.type.latlng_pb2.LatLng`. :returns: The current point as a protobuf. re)rZLatLngr^r_)rfr r rr6szGeoPoint.to_protobufcCs&t|tsdS|j|jko$|j|jkS)zCompare two geo points for equality. :rtype: boolean :returns: True if the points compare equal, else False. F)r'rPr^r_rfotherr r r__eq__s    zGeoPoint.__eq__cCs || S)zCompare two geo points for inequality. :rtype: boolean :returns: False if the points compare equal, else True. )rjrhr r r__ne__szGeoPoint.__ne__N)__name__ __module__ __qualname____doc__rgr6rjrkr r r rrPs    rP)F)F)rorIr0Zgoogle.protobufrZ google.typerrZgcloud._helpersrrZgcloud.datastore._generatedrr5Zgcloud.datastore.entityrZgcloud.datastore.keyr__all__r rr r r4r:r rSr&r9objectrPr r r rs.        ) 6 *& >=$