Add a parse_optional_expr function and use it instead of let var = if var == py.None() { None } else { Some(parse_expr(var) };

value = 2

def function(a, b):
    return a + b

if __name__ == '__main__':
    print(function(value, 2 + 2))