From c66aa3baf47f6b034373612021e381718ff77a40 Mon Sep 17 00:00:00 2001
From: katboi01 <katboiofficial@gmail.com>
Date: Thu, 2 Jan 2025 13:17:17 +0100
Subject: [PATCH] CORS test

---
 app.py                                  | 2 ++
 modules/Archive/endpoints/set_action.py | 8 ++++++--
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/app.py b/app.py
index 3369474..f09584b 100644
--- a/app.py
+++ b/app.py
@@ -1,4 +1,5 @@
 from flask import Flask
+from flask_cors import CORS
 from flask_restful import Api
 
 from modules.KF3.database import Database as KF3DB
@@ -6,6 +7,7 @@ from modules.Kingdom.database import Database as KFKDB
 from modules.Archive.database import Database as KFADB
 
 app = Flask(__name__)
+CORS(app, support_credentials=True)
 app.config['JSON_AS_ASCII'] = False
 app.databases = {}
 
diff --git a/modules/Archive/endpoints/set_action.py b/modules/Archive/endpoints/set_action.py
index 1073faa..35132a0 100644
--- a/modules/Archive/endpoints/set_action.py
+++ b/modules/Archive/endpoints/set_action.py
@@ -1,5 +1,6 @@
 from __future__ import annotations
 import json
+from flask_cors import cross_origin
 from flask_restful import Resource
 from flask import current_app as app, jsonify, request
 from typing import TYPE_CHECKING
@@ -8,11 +9,12 @@ if TYPE_CHECKING:
     from modules.Archive.database import Database
 
 class Archive_SetAction(Resource):
+    @cross_origin(supports_credentials=True)
     def post(self):
-        data = request.json
+        data = request.get_json()
 
         try:
-            bypass = "allow_override" in data and data["allow_override"]
+            bypass = ("allow_override" in data) and data["allow_override"]
             action = data["action_taken"]
             if "id_str" in data:
                 id = int(data["id_str"])
@@ -23,6 +25,8 @@ class Archive_SetAction(Resource):
         except Exception as e:
             print(e)
             response = app.response_class(response=e, status=400)
+            response.headers.add("Access-Control-Allow-Origin", "*")
+            return response
 
         db : Database = app.databases["Archive"]
         query = f"UPDATE x_posts SET action_taken = {action} WHERE id = {id}"