Merge pull request #8 from yoyicue/master

register extension with app
closeio · Aug 22, 2012 · 95a542c · 95a542c
2 parents d387ca6 + de4496d
commit 95a542c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/flask_mongoengine/__init__.py b/flask_mongoengine/__init__.py
@@ -46,6 +46,10 @@ def init_app(self, app):
 
         self.connection = mongoengine.connect(**conn_settings)
 
+        app.extensions = getattr(app, 'extensions', {})
+        app.extensions['mongoengine'] = self
+        self.app = app
+
 
 class BaseQuerySet(QuerySet):
     """