WIP

Author: Thirumalai-Shaktivel

a/__init__.py

@@ -1,2 +1,2 @@
-from .b import print_b
+from .b import print_a, print_b
 from .c import print_c

a/b.py

@@ -1,2 +1,5 @@
+def print_a() -> str:
+    return "A"
+
 def print_b() -> str:
     return "B"

src/lpython/semantics/python_ast_to_asr.cpp

@@ -3070,6 +3070,13 @@ class SymbolTableVisitor : public CommonVisitor<SymbolTableVisitor> {
                 st = st->parent;
             }
             bool ltypes, enum_py;
+            if (!std::filesystem::exists(std::filesystem::path(paths[0]+'/'+msym+".py"))
+                    && import_path != "") {
+                paths = {import_path + '/' + msym};
+                if(std::filesystem::is_directory(std::filesystem::path(paths[0]))) {
+                    msym = "__init__";
+                }
+            }
             t = (ASR::symbol_t*)(load_module(al, st,
                 msym, x.base.base.loc, false, paths, ltypes, enum_py,
                 [&](const std::string &msg, const Location &loc) { throw SemanticError(msg, loc); }

test_a.py

@@ -1,8 +1,10 @@
 import a
+from a import print_a
 
+print(print_a())
+print(a.print_a())
 print(a.print_b())
 print(a.print_c())
 
 # import platform
 # print(platform.python_version())
-