From 0bc6e4312cad0f997751739e750954ac39dfa6cc Mon Sep 17 00:00:00 2001 From: Scott Gasch Date: Mon, 27 Sep 2021 09:06:50 -0700 Subject: Move collections into collect space --- collect/bidict.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 collect/bidict.py (limited to 'collect/bidict.py') 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) -- cgit v1.3