Refactor python2js

src/python2js.c

@@ -118,15 +118,18 @@ int
 _python2js_cache(PyObject* x, PyObject* map);
 
 static int
-_python2js(PyObject* x, PyObject* map)
+_python2js_float(PyObject* x)
+{
+  double x_double = PyFloat_AsDouble(x);
+  if (x_double == -1.0 && PyErr_Occurred()) {
+    return HW_ERROR;
+  }
+  return hiwire_double(x_double);
+}
+
+static int
+_python2js_long(PyObject* x)
 {
-  if (x == Py_None) {
-    return hiwire_undefined();
-  } else if (x == Py_True) {
-    return hiwire_true();
-  } else if (x == Py_False) {
-    return hiwire_false();
-  } else if (PyLong_Check(x)) {
   int overflow;
   long x_long = PyLong_AsLongAndOverflow(x, &overflow);
   if (x_long == -1) {
@@ -135,24 +138,17 @@ _python2js(PyObject* x, PyObject* map)
       if (py_float == NULL) {
         return HW_ERROR;
       }
-        double x_double = PyFloat_AsDouble(py_float);
-        Py_DECREF(py_float);
-        if (x_double == -1.0 && PyErr_Occurred()) {
-          return HW_ERROR;
-        }
-        return hiwire_double(x_double);
+      return _python2js_float(py_float);
     } else if (PyErr_Occurred()) {
       return HW_ERROR;
     }
   }
   return hiwire_int(x_long);
-  } else if (PyFloat_Check(x)) {
-    double x_double = PyFloat_AsDouble(x);
-    if (x_double == -1.0 && PyErr_Occurred()) {
-      return HW_ERROR;
-    }
-    return hiwire_double(x_double);
-  } else if (PyUnicode_Check(x)) {
+}
+
+static int
+_python2js_unicode(PyObject* x)
+{
   int kind = PyUnicode_KIND(x);
   int data = (int)PyUnicode_DATA(x);
   int length = (int)PyUnicode_GET_LENGTH(x);
@@ -167,16 +163,22 @@ _python2js(PyObject* x, PyObject* map)
       PyErr_SetString(PyExc_ValueError, "Unknown Unicode KIND");
       return HW_ERROR;
   }
-  } else if (PyBytes_Check(x)) {
+}
+
+static int
+_python2js_bytes(PyObject* x)
+{
   char* x_buff;
   Py_ssize_t length;
   if (PyBytes_AsStringAndSize(x, &x_buff, &length)) {
     return HW_ERROR;
   }
   return hiwire_bytes((int)(void*)x_buff, length);
-  } else if (JsProxy_Check(x)) {
-    return JsProxy_AsJs(x);
-  } else if (PyList_Check(x) || is_type_name(x, "<class 'numpy.ndarray'>")) {
+}
+
+static int
+_python2js_sequence(PyObject* x, PyObject* map)
+{
   int jsarray = hiwire_array();
   if (_python2js_add_to_cache(map, x, jsarray)) {
     hiwire_decref(jsarray);
@@ -210,7 +212,11 @@ _python2js(PyObject* x, PyObject* map)
     return HW_ERROR;
   }
   return jsarray;
-  } else if (PyDict_Check(x)) {
+}
+
+static int
+_python2js_dict(PyObject* x, PyObject* map)
+{
   int jsdict = hiwire_object();
   if (_python2js_add_to_cache(map, x, jsdict)) {
     hiwire_decref(jsdict);
@@ -241,6 +247,32 @@ _python2js(PyObject* x, PyObject* map)
     return HW_ERROR;
   }
   return jsdict;
+}
+
+static int
+_python2js(PyObject* x, PyObject* map)
+{
+  if (x == Py_None) {
+    return hiwire_undefined();
+  } else if (x == Py_True) {
+    return hiwire_true();
+  } else if (x == Py_False) {
+    return hiwire_false();
+  } else if (PyLong_Check(x)) {
+    return _python2js_long(x);
+  } else if (PyFloat_Check(x)) {
+    return _python2js_float(x);
+  } else if (PyUnicode_Check(x)) {
+    return _python2js_unicode(x);
+  } else if (PyBytes_Check(x)) {
+    return _python2js_bytes(x);
+  } else if (JsProxy_Check(x)) {
+    return JsProxy_AsJs(x);
+  } else if (PyList_Check(x) || PyTuple_Check(x) ||
+             is_type_name(x, "<class 'numpy.ndarray'>")) {
+    return _python2js_sequence(x, map);
+  } else if (PyDict_Check(x)) {
+    return _python2js_dict(x, map);
   } else {
     Py_INCREF(x);
     return pyproxy_new((int)x);
@@ -265,25 +297,21 @@ _python2js_add_to_cache(PyObject* map, PyObject* pyparent, int jsparent)
   /* Use the pointer converted to an int so cache is by identity, not hash */
   PyObject* pyparentid = PyLong_FromSize_t((size_t)pyparent);
   PyObject* jsparentid = PyLong_FromLong(jsparent);
-  if (PyDict_SetItem(map, pyparentid, jsparentid)) {
+  int result = PyDict_SetItem(map, pyparentid, jsparentid);
   Py_DECREF(pyparentid);
   Py_DECREF(jsparentid);
-    return HW_ERROR;
-  }
-  Py_DECREF(pyparentid);
-  Py_DECREF(jsparentid);
-  return 0;
+
+  return result ? HW_ERROR : 0;
 }
 
 int
 _python2js_remove_from_cache(PyObject* map, PyObject* pyparent)
 {
   PyObject* pyparentid = PyLong_FromSize_t((size_t)pyparent);
-  if (PyDict_DelItem(map, pyparentid)) {
-    return HW_ERROR;
-  }
+  int result = PyDict_DelItem(map, pyparentid);
   Py_DECREF(pyparentid);
-  return 0;
+
+  return result ? HW_ERROR : 0;
 }
 
 int