additional archive endpoints

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/<id>")
         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):

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()

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   

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:

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