feat: add tomllib module

GreasySlug · Oct 16, 2023 · 0fea05a · 0fea05a
1 parent 55f24ae
commit 0fea05a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/crates/erg_common/python_util.rs b/crates/erg_common/python_util.rs
@@ -14,7 +14,7 @@ use crate::random::random;
 use crate::serialize::get_magic_num_from_bytes;
 
 #[cfg(unix)]
-pub const BUILTIN_PYTHON_MODS: [&str; 176] = [
+pub const BUILTIN_PYTHON_MODS: [&str; 177] = [
     "abc",
     "argparse",
     "array",
@@ -166,6 +166,7 @@ pub const BUILTIN_PYTHON_MODS: [&str; 176] = [
     "tkinter",
     "token",
     "tokenize",
+    "tomllib",
     "trace",
     "traceback",
     "tracemalloc",
@@ -193,7 +194,7 @@ pub const BUILTIN_PYTHON_MODS: [&str; 176] = [
     "zoneinfo",
 ];
 #[cfg(windows)]
-pub const BUILTIN_PYTHON_MODS: [&str; 171] = [
+pub const BUILTIN_PYTHON_MODS: [&str; 172] = [
     "abc",
     "argparse",
     "array",
@@ -339,6 +340,7 @@ pub const BUILTIN_PYTHON_MODS: [&str; 171] = [
     "tkinter",
     "token",
     "tokenize",
+    "tomllib",
     "trace",
     "traceback",
     "tracemalloc",
@@ -367,7 +369,7 @@ pub const BUILTIN_PYTHON_MODS: [&str; 171] = [
     "zoneinfo",
 ];
 #[cfg(not(any(windows, unix)))]
-pub const BUILTIN_PYTHON_MODS: [&str; 165] = [
+pub const BUILTIN_PYTHON_MODS: [&str; 166] = [
     "abc",
     "argparse",
     "array",
@@ -509,6 +511,7 @@ pub const BUILTIN_PYTHON_MODS: [&str; 165] = [
     "tkinter",
     "token",
     "tokenize",
+    "tomllib",
     "trace",
     "traceback",
     "tracemalloc",

diff --git a/crates/erg_compiler/lib/pystd/tomllib.d.er b/crates/erg_compiler/lib/pystd/tomllib.d.er
@@ -0,0 +1,2 @@
+.load: Bytes -> {Str: Obj}
+.loads: Str -> {Str: Obj}