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

py: Implement __delitem__ method for classes.

Browse Source
This commit is contained in:
Paul Sokolovsky 2014-04-15 01:20:52 +03:00
parent bc9ec5002d
commit 75ffcaeace
2 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
py/objtype.c
View File

@ -309,16 +309,20 @@ STATIC bool class_store_attr(mp_obj_t self_in, qstr attr, mp_obj_t value) {
}
bool class_store_item(mp_obj_t self_in, mp_obj_t index, mp_obj_t value) {
mp_obj_class_t *self = self_in;
mp_obj_t member;
uint meth_args;
if (value == MP_OBJ_NULL) {
// delete item
// TODO implement me!
return false;
member = mp_obj_class_lookup(self->base.type, MP_QSTR___delitem__);
meth_args = 2;
} else {
member = mp_obj_class_lookup(self->base.type, MP_QSTR___setitem__);
meth_args = 3;
}
mp_obj_class_t *self = self_in;
mp_obj_t member = mp_obj_class_lookup(self->base.type, MP_QSTR___setitem__);
if (member != MP_OBJ_NULL) {
mp_obj_t args[3] = {self_in, index, value};
mp_call_function_n_kw(member, 3, 0, args);
mp_call_function_n_kw(member, meth_args, 0, args);
return true;
} else {
return false;

6
tests/basics/class_item.py
View File

@ -1,4 +1,4 @@
# test class with __getitem__ and __setitem__ methods
# test class with __getitem__, __setitem__, __delitem__ methods
class C:
def __getitem__(self, item):
@ -8,6 +8,10 @@ class C:
def __setitem__(self, item, value):
print('set', item, value)
def __delitem__(self, item):
print('del', item)
c = C()
print(c[1])
c[1] = 2
del c[3]