Exemplo n.º 1
0
 def run_query(self, query):
     query_pb = query._ToPb()
     encoded_response = yield self._make_request('RunQuery',
                                                 query_pb.Encode())
     results_pb = datastore_pb.QueryResult(encoded_response)
     raise gen.Return(
         [Entity.FromPb(entity) for entity in results_pb.result_list()])
Exemplo n.º 2
0
 def run_query(self, query):
     entities = []
     query_pb = query._ToPb()
     encoded_response = yield self._make_request('RunQuery',
                                                 query_pb.Encode())
     results_pb = datastore_pb.QueryResult(encoded_response)
     entities.extend(
         [Entity.FromPb(entity) for entity in results_pb.result_list()])
     while results_pb.has_more_results() and len(
             results_pb.result_list()) >= 100:
         query_pb.set_offset(query_pb.offset() +
                             len(results_pb.result_list()))
         encoded_response = yield self._make_request(
             'RunQuery', query_pb.Encode())
         results_pb = datastore_pb.QueryResult(encoded_response)
         entities.extend(
             [Entity.FromPb(entity) for entity in results_pb.result_list()])
     raise gen.Return(entities)
Exemplo n.º 3
0
    def get(self, key, txid=None):
        request = datastore_pb.GetRequest()
        req_key = request.add_key()
        req_key.MergeFrom(key._ToPb())

        if txid is not None:
            req_tx = request.mutable_transaction()
            req_tx.set_app(self.project_id)
            req_tx.set_handle(txid)

        encoded_response = yield self._make_request('Get', request.Encode())
        get_response = datastore_pb.GetResponse(encoded_response)
        response_entity = get_response.entity(0).entity()
        if not response_entity.has_key():
            return

        raise gen.Return(Entity.FromPb(response_entity))