Dump/load dictionary with instances of different classes as values.

Author: rohe

src/oidcmsg/item.py

@@ -0,0 +1,43 @@
+from typing import List
+from typing import Optional
+
+from oidcmsg.impexp import ImpExp
+from oidcmsg.storage import importer
+from oidcmsg.storage.utils import qualified_name
+
+
+class DLDict(ImpExp):
+    parameter = {
+        "item_class": "",
+        "db": {}
+    }
+
+    def __init__(self):
+        ImpExp.__init__(self)
+        self.db = {}
+
+    def __setitem__(self, key: str, val):
+        self.db[key] = val
+
+    def __getitem__(self, key: str):
+        return self.db[key]
+
+    def dump(self, exclude_attributes: Optional[List[str]] = None) -> dict:
+        res = {}
+
+        for k,v in self.db.items():
+            _class = qualified_name(v.__class__)
+            res[k] = [_class, v.dump(exclude_attributes=exclude_attributes)]
+
+        return res
+
+    def load(self, spec: dict, **kwargs) -> "DLDict":
+        for attr, (_item_cls, _item) in spec.items():
+            self.db[attr] = importer(_item_cls)(**kwargs).load(_item)
+        return self
+
+    def keys(self):
+        return self.db.keys()
+
+    def items(self):
+        return self.db.items()

tests/test_13_dump_item.py

@@ -0,0 +1,34 @@
+from cryptojwt import KeyBundle
+from cryptojwt.key_bundle import build_key_bundle
+from cryptojwt.utils import qualified_name
+
+from oidcmsg.item import DLDict
+
+KEYSPEC = [
+    {"type": "RSA", "use": ["sig"]},
+    {"type": "EC", "crv": "P-256", "use": ["sig"]},
+]
+
+KEYSPEC_2 = [
+    {"type": "RSA", "use": ["sig"]},
+    {"type": "EC", "crv": "P-256", "use": ["sig"]},
+    {"type": "EC", "crv": "P-384", "use": ["sig"]},
+]
+
+
+
+def test_dl_dict():
+    _dict = DLDict()
+    _kb1 = build_key_bundle(key_conf=KEYSPEC)
+    _dict['a'] = _kb1
+    _kb2 = build_key_bundle(key_conf=KEYSPEC_2)
+    _dict['b'] = _kb2
+
+    dump = _dict.dump()
+
+    _dict_copy = DLDict().load(dump)
+
+    assert set(_dict_copy.keys()) == {"a", "b"}
+
+    kb1_copy = _dict_copy['a']
+    assert len(kb1_copy.keys()) == 2