142 lines
4.8 KiB
Python
142 lines
4.8 KiB
Python
from nonebot import on_command, get_plugin_config,logger
|
|
from nonebot.permission import SUPERUSER
|
|
from nonebot.rule import to_me
|
|
from nonebot.adapters.onebot.v11 import PrivateMessageEvent, GroupMessageEvent, MessageSegment
|
|
from nonebot.params import Depends
|
|
from .config import Config
|
|
from .utils import post, get_classes, post_vcode, get_log
|
|
import random
|
|
import asyncio
|
|
import time
|
|
|
|
plugin_config = get_plugin_config(Config)
|
|
|
|
|
|
help = on_command("咸鸭蛋",rule=to_me(),aliases={"apihelp", "sudhelp"},permission=SUPERUSER, priority=0, block=True)
|
|
@help.handle()
|
|
async def _():
|
|
await asyncio.sleep(random.uniform(2, 3))
|
|
await help.finish(plugin_config.HelpStr)
|
|
|
|
ddonline = on_command("在线人数",rule=to_me(),aliases={"ddonline", "ddop"}, priority=0, block=True)
|
|
@ddonline.handle()
|
|
async def _(event:PrivateMessageEvent):
|
|
id:str = str(event.user_id)
|
|
msg:str = await post("在线人数",id)
|
|
await asyncio.sleep(random.uniform(2, 3))
|
|
await ddonline.finish(msg)
|
|
|
|
addkami = on_command("添加卡密",rule=to_me(),aliases={"addkami", "akm"},permission=SUPERUSER, priority=0, block=True)
|
|
@addkami.handle()
|
|
async def _(event:PrivateMessageEvent):
|
|
id:str = str(event.user_id)
|
|
msg:str = event.get_plaintext()
|
|
if len(msg.split(' ')) != 3:
|
|
await asyncio.sleep(random.uniform(2, 3))
|
|
await ddonline.finish("参数不正确！")
|
|
|
|
classes:str = msg.split(' ')[1]
|
|
classes = get_classes(classes)
|
|
if classes == '':
|
|
await ddonline.finish("卡密类型不正确！")
|
|
|
|
kami:str = msg.split(' ')[2]
|
|
msg:str = await post("添加卡密",id,{
|
|
"classes":classes,
|
|
"kami":kami
|
|
})
|
|
await asyncio.sleep(random.uniform(2, 3))
|
|
await ddonline.finish(msg)
|
|
|
|
createkami = on_command("生成卡密",rule=to_me(),aliases={"createkami", "ckm"},permission=SUPERUSER, priority=0, block=True)
|
|
@createkami.handle()
|
|
async def _(event:PrivateMessageEvent):
|
|
id:str = str(event.user_id)
|
|
msg:str = event.get_plaintext()
|
|
if len(msg.split(' ')) != 2:
|
|
await asyncio.sleep(random.uniform(2, 3))
|
|
await ddonline.finish("参数不正确！")
|
|
|
|
classes:str = msg.split(' ')[1]
|
|
classes = get_classes(classes)
|
|
if classes == '':
|
|
await ddonline.finish("卡密类型不正确！")
|
|
|
|
msg:str = await post("生成卡密",id,{
|
|
"classes":classes
|
|
})
|
|
await asyncio.sleep(random.uniform(2, 3))
|
|
await ddonline.finish(msg)
|
|
|
|
addviptime = on_command("用户加时",rule=to_me(),aliases={"addviptime", "avt"},permission=SUPERUSER, priority=0, block=True)
|
|
@addviptime.handle()
|
|
async def _(event:PrivateMessageEvent):
|
|
id:str = str(event.user_id)
|
|
msg:str = event.get_plaintext()
|
|
if len(msg.split(' ')) != 3:
|
|
await asyncio.sleep(random.uniform(2, 3))
|
|
await ddonline.finish("参数不正确！")
|
|
|
|
username:str = msg.split(' ')[1]
|
|
classes:str = msg.split(' ')[2]
|
|
classes = get_classes(classes)
|
|
if classes == '':
|
|
await ddonline.finish("卡密类型不正确！")
|
|
|
|
msg:str = await post("用户加时",id,{
|
|
"username":username,
|
|
"classes":classes
|
|
})
|
|
await asyncio.sleep(random.uniform(2, 3))
|
|
await ddonline.finish(msg)
|
|
|
|
generate_qq_vcode = on_command("绑定QQ",aliases={"bindqq", "绑定qq"}, priority=0, block=True)
|
|
|
|
# 添加用户使用时间记录字典
|
|
user_last_use_time = {}
|
|
|
|
@generate_qq_vcode.handle()
|
|
async def _(event: GroupMessageEvent): # GroupMessageEvent PrivateMessageEvent
|
|
# 检查是否来自指定群组
|
|
if event.group_id != 621016172:
|
|
return
|
|
# if event.user_id != 1424473282:
|
|
# return
|
|
|
|
id:str = str(event.user_id)
|
|
|
|
# 限流检查：检查用户上次使用时间
|
|
current_time = time.time()
|
|
if id in user_last_use_time:
|
|
time_diff = current_time - user_last_use_time[id]
|
|
if time_diff < 60: # 60秒内已使用过
|
|
await generate_qq_vcode.finish(f"请求过于频繁，请在{int(60 - time_diff)}秒后再试")
|
|
return
|
|
|
|
# 更新用户最后使用时间
|
|
user_last_use_time[id] = current_time
|
|
|
|
msg:str = await post_vcode(id)
|
|
await asyncio.sleep(random.uniform(2, 3))
|
|
# 在消息前添加@用户
|
|
at_user = MessageSegment.at(event.user_id)
|
|
await generate_qq_vcode.finish(at_user + " " + msg)
|
|
|
|
|
|
|
|
|
|
view_logs = on_command("查看日志",aliases={"logs", "查询日志"}, priority=0, block=True)
|
|
@view_logs.handle()
|
|
async def _(event:GroupMessageEvent): # GroupMessageEvent PrivateMessageEvent
|
|
# 检查是否来自指定群组
|
|
if event.group_id != 621016172:
|
|
return
|
|
# if event.user_id != 1424473282:
|
|
# return
|
|
|
|
id:str = str(event.user_id)
|
|
msg:str = await get_log(id)
|
|
await asyncio.sleep(random.uniform(2, 3))
|
|
# 在消息前添加@用户
|
|
at_user = MessageSegment.at(event.user_id)
|
|
await view_logs.finish(at_user + " " + msg) |