mirror of
https://github.com/woodchen-ink/Q58Bot.git
synced 2025-07-18 05:42:06 +08:00
71 lines
2.8 KiB
Python
71 lines
2.8 KiB
Python
import os
|
|
import json
|
|
import logging
|
|
import time
|
|
from telethon import TelegramClient, events
|
|
|
|
BOT_TOKEN = os.environ.get('BOT_TOKEN')
|
|
ADMIN_ID = int(os.environ.get('ADMIN_ID')) # 从环境变量获取 ADMIN_ID 并转换为整数
|
|
KEYWORDS_FILE = '/app/data/keywords.json'
|
|
|
|
def load_keywords():
|
|
if os.path.exists(KEYWORDS_FILE):
|
|
with open(KEYWORDS_FILE, 'r') as f:
|
|
return json.load(f)
|
|
return ['推广', '广告', 'ad', 'promotion']
|
|
|
|
def save_keywords(keywords):
|
|
with open(KEYWORDS_FILE, 'w') as f:
|
|
json.dump(keywords, f)
|
|
|
|
KEYWORDS = load_keywords()
|
|
|
|
client = TelegramClient('bot', api_id=6, api_hash='eb06d4abfb49dc3eeb1aeb98ae0f581e')
|
|
client.start(bot_token=BOT_TOKEN)
|
|
|
|
@client.on(events.NewMessage(pattern=''))
|
|
async def handler(event):
|
|
global KEYWORDS
|
|
if event.is_private and event.sender_id == ADMIN_ID:
|
|
command = event.message.text.split()
|
|
if command[0].lower() == '/add' and len(command) > 1:
|
|
new_keyword = command[1].lower()
|
|
if new_keyword not in KEYWORDS:
|
|
KEYWORDS.append(new_keyword)
|
|
save_keywords(KEYWORDS)
|
|
await event.respond(f"关键词 '{new_keyword}' 已添加到列表中。")
|
|
else:
|
|
await event.respond(f"关键词 '{new_keyword}' 已经在列表中。")
|
|
elif command[0].lower() == '/delete' and len(command) > 1:
|
|
keyword_to_delete = command[1].lower()
|
|
if keyword_to_delete in KEYWORDS:
|
|
KEYWORDS.remove(keyword_to_delete)
|
|
save_keywords(KEYWORDS)
|
|
await event.respond(f"关键词 '{keyword_to_delete}' 已从列表中删除。")
|
|
else:
|
|
await event.respond(f"关键词 '{keyword_to_delete}' 不在列表中。")
|
|
elif command[0].lower() == '/list':
|
|
await event.respond(f"当前关键词列表：{', '.join(KEYWORDS)}")
|
|
return
|
|
|
|
if not event.is_private and any(keyword in event.message.text.lower() for keyword in KEYWORDS):
|
|
if event.sender_id != ADMIN_ID:
|
|
await event.delete()
|
|
await event.respond("已撤回该消息。注:已发送的推广链接不要多次发送,置顶已有项目的推广链接也会自动撤回。")
|
|
|
|
def run():
|
|
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
logger = logging.getLogger('TeleGuard')
|
|
|
|
while True:
|
|
try:
|
|
logger.info("TeleGuard is starting...")
|
|
client.run_until_disconnected()
|
|
except Exception as e:
|
|
logger.error(f"An error occurred in TeleGuard: {str(e)}")
|
|
logger.info("Attempting to restart TeleGuard in 60 seconds...")
|
|
time.sleep(60) # 等待60秒后重试
|
|
|
|
if __name__ == '__main__':
|
|
run()
|