py/objtype: Add __dict__ attribute for class objects.

pi-anl · pi-anl · commit cb925cdd5178 · 2020-04-30T12:40:05.000+10:00
diff --git a/py/obj.h b/py/obj.h
@@ -895,6 +895,7 @@ size_t mp_obj_dict_len(mp_obj_t self_in);
 mp_obj_t mp_obj_dict_get(mp_obj_t self_in, mp_obj_t index);
 mp_obj_t mp_obj_dict_store(mp_obj_t self_in, mp_obj_t key, mp_obj_t value);
 mp_obj_t mp_obj_dict_delete(mp_obj_t self_in, mp_obj_t key);
+mp_obj_t mp_obj_dict_copy(mp_obj_t self_in);
 static inline mp_map_t *mp_obj_dict_get_map(mp_obj_t dict) {
     return &((mp_obj_dict_t *)MP_OBJ_TO_PTR(dict))->map;
 }
diff --git a/py/objdict.c b/py/objdict.c
@@ -227,7 +227,7 @@ STATIC mp_obj_t dict_clear(mp_obj_t self_in) {
 }
 STATIC MP_DEFINE_CONST_FUN_OBJ_1(dict_clear_obj, dict_clear);
 
-STATIC mp_obj_t dict_copy(mp_obj_t self_in) {
+mp_obj_t mp_obj_dict_copy(mp_obj_t self_in) {
     mp_check_self(mp_obj_is_dict_type(self_in));
     mp_obj_dict_t *self = MP_OBJ_TO_PTR(self_in);
     mp_obj_t other_out = mp_obj_new_dict(self->map.alloc);
@@ -240,7 +240,7 @@ STATIC mp_obj_t dict_copy(mp_obj_t self_in) {
     memcpy(other->map.table, self->map.table, self->map.alloc * sizeof(mp_map_elem_t));
     return other_out;
 }
-STATIC MP_DEFINE_CONST_FUN_OBJ_1(dict_copy_obj, dict_copy);
+STATIC MP_DEFINE_CONST_FUN_OBJ_1(dict_copy_obj, mp_obj_dict_copy);
 
 #if MICROPY_PY_BUILTINS_DICT_FROMKEYS
 // this is a classmethod
diff --git a/py/objtype.c b/py/objtype.c
@@ -1013,6 +1013,12 @@ STATIC void type_attr(mp_obj_t self_in, qstr attr, mp_obj_t *dest) {
             dest[0] = MP_OBJ_NEW_QSTR(self->name);
             return;
         }
+        #if MICROPY_CPYTHON_COMPAT
+        if (attr == MP_QSTR___dict__) {
+            dest[0] = mp_obj_dict_copy(MP_OBJ_FROM_PTR(self->locals_dict));
+            return;
+        }
+        #endif
         if (attr == MP_QSTR___bases__) {
             if (self == &mp_type_object) {
                 dest[0] = mp_const_empty_tuple;

Original file line number	Diff line number	Diff line change
`@@ -895,6 +895,7 @@ size_t mp_obj_dict_len(mp_obj_t self_in);`
`895`	`895`	`mp_obj_t mp_obj_dict_get(mp_obj_t self_in, mp_obj_t index);`
`896`	`896`	`mp_obj_t mp_obj_dict_store(mp_obj_t self_in, mp_obj_t key, mp_obj_t value);`
`897`	`897`	`mp_obj_t mp_obj_dict_delete(mp_obj_t self_in, mp_obj_t key);`
	`898`	`+mp_obj_t mp_obj_dict_copy(mp_obj_t self_in);`
`898`	`899`	`static inline mp_map_t *mp_obj_dict_get_map(mp_obj_t dict) {`
`899`	`900`	`return &((mp_obj_dict_t *)MP_OBJ_TO_PTR(dict))->map;`
`900`	`901`	`}`