Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-20 03:20:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

improve main.c

Browse Source
This commit is contained in:
blueloveTH 2025-03-12 12:20:03 +08:00
parent 9ecfc0196f
commit 6264776999
1 changed files with 41 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
src2/main.c
View File

@ -2,6 +2,7 @@
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <assert.h> #include <assert.h>
#include <string.h>
#include "pocketpy.h" #include "pocketpy.h"
@ -29,7 +30,7 @@ static void sigint_handler(int sig) { py_interrupt(); }
#endif #endif
char* read_file(const char* path) { static char* read_file(const char* path) {
FILE* file = fopen(path, "rb"); FILE* file = fopen(path, "rb");
if(file == NULL) { if(file == NULL) {
printf("Error: file not found\n"); printf("Error: file not found\n");
@ -44,6 +45,27 @@ char* read_file(const char* path) {
return buffer; return buffer;
} }
static void tracefunc(py_Frame* frame, enum py_TraceEvent event) {
int line;
const char* filename = py_Frame_sourceloc(frame, &line);
const char* event_str;
switch(event) {
case TRACE_EVENT_LINE:
event_str = "line";
break;
case TRACE_EVENT_EXCEPTION:
event_str = "exception";
break;
case TRACE_EVENT_PUSH:
event_str = "push";
break;
case TRACE_EVENT_POP:
event_str = "pop";
break;
}
printf("\x1b[30m%s:%d, event=%s\x1b[0m\n", filename, line, event_str);
}
static char buf[2048]; static char buf[2048];
int main(int argc, char** argv) { int main(int argc, char** argv) {
@ -55,15 +77,27 @@ int main(int argc, char** argv) {
// signal(SIGINT, sigint_handler); // signal(SIGINT, sigint_handler);
#endif #endif
if(argc > 2) { bool trace = false;
printf("Usage: pocketpy [filename]\n"); const char* filename = NULL;
return 0;
for(int i = 1; i < argc; i++) {
if(strcmp(argv[i], "--trace") == 0) {
trace = true;
continue;
}
if(filename == NULL) {
filename = argv[i];
continue;
}
printf("Usage: pocketpy [--trace] filename\n");
} }
py_initialize(); py_initialize();
py_sys_setargv(argc, argv); py_sys_setargv(argc, argv);
if(argc == 1) { if(trace) py_sys_settrace(tracefunc);
if(filename == NULL) {
printf("pocketpy " PK_VERSION " (" __DATE__ ", " __TIME__ ") "); printf("pocketpy " PK_VERSION " (" __DATE__ ", " __TIME__ ") ");
printf("[%d bit] on %s", (int)(sizeof(void*) * 8), PY_SYS_PLATFORM_STRING); printf("[%d bit] on %s", (int)(sizeof(void*) * 8), PY_SYS_PLATFORM_STRING);
#ifndef NDEBUG #ifndef NDEBUG
@ -89,9 +123,9 @@ int main(int argc, char** argv) {
} }
} }
} else { } else {
char* source = read_file(argv[1]); char* source = read_file(filename);
if(source) { if(source) {
if(!py_exec(source, argv[1], EXEC_MODE, NULL)) py_printexc(); if(!py_exec(source, filename, EXEC_MODE, NULL)) py_printexc();
free(source); free(source);
} }
} }