PagerMaid-Pyro/pagermaid/web/api/ignore_groups.py

from fastapi import APIRouter
from starlette.responses import JSONResponse

from pagermaid.common.ignore import ignore_groups_manager, get_group_list
from pagermaid.web.api import authentication

route = APIRouter()


@route.get(
    "/get_ignore_group_list",
    response_class=JSONResponse,
    dependencies=[authentication()],
)
async def get_ignore_group_list():
    try:
        groups = []
        for data in await get_group_list():
            data["status"] = ignore_groups_manager.check_id(data["id"])
            groups.append(data)
        return {"status": 0, "msg": "ok", "data": {"groups": groups}}
    except BaseException:
        return {"status": -100, "msg": "获取群组列表失败"}


@route.post(
    "/set_ignore_group_status",
    response_class=JSONResponse,
    dependencies=[authentication()],
)
async def set_ignore_group_status(data: dict):
    cid: int = data.get("id")
    status: bool = data.get("status")
    if status:
        ignore_groups_manager.add_id(cid)
    else:
        ignore_groups_manager.del_id(cid)
    return {"status": 0, "msg": f'成功{"忽略" if status else "取消忽略"} {cid}'}


@route.post(
    "/clear_ignore_group", response_class=JSONResponse, dependencies=[authentication()]
)
async def clear_ignore_group():
    ignore_groups_manager.clear_subs()
    return {"status": 0, "msg": "成功清空忽略列表"}