Add global shutdown function

Author: dvdkon

pygeodiff/__init__.py

@@ -7,7 +7,7 @@
     :license: MIT, see LICENSE for more details.
 """
 
-from .main import GeoDiff
+from .main import GeoDiff, shutdown
 from .geodifflib import (
     GeoDiffLibError,
     GeoDiffLibConflictError,
@@ -20,6 +20,7 @@
 
 __all__ = [
     "GeoDiff",
+    "shutdown",
     "GeoDiffLibError",
     "GeoDiffLibConflictError",
     "GeoDiffLibUnsupportedChangeError",

pygeodiff/main.py

@@ -55,7 +55,7 @@ def _lazy_load(self):
             self.context = self.clib.create_context()
 
     def shutdown(self):
-        if self.context is not None:
+        if self.context is not None and self.clib.lib is not None:
             self.clib.destroy_context(self.context)
         self.context = None
 
@@ -469,3 +469,13 @@ def create_wkb_from_gpkg_header(self, geometry):
         """
         self._lazy_load()
         return self.clib.create_wkb_from_gpkg_header(self.context, geometry)
+
+
+def shutdown():
+    """
+    Unloads global GeoDiffLib instance. This breaks all existing GeoDiff
+    instances!
+    """
+    if GeoDiff._clib_weakref is not None and GeoDiff._clib_weakref() is not None:
+        GeoDiff._clib_weakref().shutdown()
+        GeoDiff._clib_weakref = None

pygeodiff/tests/test_lib_iface.py

@@ -6,7 +6,7 @@
 
 import unittest
 
-from pygeodiff import GeoDiff
+from pygeodiff import GeoDiff, GeoDiffLibError, shutdown
 
 
 class UnitTestsLibLazyloading(unittest.TestCase):
@@ -41,3 +41,12 @@ def test_multiple(self):
         self.assertIsNotNone(GeoDiff._clib_weakref())
         del geodiff2
         self.assertIsNone(GeoDiff._clib_weakref())
+
+    def test_global_shutdown(self):
+        geodiff = GeoDiff()
+        geodiff.version()
+        self.assertIsNotNone(GeoDiff._clib_weakref)
+        self.assertIsNotNone(GeoDiff._clib_weakref())
+        shutdown()
+        self.assertIsNone(GeoDiff._clib_weakref)
+        self.assertRaises(AttributeError, geodiff.version)