sys/gettrace: Add gettrace function to mirror settrace.

pi-anl · pi-anl · commit 1c58bbbacd20 · 2023-08-22T09:09:32.000+09:30
Refer to https://docs.python.org/3/library/sys.html#sys.gettrace
diff --git a/py/modsys.c b/py/modsys.c
@@ -194,6 +194,11 @@ STATIC mp_obj_t mp_sys_settrace(mp_obj_t obj) {
 }
 MP_DEFINE_CONST_FUN_OBJ_1(mp_sys_settrace_obj, mp_sys_settrace);
 
+STATIC mp_obj_t mp_sys_gettrace() {
+    return mp_prof_gettrace();
+}
+MP_DEFINE_CONST_FUN_OBJ_0(mp_sys_gettrace_obj, mp_sys_gettrace);
+
 // _getframe(): Return current frame object.
 STATIC mp_obj_t mp_sys_getframe(size_t n_args, const mp_obj_t *args) {
     int depth = 0;
@@ -283,6 +288,7 @@ STATIC const mp_rom_map_elem_t mp_module_sys_globals_table[] = {
 
     #if MICROPY_PY_SYS_SETTRACE
     { MP_ROM_QSTR(MP_QSTR_settrace), MP_ROM_PTR(&mp_sys_settrace_obj) },
+    { MP_ROM_QSTR(MP_QSTR_gettrace), MP_ROM_PTR(&mp_sys_gettrace_obj) },
     { MP_ROM_QSTR(MP_QSTR__getframe), MP_ROM_PTR(&mp_sys_getframe_obj) },
     #endif
 
diff --git a/py/profile.c b/py/profile.c
@@ -355,6 +355,13 @@ mp_obj_t mp_prof_settrace(mp_obj_t callback) {
     return mp_const_none;
 }
 
+mp_obj_t mp_prof_gettrace() {
+    if (prof_trace_cb == MP_OBJ_NULL) {
+        return mp_const_none;
+    }
+    return prof_trace_cb;
+}
+
 mp_obj_t mp_prof_get_frame(int depth) {
 
     mp_code_state_t *code_state = MP_STATE_THREAD(current_code_state);
diff --git a/py/profile.h b/py/profile.h
@@ -61,6 +61,7 @@ mp_obj_t mp_obj_new_frame(const mp_code_state_t *code_state);
 
 // This is the implementation for the sys.settrace
 mp_obj_t mp_prof_settrace(mp_obj_t callback);
+mp_obj_t mp_prof_gettrace();
 
 mp_obj_t mp_prof_get_frame(int depth);
 mp_obj_t mp_prof_frame_enter(mp_code_state_t *code_state);
