str: Implement proper string (instead of byte string) indexing.

Also, support negative indexes.
pull/202/head
Paul Sokolovsky 2014-01-21 02:22:02 +02:00
rodzic 545591a696
commit 7380a83780
2 zmienionych plików z 13 dodań i 3 usunięć

Wyświetl plik

@ -44,9 +44,8 @@ mp_obj_t str_binary_op(int op, mp_obj_t lhs_in, mp_obj_t rhs_in) {
// TODO: need predicate to check for int-like type (bools are such for example)
// ["no", "yes"][1 == 2] is common idiom
if (MP_OBJ_IS_SMALL_INT(rhs_in)) {
// TODO: This implements byte string access for single index so far
// TODO: Handle negative indexes.
return mp_obj_new_int(lhs_str[mp_obj_get_int(rhs_in)]);
uint index = mp_get_index(lhs->base.type, strlen(lhs_str), rhs_in);
return mp_obj_new_str(qstr_from_strn_copy(lhs_str + index, 1));
#if MICROPY_ENABLE_SLICE
} else if (MP_OBJ_IS_TYPE(rhs_in, &slice_type)) {
machine_int_t start, stop, step;

Wyświetl plik

@ -10,6 +10,17 @@ print('123' + "456")
print('123' * 5)
print('abc'[1])
print('abc'[-1])
try:
'abc'[100]
except IndexError:
print('caught')
try:
'abc'[-4]
except IndexError:
print('caught2')
# iter
print(list('str'))