diff --git a/modules/Archive/database.py b/modules/Archive/database.py index 09c0b10..9352f5c 100644 --- a/modules/Archive/database.py +++ b/modules/Archive/database.py @@ -2,7 +2,21 @@ import json from .databaseController import DatabaseController from .endpoints.query import Archive_Query +from .endpoints.get_post import Archive_GetPost from .endpoints.get_posts import Archive_GetPosts +from .endpoints.set_action import Archive_SetAction + +def get_post_dictionary(id, error_id, action_taken, text, files, date, x_handle, x_rating): + if files == "" or files is None: + files = [] + else: + files = files.split(',') + dict = {"id": id, "id_str": str(id), "error_id": error_id, "action_taken": action_taken, "text": text, "files": files, "date": date, "handle": x_handle, "rating": x_rating} + if error_id == -1: + del(dict["error_id"]) + if action_taken == -1: + del(dict["action_taken"]) + return dict class Database: db : DatabaseController = None @@ -17,10 +31,26 @@ class Database: app.databases["Archive"] = self api.add_resource(Archive_Query, "/Archive/Query") + api.add_resource(Archive_GetPost, "/Archive/GetPost/") api.add_resource(Archive_GetPosts, "/Archive/GetPosts") + api.add_resource(Archive_SetAction, "/Archive/SetAction") + def get_post(self, id): + query = f'''SELECT x_posts.id, x_posts.error_id, x_posts.action_taken, x_post_details.text, x_post_details.files, x_post_details.date, accounts.x_handle, accounts.rating from x_posts + LEFT JOIN x_post_details + ON x_posts.id = x_post_details.id + LEFT JOIN accounts + ON x_posts.account_id = accounts.id WHERE x_posts.id = {id}''' + result = self.db.run_query(query) + if len(result) == 0: + return None + else: + #return most recent post + result = result[-1] + return get_post_dictionary(result[0], result[1], result[2], result[3], result[4], result[5], result[6], result[7]) + def get_posts(self, num_posts, actions_taken = [0, 1, 2, 3], last_id = -1, offset = 0): - num_posts = max(1, min(num_posts, 30)) + num_posts = max(1, min(num_posts, 100)) where_query = "WHERE x_post_details.id NOT NULL AND x_posts.error_id = 0" if last_id != -1: @@ -29,7 +59,7 @@ class Database: where_query += " AND (" + " OR ".join([f"x_posts.action_taken = {action}" for action in actions_taken]) + ")" query = f''' - SELECT x_posts.id, x_post_details.text, x_post_details.files, accounts.x_handle, x_post_details.date FROM x_posts + SELECT x_posts.id, x_posts.action_taken, x_post_details.text, x_post_details.files, x_post_details.date, accounts.x_handle, accounts.rating FROM x_posts LEFT JOIN x_post_details ON x_posts.id = x_post_details.id LEFT JOIN accounts @@ -42,7 +72,7 @@ class Database: posts = [] if result is not None: for result in self.db.run_query(query): - posts.append({"id": result[0], "id_str": str(result[0]), "text": result[1], "files": result[2].split(','), "handle": result[3], "date": result[4]}) + posts.append(get_post_dictionary(result[0], -1, result[1], result[2], result[3], result[4], result[5], result[6])) return posts def reload_data(self): diff --git a/modules/Archive/databaseController.py b/modules/Archive/databaseController.py index 91821e6..6b5466e 100644 --- a/modules/Archive/databaseController.py +++ b/modules/Archive/databaseController.py @@ -17,6 +17,16 @@ class DatabaseController: except Exception as e: print(e) return None + + def run_command(self, commnd): + try: + self.cursor.execute(commnd) + result = self.cursor.rowcount + self.conn.commit() + return result > 0 + except Exception as e: + print(e) + return False def close(self): self.conn.close() \ No newline at end of file diff --git a/modules/Archive/endpoints/get_post.py b/modules/Archive/endpoints/get_post.py new file mode 100644 index 0000000..405ba39 --- /dev/null +++ b/modules/Archive/endpoints/get_post.py @@ -0,0 +1,27 @@ +from __future__ import annotations +import json +from flask_restful import Resource +from flask import current_app as app, jsonify, request +from typing import TYPE_CHECKING + +if TYPE_CHECKING: + from modules.Archive.database import Database + +class Archive_GetPost(Resource): + def get(self, id): + id = int(id) + db : Database = app.databases["Archive"] + + result = db.get_post(id) + + if result is None: + response = app.response_class(status=404) + else: + response = app.response_class( + response=json.dumps(result, ensure_ascii=False, indent=1), + status=200, + mimetype='application/json' + ) + + response.headers.add("Access-Control-Allow-Origin", "*") + return response \ No newline at end of file diff --git a/modules/Archive/endpoints/get_posts.py b/modules/Archive/endpoints/get_posts.py index 60bcb18..ab546e0 100644 --- a/modules/Archive/endpoints/get_posts.py +++ b/modules/Archive/endpoints/get_posts.py @@ -11,7 +11,7 @@ class Archive_GetPosts(Resource): def get(self): db : Database = app.databases["Archive"] try: - count = int(request.args["count"]) if "count" in request.args else 10 + count = int(request.args["count"]) if "count" in request.args else 20 page = int(request.args["page"]) if "page" in request.args else 1 last_id = int(request.args["last_id"]) if "last_id" in request.args else -1 except: diff --git a/modules/Archive/endpoints/set_action.py b/modules/Archive/endpoints/set_action.py new file mode 100644 index 0000000..1073faa --- /dev/null +++ b/modules/Archive/endpoints/set_action.py @@ -0,0 +1,41 @@ +from __future__ import annotations +import json +from flask_restful import Resource +from flask import current_app as app, jsonify, request +from typing import TYPE_CHECKING + +if TYPE_CHECKING: + from modules.Archive.database import Database + +class Archive_SetAction(Resource): + def post(self): + data = request.json + + try: + bypass = "allow_override" in data and data["allow_override"] + action = data["action_taken"] + if "id_str" in data: + id = int(data["id_str"]) + elif "id" in data: + id = data["id"] + else: + raise Exception("no id (int) or id_str (str) specified") + except Exception as e: + print(e) + response = app.response_class(response=e, status=400) + + db : Database = app.databases["Archive"] + query = f"UPDATE x_posts SET action_taken = {action} WHERE id = {id}" + if not bypass: + query += " AND action_taken = 0" + + result = db.db.run_command(query) + + response = app.response_class( + response=result, + status=200, + mimetype='application/json' + ) + + response.headers.add("Access-Control-Allow-Origin", "*") + return response \ No newline at end of file