commit b97669ab9432146c1303dedfaa7b98f7ede3f8db
author Damien George <damien.p.george@gmail.com> Wed Jan 08 11:47:55 2014 +0000
committer Damien George <damien.p.george@gmail.com> Wed Jan 08 11:47:55 2014 +0000
tree 773cf5fe2a0cae7951751fd01cd7ca9c9c5fe6f5
parent 6d6bc9efc17c0fd4f28f935280db6d8dee100d4f

py: Improve __build_class__.

py/obj.c

diff --git a/py/obj.c b/py/obj.c
index 77580e1..7d0122c 100644
--- a/py/obj.c
+++ b/py/obj.c
@@ -13,6 +13,15 @@
 #include "runtime.h"
 #include "map.h"
 
+mp_obj_t mp_obj_get_type(mp_obj_t o_in) {
+    if (MP_OBJ_IS_SMALL_INT(o_in)) {
+        return (mp_obj_t)&int_type;
+    } else {
+        mp_obj_base_t *o = o_in;
+        return (mp_obj_t)o->type;
+    }
+}
+
 const char *mp_obj_get_type_str(mp_obj_t o_in) {
     if (MP_OBJ_IS_SMALL_INT(o_in)) {
         return "int";