More importlib compat fixes

adamcharnock · Aug 17, 2023 · 531b556 · 531b556
1 parent 9c9ada2
commit 531b556
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/lightbus/commands/version.py b/lightbus/commands/version.py
@@ -3,6 +3,8 @@
 import sys
 from pathlib import Path
 
+from importlib.metadata import version as importlib_version
+
 if sys.version_info < (3, 10):
     from importlib.metadata import entry_points as _entry_points
 

diff --git a/lightbus/utilities/importing.py b/lightbus/utilities/importing.py
@@ -41,5 +41,8 @@ def load_entrypoint_classes(entrypoint_name) -> Sequence[Tuple[str, str, Callabl
     found_classes = []
     for entrypoint in entry_points(group=entrypoint_name):
         class_ = entrypoint.load()
-        found_classes.append((entrypoint.module, entrypoint.name, class_))
+        if sys.version_info < (3, 9):
+            found_classes.append((entrypoint.module_name, entrypoint.name, class_))
+        else:
+            found_classes.append((entrypoint.module, entrypoint.name, class_))
     return found_classes