Simplify references to functions in modules

gaecom · Apr 27, 2022 · 237ad4f · 237ad4f
1 parent 05e7d92
commit 237ad4f
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/jurigged/live.py b/jurigged/live.py
@@ -326,11 +326,12 @@ def _getcode(ref):
                 if ref.startswith("/"):
                     _, module, *hierarchy = ref.split("/")
                     return codefind.find_code(*hierarchy, module=module)
+                elif ":" in ref:
+                    module, hierarchy_s = ref.split(":")
+                    hierarchy = hierarchy_s.split(".")
+                    return codefind.find_code(*hierarchy, module=module)
                 else:
-                    if "/" in ref:
-                        hierarchy = ref.split("/")
-                    else:
-                        hierarchy = ref.split(".")
+                    hierarchy = ref.split(".")
                     return codefind.find_code(*hierarchy, filename=filename)
 
             for ref in opts.loop or []: