lib/utils: Add pyexec_frozen_module to load and execute frozen module.

This is a convenience function similar to pyexec_file.  It should be used
instead of raw mp_parse_compile_execute because the latter does not catch
and report exceptions.
This commit is contained in:
Damien George 2015-12-26 12:32:33 +00:00
parent 7203b58e87
commit 84b245f187
4 changed files with 19 additions and 16 deletions

View File

@ -32,7 +32,6 @@
#include "py/runtime0.h"
#include "py/runtime.h"
#include "py/stackctrl.h"
#include "py/frozenmod.h"
#include "py/mphal.h"
#include "py/gc.h"
#include "lib/utils/pyexec.h"
@ -49,8 +48,7 @@ STATIC void mp_reset(void) {
mp_obj_list_init(mp_sys_path, 0);
mp_obj_list_init(mp_sys_argv, 0);
#if MICROPY_MODULE_FROZEN
mp_lexer_t *lex = mp_find_frozen_module("main", 4);
mp_parse_compile_execute(lex, MP_PARSE_FILE_INPUT, mp_globals_get(), mp_locals_get());
pyexec_frozen_module("main");
#endif
}

View File

@ -27,12 +27,14 @@
#include <stdlib.h>
#include <stdio.h>
#include <stdint.h>
#include <string.h>
#include "py/nlr.h"
#include "py/compile.h"
#include "py/runtime.h"
#include "py/repl.h"
#include "py/gc.h"
#include "py/frozenmod.h"
#include "py/mphal.h"
#if defined(USE_DEVICE_MODE)
#include "irq.h"
@ -476,6 +478,19 @@ int pyexec_file(const char *filename) {
return parse_compile_execute(lex, MP_PARSE_FILE_INPUT, 0);
}
#if MICROPY_MODULE_FROZEN
int pyexec_frozen_module(const char *name) {
mp_lexer_t *lex = mp_find_frozen_module(name, strlen(name));
if (lex == NULL) {
printf("could not find module '%s'\n", name);
return false;
}
return parse_compile_execute(lex, MP_PARSE_FILE_INPUT, 0);
}
#endif
mp_obj_t pyb_set_repl_info(mp_obj_t o_value) {
repl_display_debugging_info = mp_obj_get_int(o_value);
return mp_const_none;

View File

@ -39,6 +39,7 @@ extern pyexec_mode_kind_t pyexec_mode_kind;
int pyexec_raw_repl(void);
int pyexec_friendly_repl(void);
int pyexec_file(const char *filename);
int pyexec_frozen_module(const char *name);
void pyexec_event_repl_init(void);
int pyexec_event_repl_process_char(int c);

View File

@ -22,11 +22,6 @@
#include "uart.h"
#include "pin.h"
#if MICROPY_MODULE_FROZEN
#include "py/compile.h"
#include "py/frozenmod.h"
#endif
extern uint32_t _heap_start;
void flash_error(int n) {
@ -306,10 +301,7 @@ soft_reset:
#endif
#if MICROPY_MODULE_FROZEN
{
mp_lexer_t *lex = mp_find_frozen_module("boot", 4);
mp_parse_compile_execute(lex, MP_PARSE_FILE_INPUT, mp_globals_get(), mp_locals_get());
}
pyexec_frozen_module("boot");
#else
if (!pyexec_file("/boot.py")) {
flash_error(4);
@ -321,10 +313,7 @@ soft_reset:
// run main script
#if MICROPY_MODULE_FROZEN
{
mp_lexer_t *lex = mp_find_frozen_module("main", 4);
mp_parse_compile_execute(lex, MP_PARSE_FILE_INPUT, mp_globals_get(), mp_locals_get());
}
pyexec_frozen_module("main");
#else
{
vstr_t *vstr = vstr_new();