commit c1bef21920d7fa03484647f2c339f53663fe0180
author John R. Lenton <jlenton@gmail.com> Sat Jan 11 12:39:33 2014 +0000
committer John R. Lenton <jlenton@gmail.com> Sat Jan 11 12:39:33 2014 +0000
tree 859bd2c4a8c4cc40e53ce2403328a889abe072ac
parent cf11c961b4d602c8b77f7d67aa976cf3fbb026b5

Implemented support for `in` and `not in` operators.

py/objstr.c

diff --git a/py/objstr.c b/py/objstr.c
index ea4f5ea..eb8b4c4 100644
--- a/py/objstr.c
+++ b/py/objstr.c
@@ -85,6 +85,15 @@
                 return mp_obj_new_str(qstr_from_str_take(val, alloc_len));
             }
             break;
+        case RT_COMPARE_OP_IN:
+        case RT_COMPARE_OP_NOT_IN:
+            /* NOTE `a in b` is `b.__contains__(a)` */
+            if (MP_OBJ_IS_TYPE(rhs_in, &str_type)) {
+                const char *rhs_str = qstr_str(((mp_obj_str_t*)rhs_in)->qstr);
+                /* FIXME \0 in strs */
+                return MP_BOOL((op == RT_COMPARE_OP_IN) ^ (strstr(lhs_str, rhs_str) == NULL));
+            }
+            break;
     }
 
     return MP_OBJ_NULL; // op not supported