First Deploy

Author: JNHFlow21

.gitignore

@@ -0,0 +1,4 @@
+bot_data.json
+.env
+__pycache__/
+.venv/

Dockerfile

@@ -0,0 +1,26 @@
+# 1. 基础镜像：使用与您本地一致的 Python 3.13 (精简版)
+FROM python:3.13-slim
+
+# 2. 关键步骤：设置时区为上海 (UTC+8)
+# 这一步确保您的“凌晨4点清理”和“限流冷却”时间准确无误
+RUN apt-get update && apt-get install -y tzdata \
+    && ln -fs /usr/share/zoneinfo/Asia/Shanghai /etc/localtime \
+    && dpkg-reconfigure -f noninteractive tzdata \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+# 3. 设置容器内的工作目录
+WORKDIR /app
+
+# 4. 复制依赖清单到容器中
+COPY requirements.txt .
+
+# 5. 安装依赖
+# --no-cache-dir 可以减小镜像体积，加快构建速度
+RUN pip install --no-cache-dir -r requirements.txt
+
+# 6. 复制当前目录下所有代码到容器中
+COPY . .
+
+# 7. 启动命令
+CMD ["python", "main.py"]

config.py

@@ -0,0 +1,21 @@
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+BOT_TOKEN = os.getenv("BOT_TOKEN")
+ADMIN_ID_RAW = os.getenv("ADMIN_GROUP_ID")
+
+try:
+    ADMIN_ID = int(ADMIN_ID_RAW) if ADMIN_ID_RAW else None
+except:
+    ADMIN_ID = None
+
+DATA_FILE = "bot_data.json"
+
+# 时间设置
+VERIFY_TIMEOUT = 60       # 验证超时 (秒)
+CACHE_TTL = 3600          # API缓存 (秒)
+BTN_RATE_LIMIT = 60       # 互动按钮冷却 (秒)
+CLEANUP_DAYS = 3          # 数据保留天数
+CLEANUP_TIME_HOUR = 4     # 每天清理时间 (小时, UTC+8)

database/storage.py

@@ -0,0 +1,79 @@
+import json
+import os
+import logging
+import time
+from config import DATA_FILE, CLEANUP_DAYS
+
+logger = logging.getLogger("Database")
+
+# 内存数据结构
+bot_data = {
+    "groups": {},
+    "pending_req": [],
+    "msg_map": {} 
+}
+
+def load_data():
+    global bot_data
+    if not os.path.exists(DATA_FILE):
+        save_data()
+        return
+
+    try:
+        with open(DATA_FILE, 'r', encoding='utf-8') as f:
+            data = json.load(f)
+            
+            # === 自动修复 Bug 1: 兼容旧数据 ===
+            if "pending_req" in data:
+                fixed_reqs = []
+                for req in data["pending_req"]:
+                    # 如果旧数据用的是 'id'，改为 'req_id'
+                    if "id" in req and "req_id" not in req:
+                        req["req_id"] = req.pop("id")
+                    fixed_reqs.append(req)
+                data["pending_req"] = fixed_reqs
+            # ================================
+
+            bot_data["groups"] = data.get("groups", {})
+            bot_data["pending_req"] = data.get("pending_req", [])
+            bot_data["msg_map"] = data.get("msg_map", {})
+            
+            logger.info("✅ 数据加载并自动修复完成")
+    except Exception as e:
+        logger.error(f"❌ 加载失败: {e}")
+
+def save_data():
+    try:
+        with open(DATA_FILE, 'w', encoding='utf-8') as f:
+            json.dump(bot_data, f, ensure_ascii=False, indent=2)
+    except Exception as e:
+        logger.error(f"❌ 保存失败: {e}")
+
+def cleanup_old_data(context=None):
+    """定时任务：清理超过3天的数据"""
+    logger.info("🧹 开始执行每日数据清理...")
+    now = time.time()
+    expire_seconds = CLEANUP_DAYS * 86400
+    
+    # 1. 清理消息映射 (msg_map)
+    # msg_map 结构: {"admin_msg_id": {"map": {...}, "ts": 123456}}
+    initial_len = len(bot_data["msg_map"])
+    keys_to_del = []
+    
+    for msg_id, info in bot_data["msg_map"].items():
+        # 兼容旧数据(没有ts的)直接删，或者给个默认时间
+        ts = info.get("ts", 0)
+        if now - ts > expire_seconds:
+            keys_to_del.append(msg_id)
+            
+    for k in keys_to_del:
+        del bot_data["msg_map"][k]
+
+    # 2. 清理过期申请
+    # ... (申请量通常不大，暂不清理，以免误删未审批的)
+
+    if keys_to_del:
+        save_data()
+        logger.info(f"✅ 清理完成。删除了 {len(keys_to_del)} 条过期广播记录。")
+    else:
+        logger.info("✅ 暂无过期数据需要清理。")

handlers/__init__.py

