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

py/objstr: Catch case of negative "maxsplit" arg to str.rsplit().

Browse Source 
Negative values mean no limit on the number of splits so should delegate to
the .split() method.
This commit is contained in:
Damien George 2017-06-02 13:07:22 +10:00
parent ab954ed513
commit 9f85c4fe48
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
py/objstr.c
View File

@ -602,6 +602,11 @@ STATIC mp_obj_t str_rsplit(size_t n_args, const mp_obj_t *args) {
GET_STR_DATA_LEN(args[0], s, len); GET_STR_DATA_LEN(args[0], s, len);
mp_int_t splits = mp_obj_get_int(args[2]); mp_int_t splits = mp_obj_get_int(args[2]);
if (splits < 0) {
// Negative limit means no limit, so delegate to split().
return mp_obj_str_split(n_args, args);
}
mp_int_t org_splits = splits; mp_int_t org_splits = splits;
// Preallocate list to the max expected # of elements, as we // Preallocate list to the max expected # of elements, as we
// will fill it from the end. // will fill it from the end.