diff --git a/flask_lambda.py b/flask_lambda.py index 2e23a16..66c894d 100644 --- a/flask_lambda.py +++ b/flask_lambda.py @@ -35,6 +35,7 @@ import six from werkzeug.wrappers import BaseRequest, BaseResponse +from elasticapm.contrib.flask import ElasticAPM __version__ = '0.0.4' @@ -217,6 +218,9 @@ def _call(self, event, context): class FlaskLambda(Flask): def __call__(self, event, context): + if not getattr(self, 'apm'): + self.apm = ElasticAPM(logging=False) + self.apm.init_app(self) return _call(self, event, context) @@ -226,4 +230,4 @@ def __init__(self, app): self.wsgi_app = app def __call__(self, event, context): - return _call(self, event, context) \ No newline at end of file + return _call(self, event, context) diff --git a/requirements.txt b/requirements.txt index f845e13..96f647c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,4 @@ Flask>=0.10 six +git+http://git.fenda.io/open/apm-agent-python.git@master +blinker