Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-12-07 10:40:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

up

Browse Source
This commit is contained in:
blueloveTH 2023-01-31 21:31:23 +08:00
parent c1485392cf
commit 6791c0784a
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/compiler.h
View File

@ -838,7 +838,7 @@ __LISTCOMP:
EXPR_TUPLE();
consumeEndStatement();
}
emit(OP_RETURN_VALUE);
emit(OP_RETURN_VALUE, -1, true);
} else if (match(TK("if"))) {
compileIfStatement();
} else if (match(TK("while"))) {

10
src/pocketpy.h
View File

@ -588,6 +588,15 @@ void __addModuleMath(VM* vm){
});
}
void __addModuleDis(VM* vm){
PyVar mod = vm->new_module("dis");
vm->bindFunc<1>(mod, "dis", [](VM* vm, const pkpy::ArgList& args) {
_Code code = vm->PyFunction_AS_C(args[0])->code;
(*vm->_stdout) << vm->disassemble(code);
return vm->None;
});
}
#define PY_CLASS(mod, name) inline static PyVar _tp(VM* vm) { return vm->_modules[#mod]->attribs[#name]; }
@ -790,6 +799,7 @@ extern "C" {
__addModuleJson(vm);
__addModuleMath(vm);
__addModuleRe(vm);
__addModuleDis(vm);
// add builtins | no exception handler | must succeed
_Code code = vm->compile(__BUILTINS_CODE, "<builtins>", EXEC_MODE);