__init__.c

// This file is part of the CircuitPython project: https://circuitpython.org
//
// SPDX-FileCopyrightText: Copyright (c) 2023 Jeff Epler for Adafruit Industries
//
// SPDX-License-Identifier: MIT
#include <string.h>

#include "py/builtin.h"
#include "py/obj.h"
#include "py/runtime.h"
#include "py/repl.h"

static const char *get_arg_str(mp_obj_t arg, qstr name) {
    return mp_obj_str_get_str(mp_arg_validate_type_string(arg, name));
}

//| """Utilities to compile possibly incomplete Python source code."""
//|
//| from types import CodeType
//|
//|

//| def compile_command(
//|     source: str, filename: str = "<input>", symbol: str = "single"
//| ) -> CodeType:
//|     """Compile a command and determine whether it is incomplete
//|
//|     The 'completeness' determination is slightly different than in standard Python
//|     (it's whatever the internal function ``mp_repl_continue_with_input`` does).
//|     In particular, it's important that the code not end with a newline character
//|     or it is likely to be treated as a complete command."""
//|
//|
static mp_obj_t compile_command(mp_uint_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args) {
    enum { ARG_source, ARG_filename, ARG_symbol };
    static const mp_arg_t allowed_args[] = {
        { MP_QSTR_source, MP_ARG_REQUIRED | MP_ARG_OBJ, { .u_obj = mp_const_none } },
        { MP_QSTR_filename, MP_ARG_OBJ, { .u_obj = MP_ROM_QSTR(MP_QSTR__lt_input_gt_) } },
        { MP_QSTR_symbol, MP_ARG_OBJ, { .u_obj = MP_ROM_QSTR(MP_QSTR_single) } },
    };
    mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
    mp_arg_parse_all(n_args, pos_args, kw_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);

    const char *source = get_arg_str(args[ARG_source].u_obj, MP_QSTR_source);
    if (mp_repl_continue_with_input(source)) {
        return mp_const_none;
    }

    return mp_call_function_n_kw((mp_obj_t)&mp_builtin_compile_obj, 3, 0, &args[0].u_obj);
}
MP_DEFINE_CONST_FUN_OBJ_KW(compile_command_obj, 1, compile_command);

static const mp_rom_map_elem_t codeop_module_globals_table[] = {
    { MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_codeop) },
    { MP_ROM_QSTR(MP_QSTR_compile_command),  MP_ROM_PTR(&compile_command_obj) },
};

static MP_DEFINE_CONST_DICT(codeop_module_globals, codeop_module_globals_table);

const mp_obj_module_t codeop_module = {
    .base = { &mp_type_module },
    .globals = (mp_obj_dict_t *)&codeop_module_globals,
};

MP_REGISTER_MODULE(MP_QSTR_codeop, codeop_module);