djsundog
/
circuitpython
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added timeout to input()

This commit is contained in:
ATMakersBill 2018-09-16 15:04:51 -04:00
parent d7517409e9
commit b61214db41
1 changed files with 34 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
py/modbuiltins.c
View File

@ -34,6 +34,7 @@
#include "py/runtime.h" #include "py/runtime.h"
#include "py/builtin.h" #include "py/builtin.h"
#include "py/stream.h" #include "py/stream.h"
#include "py/obj.h" /* For get_int */
#include "supervisor/shared/translate.h" #include "supervisor/shared/translate.h"
@ -233,10 +234,41 @@ MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_hex_obj, mp_builtin_hex);
#define mp_hal_readline readline #define mp_hal_readline readline
#endif #endif
#include "usb.h"
STATIC mp_obj_t mp_builtin_input(size_t n_args, const mp_obj_t *args) { STATIC mp_obj_t mp_builtin_input(size_t n_args, const mp_obj_t *args) {
if (n_args == 1) { if (n_args >= 1) {
mp_obj_print(args[0], PRINT_STR); mp_obj_print(args[0], PRINT_STR);
} }
if (n_args == 2)
{
if (!mp_obj_is_true(args[1]))
{ /* If they pass 0 or False, return immediately if there's no text available */
if (!usb_bytes_available())
{
return mp_const_none;
}
}
else if (MP_OBJ_IS_INT(args[1]))
{
/* Timeout has been sent... check for USB input for that # of millis */
mp_uint_t target = mp_hal_ticks_ms() + mp_obj_get_int(args[1]);
bool bytesAvaliable = false;
while (mp_hal_ticks_ms() < target)
{
if (usb_bytes_available())
{
bytesAvaliable = true;
break;
}
}
if (!bytesAvaliable)
return mp_const_none;
}
}
vstr_t line; vstr_t line;
vstr_init(&line, 16); vstr_init(&line, 16);
int ret = mp_hal_readline(&line, ""); int ret = mp_hal_readline(&line, "");
@ -248,7 +280,7 @@ STATIC mp_obj_t mp_builtin_input(size_t n_args, const mp_obj_t *args) {
} }
return mp_obj_new_str_from_vstr(&mp_type_str, &line); return mp_obj_new_str_from_vstr(&mp_type_str, &line);
} }
MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mp_builtin_input_obj, 0, 1, mp_builtin_input); MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mp_builtin_input_obj, 0, 2, mp_builtin_input);
#endif #endif