优化功能&修复bug

Author: z-mio

bot.py

@@ -28,7 +28,7 @@
     scheme,
     hostname,
     port,
-    cloudflare_cfg,
+    cloudflare_cfg, member,
 )
 from tool.scheduler_manager import aps
 from tool.utils import is_admin, parse_cron
@@ -63,6 +63,8 @@
 # 开始
 @app.on_message(filters.command("start"))
 async def start(_, message: Message):
+    if member and message.chat.id not in member:
+        return
     await message.reply("发送 `/s+文件名` 进行搜索")
 
 

config/config.py

@@ -37,6 +37,7 @@ def z_url():
 
 # user
 admin = config["user"]["admin"]
+member = config["user"]["member"]
 alist_host = config["user"]["alist_host"].removesuffix("/")
 alist_web = config["user"]["alist_web"].removesuffix("/")
 alist_token = config["user"]["alist_token"]

module/cloudflare/cloudflare.py

@@ -15,7 +15,14 @@
 )
 
 from api.cloudflare_api import graphql_api
-from config.config import nodee, cronjob, cloudflare_cfg, chat_data, write_config
+from config.config import (
+    nodee,
+    cronjob,
+    cloudflare_cfg,
+    chat_data,
+    write_config,
+    member,
+)
 from tool.scheduler_manager import aps
 from tool.utils import is_admin
 from tool.utils import pybyte
@@ -233,6 +240,8 @@ async def send_node_status(query: CallbackQuery, day):
 # 使用指令查看节点信息
 @Client.on_message(filters.command("vb"))
 async def view_bandwidth(_, message: Message):
+    if member and message.chat.id not in member:
+        return
     chat_data["node_status_mode"] = "command"
     chat_data["packUp"] = True
     chat_data[f"cd_{message.chat.id}"] = {}

module/roll/roll.py

@@ -19,7 +19,7 @@
     path,
     write_config,
     roll_cfg,
-    alist_web,
+    alist_web, member,
 )
 from module.roll.random_kaomoji import random_kaomoji
 from tool.utils import is_admin
@@ -55,6 +55,8 @@ async def sr_menu(_, message: Message):
 # 随机推荐
 @Client.on_message(filters.command("roll"))
 async def roll(_, message: Message):
+    if member and message.chat.id not in member:
+        return
     if not roll_disable():
         return
     roll_str = " ".join(message.command[1:])

module/search.py

@@ -11,7 +11,7 @@
 )
 
 from api.alist_api import AListAPI
-from config.config import config, per_page, z_url, alist_web, write_config
+from config.config import config, per_page, z_url, alist_web, write_config, member
 from tool.utils import is_admin
 from tool.utils import pybyte
 
@@ -53,6 +53,8 @@ async def zl(_, message: Message):
 # 搜索
 @Client.on_message(filters.command("s"))
 async def s(_, message: Message):
+    if member and message.chat.id not in member:
+        return
     s_str = " ".join(message.command[1:])
     if not s_str or "_bot" in s_str:
         return await message.reply("请加上文件名，例：`/s 巧克力`")