ngfchl
121 lines
4.2 KiB
Python
121 lines
4.2 KiB
Python
"""气氛组 + 小游戏 handler"""
|
|
import asyncio
|
|
import logging
|
|
|
|
from telegram import Update
|
|
from telegram.ext import ContextTypes, CommandHandler, MessageHandler, filters
|
|
|
|
from db.models import Action
|
|
from services.game_service import (
|
|
get_trivia_question, check_trivia_answer,
|
|
get_leaderboard, daily_checkin,
|
|
)
|
|
|
|
logger = logging.getLogger("spam_guard")
|
|
|
|
# 关键词触发
|
|
TRIGGER_REACTIONS = {
|
|
"谢谢": "👍❤️",
|
|
"感谢": "👍❤️🎉",
|
|
"哈哈": "😂",
|
|
"笑死": "😂💀",
|
|
"牛逼": "🔥💪",
|
|
"nb": "🔥💪",
|
|
"666": "🔥",
|
|
"晚安": "🌙✨",
|
|
}
|
|
|
|
|
|
async def handle_trigger(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""关键词触发气氛组"""
|
|
msg = update.effective_message
|
|
if not msg or not msg.text:
|
|
return
|
|
|
|
text = msg.text.strip()
|
|
for keyword, emoji in TRIGGER_REACTIONS.items():
|
|
if text == keyword or text.startswith(keyword):
|
|
try:
|
|
await msg.react(emoji)
|
|
except Exception:
|
|
pass
|
|
break
|
|
|
|
|
|
async def cmd_checkin(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""每日签到"""
|
|
user = update.effective_user
|
|
result = await daily_checkin(user.id, user.username or user.first_name)
|
|
if result["ok"]:
|
|
reply = await update.message.reply_text(
|
|
f"✅ 签到成功！\n"
|
|
f"📊 积分: {result['score']}\n"
|
|
f"🔥 连续签到: {result['streak']} 天\n"
|
|
f"🎁 签到奖励: +{result['bonus']} 分"
|
|
)
|
|
else:
|
|
reply = await update.message.reply_text("❌ 你今天已经签到过了")
|
|
asyncio.create_task(auto_delete(update.message))
|
|
asyncio.create_task(auto_delete(reply))
|
|
|
|
|
|
async def cmd_trivia(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""收割机知识答题"""
|
|
q = await get_trivia_question()
|
|
text = f"❓ {q['q']}\n\n"
|
|
for i, opt in enumerate(q["opts"], 1):
|
|
text += f"{i}. {opt}\n"
|
|
text += f"\n💡 请回复序号或答案文字"
|
|
context.user_data["trivia_answer"] = q["a"]
|
|
reply = await update.message.reply_text(text)
|
|
asyncio.create_task(auto_delete(update.message))
|
|
asyncio.create_task(auto_delete(reply))
|
|
|
|
|
|
async def cmd_score(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""查看积分"""
|
|
from db.models import GameScore
|
|
user = update.effective_user
|
|
scores = await GameScore.filter(user_id=user.id).all()
|
|
if not scores:
|
|
reply = await update.message.reply_text("📊 你还没有积分记录\n💡 签到 /checkin 或答题 /trivia 开始赚积分")
|
|
else:
|
|
text = "📊 你的积分\n━━━━━━━━━━━━\n"
|
|
for s in scores:
|
|
text += f"🎮 {s.game_type}: {s.score} 分 (胜率 {s.wins}/{s.total})\n"
|
|
reply = await update.message.reply_text(text)
|
|
asyncio.create_task(auto_delete(update.message))
|
|
asyncio.create_task(auto_delete(reply))
|
|
|
|
|
|
async def cmd_rank(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""排行榜"""
|
|
top = await get_leaderboard("trivia", 10)
|
|
if not top:
|
|
reply = await update.message.reply_text("🏆 暂无排行数据")
|
|
else:
|
|
text = "🏆 知识答题排行榜\n━━━━━━━━━━━━━━━━━━\n"
|
|
medals = ["🥇", "🥈", "🥉"]
|
|
for i, s in enumerate(top):
|
|
medal = medals[i] if i < 3 else f" {i+1}."
|
|
text += f"{medal} @{s.username or s.user_id} — {s.score} 分\n"
|
|
reply = await update.message.reply_text(text)
|
|
asyncio.create_task(auto_delete(update.message))
|
|
asyncio.create_task(auto_delete(reply))
|
|
|
|
|
|
async def auto_delete(message, seconds=60):
|
|
try:
|
|
await asyncio.sleep(seconds)
|
|
await message.delete()
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
def register_fun(app):
|
|
app.add_handler(CommandHandler("checkin", cmd_checkin, filters.ChatType.GROUPS))
|
|
app.add_handler(CommandHandler("trivia", cmd_trivia, filters.ChatType.GROUPS))
|
|
app.add_handler(CommandHandler("score", cmd_score, filters.ChatType.GROUPS))
|
|
app.add_handler(CommandHandler("rank", cmd_rank, filters.ChatType.GROUPS))
|
|
app.add_handler(MessageHandler(filters.TEXT & filters.ChatType.GROUPS, handle_trigger), group=2)
|