store app instance in class

Author: TennyZhuang

flask_mongoengine/__init__.py

@@ -104,6 +104,7 @@ class MongoEngine(object):
     def __init__(self, app=None, config=None):
         _include_mongoengine(self)
 
+        self.app = None
         self.Document = Document
         self.DynamicDocument = DynamicDocument
 
@@ -115,6 +116,8 @@ def init_app(self, app, config=None):
         if not app or not isinstance(app, Flask):
             raise Exception('Invalid Flask application instance')
 
+        self.app = app
+
         app.extensions = getattr(app, 'extensions', {})
 
         # Make documents JSON serializable