def make_connection(config=None, default_model=None): """Create a new Connection object with the right adapter. Optionally you can pass in a datastore_rpc.Configuration object. """ return datastore_rpc.Connection( adapter=ModelAdapter(default_model), config=config)
def __iter__(self): query = self._key_range.make_ascending_datastore_query( self._query_spec.entity_kind, filters=self._query_spec.filters) connection = datastore_rpc.Connection() query_options = datastore_query.QueryOptions( batch_size=self._query_spec.batch_size, start_cursor=self._cursor, produce_cursors=True) self._query = datastore_query.ResultsIterator(query.GetQuery().run( connection, query_options)) for entity_proto in self._query: yield entity_proto
def testRunWithRpcs(self): apiproxy_stub_map.apiproxy = apiproxy_stub_map.APIProxyStubMap() stub = datastore_file_stub.DatastoreFileStub('_', None) apiproxy_stub_map.apiproxy.RegisterStub('datastore_v3', stub) record = [] def foo(arg): record.append(arg) eventloop.queue_call(0.1, foo, 42) conn = datastore_rpc.Connection() config = datastore_rpc.Configuration(on_completion=foo) rpc = conn.async_get(config, []) self.assertEqual(len(rpc.rpcs), 1) eventloop.queue_rpc(rpc) eventloop.run() self.assertEqual(record, [rpc.rpcs[0], 42]) self.assertEqual(rpc.state, 2) # TODO: Use apiproxy_rpc.RPC.FINISHING.
def flush(self): """Flush all puts to datastore.""" if self.puts: datastore_rpc.Connection(config=_CreateDatastoreConfig()).put( self.puts) self.puts = []
def flush(self): """Flush all puts to datastore.""" if self.puts: datastore_rpc.Connection(config=datastore_rpc.Configuration( deadline=60)).put(self.puts) self.puts = []