From 6d45b633d81b2370f19c91c389ebf25ce564985a Mon Sep 17 00:00:00 2001
From: blueloveTH <blueloveth@foxmail.com>
Date: Tue, 11 Aug 2026 17:01:42 +0800
Subject: [PATCH] fix import multi

---
 src/compiler/compiler.c   |  2 ++
 src/public/ModuleSystem.c | 11 +++++++++++
 tests/302_import_multi.py |  4 ++++
 3 files changed, 17 insertions(+)
 create mode 100644 tests/302_import_multi.py

diff --git a/src/compiler/compiler.c b/src/compiler/compiler.c
index 3a3491ed..62f27763 100644
--- a/src/compiler/compiler.c
+++ b/src/compiler/compiler.c
@@ -2547,6 +2547,8 @@ static Error* compile_normal_import(Compiler* self, c11_sbuf* buf) {
         }
 
         c11_string* path = c11_sbuf__submit(buf);
+        c11_sbuf__ctor(buf);
+        
         int path_index = Ctx__add_const_string(ctx(), c11_string__sv(path));
         c11_string__delete(path);
 
diff --git a/src/public/ModuleSystem.c b/src/public/ModuleSystem.c
index e50754bf..ebd6eb27 100644
--- a/src/public/ModuleSystem.c
+++ b/src/public/ModuleSystem.c
@@ -56,8 +56,19 @@ static void py_ModuleInfo__dtor(py_ModuleInfo* mi) {
     c11_string__delete(mi->path);
 }
 
+static bool module__repr__(int argc, py_Ref argv) {
+    PY_CHECK_ARGC(1);
+    py_ModuleInfo* self = py_touserdata(argv);
+    c11_sbuf buf;
+    c11_sbuf__ctor(&buf);
+    pk_sprintf(&buf, "<module '%s'>", self->path->data);
+    c11_sbuf__py_submit(&buf, py_retval());
+    return true;
+}
+
 py_Type pk_module__register() {
     py_Type type = pk_newtype("module", tp_object, NULL, (py_Dtor)py_ModuleInfo__dtor, false, true);
+    py_bindmagic(type, __repr__, module__repr__);
     return type;
 }
 
diff --git a/tests/302_import_multi.py b/tests/302_import_multi.py
new file mode 100644
index 00000000..3bfa041a
--- /dev/null
+++ b/tests/302_import_multi.py
@@ -0,0 +1,4 @@
+import os, math
+
+assert repr(os) == "<module 'os'>"
+assert repr(math) == "<module 'math'>"