Merge pull request #4 from kokseen1/fix_keyerror

fix: delete from `virtual_ip_map` only if present
kokseen1 · Jan 12, 2024 · 807dfbe · 807dfbe
2 parents 3b7d440 + e36e644
commit 807dfbe
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/iro/main.py b/iro/main.py
@@ -152,10 +152,9 @@ def disconnect(sid):
 
     virtual_ip = virtual_ip_map.get(sid)
     if sid is not None:
-        vip_map_value = virtual_ip_map.get(virtual_ip)
-        del virtual_ip_map[vip_map_value]
-        del virtual_ip_map[virtual_ip]
-        del virtual_ip_map[sid]
+        vip_map_value = virtual_ip_map.pop(virtual_ip, None)
+        virtual_ip_map.pop(vip_map_value, None)
+        virtual_ip_map.pop(sid, None)
 
 
     for rnat_key, rnat_value in return_nat.copy().items():

diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
 import pathlib
 import subprocess
 
-__version__ = "1.1.9"
+__version__ = "1.1.10"
 
 # The main interface is through Pybind11Extension.
 # * You can add cxx_std=11/14/17, and then build_ext can be removed.