diff --git a/unix/Makefile b/unix/Makefile index 00e7e946028d9763a5710ba18370b1344a8be7b8..a1a6a7102c60e4f6334a24bf8a289f20992503e6 100644 --- a/unix/Makefile +++ b/unix/Makefile @@ -14,6 +14,10 @@ include ../py/py.mk CFLAGS = -I. -I$(PY_SRC) -Wall -Werror -ansi -std=gnu99 -DUNIX $(CFLAGS_MOD) LDFLAGS = $(LDFLAGS_MOD) -lm +ifeq ($(MICROPY_MOD_TIME),1) +CFLAGS_MOD += -DMICROPY_MOD_TIME=1 +SRC_MOD += time.c +endif ifeq ($(MICROPY_MOD_FFI),1) CFLAGS_MOD += `pkg-config --cflags libffi` -DMICROPY_MOD_FFI=1 LDFLAGS_MOD += -ldl -lffi diff --git a/unix/main.c b/unix/main.c index 5ca81153687972429a07cfe4881306c5950007fd..cc942163f9871a548f01d8bafb96e3b8a6cbbb18 100644 --- a/unix/main.c +++ b/unix/main.c @@ -24,6 +24,7 @@ extern const mp_obj_fun_native_t mp_builtin_open_obj; void file_init(); void rawsocket_init(); +void time_init(); void ffi_init(); static void execute_from_lexer(mp_lexer_t *lex, mp_parse_input_kind_t input_kind, bool is_repl) { @@ -242,6 +243,9 @@ int main(int argc, char **argv) { file_init(); rawsocket_init(); +#if MICROPY_MOD_TIME + time_init(); +#endif #if MICROPY_MOD_FFI ffi_init(); #endif diff --git a/unix/mpconfigport.mk b/unix/mpconfigport.mk index d220339397af72869c75c0b3d38b49229edf436d..6c7be237bc65be19476cf3b263a8dfa515c1df49 100644 --- a/unix/mpconfigport.mk +++ b/unix/mpconfigport.mk @@ -1,4 +1,7 @@ # Enable/disable modules to be included in interpreter +# Subset of CPython time module +MICROPY_MOD_TIME = 1 + # ffi module requires libffi (libffi-dev Debian package) MICROPY_MOD_FFI = 0 diff --git a/unix/time.c b/unix/time.c new file mode 100644 index 0000000000000000000000000000000000000000..acea69178a3e1464905f2d9fe00c2b66620ba00c --- /dev/null +++ b/unix/time.c @@ -0,0 +1,25 @@ +#include <string.h> +#include <time.h> + +#include "misc.h" +#include "mpconfig.h" +#include "qstr.h" +#include "obj.h" +#include "runtime.h" + +static mp_obj_t mod_time_time() { + return mp_obj_new_int((machine_int_t)time(NULL)); +} +static MP_DEFINE_CONST_FUN_OBJ_0(mod_time_time_obj, mod_time_time); + +// Note: this is deprecated since CPy3.3, but pystone still uses it. +static mp_obj_t mod_time_clock() { + return mp_obj_new_int((machine_int_t)clock()); +} +static MP_DEFINE_CONST_FUN_OBJ_0(mod_time_clock_obj, mod_time_clock); + +void time_init() { + mp_obj_t m = mp_obj_new_module(QSTR_FROM_STR_STATIC("time")); + rt_store_attr(m, QSTR_FROM_STR_STATIC("time"), (mp_obj_t)&mod_time_time_obj); + rt_store_attr(m, QSTR_FROM_STR_STATIC("clock"), (mp_obj_t)&mod_time_clock_obj); +}