Build KwArgs for function call in embedded #5241

subatiq · 2024-04-19T06:45:31Z

subatiq
Apr 19, 2024

I'm stuck on trying to build KwArgs or PyDict to pass to function call. In python module I have

def run(**kwargs) -> None:
    pass

kwargs are all strings. I try to call it from Rust:

interpreter.enter(move |vm| {
            let module = vm.import("main", 0).unwrap();
            module.get_attr("run", vm).unwrap().call(config, vm);

        });

config initially has type HashMap<String, String>, though I've tried converting it into IndexMap to then build KwArgs from it. The closest I've got is

struct ConfigArg(HashMap<String, String>);

impl Into<KwArgs> for ConfigArg {
    fn into(self) -> KwArgs {
        let mut map = IndexMap::new();

        let mut args = ArgMapping::
        for (key, value) in self.0 {
            let value = PyStr::from(value);
            map.insert(key, value);
        }

        KwArgs::new(map)
    }
}

The problem above is that KwArgs::new(...) expects IndexMap<String, PyObjectRef> and after reading docs I'm still not sure how to achieve this.

youknowone · 2024-04-20T07:07:37Z

youknowone
Apr 20, 2024
Maintainer

For constants, probably FromIterator will be the easiest way to build it. https://docs.rs/rustpython-vm/latest/rustpython_vm/function/struct.KwArgs.html#impl-FromIterator%3C(String,+T)%3E-for-KwArgs%3CT%3E

let kwargs: KwArgs = [
    ("a".to_owned(), vm.new_pyobj("v1".to_owned()),
    ("b".to_owned(), vm.new_pyobj("v2".to_owned()),
].into_iter().collect();

Otherwise, building IndexMap will be also possible. Don't forget to convert String to PyObejctRef.

impl Into<KwArgs> for ConfigArg {
    fn into(self) -> KwArgs {
        let mut map = IndexMap::new();
        for (key, value) in self.0 {
            let value = vm.new_pyobj(value);  // or `vm.ctx.new_str(value).into()`
            map.insert(key, value);
        }
        KwArgs::new(map)
    }
}

0 replies

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Build KwArgs for function call in embedded #5241

Uh oh!

{{title}}

Uh oh!

Replies: 1 comment

Uh oh!

{{title}}

Uh oh!

Select a reply

Uh oh!

Build KwArgs for function call in embedded #5241

Uh oh!

subatiq Apr 19, 2024

Replies: 1 comment

Uh oh!

youknowone Apr 20, 2024 Maintainer

subatiq
Apr 19, 2024

youknowone
Apr 20, 2024
Maintainer