1 files changed, 18 insertions, 1 deletions
diff --git a/api/main.py b/api/main.py
index 387b516a..ba79d9ce 100755
--- a/api/main.py
+++ b/api/main.py
@@ -21,6 +21,16 @@ load_dotenv()
 
 TEST_MODE = "OPENDC_FLASK_TESTING" in os.environ
 
+# Setup Sentry if DSN is specified
+if 'SENTRY_DSN' in os.environ:
+    import sentry_sdk
+    from sentry_sdk.integrations.flask import FlaskIntegration
+
+    sentry_sdk.init(
+        integrations=[FlaskIntegration()],
+        traces_sample_rate=0.5
+    )
+
 # Set up database if not testing
 if not TEST_MODE:
     database.DB.initialize_database(
@@ -31,6 +41,7 @@ if not TEST_MODE:
 
 # Set up the core app
 FLASK_CORE_APP = Flask(__name__)
+FLASK_CORE_APP.testing = TEST_MODE
 FLASK_CORE_APP.config['SECRET_KEY'] = os.environ['OPENDC_FLASK_SECRET']
 FLASK_CORE_APP.json_encoder = JSONEncoder
 
@@ -42,6 +53,8 @@ compress.init_app(FLASK_CORE_APP)
 
 SOCKET_IO_CORE = flask_socketio.SocketIO(FLASK_CORE_APP, cors_allowed_origins="*")
 
+API_VERSIONS = {'v2'}
+
 
 @FLASK_CORE_APP.route('/tokensignin', methods=['POST'])
 def sign_in():
@@ -83,6 +96,10 @@ def sign_in():
 def api_call(version, endpoint_path):
     """Call an API endpoint directly over HTTP."""
 
+    # Check whether given version is valid
+    if version not in API_VERSIONS:
+        return jsonify(error='API version not found'), 404
+
     # Get path and parameters
     (path, path_parameters) = path_parser.parse(version, endpoint_path)
 
@@ -168,4 +185,4 @@ def _process_message(message):
 
 if __name__ == '__main__':
     print("Web server started on 8081")
-    SOCKET_IO_CORE.run(FLASK_CORE_APP, host='0.0.0.0', port=8081)
+    SOCKET_IO_CORE.run(FLASK_CORE_APP, host='0.0.0.0', port=8081, use_reloader=False)