Added CORS library

requirements.txt

@@ -1,6 +1,7 @@
 django-openid-auth==0.17
-Flask-OpenID==1.3.1
+Flask-Cors==5.0.0
 Flask-Migrate==4.0.7
+Flask-OpenID==1.3.1
 Flask-Pydantic==0.12.0
 Flask-SQLAlchemy==3.1.1
 gunicorn[gevent]==23.0.0

webapp/__init__.py

@@ -2,6 +2,7 @@
 import os
 
 import flask
+from flask_cors import CORS
 from werkzeug.debug import DebuggedApplication
 from werkzeug.middleware.proxy_fix import ProxyFix
 
@@ -24,6 +25,9 @@ def create_app():
     app.config.from_pyfile("settings.py")
     app.context_processor(base_context)
 
+    # Add CORS headers
+    CORS(app, origins=["login.ubuntu.com"])
+
     # Initialize database
     init_db(app)
 

webapp/sso.py

@@ -47,12 +47,6 @@ def logout():
 
         return flask.redirect("/login")
 
-    # Allow CORS
-    @app.after_request
-    def after_request_func(response):
-        response.headers["Access-Control-Allow-Origin"] = SSO_LOGIN_URL
-        return response
-
 
 def login_required(func):
     """