Register Device.close with atexit

belay/device.py

@@ -1,4 +1,5 @@
 import ast
+import atexit
 import concurrent.futures
 import importlib.resources
 import linecache
@@ -311,6 +312,8 @@ def __init__(
         for executer in autoinit_executers:
             executer()
 
+        atexit.register(self.close)
+
         self.__post_init__()
 
     def __pre_autoinit__(self):
@@ -647,6 +650,8 @@ def __exit__(self, exc_type, exc_value, exc_tb):
     def close(self) -> None:
         """Close the connection to device."""
         # Invoke all teardown executers prior to closing out connection.
+        atexit.unregister(self.close)
+
         for executer in _sort_executers(self._belay_teardown._belay_executers):
             executer()