Custom serializer in connection object; default is msgpack

[mojodb.git] / connection.py

diff --git a/connection.py b/connection.py
index add83e3b116c881979c1eaf0ec4369f2dc441975..7623047ca1bd0634081a1f0ba8c696e2af85e140 100644 (file)
--- a/connection.py
+++ b/connection.py
@@ -29,8 +29,12 @@ class Connection(object):
     Constraint = dbutils.Constraint
     Literal = dbutils.Literal
 
-    def __init__(self, *args, **kwargs):
-        self._db_con = None
+    def __init__(self, serializer=None, *args, **kwargs):
+        if serializer is None:
+            import msgpack
+            self.serializer = msgpack
+        else:
+            self.serializer = serializer
 
     def __getattr__(self, db_name):
         return Database(self, db_name)