From 217c9720eaba7c1692bb8d9e6c8dbdbc64fc97b1 Mon Sep 17 00:00:00 2001 From: mrigal Date: Wed, 29 Apr 2015 19:47:47 +0200 Subject: [PATCH] added iterkeys method and optimized repr, still very ugly --- mongoengine/base/datastructures.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mongoengine/base/datastructures.py b/mongoengine/base/datastructures.py index 7ec67797..8c71c587 100644 --- a/mongoengine/base/datastructures.py +++ b/mongoengine/base/datastructures.py @@ -409,6 +409,9 @@ class StrictDict(object): def items(self): return [(k, self[k]) for k in iter(self)] + def iterkeys(self): + return iter(self) + def keys(self): return list(iter(self)) @@ -433,7 +436,7 @@ class StrictDict(object): __slots__ = allowed_keys_tuple def __repr__(self): - return "{%s}" % ', '.join('"{0!s}": {0!r}'.format(k, v) for (k, v) in self.iteritems()) + return "{%s}" % ', '.join('"{0!s}": {0!r}'.format(k) for k in self.iterkeys()) cls._classes[allowed_keys] = SpecificStrictDict return cls._classes[allowed_keys]