Merge pull request #92 from IndiciumSRL/master

Add support to multiple databases with aliases

Author: rozza

flask_mongoengine/__init__.py

@@ -44,19 +44,30 @@ def init_app(self, app):
                 'port': int(app.config.get('MONGODB_PORT', 0)) or None
             }
 
-        conn_settings = dict([(k.lower(), v) for k, v in conn_settings.items() if v])
+        if isinstance(conn_settings, list):
+            self.connection = {}
+            for conn in conn_settings:
+                conn = dict([(k.lower(), v) for k, v in conn.items() if v])
 
-        if 'replicaset' in conn_settings:
-            conn_settings['replicaSet'] = conn_settings['replicaset']
-            del conn_settings['replicaset']
+                if 'replicaset' in conn:
+                    conn['replicaSet'] = conn['replicaset']
+                    del conn['replicaset']
 
-        self.connection = mongoengine.connect(**conn_settings)
+                self.connection[conn.get('alias')] = mongoengine.connect(**conn)
+
+        else:
+            conn_settings = dict([(k.lower(), v) for k, v in conn_settings.items() if v])
+
+            if 'replicaset' in conn_settings:
+                conn_settings['replicaSet'] = conn_settings['replicaset']
+                del conn_settings['replicaset']
+
+            self.connection = mongoengine.connect(**conn_settings)
 
         app.extensions = getattr(app, 'extensions', {})
         app.extensions['mongoengine'] = self
         self.app = app
 
-
 class BaseQuerySet(QuerySet):
     """
     A base queryset with handy extras

tests/test_basic_app.py

@@ -52,7 +52,7 @@ def test_connection_kwargs(self):
         app.config['MONGODB_SETTINGS'] = {
             'DB': 'testing_tz_aware',
             'alias': 'tz_aware_true',
-            'TZ_AWARE': True,
+            'TZ_AWARE': True
         }
         app.config['TESTING'] = True
         db = MongoEngine()
@@ -66,6 +66,23 @@ def test_connection_kwargs(self):
         db.init_app(app)
         self.assertFalse(db.connection.tz_aware)
 
+    def test_connection_kwargs_as_list(self):
+        app = flask.Flask(__name__)
+        app.config['MONGODB_SETTINGS'] = [{
+            'DB': 'testing_tz_aware',
+            'alias': 'tz_aware_true',
+            'TZ_AWARE': True
+        }, {
+            'DB': 'testing_tz_aware_off',
+            'alias' : 'tz_aware_false',
+            'TZ_AWARE' : False
+        }]
+        app.config['TESTING'] = True
+        db = MongoEngine()
+        db.init_app(app)
+        self.assertTrue(db.connection['tz_aware_true'].tz_aware)
+        self.assertFalse(db.connection['tz_aware_false'].tz_aware)
+
     def test_with_id(self):
         c = self.app.test_client()
         resp = c.get('/show/38783728378090/')