dev(hansbug): add safe protection

HansBug · Oct 27, 2023 · 844db19 · 844db19
1 parent ce336bb
commit 844db19
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/hbutils/system/filesystem/tempfile.py b/hbutils/system/filesystem/tempfile.py
@@ -6,10 +6,14 @@
 import platform
 import shutil
 import tempfile
-import types
 import warnings
 import weakref
 
+try:
+    from types import GenericAlias
+except (ImportError, ModuleNotFoundError):
+    GenericAlias = None
+
 __all__ = [
     'TemporaryDirectory',
 ]
@@ -90,7 +94,8 @@ def cleanup(self):
             if self._finalizer.detach() or os.path.exists(self.name):
                 self._rmtree(self.name, ignore_errors=self._ignore_cleanup_errors)
 
-        __class_getitem__ = classmethod(types.GenericAlias)
+        if GenericAlias is not None:
+            __class_getitem__ = classmethod(GenericAlias)
 
 else:
     class TemporaryDirectory(object):