Develop first time

.gitignore

@@ -0,0 +1,19 @@
+# 字节码文件
+__pycache__/
+*.py[cod]
+*$py.class
+
+# 环境配置文件 (非常重要！不要上传 Key)
+.env
+.venv
+venv/
+env/
+bin/
+lib/
+
+# IDE 配置
+.vscode/
+.idea/
+
+# 日志文件
+*.log

Dockerfile

@@ -0,0 +1,7 @@
+FROM python:3.11-slim
+WORKDIR /app
+COPY requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+COPY . .
+EXPOSE 7860
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,275 @@
+import os
+import discord
+import time
+from fastapi import FastAPI
+import uvicorn
+from threading import Thread
+from discord import app_commands, ui
+# 导入数据库基础类
+from mongo import MongoManager
+# 导入学科业务函数和学科列表
+from material import (
+    CATEGORIES, 
+    upload_material_logic, 
+    approve_material_logic, 
+    delete_material_logic, 
+    list_materials_logic
+)
+# 初始化数据库管理器
+# 假设你已经设置了环境变量 MONGO_URL
+db_manager = MongoManager(os.getenv("MONGO_URL"), "tonyd")
+
+# 身份组权限列表
+mod = [1357078628969746462, 1357079285113819146, 1436537252317626449, 1357079707539214417, 1357079889160835183]
+
+# --- 1. Flask 保活 ---
+app = FastAPI()
+@app.get('/')
+def home(): return {"Stauts": "Alive"}
+
+def run_flask():
+    port = int(os.environ.get('PORT', 7860))
+    uvicorn.run(app, host='0.0.0.0', port=port)
+
+# --- 2. 辅助函数：处理多个 Role Ping ---
+def format_pings(role_input: str):
+    pass
+
+# --- 3. Modal 表单定义 ---
+
+class AnnounceModal(ui.Modal):
+    def __init__(self, channel: discord.TextChannel, ping_roles_str: str = None):
+        super().__init__(title="New Announcement")
+        self.channel = channel
+        self.ping_roles_str = ping_roles_str
+
+    announcement_title = ui.TextInput(label="Title", placeholder="Enter title...", style=discord.TextStyle.short)
+    announcement_body = ui.TextInput(label="Body", placeholder="Enter message...", style=discord.TextStyle.long)
+
+    async def on_submit(self, interaction: discord.Interaction):
+        # 准备正文
+        content = self.announcement_body.value
+        pings = self.ping_roles_str
+        
+        # 按照你的要求：Ping 放在文本最下面，但是在 --- 上面
+        if pings:
+            pass
+        
+        embed = discord.Embed(
+            title=self.announcement_title.value,
+            description=content,
+            color=discord.Color.blue()
+        )
+        
+        # Footer 只留发送人和时间
+        footer = f"\n\n||{pings}||\n\n---\n**Sent by:** ||{interaction.user.mention}||\n**Time:** <t:{int(time.time())}:f>"
+        embed.description += footer
+
+        await self.channel.send(embed=embed)
+        await interaction.response.send_message(f"✅ Sent to {self.channel.mention}!", ephemeral=False)
+
+class EditAnnounceModal(ui.Modal):
+    def __init__(self, message: discord.Message, new_ping_roles_str: str = None):
+        super().__init__(title="Edit Announcement")
+        self.message = message
+        self.new_ping_roles_str = new_ping_roles_str
+        
+        old_title = message.embeds[0].title if message.embeds else ""
+        # 裁掉旧的 Footer 
+        old_body = message.embeds[0].description.split("\n\n---")[0] if message.embeds else ""
+
+        self.edit_title = ui.TextInput(label="Title", default=old_title, style=discord.TextStyle.short)
+        self.edit_body = ui.TextInput(label="Body", default=old_body, style=discord.TextStyle.long)
+        
+        self.add_item(self.edit_title)
+        self.add_item(self.edit_body)
+
+    async def on_submit(self, interaction: discord.Interaction):
+        content = self.edit_body.value
+        pings = self.new_ping_roles_str
+        
+        if pings:
+            pass
+
+        new_embed = discord.Embed(
+            title=self.edit_title.value,
+            description=content,
+            color=discord.Color.blue()
+        )
+        
+        footer = f"\n\n||{pings}||\n\n---\n**Edited by:** ||{interaction.user.mention}||\n**Time:** <t:{int(time.time())}:f>"
+        new_embed.description += footer
+
+        await self.message.edit(embed=new_embed)
+        await interaction.response.send_message(f"✅ Updated announcement {message.id} !", ephemeral=False)
+
+# --- 4. 机器人主体 ---
+class MyBot(discord.Client):
+    def __init__(self):
+        super().__init__(intents=discord.Intents.default())
+        self.tree = app_commands.CommandTree(self)
+
+    async def setup_hook(self):
+        await self.tree.sync()
+        print(f"✅ Slash commands synced")
+
+client = MyBot()
+# 如果你使用的是 client.tree，请确保将 db_manager 挂载到 client 上
+# 这样在指令内部可以通过 interaction.client.db 访问
+client.db = db_manager 
+# --- 5. 指令定义 ---
+
+@client.tree.command(name="announce", description="Post announcement")
+@app_commands.describe(
+    channel="Target channel", 
+    role_ids="Input Role Pings"
+)
+async def announce(interaction: discord.Interaction, channel: discord.TextChannel, role_ids: str = None):
+    if not any(role.id in mod for role in interaction.user.roles):
+        await interaction.response.send_message("❌ **Access Denied**", ephemeral=True)
+        return
+    await interaction.response.send_modal(AnnounceModal(channel, role_ids))
+
+@client.tree.command(name="announce_edit", description="Edit announcement")
+@app_commands.describe(
+    channel="The channel", 
+    message_id="The ID of the message", 
+    new_role_ids="New Role Pings to add to the bottom"
+)
+async def announce_edit(interaction: discord.Interaction, channel: discord.TextChannel, message_id: str, new_role_ids: str = None):
+    if not any(role.id in mod for role in interaction.user.roles):
+        await interaction.response.send_message("❌ **Access Denied**", ephemeral=True)
+        return
+
+    try:
+        message = await channel.fetch_message(int(message_id))
+        await interaction.response.send_modal(EditAnnounceModal(message, new_role_ids))
+    except Exception as e:
+        await interaction.response.send_message(f"❌ Error: {e}", ephemeral=False)
+
+@client.tree.command(name="wismer_commands", description="Show all avaliable commands for Wismer Bot")
+async def wismer_commands(interaction: discord.Interaction):
+    await interaction.response.send_message(
+        '''**Avaliable Commands**:
+        `/announce`: Post an announcement in selected channel. Only Moderator+ can use it.
+        `/announce_edit`: Edit an announcement that sent by Wismer Bot. Only Moderator+ can use it.
+        `/mats`: Show all materials.
+        `/uploadmat`: Upload a material.''',
+        ephemeral=False
+    )
+
+
+
+ADMIN_LOG_CHANNEL_ID = 1468391869686878455  # Replace with your Admin/Staff channel ID
+
+# --- 1. Upload Material ---
+# --- 1. Upload Material ---
+@client.tree.command(name="uploadmat", description="Upload a study link (Requires Approval)")
+@app_commands.describe(category="Subject", title="Resource Title", url="Resource Link")
+@app_commands.choices(category=[app_commands.Choice(name=c, value=c) for c in CATEGORIES])
+async def uploadmat(interaction: discord.Interaction, category: str, title: str, url: str):
+    # 改为 ephemeral=True，因为你的反馈消息是发给个人看的
+    await interaction.response.defer(ephemeral=True)
+    
+    if not url.startswith(("http://", "https://")):
+        return await interaction.followup.send("❌ Invalid URL! Must start with http or https.", ephemeral=True)
+
+    try:
+        # 执行逻辑
+        new_id = await upload_material_logic(client.db, category, title, url, interaction.user.id)
+        
+        # 1. Feedback to User (保持和 defer 一致的 ephemeral)
+        await interaction.followup.send(
+            f"✅ Submitted! **{category}** ID: `#{new_id}`. Please wait for Moderator approval.", 
+            ephemeral=True
+        )
+
+        # 2. Alert to Staff Channel (这是发送到频道，不涉及交互反馈)
+        log_channel = client.get_channel(ADMIN_LOG_CHANNEL_ID)
+        if log_channel:
+            embed = discord.Embed(title="🔔 New Material Pending Approval", color=discord.Color.orange())
+            embed.add_field(name="Subject", value=category, inline=True)
+            embed.add_field(name="Index ID", value=f"`#{new_id}`", inline=True)
+            embed.add_field(name="Title", value=title, inline=False)
+            embed.add_field(name="URL", value=url, inline=False)
+            embed.add_field(name="Uploader", value=interaction.user.mention, inline=False)
+            embed.set_footer(text="Use /approvemat to authorize this link.")
+            await log_channel.send(embed=embed)
+            
+    except Exception as e:
+        print(f"Error in uploadmat: {e}")
+        # 如果报错了也给用户一个反馈，防止他们因为没反应而狂点
+        try:
+            await interaction.followup.send(f"❌ An error occurred: {e}", ephemeral=True)
+        except:
+            pass
+        
+# --- 2. Approve Material ---
+@client.tree.command(name="approvemat", description="[Mod Only] Approve a pending material")
+@app_commands.choices(category=[app_commands.Choice(name=c, value=c) for c in CATEGORIES])
+async def approvemat(interaction: discord.Interaction, category: str, index_id: str):
+    await interaction.response.defer(ephemeral=False)
+    # Changed MOD_ROLES to mod
+    if not any(role.id in mod for role in interaction.user.roles):
+        return await interaction.followup.send("❌ Access Denied: Staff Only", ephemeral=True)
+
+    await approve_material_logic(client.db, category, index_id)
+    await interaction.followup.send(
+        f"✅ Approved **{category}** `#{index_id}`. It is now visible to everyone.", 
+        ephemeral=False
+    )
+
+# --- 3. Delete Material ---
+@client.tree.command(name="deletemat", description="[Mod Only] Delete a material")
+@app_commands.choices(category=[app_commands.Choice(name=c, value=c) for c in CATEGORIES])
+async def deletemat(interaction: discord.Interaction, category: str, index_id: str):
+    # Changed MOD_ROLES to mod
+    await interaction.response.defer(ephemeral=False)
+    if not any(role.id in mod for role in interaction.user.roles):
+        return await interaction.followup.send("❌ Access Denied: Staff Only", ephemeral=True)
+
+    await delete_material_logic(client.db, category, index_id)
+    await interaction.followup.send(
+        f"🗑️ Deleted ID `#{index_id}` from **{category}**.", 
+        ephemeral=False
+    )
+
+# --- 4. List Approved Materials ---
+@client.tree.command(name="mats", description="View approved study materials")
+@app_commands.choices(category=[app_commands.Choice(name=c, value=c) for c in CATEGORIES])
+async def mats(interaction: discord.Interaction, category: str):
+    await interaction.response.defer(ephemeral=False)
+    materials = await list_materials_logic(client.db, category, status="approved")
+    if not materials:
+        return await interaction.followup.send(f"📭 No approved materials found in **{category}**.", ephemeral=True)
+
+    list_text = "\n".join([f"`#{m['_id']}` [{m['title']}]({m['url']})" for m in materials])
+    embed = discord.Embed(title=f"📚 {category} Resource Directory", description=list_text, color=discord.Color.blue())
+    await interaction.followup.send(embed=embed)
+
+# --- 5. List Pending Materials ---
+@client.tree.command(name="unappmats", description="[Mod Only] View pending materials")
+@app_commands.choices(category=[app_commands.Choice(name=c, value=c) for c in CATEGORIES])
+async def unappmats(interaction: discord.Interaction, category: str):
+    # Changed MOD_ROLES to mod
+    await interaction.response.defer(ephemeral=False)
+    if not any(role.id in mod for role in interaction.user.roles):
+        return await interaction.followup.send("❌ Access Denied: Staff Only", ephemeral=True)
+
+    materials = await list_materials_logic(client.db, category, status="pending")
+    if not materials:
+        return await interaction.followup.send(f"✅ No pending items in **{category}**.", ephemeral=True)
+
+    list_text = "\n".join([f"`#{m['_id']}` **{m['title']}**\n🔗 {m['url']}" for m in materials])
+    embed = discord.Embed(title=f"⏳ {category} Pending List", description=list_text, color=discord.Color.orange())
+    await interaction.followup.send(embed=embed)
+
+
+
+# --- 启动 ---
+if __name__ == "__main__":
+    t = Thread(target=run_flask)
+    t.daemon = True
+    t.start()
+    token = os.getenv('DISCORD_TOKEN')
+    client.run(token)

