Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2026-02-04 06:30:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

add py_execo

Browse Source
This commit is contained in:
blueloveTH 2026-01-06 13:24:06 +08:00
parent b4ba0d91d5
commit 07a65a8f21
4 changed files with 44 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/pocketpy/pocketpy.h
View File

@ -185,6 +185,8 @@ PK_API bool py_compile(const char* source,
/// Compile a `.py` file into a `.pyc` file.
PK_API bool py_compilefile(const char* src_path,
const char* dst_path) PY_RAISE;
/// Run a compiled code object.
PK_API bool py_execo(const void* data, int size, const char* filename, py_Ref module) PY_RAISE PY_RETURN;
/// Run a source string.
/// @param source source string.
/// @param filename filename (for error messages).

14
src/public/CodeExecution.c
View File

@ -149,6 +149,20 @@ bool py_exec(const char* source, const char* filename, enum py_CompileMode mode,
return ok;
}
bool py_execo(const void* data, int size, const char* filename, py_Ref module) {
CodeObject co;
char* err = CodeObject__loads(data, size, filename, &co);
if(err == NULL) {
c11__rtassert(co.src->mode == EXEC_MODE);
c11__rtassert(co.src->is_dynamic == false);
bool ok = pk_exec(&co, module);
CodeObject__dtor(&co);
return ok;
} else {
return RuntimeError("bad code object %s: %s", filename, err);
}
}
bool py_eval(const char* source, py_Ref module) {
return py_exec(source, "<string>", EVAL_MODE, module);
}

12
src/public/ModuleSystem.c
View File

@ -188,20 +188,10 @@ __SUCCESS:
bool ok;
if(is_pyc) {
CodeObject co;
char* err = CodeObject__loads(data, data_size, filename->data, &co);
if(err == NULL) {
c11__rtassert(co.src->mode == EXEC_MODE);
c11__rtassert(co.src->is_dynamic == false);
ok = pk_exec(&co, mod);
} else {
RuntimeError("failed to load %s: %s", filename->data, err);
ok = false;
}
ok = py_execo(data, data_size, filename->data, mod);
} else {
ok = py_exec(data, filename->data, EXEC_MODE, mod);
}
py_assign(py_retval(), mod);
c11_string__delete(filename);

47
src2/main.c
View File

@ -9,21 +9,6 @@
#include <windows.h>
#endif
static char* read_file(const char* path) {
FILE* file = fopen(path, "rb");
if(file == NULL) {
printf("Error: file not found\n");
return NULL;
}
fseek(file, 0, SEEK_END);
long size = ftell(file);
fseek(file, 0, SEEK_SET);
char* buffer = PK_MALLOC(size + 1);
size = fread(buffer, 1, size, file);
buffer[size] = 0;
return buffer;
}
static char buf[2048];
int main(int argc, char** argv) {
@ -115,9 +100,28 @@ int main(int argc, char** argv) {
if(profile) py_profiler_begin();
if(debug) py_debugger_waitforattach("127.0.0.1", 6110);
char* source = read_file(filename);
if(source) {
if(!py_exec(source, filename, EXEC_MODE, NULL)) py_printexc();
int data_size;
char* data = py_callbacks()->importfile(filename, &data_size);
// check filename endswith .pyc
bool is_pyc = false;
int filename_len = (int)strlen(filename);
if(filename_len >= 4) {
if(filename[filename_len - 4] == '.' &&
filename[filename_len - 3] == 'p' &&
filename[filename_len - 2] == 'y' &&
filename[filename_len - 1] == 'c') {
is_pyc = true;
}
}
if(data) {
bool ok;
if(is_pyc) {
ok = py_execo(data, data_size, filename, NULL);
} else {
ok = py_exec(data, filename, EXEC_MODE, NULL);
}
if(!ok) py_printexc();
if(profile) {
char* json_report = py_profiler_report();
@ -128,8 +132,11 @@ int main(int argc, char** argv) {
}
PK_FREE(json_report);
}
PK_FREE(source);
PK_FREE(data);
} else {
printf("Error: cannot open file '%s'\n", filename);
py_finalize();
return 1;
}
}