From 0489988947a3271c61cb6212c38fb5d1792d14c6 Mon Sep 17 00:00:00 2001
From: Daniel Dourvaris <dourvaris@gmail.com>
Date: Wed, 12 Aug 2015 15:32:30 +0300
Subject: [PATCH] Allow setting the resulting dict class.

---
 dictalchemy/utils.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/dictalchemy/utils.py b/dictalchemy/utils.py
index 1a5ca09..bdc3521 100644
--- a/dictalchemy/utils.py
+++ b/dictalchemy/utils.py
@@ -7,6 +7,7 @@
 from __future__ import absolute_import, division
 
 import copy
+
 from sqlalchemy import inspect
 from sqlalchemy.ext.associationproxy import _AssociationList
 
@@ -42,7 +43,8 @@ def arg_to_dict(arg):
 
 
 def asdict(model, exclude=None, exclude_underscore=None, exclude_pk=None,
-           follow=None, include=None, only=None, method='asdict', **kwargs):
+           follow=None, include=None, only=None, method='asdict',
+           DictClass=dict, **kwargs):
     """Get a dict from a model
 
     Using the `method` parameter makes it possible to have multiple methods
@@ -75,6 +77,7 @@ def asdict(model, exclude=None, exclude_underscore=None, exclude_pk=None,
     :param method: Name of the method that is currently called. This will be \
             the default method used in 'follow' unless another method is\
             set.
+    :param DictClass: Class to be used for resulting dict.
 
     :raises: :class:`dictalchemy.errors.MissingRelationError` \
             if `follow` contains a non-existent relationship.
@@ -115,7 +118,7 @@ def asdict(model, exclude=None, exclude_underscore=None, exclude_pk=None,
                                                      None)) or [])
         attrs = [k for k in columns + synonyms + include if k not in exclude]
 
-    data = dict([(k, getattr(model, k)) for k in attrs])
+    data = DictClass([(k, getattr(model, k)) for k in attrs])
 
     for (rel_key, orig_args) in follow.iteritems():
 
@@ -139,7 +142,7 @@ def asdict(model, exclude=None, exclude_underscore=None, exclude_pk=None,
                     rel_data.append(getattr(child, method)(**args))
                 else:
                     try:
-                        rel_data.append(dict(child))
+                        rel_data.append(DictClass(child))
                         # TypeError is for non-dictable children
                     except TypeError:
                         rel_data.append(copy.copy(child))
@@ -152,7 +155,7 @@ def asdict(model, exclude=None, exclude_underscore=None, exclude_pk=None,
                     rel_data[child_key] = getattr(child, method)(**args)
                 else:
                     try:
-                        rel_data[child_key] = dict(child)
+                        rel_data[child_key] = DictClass(child)
                     except ValueError:
                         rel_data[child_key] = copy.copy(child)
 
@@ -163,7 +166,7 @@ def asdict(model, exclude=None, exclude_underscore=None, exclude_pk=None,
                 if hasattr(child, method):
                     rel_data.append(getattr(child, method)(**args))
                 else:
-                    rel_data.append(dict(child))
+                    rel_data.append(DictClass(child))
 
         elif rel is None:
             rel_data = None