@@ -0,0 +1,148 @@
+from typing import List, Dict, Optional
+from datetime import datetime
+import html
+from telegram import (
+    Update, 
+    InlineKeyboardButton, 
+    InlineKeyboardMarkup, 
+    BotCommand,
+    BotCommandScopeChat,
+    BotCommandScopeAllPrivateChats,
+    BotCommandScopeAllChatAdministrators,
+    BotCommandScopeAllGroupChats,
+    BotCommandScopeDefault
+)
+from telegram.ext import ContextTypes, Application
+from telegram.constants import ParseMode
+from config import ADMIN_ID
+from database.storage import bot_data, save_data
+
+# 检查管理员权限
+async def check_is_admin(update: Update) -> bool:
+    if not ADMIN_ID: return False
+    user_id = update.effective_user.id
+    chat_id = update.effective_chat.id
+    return user_id == ADMIN_ID or chat_id == ADMIN_ID
+
+# ================= 补回丢失的函数：获取用户管理的群组 =================
+async def get_user_managed_groups(context: ContextTypes.DEFAULT_TYPE, user_id: int) -> List[dict]:
+    """获取用户有管理权的群组列表 (用于 /setup 菜单)"""
+    managed = []
+    # 遍历数据库中已登记的群组
+    for chat_id_str, info in bot_data["groups"].items():
+        try:
+            # 调用 Telegram API 检查该用户在群里的身份
+            member = await context.bot.get_chat_member(chat_id=int(chat_id_str), user_id=user_id)
+            if member.status in ['creator', 'administrator']:
+                managed.append({"id": chat_id_str, "title": info.get("title", "未知群组")})
+        except Exception:
+            # 如果Bot被踢了或者没权限看，跳过
+            continue 
+    return managed
+# ===================================================================
+
+async def post_init(application: Application):
+    """配置指令菜单 (Bot启动时执行)"""
+    bot = application.bot
+    await bot.delete_my_commands(scope=BotCommandScopeDefault())
+    
+    # 1. 公共指令
+    public_cmds = [
+        BotCommand("fear", "😨 恐慌指数"),
+        BotCommand("help", "💡 帮助")
+    ]
+    await bot.set_my_commands(public_cmds, scope=BotCommandScopeAllGroupChats())
+    
+    # 2. 私聊指令
+    private_cmds = public_cmds + [
+        BotCommand("start", "开始"),
+        BotCommand("setup", "⚙️ 配置"),
+        BotCommand("id", "获取ID")
+    ]
+    await bot.set_my_commands(private_cmds, scope=BotCommandScopeAllPrivateChats())
+
+    # 3. 群管指令
+    admin_cmds = public_cmds + [
+        BotCommand("request_broadcast", "📢 申请广播"),
+        BotCommand("id", "🆔 群ID")
+    ]
+    await bot.set_my_commands(admin_cmds, scope=BotCommandScopeAllChatAdministrators())
+
+    # 4. 终极管理员指令
+    if ADMIN_ID:
+        root_cmds = public_cmds + [
+            BotCommand("report", "📊 审计"),
+            BotCommand("revoke", "🚫 移除"),
+            BotCommand("id", "🆔 获取ID")
+        ]
+        try:
+            await bot.set_my_commands(root_cmds, scope=BotCommandScopeChat(chat_id=ADMIN_ID))
+        except Exception:
+            pass
+
+# 申请广播
+async def cmd_request_broadcast(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    chat = update.effective_chat
+    if chat.type == 'private' or not ADMIN_ID: return
+    
+    req_id = f"{chat.id}_{int(datetime.now().timestamp())}"
+    bot_data["pending_req"].append({
+        "req_id": req_id, 
+        "chat_id": chat.id, 
+        "title": chat.title
+    })
+    save_data()
+    
+    kb = [[InlineKeyboardButton("✅ 批准", callback_data=f"appr_yes_{req_id}"), InlineKeyboardButton("❌ 拒绝", callback_data=f"appr_no_{req_id}")]]
+    await context.bot.send_message(ADMIN_ID, f"📩 **广播申请**\n群组: {chat.title}", reply_markup=InlineKeyboardMarkup(kb), parse_mode=ParseMode.MARKDOWN)
+    await update.message.reply_text("✅ 已提交申请，等待审核。")
+
+# 审批回调
+async def callback_approval(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    query = update.callback_query
+    parts = query.data.split("_", 2)
+    action = parts[1]
+    req_id = parts[2]
+    
+    req = next((r for r in bot_data["pending_req"] if r.get("req_id") == req_id), None)
+    
+    if not req:
+        await query.edit_message_text("⚠️ 请求失效或已处理")
+        return
+    
+    if action == "yes":
+        cid = str(req["chat_id"])
+        if cid not in bot_data["groups"]: bot_data["groups"][cid] = {"title": req["title"]}
+        
+        bot_data["groups"][cid].update({
+            "is_broadcast": True,
+            "is_paused": False,
+            "sub_news": True, "sub_marketing": True, "sub_analysis": True, "sub_signal": True
+        })
+        save_data()
+        await query.edit_message_text(f"✅ 已批准: {req['title']}")
+        try: await context.bot.send_message(req["chat_id"], "🎉 恭喜！广播申请已通过。管理员请私聊 Bot 使用 /setup 配置。")
+        except: pass
+    else:
+        await query.edit_message_text(f"❌ 已拒绝: {req['title']}")
+        
+    bot_data["pending_req"].remove(req)
+    save_data()
+
+async def cmd_report(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    if not await check_is_admin(update): return
+    txt = "📊 **群组审计**\n\n"
+    for cid, i in bot_data["groups"].items():
+        if not i.get("is_broadcast"): continue
+        status = "🔴暂停" if i.get("is_paused") else "🟢正常"
+        txt += f"**{html.escape(i['title'])}**\nID: `{cid}`\n状态: {status}\n话题: News:`{i.get('topic_news')}` Sig:`{i.get('topic_signal')}`\n\n"
+    if len(txt) > 4000: txt = txt[:4000]
+    await update.message.reply_text(txt, parse_mode=ParseMode.MARKDOWN)
+
+async def cmd_revoke(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    if not await check_is_admin(update) or not context.args: return
+    tid = context.args[0]
+    if tid in bot_data["groups"]:
+        bot_data["groups"][tid]["is_broadcast"] = False
+        save_data()
+        await update.message.reply_text(f"✅ 已移除权限: {tid}")