Add a try-finally for builtins_import

Author: bitterpanda63

aikido_zen/sinks/builtins_import.py

@@ -9,37 +9,38 @@
 
 @after
 def _import(func, instance, args, kwargs, return_value):
-    if running_import_scan.get():
-        return
-    running_import_scan.set(True)
-
-    if not hasattr(return_value, "__file__"):
-        return  # Would be built-in into the interpreter (system package)
-
-    if not hasattr(return_value, "__package__"):
-        return
-    name = getattr(return_value, "__package__")
-
-    if not name:
-        # Make sure the name exists
-        return
-    name = name.split(".")[0]  # Remove submodules
-    if name == "importlib" or name == "importlib_metadata":
-        # Avoid circular dependencies, this is a double safety-check for if contextvar check fails.
-        return
-
-    if PackagesStore.get_package(name):
-        return
-
-    version = None
     try:
-        version = importlib.metadata.version(name)
-    except PackageNotFoundError:
-        pass
-    if version:
-        PackagesStore.add_package(name, version)
-
-    running_import_scan.set(False)
+        if running_import_scan.get():
+            return
+        running_import_scan.set(True)
+
+        if not hasattr(return_value, "__file__"):
+            return  # Would be built-in into the interpreter (system package)
+
+        if not hasattr(return_value, "__package__"):
+            return
+        name = getattr(return_value, "__package__")
+
+        if not name:
+            # Make sure the name exists
+            return
+        name = name.split(".")[0]  # Remove submodules
+        if name == "importlib" or name == "importlib_metadata":
+            # Avoid circular dependencies, this is a double safety-check for if contextvar check fails.
+            return
+
+        if PackagesStore.get_package(name):
+            return
+
+        version = None
+        try:
+            version = importlib.metadata.version(name)
+        except PackageNotFoundError:
+            pass
+        if version:
+            PackagesStore.add_package(name, version)
+    finally:
+        running_import_scan.set(False)
 
 
 @on_import("builtins")