Question about call python function in C

[errtiocv]

Hi, I'm trying on embedding micropython in C program.
I have a test.py file:

Code: Select all

 def test(arg1,arg2):
    a = int(arg1)
    b = int(arg2)
    c= a+b
    print("a+b=",c)
    
test(5,6)

And I use mp_call_function_0 to call test.py in C program:

Code: Select all

STATIC int do_file(const char *file) {
    nlr_buf_t nlr;
    if (nlr_push(&nlr) == 0) {
        // create lexer based on source kind
        mp_lexer_t *lex;
        lex = mp_lexer_new_from_file((const char*)file);
        qstr source_name = lex->source_name;
        mp_parse_tree_t parse_tree = mp_parse(lex, MP_PARSE_FILE_INPUT);
        mp_obj_t module_fun = mp_compile(&parse_tree, source_name, false);
        mp_obj_t ret = mp_call_function_0(module_fun);
        
        nlr_pop();
        return 0;

    } else {
        // uncaught exception
        printf("uncaught exception\n");
        return (mp_obj_t)nlr.ret_val;
    }
}

Then I call do_file("path/test.py") in main function, it work well.

Now I want to call python file with input parameters and get the result return, I have tried mp_call_function_2(module_fun,mp_obj_new_int(arg1),mp_obj_new_int(arg2)) but it go uncaught exception.
And I don't know how to let C program recognize the function defined in the python file, is there any example?

[jimmo]

What does it mean to call a python file with input parameters?

If what you want is to load your python file and then call individual functions from inside it (i.e. call the test function directly) then you'll need to lookup the test function in globals and execute it.

Look at runtime.c:mp_parse_compile_execute for the code path that handles this for the import statement.

The key detail though is that mp_compile returns a function that when executed will run all the top-level code in the file and assign all variables and defs into the current globals dict. In your example, the last line also calls the test function.

So if you want to capture just the variables from the file without modifying the current globals, you need to temporarily replace the globals dict with your own one.

But the simple answer to your question is you can lookup "test" in globals and find your function, then execute it the way you suggested using call_function_2