snail-lang 0.6.2__tar.gz → 0.6.3__tar.gz

{snail_lang-0.6.2 → snail_lang-0.6.3}/Cargo.lock

@@ -485,11 +485,11 @@ checksum = "67b1b7a3b5fe4f1376887184045fcf45c69e92af734b7aaddc05fb777b6fbd03"
 
 [[package]]
 name = "snail-ast"
-version = "0.6.2"
+version = "0.6.3"
 
 [[package]]
 name = "snail-core"
-version = "0.6.2"
+version = "0.6.3"
 dependencies = [
  "pyo3",
  "snail-ast",
@@ -500,14 +500,14 @@ dependencies = [
 
 [[package]]
 name = "snail-error"
-version = "0.6.2"
+version = "0.6.3"
 dependencies = [
  "snail-ast",
 ]
 
 [[package]]
 name = "snail-lower"
-version = "0.6.2"
+version = "0.6.3"
 dependencies = [
  "pyo3",
  "snail-ast",
@@ -516,7 +516,7 @@ dependencies = [
 
 [[package]]
 name = "snail-parser"
-version = "0.6.2"
+version = "0.6.3"
 dependencies = [
  "pest",
  "pest_derive",
@@ -526,7 +526,7 @@ dependencies = [
 
 [[package]]
 name = "snail-proptest"
-version = "0.6.2"
+version = "0.6.3"
 dependencies = [
  "proptest",
  "pyo3",
@@ -540,7 +540,7 @@ dependencies = [
 
 [[package]]
 name = "snail-python"
-version = "0.6.2"
+version = "0.6.3"
 dependencies = [
  "pyo3",
  "snail-core",

{snail_lang-0.6.2 → snail_lang-0.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snail-lang
-Version: 0.6.2
+Version: 0.6.3
 Requires-Dist: jmespath>=1.0.1
 Requires-Dist: maturin>=1.5 ; extra == 'dev'
 Requires-Dist: pytest ; extra == 'dev'

{snail_lang-0.6.2 → snail_lang-0.6.3}/crates/snail-ast/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "snail-ast"
-version = "0.6.2"
+version = "0.6.3"
 edition = "2024"
 readme = "README.md"
 

{snail_lang-0.6.2 → snail_lang-0.6.3}/crates/snail-core/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "snail-core"
-version = "0.6.2"
+version = "0.6.3"
 edition.workspace = true
 readme = "README.md"
 

{snail_lang-0.6.2 → snail_lang-0.6.3}/crates/snail-error/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "snail-error"
-version = "0.6.2"
+version = "0.6.3"
 edition = "2024"
 readme = "README.md"
 

{snail_lang-0.6.2 → snail_lang-0.6.3}/crates/snail-lower/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "snail-lower"
-version = "0.6.2"
+version = "0.6.3"
 edition = "2024"
 readme = "README.md"
 

{snail_lang-0.6.2 → snail_lang-0.6.3}/crates/snail-lower/src/constants.rs

@@ -41,6 +41,7 @@ pub(crate) const SNAIL_MAP_SRC_PYVAR: &str = "__snail_src";
 pub(crate) const SNAIL_MAP_FD_PYVAR: &str = "__snail_fd";
 pub(crate) const SNAIL_MAP_TEXT_PYVAR: &str = "__snail_text";
 pub const SNAIL_LAZY_TEXT_CLASS: &str = "__SnailLazyText";
+pub const SNAIL_LAZY_FILE_CLASS: &str = "__SnailLazyFile";
 
 pub(crate) fn injected_py_name(name: &str) -> Option<&'static str> {
     match name {

{snail_lang-0.6.2 → snail_lang-0.6.3}/crates/snail-lower/src/map.rs

@@ -156,7 +156,7 @@ fn lower_map_file_loop(
     auto_print_last: bool,
 ) -> Result<PyObject, LowerError> {
     // for __snail_src in __snail_paths:
-    //     with open(__snail_src, 'r') as __snail_fd:
+    //     with __SnailLazyFile(__snail_src, 'r') as __snail_fd:
     //         __snail_text = __SnailLazyText(__snail_fd)
     //         # user code
 
@@ -201,14 +201,14 @@ fn lower_map_file_loop(
         lower_block_with_auto_print(builder, &program.stmts, auto_print_last, &program.span)?;
     with_body.extend(user_code);
 
-    // open(__snail_src, 'r')
-    let open_call = builder
+    // __SnailLazyFile(__snail_src, 'r')
+    let lazy_file_call = builder
         .call_node(
             "Call",
             vec![
                 name_expr(
                     builder,
-                    "open",
+                    SNAIL_LAZY_FILE_CLASS,
                     span,
                     builder.load_ctx().map_err(py_err_to_lower)?,
                 )?,
@@ -231,12 +231,12 @@ fn lower_map_file_loop(
         )
         .map_err(py_err_to_lower)?;
 
-    // with open(...) as __snail_fd:
+    // with __SnailLazyFile(...) as __snail_fd:
     let with_item = builder
         .call_node_no_loc(
             "withitem",
             vec![
-                open_call,
+                lazy_file_call,
                 name_expr(
                     builder,
                     SNAIL_MAP_FD_PYVAR,

{snail_lang-0.6.2 → snail_lang-0.6.3}/crates/snail-parser/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "snail-parser"
-version = "0.6.2"
+version = "0.6.3"
 edition = "2024"
 readme = "README.md"
 

{snail_lang-0.6.2 → snail_lang-0.6.3}/crates/snail-python/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "snail-python"
-version = "0.6.2"
+version = "0.6.3"
 edition.workspace = true
 build = "build.rs"
 

{snail_lang-0.6.2 → snail_lang-0.6.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "snail-lang"
-version = "0.6.2"
+version = "0.6.3"
 description = "Snail programming language interpreter"
 readme = "README.md"
 requires-python = ">=3.10"

{snail_lang-0.6.2 → snail_lang-0.6.3}/python/snail/runtime/__init__.py

@@ -42,6 +42,7 @@ _subprocess_status = None
 _jmespath_query = None
 _js = None
 _lazy_text_class = None
+_lazy_file_class = None
 _incr_attr = None
 _incr_index = None
 _aug_attr = None
@@ -120,6 +121,15 @@ def _get_lazy_text_class():
     return _lazy_text_class
 
 
+def _get_lazy_file_class():
+    global _lazy_file_class
+    if _lazy_file_class is None:
+        from .lazy_file import LazyFile
+
+        _lazy_file_class = LazyFile
+    return _lazy_file_class
+
+
 def _get_incr_attr():
     global _incr_attr
     if _incr_attr is None:
@@ -236,3 +246,4 @@ def install_helpers(globals_dict: dict) -> None:
     globals_dict["__snail_aug_index"] = _lazy_aug_index
     globals_dict["js"] = _lazy_js
     globals_dict["__SnailLazyText"] = _get_lazy_text_class()
+    globals_dict["__SnailLazyFile"] = _get_lazy_file_class()

snail_lang-0.6.3/python/snail/runtime/lazy_file.py

@@ -0,0 +1,41 @@
+"""Lazy file opener for map mode."""
+
+from __future__ import annotations
+
+
+class LazyFile:
+    """Context manager that opens the file on first access."""
+
+    __slots__ = ("_path", "_mode", "_kwargs", "_fd", "_closed")
+
+    def __init__(self, path, mode="r", **kwargs):
+        self._path = path
+        self._mode = mode
+        self._kwargs = kwargs
+        self._fd = None
+        self._closed = False
+
+    def _ensure_open(self):
+        if self._closed:
+            raise ValueError("I/O operation on closed file.")
+        if self._fd is None:
+            self._fd = open(self._path, self._mode, **self._kwargs)
+        return self._fd
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc, tb):
+        self._closed = True
+        if self._fd is not None:
+            self._fd.close()
+        return False
+
+    def __getattr__(self, name):
+        return getattr(self._ensure_open(), name)
+
+    def __iter__(self):
+        return iter(self._ensure_open())
+
+    def __next__(self):
+        return next(self._ensure_open())