diff --git a/flask_mongoengine/__init__.py b/flask_mongoengine/__init__.py index 6c15894e..93ca80ba 100644 --- a/flask_mongoengine/__init__.py +++ b/flask_mongoengine/__init__.py @@ -99,6 +99,7 @@ class MongoEngine(object): def __init__(self, app=None, config=None): _include_mongoengine(self) + self.app = None self.Document = Document self.DynamicDocument = DynamicDocument @@ -109,6 +110,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