X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=collect%2Fbidict.py;fp=collect%2Fbidict.py;h=5ba3fc30f6780537509ba26f49202852ef8a5e05;hb=0bc6e4312cad0f997751739e750954ac39dfa6cc;hp=0000000000000000000000000000000000000000;hpb=37d09d6ac30c8c66477149b7c73139c3e6782468;p=python_utils.git

diff --git a/collect/bidict.py b/collect/bidict.py
new file mode 100644
index 0000000..5ba3fc3
--- /dev/null
+++ b/collect/bidict.py
@@ -0,0 +1,20 @@
+#!/usr/bin/env python3
+
+class bidict(dict):
+    def __init__(self, *args, **kwargs):
+        super(bidict, self).__init__(*args, **kwargs)
+        self.inverse = {}
+        for key, value in self.items():
+            self.inverse.setdefault(value, []).append(key)
+
+    def __setitem__(self, key, value):
+        if key in self:
+            self.inverse[self[key]].remove(key)
+        super(bidict, self).__setitem__(key, value)
+        self.inverse.setdefault(value, []).append(key)
+
+    def __delitem__(self, key):
+        self.inverse.setdefault(self[key], []).remove(key)
+        if self[key] in self.inverse and not self.inverse[self[key]]:
+            del self.inverse[self[key]]
+        super(bidict, self).__delitem__(key)