K2cr2O1
351ed0b063
移除code_py.py中py命令的ADMIN权限限制，使其对所有用户可用 在jrcd.py中为handle_jrcd和handle_bbcd命令添加对特定事件ID(831797331)的过滤
189 lines
5.7 KiB
Python
189 lines
5.7 KiB
Python
"""
|
||
今日人品插件
|
||
|
||
提供 /jrcd 和 /bbcd 指令，用于娱乐。
|
||
"""
|
||
|
||
import random
|
||
from datetime import datetime
|
||
|
||
from core.bot import Bot
|
||
from core.managers.command_manager import matcher
|
||
from core.managers.redis_manager import redis_manager
|
||
from core.utils.executor import run_in_thread_pool
|
||
from models.events.message import MessageEvent, MessageSegment
|
||
from core.utils.logger import logger
|
||
|
||
|
||
__plugin_meta__ = {
|
||
"name": "jrcd",
|
||
"description": "来看看你的长度吧！",
|
||
"usage": "/jrcd\n/bbcd [@某人]",
|
||
}
|
||
|
||
# jrcd
|
||
JRCDMSG_1 = [
|
||
"今天的长度是%scm，可以让我一口吃掉吗罒ω罒",
|
||
"今天的长度是%scm，啥啊？怎么这么小啊？(*°ｰ°)v",
|
||
"今天的长度是%scm，什么嘛，原来是可爱的小豆丁呀(*°ｰ°)v",
|
||
]
|
||
JRCDMSG_2 = [
|
||
"今天的长度是%scm，还行，也不是不能接受(๑´ㅂ´๑)",
|
||
"今天的长度是%scm，小老弟不错啊，和哥哥一起玩会儿吗(〃∇〃)",
|
||
"今天的长度是%scm，也许我们今晚能做很多很多事情呢(〃∇〃)",
|
||
]
|
||
JRCDMSG_3 = [
|
||
"今天的长度是%scm，哦豁？听说你很勇哦？(✧◡✧)",
|
||
"今天的长度是%scm，嘶哈嘶哈(((o(*°▽°*)o)))...",
|
||
"今天的长度是%scm，我靠，让哥哥爽一-爽吧！(((o(*°▽°*)o)))...",
|
||
"今天的长度是%scm，单是看到哥哥的长度就....(〃w〃)",
|
||
]
|
||
|
||
# bbcd long
|
||
BBCDMSG1 = ["还行,可以尝试一下(๑‾ ꇴ ‾๑)"]
|
||
BBCDMSG2 = ["不错的成绩,努力一下或许可以让他受孕哦..(〃w〃)"]
|
||
BBCDMSG3 = ["好猛,试试强制让他受孕吧！！！(((o(*°▽°*)o)))"]
|
||
|
||
# bbcd short
|
||
BBCDMSG4 = ["差的不多,富贵险中求一下(*°ｰ°)v?"]
|
||
BBCDMSG5 = ["还行,可以尝试一下(๑‾ ꇴ ‾๑)"]
|
||
BBCDMSG6 = ["快逃!!!!!!!!(o(*°▽°*)o)"]
|
||
|
||
# bbcd equal
|
||
BBCDMSG7 = ["试试刺刀看看谁能赢吧！"]
|
||
|
||
|
||
def get_jrcd(user_id: int) -> int:
|
||
"""
|
||
根据用户ID和当前日期生成一个伪随机的“长度”值。
|
||
|
||
:param user_id: 用户QQ号。
|
||
:return: 返回一个1到30之间的整数。
|
||
"""
|
||
current_time = (
|
||
datetime.now().year * 100 + datetime.now().month * 100 + datetime.now().day
|
||
)
|
||
|
||
random.seed(hash(user_id + current_time))
|
||
jrcd = random.randint(1, 30)
|
||
random.seed()
|
||
|
||
return jrcd
|
||
|
||
|
||
@matcher.command("jrcd")
|
||
async def handle_jrcd(bot: Bot, event: MessageEvent, args: list[str]):
|
||
if event.id == 831797331:
|
||
return None
|
||
"""
|
||
处理 jrcd 指令，回复用户的“今日长度”。
|
||
|
||
:param bot: Bot 实例。
|
||
:param event: 消息事件对象。
|
||
:param args: 指令参数列表（未使用）。
|
||
"""
|
||
user_id = event.user_id
|
||
jrcd = await run_in_thread_pool(get_jrcd, user_id)
|
||
|
||
msg_text = ""
|
||
if jrcd <= 9:
|
||
msg_text = random.choice(JRCDMSG_1) % jrcd
|
||
elif jrcd <= 19:
|
||
msg_text = random.choice(JRCDMSG_2) % jrcd
|
||
else:
|
||
msg_text = random.choice(JRCDMSG_3) % jrcd
|
||
|
||
reply_segments = [MessageSegment.at(user_id), MessageSegment.from_text(msg_text)]
|
||
await event.reply(reply_segments)
|
||
|
||
# 使用 Lua 脚本原子化地增加总调用次数
|
||
lua_script = "return redis.call('INCR', KEYS[1])"
|
||
try:
|
||
total_calls = await redis_manager.execute_lua_script(
|
||
script=lua_script,
|
||
keys=["neobot:jrcd:total_calls"],
|
||
args=[]
|
||
)
|
||
if total_calls:
|
||
logger.info(f"jrcd 总调用次数: {total_calls}")
|
||
except Exception as e:
|
||
logger.error(f"jrcd 插件增加调用次数失败: {e}")
|
||
|
||
|
||
@matcher.command("jrcd_stats")
|
||
async def handle_jrcd_stats(bot: Bot, event: MessageEvent, args: list[str]):
|
||
"""
|
||
处理 jrcd_stats 指令，查询 /jrcd 的总调用次数。
|
||
|
||
:param bot: Bot 实例。
|
||
:param event: 消息事件对象。
|
||
:param args: 指令参数列表（未使用）。
|
||
"""
|
||
total_calls = await redis_manager.get("neobot:jrcd:total_calls")
|
||
if not total_calls:
|
||
total_calls = 0
|
||
|
||
reply_text = f"/jrcd 指令已被大家调用了 {total_calls} 次啦！"
|
||
await event.reply(reply_text)
|
||
|
||
|
||
@matcher.command("bbcd")
|
||
async def handle_bbcd(bot: Bot, event: MessageEvent, args: list[str]):
|
||
if event.id == 831797331:
|
||
return None
|
||
"""
|
||
处理 bbcd 指令，比较两位用户的“长度”。
|
||
|
||
:param bot: Bot 实例。
|
||
:param event: 消息事件对象。
|
||
:param args: 指令参数列表（未使用）。
|
||
"""
|
||
message = event.message
|
||
print(message)
|
||
if len(message) < 2:
|
||
return
|
||
|
||
user_id1 = event.user_id
|
||
try:
|
||
user_id2 = int(message[1].data.get("qq", 0))
|
||
except Exception:
|
||
return
|
||
|
||
if user_id1 == user_id2:
|
||
await event.reply("不能和自己比！")
|
||
return
|
||
|
||
jrcd1 = await run_in_thread_pool(get_jrcd, user_id1)
|
||
jrcd2 = await run_in_thread_pool(get_jrcd, user_id2)
|
||
|
||
jrcz = jrcd1 - jrcd2
|
||
|
||
text_part = ""
|
||
if jrcz == 0:
|
||
text_part = f" 一样长。{random.choice(BBCDMSG7)}"
|
||
elif jrcz > 0:
|
||
text_part = f" 长{jrcz}cm。"
|
||
if jrcz <= 9:
|
||
text_part += random.choice(BBCDMSG1)
|
||
elif jrcz <= 19:
|
||
text_part += random.choice(BBCDMSG2)
|
||
else:
|
||
text_part += random.choice(BBCDMSG3)
|
||
else: # jrcz < 0
|
||
text_part = f" 短{abs(jrcz)}cm。"
|
||
if jrcz >= -9:
|
||
text_part += random.choice(BBCDMSG4)
|
||
elif jrcz >= -19:
|
||
text_part += random.choice(BBCDMSG5)
|
||
else:
|
||
text_part += random.choice(BBCDMSG6)
|
||
|
||
segments = [
|
||
MessageSegment.at(user_id1),
|
||
MessageSegment.from_text(" 你的长度比 "),
|
||
MessageSegment.at(user_id2),
|
||
MessageSegment.from_text(text_part),
|
||
]
|
||
|
||
await event.reply(segments)
|