material.py

@@ -0,0 +1,37 @@
+import time
+
+CATEGORIES = ["Language", "Math", "Science", "Music", "Geography", "History", "Transition", "Health"]
+
+# --- 内部辅助：获取自增 ID ---
+async def _get_next_id(db, category):
+    await db.inc("counters", category, "current_value", 1)
+    doc = await db.get("counters", category)
+    return str(doc["current_value"])
+
+# --- 上传资料 (默认 pending 状态) ---
+async def upload_material_logic(db, category, title, url, user_id):
+    if category not in CATEGORIES: return None
+    new_id = await _get_next_id(db, category)
+    data = {
+        "title": title,
+        "url": url,
+        "author": user_id,
+        "status": "pending",  # 初始状态为待审核
+        "time": int(time.time())
+    }
+    await db.set(category, new_id, data)
+    return new_id
+
+# --- 审核资料 ---
+async def approve_material_logic(db, category, index_id):
+    return await db.update(category, index_id, {"status": "approved"})
+
+# --- 删除资料 ---
+async def delete_material_logic(db, category, index_id):
+    return await db.delete(category, index_id)
+
+# --- 获取列表 (根据状态过滤) ---
+async def list_materials_logic(db, category, status="approved"):
+    # 如果传入了 category，只查该学科；否则可能需要遍历（为了简洁，这里建议指定 category）
+    query = {"status": status}
+    return await db.find_many(category, query=query, limit=20)

