main.py

import sys
import asyncio
if sys.platform != "win32":
    import uvloop
    asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())

from fastapi import FastAPI, Response
from fastapi.middleware.cors import CORSMiddleware

from src.utils.asyncMySQL import AsyncMySQL


app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=["http://localhost:5173", "*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"]
)


@app.on_event("startup")
async def startup():
    global db
    db = AsyncMySQL()
    await db.init()


@app.on_event("shutdown")
async def shutdown():
    await db.close()


@app.get("/history/user")
async def get_history(uid: str, page: int = 1, isReverse: bool = False):

    res = await db.search("AID,CID,update_time", "history", f"UID={uid}",
                          f"{(page-1)*20},{page*20}",
                          f"update_time {'ASC' if isReverse else 'DESC'}")
    

    ret_list = [{
        "aid": item[0],
        "cid": item[1],
        "update_time":item[2].strftime("%Y-%m-%d %H:%M:%S")
    } for item in res]

    return ret_list


@app.get("/history/album")
async def get_history(uid: int, aid: int):

    res = await db.search("UID,CID,update_time", "history", f"UID={uid} and AID={aid}",
                          sort=f"update_time ASC")

    if res == ():
        return None

    ret_dict = {
        "uid": res[0][0],
        "cid": res[0][1],
        "update_time": res[0][2].strftime("%Y-%m-%d %H:%M:%S")
    }

    return ret_dict


@app.delete("/history")
async def delete_history(uid: str):
    await db.delete("history", f"UID={uid}")
    return Response(status_code=200)


@app.post("/record")
async def update_history(uid: str, aid: int, cid: int):

    res = await db.search("*", "history", f"UID={uid} and AID={aid}", "1", "update_time DESC")

    if(res == ()):
        await db.insert("history", "UID,AID,CID,update_time", f"{uid},{aid},{cid},NOW()")
    else:
        await db.update("history", f"CID={cid}, update_time=NOW()", f"UID={uid} and AID={aid}")

    return Response(status_code=201)


@app.get("/tags")
async def get_tags(uid: str):

    res = await db.search("tag", "tags", f"UID={uid}")

    return res


@app.post("/tag")
async def update_tag(uid: str, tag: str):
    if(await db.search("tag", "tags", f"UID={uid} AND tag='{tag}'")):
        return Response(status_code=200)
    else:
        await db.insert("tags", "UID,tag", f"{uid},'{tag}'")
    return Response(status_code=201)


@app.get("/comments")
async def get_comments(aid: str = None, uid: str = None, page: int = 1, isReverse: bool = False):

    if(aid and uid):
        return Response(status_code=400)
    elif(aid):
        res = await db.search("UID,AID,comment,comment_time", "comments", f"AID={aid}",
                              f"{(page-1)*20},{page*20}",
                              f"comment_time {'ASC' if isReverse else 'DESC'}")
    elif(uid):
        res = await db.search("UID,AID,comment,comment_time", "comments", f"UID={uid}",
                              f"{(page-1)*20},{page*20}",
                              f"comment_time {'ASC' if isReverse else 'DESC'}")
    else:
        return Response(status_code=400)

    ret_list = [{
        "uid": item[0],
        "aid": item[1],
        "comment": item[2],
        "comment_time":item[3].strftime("%Y-%m-%d %H:%M:%S")
    } for item in res]

    return ret_list