fix: import error messages

boa/interpret.py

@@ -27,29 +27,38 @@
 
 
 class BoaImporter(importlib.abc.MetaPathFinder):
+    def __init__(self):
+        self._path_lookup = {}
+
     def find_module(self, fullname, package_path, target=None):
-        # Return a loader
-        return self
+        # note: maybe instead of looping, append path to package_path
+        path = Path(fullname.replace(".", "/")).with_suffix(".vy")
+
+        for prefix in package_path:
+            # for fullname == "x.y.z"
+            # prefix looks something like "<...>/site-packages/x"
+            to_try = Path(prefix).parent / path
+
+            if to_try.exists():
+                self._path_lookup[fullname] = to_try
+                return self
+
+        return None
 
     def load_module(self, fullname):
-        # Return a module
         if fullname in sys.modules:
             return sys.modules[fullname]
 
-        path = Path(fullname.replace(".", "/")).with_suffix(".vy")
-        for prefix in sys.path:
-            to_try = Path(prefix) / path
-            try:
-                ret = load_partial(to_try)
-                break
-            except (FileNotFoundError, NotADirectoryError):
-                pass
-        else:
-            raise ImportError(fullname)
+        # import system should guarantee this, but be paranoid
+        if fullname not in self._path_lookup:
+            raise ImportError(f"invariant violated: no lookup for {fullname}")
+
+        path = self._path_lookup[fullname]
+        ret = load_partial(path)
 
         # comply with PEP-302:
-        ret.__name__ = to_try.name
-        ret.__file__ = str(to_try)
+        ret.__name__ = path.name
+        ret.__file__ = str(path)
         sys.modules[fullname] = ret
         return ret