fix: cors

michaelfromyeg · Jun 18, 2023 · d7003db · d7003db
1 parent 69dddba
commit d7003db
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/docs/TODO.md b/docs/TODO.md
@@ -4,3 +4,4 @@
 - [ ] `CONTRIBUTING.md`
 - [ ] `CHANGELOG.md`
 - [ ] Remove "whoami" step from `deploy.yml`
+- [ ] Better CORS in server
diff --git a/server/main.py b/server/main.py
@@ -4,17 +4,22 @@
 
 import json
 import re
+import os
 
 import lyricsgenius
 from flask import Flask, jsonify, request
+from flask_cors import CORS
 from dotenv import load_dotenv
 
 from song import Song
 
-app = Flask("__main__")
+app = Flask(__name__)
+CORS(app)
 
 load_dotenv()
 
+IS_DEVELOPMENT = os.getenv("PYTHON_ENV") == "development"
+
 
 @app.route("/lyrics/<artist>/<song_name>", methods=["GET"])
 def get_lyrics(artist, song_name):
@@ -42,4 +47,4 @@ def find_rhymes():
     return json.dumps(song_instance.find_all_rhyme_clusters())
 
 
-app.run(debug=True)
+app.run(debug=IS_DEVELOPMENT)
diff --git a/server/requirements.txt b/server/requirements.txt
@@ -7,6 +7,7 @@ charset-normalizer==3.1.0
 click==8.1.3
 dill==0.3.6
 Flask==2.3.2
+Flask-Cors==3.0.10
 idna==3.4
 isort==5.12.0
 itsdangerous==2.1.2
@@ -24,6 +25,7 @@ pylint==2.17.4
 python-datamuse==2.0.1
 python-dotenv==1.0.0
 requests==2.31.0
+six==1.16.0
 soupsieve==2.4.1
 tomli==2.0.1
 tomlkit==0.11.8