🚚 Move flask_admin, mongo_profiler wrapper to example

eduflow · Oct 28, 2018 · edf7b5f · edf7b5f
1 parent fdbc32e
commit edf7b5f
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 44 deletions.
diff --git a/examples/flask_todo/app.py b/examples/flask_todo/app.py
@@ -24,4 +24,43 @@ def index():
     return app
 
 
+def setup_flask_mongo_profiler(app):
+    from flask_mongo_profiler.contrib.werkzeug.mongo import QueuedMongoCommandLogger
+    from flask_mongo_profiler.contrib.werkzeug.werkzeug_middleware import (
+        ProfilerMiddleware
+    )
+    import pymongo
+
+    mongo_logger = QueuedMongoCommandLogger()
+    pymongo.monitoring.register(mongo_logger)
+    app.wsgi_app = ProfilerMiddleware(
+        app.wsgi_app,
+        pymongo_logger=mongo_logger,
+        ignored_url_patterns=['admin', 'favicon.ico'],
+    )
+    return app
+
+
+def setup_flask_admin(app):
+    from flask_admin import Admin
+    from flask_mongo_profiler.contrib.flask_admin.views import (
+        profiling as profiling_views
+    )
+    from flask_mongo_profiler.contrib.mongoengine import profiling as profiling_models
+
+    admin = Admin(name='Peergrade', template_mode='bootstrap3')
+    admin.add_view(
+        profiling_views.ProfilingRequestView(
+            profiling_models.ProfilingRequest, category='Profiling', name='Requests'
+        )
+    )
+    admin.add_view(
+        profiling_views.ProfilingQueryView(
+            profiling_models.ProfilingQuery, category='Profiling', name='Queries'
+        )
+    )
+    admin.init_app(app)
+    return app
+
+
 app = create_app()
diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -3,57 +3,19 @@
 
 import pytest
 
-from examples.flask_todo.app import create_app
+from examples.flask_todo.app import (
+    create_app,
+    setup_flask_admin,
+    setup_flask_mongo_profiler,
+)
 from flask import url_for
 
 
-def setup_profiler(app):
-    from flask_mongo_profiler.contrib.werkzeug.mongo import (
-        LoggingMongoCommandLogger,
-        QueuedMongoCommandLogger,
-    )
-    from flask_mongo_profiler.contrib.werkzeug.werkzeug_middleware import (
-        ProfilerMiddleware
-    )
-    import pymongo
-
-    mongo_logger = QueuedMongoCommandLogger()
-    pymongo.monitoring.register(mongo_logger)
-    app.wsgi_app = ProfilerMiddleware(
-        app.wsgi_app,
-        pymongo_logger=mongo_logger,
-        ignored_url_patterns=['admin', 'favicon.ico'],
-    )
-    return app
-
-
-def setup_flask_admin(app):
-    from flask_admin import Admin
-    from flask_mongo_profiler.contrib.flask_admin.views import (
-        profiling as profiling_views
-    )
-    from flask_mongo_profiler.contrib.mongoengine import profiling as profiling_models
-
-    admin = Admin(name='Peergrade', template_mode='bootstrap3')
-    admin.add_view(
-        profiling_views.ProfilingRequestView(
-            profiling_models.ProfilingRequest, category='Profiling', name='Requests'
-        )
-    )
-    admin.add_view(
-        profiling_views.ProfilingQueryView(
-            profiling_models.ProfilingQuery, category='Profiling', name='Queries'
-        )
-    )
-    admin.init_app(app)
-    return app
-
-
 @pytest.fixture
 def app():
     app = create_app()
     app.debug = True
-    setup_profiler(app)
+    setup_flask_mongo_profiler(app)
     setup_flask_admin(app)
     return app