From 068b1bc3d0793a2fc3e5fc3cd8267d90e15d3b24 Mon Sep 17 00:00:00 2001
From: Hilton Bristow <hilton.bristow@gmail.com>
Date: Thu, 27 Mar 2014 11:34:20 +1000
Subject: [PATCH] More generic todict

---
 modules/matlab/generator/parse_tree.py | 33 ++++++++++++++++----------
 1 file changed, 21 insertions(+), 12 deletions(-)

diff --git a/modules/matlab/generator/parse_tree.py b/modules/matlab/generator/parse_tree.py
index 21c4899ff..bb1d45665 100644
--- a/modules/matlab/generator/parse_tree.py
+++ b/modules/matlab/generator/parse_tree.py
@@ -1,3 +1,4 @@
+import collections
 from textwrap import fill
 from filters import *
 
@@ -333,23 +334,31 @@ def constants(tree):
             for gen in constants(val):
                 yield gen
 
+def isstring(s):
+    """
+    Check if variable is string representable (regular/unicode/raw)
+    in a Python 2 and 3 compatible manner
+    """
+    try:
+        return isinstance(s, basestring)
+    except NameError:
+        return isinstance(s, str)
+
 def todict(obj, classkey=None):
     """
     Convert the ParseTree to a dictionary, stripping all objects of their
     methods and converting class names to strings
     """
+    if isstring(obj):
+        return obj
     if isinstance(obj, dict):
-        for k in obj.keys():
-            obj[k] = todict(obj[k], classkey)
+        obj.update((key, todict(val, classkey)) for key, val in obj.items())
         return obj
-    elif isinstance(obj, list):
-        return [todict(v, classkey) for v in obj]
-    elif hasattr(obj, "__dict__"):
-        data = dict([(key, todict(value, classkey))
-            for key, value in obj.__dict__.items()
-            if not callable(value) and not key.startswith('_')])
-        if classkey is not None and hasattr(obj, "__class__"):
+    if isinstance(obj, collections.Iterable):
+        return [todict(val, classkey) for val in obj]
+    if hasattr(obj, '__dict__'):
+        attrs = dict((key, todict(val, classkey)) for key, val in vars(obj).items())
+        if classkey is not None and hasattr(obj, '__class__'):
             data[classkey] = obj.__class__.__name__
-        return data
-    else:
-        return obj
+        return attrs
+    return obj