Robust key fetching

Fixes #31
This commit is contained in:
Bouke Haarsma 2013-03-19 11:38:37 +01:00
parent 476718f5f4
commit 847ba5a060

View file

@ -37,10 +37,10 @@ class DatabaseBackend(Backend):
def mget(self, keys):
if not keys:
return
prefixed_keys = [self.add_prefix(key) for key in keys]
stored = self._model._default_manager.filter(key__in=prefixed_keys)
for key, const in itertools.izip(keys, stored):
yield key, const.value
keys = dict((self.add_prefix(key), key) for key in keys)
stored = self._model._default_manager.filter(key__in=keys.keys())
for const in stored:
yield keys[const.key], const.value
def get(self, key):
key = self.add_prefix(key)