From 96851f5db8542a4630aaa9b0dbe7cb0495b5ab8b Mon Sep 17 00:00:00 2001 Message-Id: <96851f5db8542a4630aaa9b0dbe7cb0495b5ab8b.1717874789.git.mdw@distorted.org.uk> From: Mark Wooding Date: Mon, 17 Nov 2008 23:53:05 +0000 Subject: [PATCH] catacomb/__init__.py: Add iterfoo methods to _groupmap. Organization: Straylight/Edgeware From: Mark Wooding This makes them look more like dictionaries, which is nice. --- catacomb/__init__.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/catacomb/__init__.py b/catacomb/__init__.py index cda656d..aaec637 100644 --- a/catacomb/__init__.py +++ b/catacomb/__init__.py @@ -331,10 +331,20 @@ class _groupmap (object): raise TypeError, "immutable object" def __iter__(me): return iter(me.map) + def iterkeys(me): + return iter(me.map) + def itervalues(me): + for k in me: + yield me[k] + def iteritems(me): + for k in me: + yield k, me[k] def keys(me): return [k for k in me] def values(me): return [me[k] for k in me] + def items(me): + return [(k, me[k]) for k in me] eccurves = _groupmap(_base._eccurves, ECInfo._curven) primegroups = _groupmap(_base._pgroups, DHInfo._groupn) bingroups = _groupmap(_base._bingroups, BinDHInfo._groupn) -- [mdw]