mongo.py

@@ -0,0 +1,75 @@
+import motor.motor_asyncio
+import logging
+
+# 设置简单的日志，方便你在 Render 控制台看到数据库报错
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger("MongoManager")
+
+class MongoManager:
+    def __init__(self, uri: str, db_name: str):
+        """
+        初始化 MongoDB 连接
+        :param uri: 你的 mongodb+srv:// 字符串
+        :param db_name: 数据库名称
+        """
+        self.client = motor.motor_asyncio.AsyncIOMotorClient(uri)
+        self.db = self.client[db_name]
+        logger.info(f"Connected to MongoDB: {db_name}")
+
+    # --- 核心简洁接口 ---
+
+    async def set(self, collection: str, doc_id: any, data: dict):
+        """设置/覆盖数据 (id可以是数字、字符串或消息ID)"""
+        try:
+            return await self.db[collection].update_one(
+                {"_id": doc_id},
+                {"$set": data},
+                upsert=True
+            )
+        except Exception as e:
+            logger.error(f"Set Error: {e}")
+
+    async def update(self, collection: str, doc_id: any, update_data: dict):
+        """局部更新数据 (只修改传入的字段)"""
+        try:
+            return await self.db[collection].update_one(
+                {"_id": doc_id},
+                {"$set": update_data}
+            )
+        except Exception as e:
+            logger.error(f"Update Error: {e}")
+
+    async def get(self, collection: str, doc_id: any):
+        """获取单个数据"""
+        try:
+            return await self.db[collection].find_one({"_id": doc_id})
+        except Exception as e:
+            logger.error(f"Get Error: {e}")
+            return None
+
+    async def delete(self, collection: str, doc_id: any):
+        """删除单个数据"""
+        try:
+            return await self.db[collection].delete_one({"_id": doc_id})
+        except Exception as e:
+            logger.error(f"Delete Error: {e}")
+
+    async def find_many(self, collection: str, query: dict = {}, limit: int = 10, sort_field: str = "_id"):
+        """获取多条数据 (默认按ID倒序，即拿最新的)"""
+        try:
+            cursor = self.db[collection].find(query).sort(sort_field, -1).limit(limit)
+            return await cursor.to_list(length=limit)
+        except Exception as e:
+            logger.error(f"FindMany Error: {e}")
+            return []
+
+    async def inc(self, collection: str, doc_id: any, field: str, amount: int = 1):
+        """数字自增 (比如统计公告修改次数)"""
+        try:
+            return await self.db[collection].update_one(
+                {"_id": doc_id},
+                {"$inc": {field: amount}},
+                upsert=True
+            )
+        except Exception as e:
+            logger.error(f"Increment Error: {e}")

requirements.txt

@@ -0,0 +1,5 @@
+discord.py
+motor
+dnspython
+fastapi
+uvicorn