Adding replica set support for mongokit

ReplicaSetConnection added
Documentation, changelog updated

Author: liyanchang

AUTHORS

@@ -22,4 +22,6 @@ Great contributors:
   Ozgur <hinoglu **at** gmail.com>
     * patch to fix atomic save issue
 
+  Liyan Chang <liyan@filepicker.io>
+    * Replica Set Support
 

CHANGELOG

@@ -1,6 +1,12 @@
 Change Log
 ==========
 
+
+To Be Released:
+-----
+ * use MongoClient with MasterSlaveConnection
+
+
 v0.8.3
 ------
 

doc/tutorial.txt

@@ -55,9 +55,22 @@ Now, to open a connection to MongoDB::
 
     connection = Connection()
 
-This a wrapped version of pymongo's Connection_ and will attempt to connect to
-a MongoDB instance running locally. For more information on how to configure
-the Connection, see pymongo_'s documentation.
+This a wrapped version of pymongo's Connection_ and will attempt to connect to a MongoDB instance running locally.
+
+e.g. Speficying a host
+
+    connection = Connection(host="HOSTNAME", port=PORT)
+
+e.g. Specifiying a Replica Set host
+
+    from mongokit import ReplicaSetConnection
+
+    connection = ReplicaSetConnection(
+        host="HOSTNAME:PORT,HOSTNAME:PORT"
+        replicaset="REPLICA_SET_NAME",
+        read_preferences=pymongo.read_preferences.ReadPreferance.SECONDARY_PREFERRED)
+
+For more information on how to configure the Connection, see pymongo_'s documentation.
 
 Once you have an active connection, you need to register your the ``BlogPost``
 object::
@@ -140,4 +153,5 @@ as well::
     my first blog post
 
 pymongo_ makes it very easy to perform complex queries on your data so for more
-information, see the :doc:`query` documentation.
+information, see the :doc:`query` documentation.
+

mongokit/__init__.py

@@ -38,6 +38,7 @@
 from collection import Collection
 from connection import Connection, MongoClient
 from master_slave_connection import MasterSlaveConnection
+from replica_set_connection import ReplicaSetConnection
 from pymongo import ASCENDING as INDEX_ASCENDING,\
   DESCENDING as INDEX_DESCENDING,\
   ALL as INDEX_ALL,\

mongokit/replica_set_connection.py

@@ -0,0 +1,72 @@
+"""
+Replica Set integration with for MongoKit
+Liyan Chang, liyan@filepicker.io
+(same license as Mongokit)
+"""
+
+from pymongo.replica_set_connection import ReplicaSetConnection as PymongoReplicaSetConnection
+from mongokit.database import Database
+from mongokit.connection import CallableMixin, _iterables
+
+class ReplicaSetConnection(PymongoReplicaSetConnection):
+    """ Replica Set support for MongoKit """
+
+    def __init__(self, *args, **kwargs):
+        """ The ReplicaSetConnection is a wrapper around the
+            pymongo.replica_set_connection implementation.
+        """
+
+        self._databases = {}
+        self._registered_documents = {}
+
+        super(ReplicaSetConnection, self).__init__(*args, **kwargs)
+
+    """
+    All following code is duplicate code copied from connection.py
+    Any edit to this code should probably be propagated over to
+    connection.py as well as master_slave_connection.py
+    """
+
+    def register(self, obj_list):
+        decorator = None
+        if not isinstance(obj_list, _iterables):
+            # we assume that the user used this as a decorator
+            # using @register syntax or using conn.register(SomeDoc)
+            # we stock the class object in order to return it later
+            decorator = obj_list
+            obj_list = [obj_list]
+        # cleanup
+        for dbname, db in self._databases.items():
+            for colname, col in db._collections.items():
+                for docname, doc in col._documents.items():
+                    del col._documents[docname]
+                for obj_name in [obj.__name__ for obj in obj_list]:
+                    if obj_name in col._registered_documents:
+                        del col._registered_documents[obj_name]
+        # register
+        for obj in obj_list:
+            CallableDocument = type(
+              "Callable%s" % obj.__name__,
+              (obj, CallableMixin),
+              {"_obj_class":obj, "__repr__":object.__repr__}
+            )
+            self._registered_documents[obj.__name__] = CallableDocument
+        # if the class object is stored, it means the user used a decorator and
+        # we must return the class object
+        if decorator is not None:
+            return decorator
+
+    def __getattr__(self, key):
+        if key in self._registered_documents:
+            document = self._registered_documents[key]
+            try:
+                return getattr(self[document.__database__][document.__collection__], key)
+            except AttributeError:
+                raise AttributeError("%s: __collection__ attribute not found. "
+                  "You cannot specify the `__database__` attribute without "
+                  "the `__collection__` attribute" % key)
+        else:
+            if key not in self._databases:
+                self._databases[key] = Database(self, key)
+            return self._databases[key]
+