From 37d09d6ac30c8c66477149b7c73139c3e6782468 Mon Sep 17 00:00:00 2001 From: Scott Gasch Date: Mon, 27 Sep 2021 08:47:39 -0700 Subject: Compress small text messages by just restricting the alphabet. Adds bidict. --- bidict.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 bidict.py (limited to 'bidict.py') diff --git a/bidict.py b/bidict.py new file mode 100644 index 0000000..5ba3fc3 --- /dev/null +++ b/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