Merge pull request #190 from innolitics/fix-strenum-import

Fix: fix broken import for python versions < 3.11
innolitics · Jul 18, 2024 · be4bb51 · be4bb51
2 parents a7ef203 + b09d52a
commit be4bb51
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -6,5 +6,6 @@ build/
 dist/
 .vscode
 venv
+.venv
 .tool-versions
-_debug_pkg
+_debug_pkg
diff --git a/n2y/errors.py b/n2y/errors.py
@@ -1,4 +1,13 @@
-from enum import StrEnum
+try:
+    from enum import StrEnum
+except ImportError:
+    # StrEnum introduced in 3.11
+    # StrEnum defined per recommendation here:
+    #   https://docs.python.org/3.10/library/enum.html#others
+    from enum import Enum
+
+    class StrEnum(str, Enum):
+        pass
 
 
 class N2YError(Exception):