katboi01
/
KemoFureApi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added account post count endpoint

This commit is contained in:
katboi01 2025-01-12 10:27:10 +01:00
parent 6b0de05a80
commit 50559223ed
8 changed files with 106 additions and 96 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

80
modules/Archive/database.py
View File

@ -1,9 +1,13 @@
import json
from flask import Flask
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
from .endpoints.get_posts_count import Archive_GetPostsCount
from .endpoints.query import Query
from .endpoints.post import GetPost
from .endpoints.posts import GetPosts
from .endpoints.set_action import SetAction
from .endpoints.posts_count import GetPostsCount
from .endpoints.account_stats import AccountStats
def get_post_dictionary(id, error_id, action_taken, text, files, date, x_handle, x_rating):
if files == "" or files is None:
@ -19,22 +23,50 @@ def get_post_dictionary(id, error_id, action_taken, text, files, date, x_handle,
class Database:
db : DatabaseController = None
app : Flask = None
def __init__(self, api) -> None:
app = api.app
if "Archive" in app.databases:
del app.databases["Archive"]
self.app = api.app
if "Archive" in self.app.databases:
del self.app.databases["Archive"]
self.reload_data()
app.databases["Archive"] = self
self.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_GetPostsCount, "/Archive/GetPosts/Count")
api.add_resource(Archive_SetAction, "/Archive/SetAction")
api.add_resource(Query, "/Archive/Query")
api.add_resource(GetPost, "/Archive/GetPost/<id>")
api.add_resource(GetPosts, "/Archive/GetPosts")
api.add_resource(GetPostsCount, "/Archive/GetPosts/Count")
api.add_resource(AccountStats, "/Archive/AccountStats")
api.add_resource(SetAction, "/Archive/SetAction")
def get_accounts(self):
query = f'''
SELECT x_handle, id FROM accounts
ORDER BY x_handle ASC'''
results = self.db.run_query(query)
accounts = []
if results is not None:
accounts = [{"x_handle": result[0], "id": result[1]} for result in results]
return accounts
def get_account_stats(self):
query = '''SELECT x_handle,
SUM(IIF(x_posts.action_taken = 0, 1, 0)) as undecided,
SUM(IIF(x_posts.action_taken = 1, 1, 0)) as approved,
SUM(IIF(x_posts.action_taken = 2, 1, 0)) as deleted,
SUM(IIF(x_posts.action_taken = 3, 1, 0)) as hidden,
SUM(IIF(x_post_details.id is NULL, 1, 0)) as invalid
FROM x_posts
LEFT JOIN accounts on accounts.id = x_posts.account_id
LEFT JOIN x_post_details on x_post_details.id = x_posts.id
GROUP BY account_id
ORDER BY x_handle'''
results = self.db.run_query(query)
return [{"x_handle": result[0], "undecided": result[1], "approved": result[2], "deleted": result[3], "hidden": result[4], "invalid": result[5]} for result in results]
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
@ -99,5 +131,25 @@ class Database:
posts.append(get_post_dictionary(result[0], -1, result[1], result[2], result[3], result[4], result[5], result[6]))
return posts
def wrap_query_response(self, result, mode = "json"):
if result is None:
response = self.app.response_class(status=400)
else:
if mode == "json":
response = self.app.response_class(
response=json.dumps(result, ensure_ascii=False, indent=1),
status=200,
mimetype='application/json'
)
elif mode == "text":
response = self.app.response_class(
response=str(result),
status=200,
mimetype='text/plain'
)
response.headers.add("Access-Control-Allow-Origin", "*")
return response
def reload_data(self):
self.db = DatabaseController("/home/pi/python/Katbots/JapariArchive/database.db")

13
modules/Archive/endpoints/account_stats.py Normal file
View File

@ -0,0 +1,13 @@
from __future__ import annotations
from flask_restful import Resource
from flask import current_app as app
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from modules.Archive.database import Database
class AccountStats(Resource):
def get(self):
db : Database = app.databases["Archive"]
result = db.get_account_stats()
return db.wrap_query_response(result)

27
modules/Archive/endpoints/get_post.py
View File

@ -1,27 +0,0 @@
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

15
modules/Archive/endpoints/post.py Normal file
View File

@ -0,0 +1,15 @@
from __future__ import annotations
from flask_restful import Resource
from flask import current_app as app
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from modules.Archive.database import Database
class GetPost(Resource):
def get(self, id):
id = int(id)
db : Database = app.databases["Archive"]
result = db.get_post(id)
return db.wrap_query_response(result)

18
modules/Archive/endpoints/get_posts.py → modules/Archive/endpoints/posts.py
View File

@ -1,13 +1,12 @@
from __future__ import annotations
import json
from flask_restful import Resource
from flask import current_app as app, jsonify, request
from flask import current_app as app, request
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from modules.Archive.database import Database
class Archive_GetPosts(Resource):
class GetPosts(Resource):
def get(self):
db : Database = app.databases["Archive"]
try:
@ -53,15 +52,4 @@ class Archive_GetPosts(Resource):
offset = real_page * count
result = db.get_posts(count, artist, actions_taken=actions, last_id= -1, offset= offset, include_ratings= ratings, order=order)
if result is None:
response = app.response_class(status=400)
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
return db.wrap_query_response(result)

18
modules/Archive/endpoints/get_posts_count.py → modules/Archive/endpoints/posts_count.py
View File

@ -1,13 +1,12 @@
from __future__ import annotations
import json
from flask_restful import Resource
from flask import current_app as app, jsonify, request
from flask import current_app as app, request
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from modules.Archive.database import Database
class Archive_GetPostsCount(Resource):
class GetPostsCount(Resource):
def get(self):
db : Database = app.databases["Archive"]
try:
@ -40,15 +39,4 @@ class Archive_GetPostsCount(Resource):
ratings = ["SFW", "NSFW"]
result = db.get_posts_count(artist, actions_taken=actions, last_id= -1, include_ratings= ratings)
if result is None:
response = app.response_class(status=400)
else:
response = app.response_class(
response=str(result),
status=200,
mimetype='text/plain'
)
response.headers.add("Access-Control-Allow-Origin", "*")
return response
return db.wrap_query_response(result, mode="text")

17
modules/Archive/endpoints/query.py
View File

@ -1,13 +1,12 @@
from __future__ import annotations
import json
from flask_restful import Resource
from flask import current_app as app, jsonify, request
from flask import current_app as app, request
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from modules.Archive.database import Database
class Archive_Query(Resource):
class Query(Resource):
def post(self):
query = request.data.decode("utf-8")
@ -15,14 +14,4 @@ class Archive_Query(Resource):
result = db.db.run_query(query)
if result is None:
response = app.response_class(status=400)
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
return db.wrap_query_response(result)

14
modules/Archive/endpoints/set_action.py
View File

@ -1,13 +1,13 @@
from __future__ import annotations
import json
from flask_restful import Resource
from flask import current_app as app, jsonify, request
from flask import current_app as app, request
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from modules.Archive.database import Database
class Archive_SetAction(Resource):
class SetAction(Resource):
def post(self):
data = json.loads(request.data)
@ -32,12 +32,4 @@ class Archive_SetAction(Resource):
query += " AND action_taken = 0"
result = db.db.run_command(query)
response = app.response_class(
response=str(result),
status=200,
mimetype='text/plain'
)
response.headers.add("Access-Control-Allow-Origin", "*")
return response
return db.wrap_query_response(result, mode="text")