Make the api mandatory in RESTfulPrometheusMetrics.for_app_factory

Fixes #62
rycus86 · Jul 22, 2020 · a3ed550 · a3ed550
1 parent 9cfbbd6
commit a3ed550
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/examples/restful-return-none/server.py b/examples/restful-return-none/server.py
@@ -5,7 +5,7 @@
 app = Flask(__name__)
 restful_api = Api(app)
 
-metrics = RESTfulPrometheusMetrics(app, restful_api)
+metrics = RESTfulPrometheusMetrics.for_app_factory(restful_api)
 
 
 class Test(Resource):
@@ -26,4 +26,5 @@ def get():
 
 
 if __name__ == '__main__':
+    metrics.init_app(app)
     app.run('0.0.0.0', 4000)
diff --git a/prometheus_flask_exporter/__init__.py b/prometheus_flask_exporter/__init__.py
@@ -818,6 +818,10 @@ def __init__(self, app, api, **kwargs):
             kwargs['response_converter'] = self._create_response_converter(api)
         super().__init__(app, **kwargs)
 
+    @classmethod
+    def for_app_factory(cls, api, **kwargs):
+        return cls(app=None, api=api, **kwargs)
+
     @staticmethod
     def _create_response_converter(api):
         def _make_response(response):