mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
from models.group_info import GroupInfo
|
|
from utils.manager import requests_manager
|
|
from utils.utils import get_bot
|
|
|
|
from ..auth import Depends, User, token_to_user
|
|
from ..config import *
|
|
|
|
|
|
@router.get("/webui/request", dependencies=[token_to_user()])
|
|
def _(type_: Optional[str]) -> Result:
|
|
req_data = requests_manager.get_data()
|
|
req_list = []
|
|
if type_ in ["group", "private"]:
|
|
req_data = req_data[type_]
|
|
for x in req_data:
|
|
req_data[x]["oid"] = x
|
|
req_list.append(RequestResult(**req_data[x]))
|
|
req_list.reverse()
|
|
return Result(code=200, data=req_list)
|
|
|
|
|
|
@router.delete("/webui/request", dependencies=[token_to_user()])
|
|
def _(type_: Optional[str]) -> Result:
|
|
"""
|
|
清空请求
|
|
:param type_: 类型
|
|
"""
|
|
requests_manager.clear(type_)
|
|
return Result(code=200)
|
|
|
|
|
|
@router.post("/webui/request", dependencies=[token_to_user()])
|
|
async def _(parma: RequestParma) -> Result:
|
|
"""
|
|
操作请求
|
|
:param parma: 参数
|
|
"""
|
|
result = "操作成功！"
|
|
flag = 3
|
|
if bot := get_bot():
|
|
if parma.handle == "approve":
|
|
if parma.type == "group":
|
|
if rid := requests_manager.get_group_id(parma.id):
|
|
# await GroupInfo.update_or_create(defaults={"group_flag": 1}, )
|
|
if group := await GroupInfo.filter(group_id=rid).first():
|
|
await group.update_or_create(group_flag=1)
|
|
else:
|
|
group_info = await bot.get_group_info(group_id=rid)
|
|
await GroupInfo.update_or_create(
|
|
group_id=group_info["group_id"],
|
|
defaults={
|
|
"group_name": group_info["group_name"],
|
|
"max_member_count": group_info["max_member_count"],
|
|
"member_count": group_info["member_count"],
|
|
"group_flag": 1,
|
|
},
|
|
)
|
|
flag = await requests_manager.approve(bot, parma.id, parma.type)
|
|
elif parma.handle == "refuse":
|
|
flag = await requests_manager.refused(bot, parma.id, parma.type)
|
|
elif parma.handle == "delete":
|
|
requests_manager.delete_request(parma.id, parma.type)
|
|
if parma.handle != "delete":
|
|
if flag == 1:
|
|
result = "该请求已失效"
|
|
requests_manager.delete_request(parma.id, parma.type)
|
|
elif flag == 2:
|
|
result = "未找到此Id"
|
|
return Result(code=200, data=result)
|