mirror of
https://github.com/yshtcn/DiaryAssistant_bot.git
synced 2025-12-14 02:00:28 +08:00
190 lines
5.7 KiB
Python
190 lines
5.7 KiB
Python
# 修改代码以记录黑名单而非删除消息
|
||
|
||
# 示例代码（请在本地环境中运行）
|
||
import json
|
||
import requests
|
||
import time
|
||
from datetime import datetime
|
||
import os
|
||
import json
|
||
|
||
# 配置代理
|
||
proxies = {
|
||
'http': 'http://127.0.0.1:7890',
|
||
'https': 'http://127.0.0.1:7890'
|
||
}
|
||
|
||
TOKEN = None
|
||
|
||
# 示范Token
|
||
example_token = "Your_Token_Here"
|
||
|
||
# 定义配置文件名
|
||
config_filename = "bot_config.json"
|
||
|
||
# 检查配置文件是否存在
|
||
config_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), config_filename)
|
||
|
||
if os.path.exists(config_path):
|
||
# 读取配置文件
|
||
with open(config_path, 'r') as f:
|
||
config = json.load(f)
|
||
TOKEN = config.get("TOKEN", "")
|
||
|
||
# 检查Token是否已设置
|
||
if TOKEN == example_token:
|
||
print("Please update your bot token in the config file.")
|
||
exit(1)
|
||
else:
|
||
# 如果配置文件不存在，则创建一个新的配置文件并写入示范Token
|
||
config = {"TOKEN": example_token}
|
||
with open(config_path, 'w') as f:
|
||
json.dump(config, f)
|
||
print(f"Config file created at {config_path}. Please update your bot token.")
|
||
exit(1)
|
||
|
||
URL = f"https://api.telegram.org/bot{TOKEN}/"
|
||
|
||
# 尝试从文件中加载已有数据
|
||
try:
|
||
with open('user_data.json', 'r') as f:
|
||
user_data = json.load(f)
|
||
except (FileNotFoundError, json.JSONDecodeError):
|
||
user_data = {}
|
||
|
||
|
||
|
||
# 尝试从文件中加载黑名单
|
||
try:
|
||
with open('blacklist.json', 'r') as f:
|
||
blacklist = json.load(f)
|
||
except FileNotFoundError:
|
||
blacklist = []
|
||
|
||
|
||
# 获取更新
|
||
def get_updates(offset=None):
|
||
try:
|
||
url = URL + "getUpdates"
|
||
params = {'offset': offset}
|
||
r = requests.get(url, params=params, proxies=proxies)
|
||
return r.json()
|
||
except Exception as e:
|
||
print(f"Error getting updates: {e}")
|
||
return None
|
||
|
||
# 发送消息
|
||
def send_message(chat_id, text):
|
||
try:
|
||
# 尝试从文件中加载消息队列
|
||
try:
|
||
with open('message_queue.json', 'r') as f:
|
||
message_queue = json.load(f)
|
||
except FileNotFoundError:
|
||
message_queue = []
|
||
|
||
# 将新消息添加到队列
|
||
message_queue.append({'chat_id': chat_id, 'text': text})
|
||
|
||
# 将更新后的消息队列保存回文件
|
||
with open('message_queue.json', 'w') as f:
|
||
json.dump(message_queue, f)
|
||
except Exception as e:
|
||
print(f"Error queuing message: {e}")
|
||
|
||
def process_message_queue():
|
||
# 尝试从文件中加载消息队列
|
||
try:
|
||
with open('message_queue.json', 'r') as f:
|
||
message_queue = json.load(f)
|
||
except FileNotFoundError:
|
||
message_queue = []
|
||
|
||
# 遍历消息队列，尝试发送消息
|
||
for message in message_queue:
|
||
chat_id = message['chat_id']
|
||
text = message['text']
|
||
try:
|
||
url = URL + "sendMessage"
|
||
params = {'chat_id': chat_id, 'text': text}
|
||
r = requests.get(url, params=params, proxies=proxies)
|
||
if r.status_code == 200:
|
||
message_queue.remove(message)
|
||
except Exception as e:
|
||
print(f"Error sending message: {e}")
|
||
|
||
# 将更新后的消息队列保存回文件
|
||
with open('message_queue.json', 'w') as f:
|
||
json.dump(message_queue, f)
|
||
|
||
|
||
|
||
# 主程序逻辑
|
||
def main():
|
||
print("Program started") # Debugging line
|
||
last_update_id = None
|
||
while True:
|
||
|
||
print("Checking for updates...") # Debugging line
|
||
updates = get_updates(last_update_id)
|
||
if updates and "result" in updates:
|
||
print(f"Received updates: {updates}") # Debugging line
|
||
for update in updates["result"]:
|
||
last_update_id = update["update_id"] + 1
|
||
chat_id = update["message"]["chat"]["id"]
|
||
message_id = update["message"]["message_id"]
|
||
message_text = update["message"]["text"]
|
||
# 获取当前日期和时间
|
||
current_datetime = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||
|
||
# 在 message_text 前面加上日期和时间
|
||
message_text_with_datetime = f"【记录时间：{current_datetime}】\n{message_text}"
|
||
|
||
# 创建唯一标识符
|
||
unique_id = f"{TOKEN}_{chat_id}_{message_id}"
|
||
|
||
|
||
|
||
if unique_id not in blacklist:
|
||
# 处理“done”命令
|
||
if message_text.lower() == "/todaydone":
|
||
send_message(chat_id, "\n".join(user_data[chat_id]))
|
||
user_data[chat_id] = []
|
||
else:
|
||
send_message(chat_id, f"{message_text_with_datetime}")
|
||
if chat_id not in user_data:
|
||
user_data[chat_id] = []
|
||
user_data[chat_id].append(message_text)
|
||
blacklist.append(unique_id) # 添加到黑名单
|
||
|
||
# 保存数据到文件
|
||
with open('user_data.json', 'w') as f:
|
||
json.dump(user_data, f)
|
||
with open('blacklist.json', 'w') as f:
|
||
json.dump(blacklist, f)
|
||
|
||
|
||
# 保存数据到文件
|
||
with open('user_data.json', 'w') as f:
|
||
json.dump(user_data, f)
|
||
with open('blacklist.json', 'w') as f:
|
||
json.dump(blacklist, f)
|
||
|
||
process_message_queue()
|
||
else:
|
||
print(f"{URL}Received updates: {updates}")
|
||
print("Error or no updates; retrying in 5 seconds...")
|
||
time.sleep(5) # 等待5秒再重试
|
||
|
||
|
||
# 每1分钟检查一次
|
||
time.sleep(15)
|
||
|
||
# 运行主函数
|
||
if __name__ == '__main__':
|
||
main()
|
||
|
||
|
||
|
||
|