@@ -465,14 +465,35 @@ impl Environment {
465
465
/// import_object = wasi_env.generate_import_object(store, wasi.Version.SNAPSHOT1)
466
466
/// ```
467
467
//#[pyo3(text_signature = "($self, store, wasi_version)")]
468
- fn generate_import_object ( & self , store : & Store , wasi_version : Version ) -> PyResult < PyObject > {
468
+ fn generate_import_object ( & self , store : & Store , wasi_version : Version ) -> ImportObject {
469
469
let import_object = wasmer_wasi:: generate_import_object_from_env (
470
470
store. inner ( ) ,
471
471
self . inner . clone ( ) ,
472
472
wasi_version. into ( ) ,
473
473
) ;
474
474
475
- ImportObject :: raw_new ( import_object) . to_dict ( )
475
+ ImportObject :: raw_new ( import_object)
476
+ }
477
+
478
+ /// Create a dictionary of import with an existing
479
+ /// `Environment`. The import object will be different according
480
+ /// to the WASI version.
481
+ ///
482
+ /// Use the `Version` enum to use a specific WASI version, or use
483
+ /// `get_version` to read the WASI version from a `wasmer.Module`.
484
+ ///
485
+ /// ## Example
486
+ ///
487
+ /// ```py
488
+ /// from wasmer import wasi, Store
489
+ ///
490
+ /// store = Store()
491
+ /// wasi_env = wasi.StateBuilder('test-program').argument('--foo').finalize()
492
+ /// imports = wasi_env.generate_imports(store, wasi.Version.SNAPSHOT1)
493
+ /// ```
494
+ //#[pyo3(text_signature = "($self, store, wasi_version)")]
495
+ fn generate_imports ( & self , store : & Store , wasi_version : Version ) -> PyResult < PyObject > {
496
+ self . generate_import_object ( store, wasi_version) . to_dict ( )
476
497
}
477
498
}
478
499
0 commit comments