This repository includes a small Rust utility for transforming Python source
code. It parses a file with Ruff's parser and rewrites binary operations and
augmented assignments (e.g., +=) into calls to the corresponding functions in
the standard library's operator module. The transformation is idempotent, so
re-running it on already rewritten code leaves the output unchanged.
Run it with:
cargo run -- path/to/file.py
To apply the transform automatically when modules are imported, install the provided import hook:
import diet_import_hook
diet_import_hook.install()After calling install(), any subsequent imports will be rewritten using the
diet-python transform before execution.
Run the included example to see the hook in action:
python example_usage.py
The script installs the hook, imports example_module, and asserts that its
bytecode calls operator.add instead of using BINARY_OP.
To run the official CPython test suite with a Python installed via uv, use:
scripts/run_cpython_tests.sh
The script clones the cpython repository if necessary, creates a virtual
environment using uv, and executes the test suite with that interpreter.