snail-lang 0.7.0__tar.gz → 0.7.2__tar.gz

{snail_lang-0.7.0 → snail_lang-0.7.2}/Cargo.lock

@@ -312,25 +312,25 @@ checksum = "67b1b7a3b5fe4f1376887184045fcf45c69e92af734b7aaddc05fb777b6fbd03"
 
 [[package]]
 name = "snail-ast"
-version = "0.7.0"
+version = "0.7.2"
 
 [[package]]
 name = "snail-error"
-version = "0.7.0"
+version = "0.7.2"
 dependencies = [
  "snail-ast",
 ]
 
 [[package]]
 name = "snail-lower"
-version = "0.7.0"
+version = "0.7.2"
 dependencies = [
  "snail-python",
 ]
 
 [[package]]
 name = "snail-parser"
-version = "0.7.0"
+version = "0.7.2"
 dependencies = [
  "pest",
  "pest_derive",
@@ -340,7 +340,7 @@ dependencies = [
 
 [[package]]
 name = "snail-python"
-version = "0.7.0"
+version = "0.7.2"
 dependencies = [
  "pyo3",
  "snail-ast",

{snail_lang-0.7.0 → snail_lang-0.7.2}/Cargo.toml

@@ -4,3 +4,9 @@ members = ["crates/snail-python"]
 
 [workspace.package]
 edition = "2024"
+
+[workspace.lints.rust]
+warnings = "deny"
+
+[workspace.lints.clippy]
+all = "deny"

{snail_lang-0.7.0 → snail_lang-0.7.2}/PKG-INFO

@@ -1,13 +1,14 @@
 Metadata-Version: 2.4
 Name: snail-lang
-Version: 0.7.0
+Version: 0.7.2
+Requires-Dist: astunparse>=1.6.3 ; python_full_version < '3.9'
 Requires-Dist: jmespath>=1.0.1
 Requires-Dist: maturin>=1.5 ; extra == 'dev'
 Requires-Dist: pytest ; extra == 'dev'
 Provides-Extra: dev
 License-File: LICENSE
 Summary: Snail programming language interpreter
-Requires-Python: >=3.10
+Requires-Python: >=3.8
 Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
 
 <p align="center">
@@ -26,10 +27,10 @@ way into something good, but its certainly not there yet.
 
 ## Installing Snail
 
-Install [uv](https://docs.astral.sh/uv/getting-started/installation/) and then run:
-
 ```bash
-uv tool install -p 3.12 snail-lang
+pip install snail-lang
+-or-
+uv tool install snail-lang
 ```
 
 That installs the `snail` CLI for your user; try it with `snail "print('hello')"` once the install completes.
@@ -75,6 +76,7 @@ END { print("done") }
 | `$p` | Current file path |
 | `$m` | Last regex match object |
 
+
 Begin/end blocks can live in the source file (`BEGIN { ... }` / `END { ... }`) or be supplied
 via CLI flags (`-b`/`--begin`, `-e`/`--end`) for setup and teardown. CLI BEGIN blocks run
 before in-file BEGIN blocks; CLI END blocks run after in-file END blocks.
@@ -112,6 +114,13 @@ BEGIN/END blocks are regular Snail blocks, so awk/map-only `$` variables are not
 snail --map --begin "print('start')" --end "print('done')" "print($src)" *.txt
 ```
 
+### Built-in Variables (All Modes)
+
+| Variable | Description |
+|----------|-------------|
+| `$e` | Exception object in `expr:fallback?` |
+| `$env` | Environment map (wrapper around `os.environ`) |
+
 ### Compact Error Handling
 
 The `?` operator makes error handling terse yet expressive:
@@ -243,7 +252,7 @@ names = js('{{"name": "Ada"}}\n{{"name": "Lin"}}') | $[[*].name]
 ### Full Python Interoperability
 
 Snail compiles to Python AST—import any Python module, use any library, in any
-environment. Assuming that you are using Python 3.10 or later.
+environment. Assuming that you are using Python 3.8 or later.
 
 ## 🚀 Quick Start
 
@@ -262,6 +271,9 @@ snail 'if let [_, user, domain] = "user@example.com" in /^[\w.]+@([\w.]+)$/ { pr
 
 # Awk mode: print line numbers for matches
 rg -n "TODO" README.md | snail --awk '/TODO/ { print("{$n}: {$0}") }'
+
+# Environment variables
+snail 'print($env.PATH)'
 ```
 
 ## 📚 Documentation
@@ -294,7 +306,7 @@ machine snail adds 5 ms of overhead above the regular python3 interpreter.
 
 ### Prerequisites
 
-**Python 3.10+** (required at runtime)
+**Python 3.8+** (required at runtime)
 
 Snail runs in-process via a Pyo3 extension module, so it uses the active Python environment.
 

{snail_lang-0.7.0 → snail_lang-0.7.2}/README.md

@@ -14,10 +14,10 @@ way into something good, but its certainly not there yet.
 
 ## Installing Snail
 
-Install [uv](https://docs.astral.sh/uv/getting-started/installation/) and then run:
-
 ```bash
-uv tool install -p 3.12 snail-lang
+pip install snail-lang
+-or-
+uv tool install snail-lang
 ```
 
 That installs the `snail` CLI for your user; try it with `snail "print('hello')"` once the install completes.
@@ -63,6 +63,7 @@ END { print("done") }
 | `$p` | Current file path |
 | `$m` | Last regex match object |
 
+
 Begin/end blocks can live in the source file (`BEGIN { ... }` / `END { ... }`) or be supplied
 via CLI flags (`-b`/`--begin`, `-e`/`--end`) for setup and teardown. CLI BEGIN blocks run
 before in-file BEGIN blocks; CLI END blocks run after in-file END blocks.
@@ -100,6 +101,13 @@ BEGIN/END blocks are regular Snail blocks, so awk/map-only `$` variables are not
 snail --map --begin "print('start')" --end "print('done')" "print($src)" *.txt
 ```
 
+### Built-in Variables (All Modes)
+
+| Variable | Description |
+|----------|-------------|
+| `$e` | Exception object in `expr:fallback?` |
+| `$env` | Environment map (wrapper around `os.environ`) |
+
 ### Compact Error Handling
 
 The `?` operator makes error handling terse yet expressive:
@@ -231,7 +239,7 @@ names = js('{{"name": "Ada"}}\n{{"name": "Lin"}}') | $[[*].name]
 ### Full Python Interoperability
 
 Snail compiles to Python AST—import any Python module, use any library, in any
-environment. Assuming that you are using Python 3.10 or later.
+environment. Assuming that you are using Python 3.8 or later.
 
 ## 🚀 Quick Start
 
@@ -250,6 +258,9 @@ snail 'if let [_, user, domain] = "user@example.com" in /^[\w.]+@([\w.]+)$/ { pr
 
 # Awk mode: print line numbers for matches
 rg -n "TODO" README.md | snail --awk '/TODO/ { print("{$n}: {$0}") }'
+
+# Environment variables
+snail 'print($env.PATH)'
 ```
 
 ## 📚 Documentation
@@ -282,7 +293,7 @@ machine snail adds 5 ms of overhead above the regular python3 interpreter.
 
 ### Prerequisites
 
-**Python 3.10+** (required at runtime)
+**Python 3.8+** (required at runtime)
 
 Snail runs in-process via a Pyo3 extension module, so it uses the active Python environment.
 

{snail_lang-0.7.0 → snail_lang-0.7.2}/crates/snail-ast/Cargo.toml

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

{snail_lang-0.7.0 → snail_lang-0.7.2}/crates/snail-error/Cargo.toml

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

{snail_lang-0.7.0 → snail_lang-0.7.2}/crates/snail-parser/Cargo.toml

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

{snail_lang-0.7.0 → snail_lang-0.7.2}/crates/snail-parser/src/snail.pest

@@ -250,9 +250,9 @@ boolean = { "True" | "False" }
 none = { "None" }
 
 // Special variables: exception, AWK fields, map vars, injected vars
-exception_var = { "$e" }
+exception_var = { "$e" ~ !ident_continue }
 field_index_var = @{ "$" ~ ASCII_DIGIT+ }
-injected_var = { "$text" | "$src" | "$fn" | "$fd" | "$n" | "$p" | "$m" | "$f" }
+injected_var = { "$text" | "$src" | "$env" | "$fn" | "$fd" | "$n" | "$p" | "$m" | "$f" }
 
 // Number, string, and regex literals
 number = @{ ASCII_DIGIT+ ~ ("." ~ ASCII_DIGIT+)? }

{snail_lang-0.7.0 → snail_lang-0.7.2}/crates/snail-parser/tests/syntax_expressions.rs

@@ -293,6 +293,15 @@ fn parses_structured_accessor_with_pipeline() {
     }
 }
 
+#[test]
+fn parses_env_var() {
+    let program = parse_ok("value = $env");
+    assert_eq!(program.stmts.len(), 1);
+
+    let (_, value) = expect_assign(&program.stmts[0]);
+    expect_name(value, "$env");
+}
+
 #[test]
 fn parses_empty_structured_accessor() {
     let program = parse_ok("result = $[]");

{snail_lang-0.7.0 → snail_lang-0.7.2}/crates/snail-python/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "snail-python"
-version = "0.7.0"
+version = "0.7.2"
 edition.workspace = true
 build = "build.rs"
 
@@ -9,7 +9,7 @@ name = "snail_python"
 crate-type = ["cdylib", "rlib"]
 
 [dependencies]
-pyo3 = { version = "0.21", features = ["extension-module", "abi3-py310"] }
+pyo3 = { version = "0.21", features = ["extension-module", "abi3-py38"] }
 snail-ast = { path = "../snail-ast" }
 snail-error = { path = "../snail-error" }
 snail-parser = { path = "../snail-parser" }

{snail_lang-0.7.0 → snail_lang-0.7.2}/crates/snail-python/src/lower/constants.rs

@@ -37,9 +37,11 @@ pub(crate) const SNAIL_AWK_MATCH_PYVAR: &str = "__snail_match";
 pub(crate) const SNAIL_MAP_SRC: &str = "$src";
 pub(crate) const SNAIL_MAP_FD: &str = "$fd";
 pub(crate) const SNAIL_MAP_TEXT: &str = "$text";
+pub(crate) const SNAIL_ENV: &str = "$env";
 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(crate) const SNAIL_ENV_PYVAR: &str = "__snail_env";
 pub const SNAIL_LAZY_TEXT_CLASS: &str = "__SnailLazyText";
 pub const SNAIL_LAZY_FILE_CLASS: &str = "__SnailLazyFile";
 
@@ -55,6 +57,7 @@ pub(crate) fn injected_py_name(name: &str) -> Option<&'static str> {
         SNAIL_MAP_SRC => Some(SNAIL_MAP_SRC_PYVAR),
         SNAIL_MAP_FD => Some(SNAIL_MAP_FD_PYVAR),
         SNAIL_MAP_TEXT => Some(SNAIL_MAP_TEXT_PYVAR),
+        SNAIL_ENV => Some(SNAIL_ENV_PYVAR),
         _ => None,
     }
 }

{snail_lang-0.7.0 → snail_lang-0.7.2}/crates/snail-python/src/lower/expr.rs

@@ -36,9 +36,7 @@ pub(crate) fn lower_assign_target(
         AssignTarget::Index { value, index, span } => {
             let value_expr = lower_expr_with_exception(builder, value, None)?;
             let index_expr = lower_expr_with_exception(builder, index, None)?;
-            builder
-                .call_node("Subscript", vec![value_expr, index_expr, store_ctx], span)
-                .map_err(py_err_to_lower)
+            subscript_expr(builder, value_expr, index_expr, store_ctx, span)
         }
         AssignTarget::Starred { target, span } => {
             let value = lower_assign_target(builder, target)?;
@@ -101,9 +99,7 @@ pub(crate) fn lower_delete_target(
         AssignTarget::Index { value, index, span } => {
             let value_expr = lower_expr_with_exception(builder, value, None)?;
             let index_expr = lower_expr_with_exception(builder, index, None)?;
-            builder
-                .call_node("Subscript", vec![value_expr, index_expr, del_ctx], span)
-                .map_err(py_err_to_lower)
+            subscript_expr(builder, value_expr, index_expr, del_ctx, span)
         }
         AssignTarget::Starred { .. } => Err(LowerError::new(
             "starred targets are not valid in del statements",
@@ -143,6 +139,19 @@ pub(crate) fn lower_delete_target(
     }
 }
 
+fn subscript_expr(
+    builder: &AstBuilder<'_>,
+    value: PyObject,
+    index: PyObject,
+    ctx: PyObject,
+    span: &SourceSpan,
+) -> Result<PyObject, LowerError> {
+    let slice = builder.wrap_index(index, span).map_err(py_err_to_lower)?;
+    builder
+        .call_node("Subscript", vec![value, slice, ctx], span)
+        .map_err(py_err_to_lower)
+}
+
 pub(crate) fn lower_regex_match(
     builder: &AstBuilder<'_>,
     value: &Expr,
@@ -450,17 +459,13 @@ pub(crate) fn lower_expr_with_exception(
                 builder.load_ctx().map_err(py_err_to_lower)?,
             )?;
             let index_expr = number_expr(builder, &python_index.to_string(), span)?;
-            builder
-                .call_node(
-                    "Subscript",
-                    vec![
-                        value,
-                        index_expr,
-                        builder.load_ctx().map_err(py_err_to_lower)?,
-                    ],
-                    span,
-                )
-                .map_err(py_err_to_lower)
+            subscript_expr(
+                builder,
+                value,
+                index_expr,
+                builder.load_ctx().map_err(py_err_to_lower)?,
+                span,
+            )
         }
         Expr::Number { value, span } => number_expr(builder, value, span),
         Expr::String {
@@ -805,17 +810,13 @@ pub(crate) fn lower_expr_with_exception(
                     span,
                 )
                 .map_err(py_err_to_lower)?;
-            builder
-                .call_node(
-                    "Subscript",
-                    vec![
-                        tuple_expr,
-                        index_expr,
-                        builder.load_ctx().map_err(py_err_to_lower)?,
-                    ],
-                    span,
-                )
-                .map_err(py_err_to_lower)
+            subscript_expr(
+                builder,
+                tuple_expr,
+                index_expr,
+                builder.load_ctx().map_err(py_err_to_lower)?,
+                span,
+            )
         }
         Expr::Regex { pattern, span } => {
             let func = name_expr(
@@ -908,17 +909,13 @@ pub(crate) fn lower_expr_with_exception(
                     .parse::<i32>()
                     .map_err(|_| LowerError::new(format!("Invalid match group index: .{attr}")))?;
                 let index_expr = number_expr(builder, &index.to_string(), span)?;
-                return builder
-                    .call_node(
-                        "Subscript",
-                        vec![
-                            value,
-                            index_expr,
-                            builder.load_ctx().map_err(py_err_to_lower)?,
-                        ],
-                        span,
-                    )
-                    .map_err(py_err_to_lower);
+                return subscript_expr(
+                    builder,
+                    value,
+                    index_expr,
+                    builder.load_ctx().map_err(py_err_to_lower)?,
+                    span,
+                );
             }
             builder
                 .call_node(
@@ -935,13 +932,13 @@ pub(crate) fn lower_expr_with_exception(
         Expr::Index { value, index, span } => {
             let value = lower_expr_with_exception(builder, value, exception_name)?;
             let index = lower_expr_with_exception(builder, index, exception_name)?;
-            builder
-                .call_node(
-                    "Subscript",
-                    vec![value, index, builder.load_ctx().map_err(py_err_to_lower)?],
-                    span,
-                )
-                .map_err(py_err_to_lower)
+            subscript_expr(
+                builder,
+                value,
+                index,
+                builder.load_ctx().map_err(py_err_to_lower)?,
+                span,
+            )
         }
         Expr::Paren { expr, .. } => lower_expr_with_exception(builder, expr, exception_name),
         Expr::List { elements, span } => {
@@ -1344,17 +1341,13 @@ fn lower_postfix_name_incr(
             span,
         )
         .map_err(py_err_to_lower)?;
-    builder
-        .call_node(
-            "Subscript",
-            vec![
-                tuple_expr,
-                index_expr,
-                builder.load_ctx().map_err(py_err_to_lower)?,
-            ],
-            span,
-        )
-        .map_err(py_err_to_lower)
+    subscript_expr(
+        builder,
+        tuple_expr,
+        index_expr,
+        builder.load_ctx().map_err(py_err_to_lower)?,
+        span,
+    )
 }
 
 fn incr_delta(op: IncrOp) -> &'static str {
@@ -1506,17 +1499,13 @@ fn lower_lambda_body_expr(
             span,
         )
         .map_err(py_err_to_lower)?;
-    builder
-        .call_node(
-            "Subscript",
-            vec![
-                tuple_expr,
-                index_expr,
-                builder.load_ctx().map_err(py_err_to_lower)?,
-            ],
-            span,
-        )
-        .map_err(py_err_to_lower)
+    subscript_expr(
+        builder,
+        tuple_expr,
+        index_expr,
+        builder.load_ctx().map_err(py_err_to_lower)?,
+        span,
+    )
 }
 
 fn lower_call_arguments(

{snail_lang-0.7.0 → snail_lang-0.7.2}/crates/snail-python/src/lower/py_ast.rs

@@ -6,13 +6,19 @@ use snail_error::LowerError;
 pub struct AstBuilder<'py> {
     py: Python<'py>,
     ast: Bound<'py, PyModule>,
+    needs_index_wrapper: bool,
 }
 
 impl<'py> AstBuilder<'py> {
     pub fn new(py: Python<'py>) -> PyResult<Self> {
+        let version_info = py.import_bound("sys")?.getattr("version_info")?;
+        let major: u8 = version_info.get_item(0)?.extract()?;
+        let minor: u8 = version_info.get_item(1)?.extract()?;
+        let needs_index_wrapper = major == 3 && minor < 9;
         Ok(Self {
             py,
             ast: py.import_bound("ast")?,
+            needs_index_wrapper,
         })
     }
 
@@ -65,6 +71,28 @@ impl<'py> AstBuilder<'py> {
         let node = self.ast.getattr(name)?.call1(tuple)?;
         Ok(node.into_py(self.py))
     }
+
+    pub fn wrap_index(&self, slice: PyObject, span: &SourceSpan) -> PyResult<PyObject> {
+        if self.needs_index_wrapper {
+            let slice_obj = slice.bind(self.py);
+            if let Ok(slice_type) = self.ast.getattr("Slice")
+                && slice_obj.is_instance(&slice_type)?
+            {
+                return Ok(slice);
+            }
+            if let Ok(ext_slice_type) = self.ast.getattr("ExtSlice")
+                && slice_obj.is_instance(&ext_slice_type)?
+            {
+                return Ok(slice);
+            }
+            if let Ok(index_type) = self.ast.getattr("Index") {
+                let index = index_type.call1((slice,))?;
+                set_location(&index, span)?;
+                return Ok(index.into_py(self.py));
+            }
+        }
+        Ok(slice)
+    }
 }
 
 pub fn set_location(node: &Bound<'_, PyAny>, span: &SourceSpan) -> PyResult<()> {

{snail_lang-0.7.0 → snail_lang-0.7.2}/pyproject.toml

@@ -4,12 +4,13 @@ build-backend = "maturin"
 
 [project]
 name = "snail-lang"
-version = "0.7.0"
+version = "0.7.2"
 description = "Snail programming language interpreter"
 readme = "README.md"
-requires-python = ">=3.10"
+requires-python = ">=3.8"
 license = { file = "LICENSE" }
 dependencies = [
+    "astunparse>=1.6.3; python_version < '3.9'",
     "jmespath>=1.0.1",
 ]
 

{snail_lang-0.7.0 → snail_lang-0.7.2}/python/snail/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from ._native import __build_info__, compile, compile_ast, exec, parse, parse_ast
 
 

{snail_lang-0.7.0 → snail_lang-0.7.2}/python/snail/cli.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import os
 import sys
+from typing import Optional
 
 from . import __build_info__, compile_ast, exec
 
@@ -79,7 +80,11 @@ def _install_trimmed_excepthook() -> None:
             colorize = _colorize.can_colorize(file=sys.stderr)
         except Exception:
             colorize = hasattr(sys.stderr, "isatty") and sys.stderr.isatty()
-        for line in tb_exc.format(colorize=colorize):
+        try:
+            formatted = tb_exc.format(colorize=colorize)
+        except TypeError:
+            formatted = tb_exc.format()
+        for line in formatted:
             sys.stderr.write(line)
 
     sys.excepthook = _snail_excepthook
@@ -87,13 +92,14 @@ def _install_trimmed_excepthook() -> None:
 
 class _Args:
     def __init__(self) -> None:
-        self.file: str | None = None
+        self.file: Optional[str] = None
         self.awk = False
         self.map = False
         self.no_print = False
         self.no_auto_import = False
         self.debug = False
         self.debug_snail_ast = False
+        self.debug_python_ast = False
         self.version = False
         self.help = False
         self.begin_code: list[str] = []
@@ -127,6 +133,7 @@ def _print_help(file=None) -> None:
     print("  -I, --no-auto-import    disable auto-imports", file=file)
     print("  --debug                 parse and compile, then print, do not run", file=file)
     print("  --debug-snail-ast       parse and print Snail AST, do not run", file=file)
+    print("  --debug-python-ast      parse and print Python AST, do not run", file=file)
     print("  -v, --version           show version and exit", file=file)
     print("  -h, --help              show this help message and exit", file=file)
 
@@ -229,6 +236,10 @@ def _parse_args(argv: list[str]) -> _Args:
             args.debug_snail_ast = True
             idx += 1
             continue
+        if token == "--debug-python-ast":
+            args.debug_python_ast = True
+            idx += 1
+            continue
         if token == "-f":
             if idx + 1 >= len(argv):
                 raise ValueError("option -f requires an argument")
@@ -252,7 +263,7 @@ def _parse_args(argv: list[str]) -> _Args:
     return args
 
 
-def _format_version(version: str, build_info: dict[str, object] | None) -> str:
+def _format_version(version: str, build_info: Optional[dict[str, object]]) -> str:
     display_version = version if version.startswith("v") else f"v{version}"
     if not build_info:
         return display_version
@@ -277,7 +288,19 @@ def _get_version() -> str:
     return version
 
 
-def main(argv: list[str] | None = None) -> int:
+def _format_python_runtime() -> str:
+    version = (
+        f"{sys.version_info.major}."
+        f"{sys.version_info.minor}."
+        f"{sys.version_info.micro}"
+    )
+    executable = sys.executable or "<unknown>"
+    if executable != "<unknown>":
+        executable = os.path.abspath(executable)
+    return f"Python {version} ({executable})"
+
+
+def main(argv: Optional[list[str]] = None) -> int:
     if argv is None:
         _install_trimmed_excepthook()
         argv = sys.argv[1:]
@@ -294,6 +317,7 @@ def main(argv: list[str] | None = None) -> int:
         return 0
     if namespace.version:
         print(_format_version(_get_version(), __build_info__))
+        print(_format_python_runtime())
         return 0
 
     # Validate --awk and --map are mutually exclusive
@@ -343,6 +367,24 @@ def main(argv: list[str] | None = None) -> int:
         print(snail_ast)
         return 0
 
+    if namespace.debug_python_ast:
+        import ast
+
+        python_ast = compile_ast(
+            source,
+            mode=mode,
+            auto_print=not namespace.no_print,
+            filename=filename,
+            begin_code=namespace.begin_code,
+            end_code=namespace.end_code,
+        )
+        try:
+            output = ast.dump(python_ast, indent=2)
+        except TypeError:
+            output = ast.dump(python_ast)
+        print(output)
+        return 0
+
     if namespace.debug:
         import ast
         import builtins
@@ -356,7 +398,13 @@ def main(argv: list[str] | None = None) -> int:
             end_code=namespace.end_code,
         )
         builtins.compile(python_ast, _display_filename(filename), "exec")
-        print(ast.unparse(python_ast))
+        try:
+            output = ast.unparse(python_ast)
+        except AttributeError:
+            import astunparse
+
+            output = astunparse.unparse(python_ast).rstrip("\n")
+        print(output)
         return 0
 
     return exec(

{snail_lang-0.7.0 → snail_lang-0.7.2}/python/snail/runtime/__init__.py

@@ -1,12 +1,13 @@
 from __future__ import annotations
 
 import importlib
+from typing import Optional
 
 __all__ = ["install_helpers", "AutoImportDict", "AUTO_IMPORT_NAMES"]
 
 # Names that can be auto-imported when first referenced.
 # Maps name -> (module, attribute) where attribute is None for whole-module imports.
-AUTO_IMPORT_NAMES: dict[str, tuple[str, str | None]] = {
+AUTO_IMPORT_NAMES: dict[str, tuple[str, Optional[str]]] = {
     # Whole module imports: import X
     "sys": ("sys", None),
     "os": ("os", None),
@@ -47,6 +48,7 @@ _incr_attr = None
 _incr_index = None
 _aug_attr = None
 _aug_index = None
+_env_map = None
 
 
 def _get_compact_try():
@@ -130,6 +132,15 @@ def _get_lazy_file_class():
     return _lazy_file_class
 
 
+def _get_env_map():
+    global _env_map
+    if _env_map is None:
+        from .env import EnvMap
+
+        _env_map = EnvMap()
+    return _env_map
+
+
 def _get_incr_attr():
     global _incr_attr
     if _incr_attr is None:
@@ -244,6 +255,7 @@ def install_helpers(globals_dict: dict) -> None:
     globals_dict["__snail_incr_index"] = _lazy_incr_index
     globals_dict["__snail_aug_attr"] = _lazy_aug_attr
     globals_dict["__snail_aug_index"] = _lazy_aug_index
+    globals_dict["__snail_env"] = _get_env_map()
     globals_dict["js"] = _lazy_js
     globals_dict["__SnailLazyText"] = _get_lazy_text_class()
     globals_dict["__SnailLazyFile"] = _get_lazy_file_class()

snail_lang-0.7.2/python/snail/runtime/env.py

@@ -0,0 +1,26 @@
+from __future__ import annotations
+
+import os
+
+
+class EnvMap:
+    __slots__ = ("_env",)
+
+    def __init__(self, env=None) -> None:
+        self._env = os.environ if env is None else env
+
+    def __fallback__(self) -> str:
+        return ""
+
+    def _lookup(self, key):
+        try:
+            return self._env[key]
+        except KeyError as exc:
+            exc.__fallback__ = self.__fallback__
+            raise
+
+    def __getitem__(self, key):
+        return self._lookup(key)
+
+    def __getattr__(self, name):
+        return self._lookup(name)