Validate API version

1 files changed, 6 insertions, 0 deletions

diff --git a/api/main.py b/api/main.py
index 1e48ec3f..ba79d9ce 100755
--- a/api/main.py
+++ b/api/main.py
@@ -53,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():
@@ -94,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)