mirror
/
micropython
kopia lustrzana https://github.com/micropython/micropython
1
0
Forkuj 0
Kod Zgłoszenia Projekty Wydania Wiki Aktywność

extmod/modbtree: Handle default value and error check.

pull/2189/head
Paul Sokolovsky 2016-06-15 04:17:45 +03:00
rodzic a1eab98eaf
commit 422396cece
1 zmienionych plików z 6 dodań i 1 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

7
extmod/modbtree.c
Wyświetl plik

@ -92,8 +92,13 @@ STATIC mp_obj_t btree_get(size_t n_args, const mp_obj_t *args) {
key.data = (void*)mp_obj_str_get_data(args[1], &key.size); key.data = (void*)mp_obj_str_get_data(args[1], &key.size);
int res = __bt_get(self->db, &key, &val, 0); int res = __bt_get(self->db, &key, &val, 0);
if (res == RET_SPECIAL) { if (res == RET_SPECIAL) {
return mp_const_none; if (n_args > 2) {
return args[2];
} else {
return mp_const_none;
}
} }
CHECK_ERROR(res);
return mp_obj_new_bytes(val.data, val.size); return mp_obj_new_bytes(val.data, val.size);
} }
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(btree_get_obj, 2, 3, btree_get); STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(btree_get_obj, 2, 3, btree_get);