Migrating to using factories and blueprints

Author: robottaway

tests/unuo_tests.py

@@ -1,14 +1,21 @@
 import unittest
 import json
+import logging
 
-from unuo.app import app
+from unuo.factories import default_factory
+
+
+logger = logging.getLogger('test')
+logging.basicConfig(level=logging.INFO)
 
 
 class UnuoTests(unittest.TestCase):
 
     def setUp(self):
+        app = default_factory()
         self.app = app.test_client()
 
     def test_profiles(self):
         rv = self.app.get('/profile')
         j = json.loads(rv.data)
+        self.assertEquals({u'builds': [u'mynewbuild']}, j)

unuo/app.py

@@ -1,50 +1,10 @@
 #!/usr/bin/env python
 
-from flask import Flask, jsonify, request
+from unuo.factories import default_factory
 
-from unuo.filebackend import post_build, run_build, get_build_profile, \
-    get_all_profiles
-from unuo.errors import ApiError
-
-
-app = Flask(__name__)
-
-
-@app.errorhandler(500)
-def error_ise(error):
-    return '{"status_code":500,"description":"oh noes!"}', 500
-
-
-@app.errorhandler(404)
-@app.errorhandler(400)
-@app.errorhandler(405)
-@app.errorhandler(ApiError)
-def error_json(error):
-    return '{"status_code":"%s","description":"%s"}' % (
-        error.code, error.description), error.code
-
-
-@app.route('/build/<name>', methods=['POST'])
-def build(name):
-    """Run given build."""
-    return run_build(name)
-
-
-@app.route('/profile', methods=['GET'])
-def list_build_profiles():
-    """Responsible for listing known builds."""
-    builds = get_all_profiles()
-    return jsonify({"builds": builds})
-
-
-@app.route('/profile/<name>', methods=['GET', 'POST'])
-def build_container(name):
-    """Responsible for creating/updating builds and launching them."""
-    if request.method == 'POST':
-        return post_build(name, request.json)
-    return jsonify(**get_build_profile(name))
+app = None
 
 
 if __name__ == '__main__':
-    app.config.from_object('unuo.config.DefaultConfig')
+    app = default_factory()
     app.run()

unuo/blueprints/__init__.py

@@ -0,0 +1,48 @@
+"""Blueprint for Docker related functions.
+
+Contains API methods for kicking off a build profile, adding a build profile,
+getting a listing of build profiles.
+"""
+from flask import Blueprint
+from flask import jsonify, request
+from unuo.errors import ApiError
+
+from unuo.filebackend import post_build, run_build, get_build_profile, \
+    get_all_profiles
+
+docker_bp = Blueprint('docker', __name__)
+
+
+#@docker_bp.errorhandler(500)
+#def error_ise(error):
+#    return '{"status_code":500,"description":"oh noes!"}', 500
+#
+
+@docker_bp.errorhandler(404)
+@docker_bp.errorhandler(400)
+@docker_bp.errorhandler(405)
+@docker_bp.errorhandler(ApiError)
+def error_json(error):
+    return '{"status_code":"%s","description":"%s"}' % (
+        error.code, error.description), error.code
+
+
+@docker_bp.route('/build/<name>', methods=['POST'])
+def build(name):
+    """Run given build."""
+    return run_build(name)
+
+
+@docker_bp.route('/profile', methods=['GET'])
+def list_build_profiles():
+    """Responsible for listing known builds."""
+    builds = get_all_profiles()
+    return jsonify({"builds": builds})
+
+
+@docker_bp.route('/profile/<name>', methods=['GET', 'POST'])
+def build_container(name):
+    """Responsible for creating/updating builds and launching them."""
+    if request.method == 'POST':
+        return post_build(name, request.json)
+    return jsonify(**get_build_profile(name))

unuo/blueprints/docker.py

@@ -0,0 +1,13 @@
+"""Flask factories for running the app and testing it.
+"""
+
+from flask import Flask
+
+from unuo.blueprints.docker import docker_bp
+
+
+def default_factory(config=None):
+    app = Flask(__name__)
+    app.config.from_object('unuo.config.DefaultConfig')
+    app.register_blueprint(docker_bp)
+    return app