mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
68 lines
1.7 KiB
Python
68 lines
1.7 KiB
Python
from configs.path_config import DATA_PATH
|
|
from utils.utils import get_bot
|
|
from datetime import datetime
|
|
import time
|
|
from services.log import logger
|
|
from utils.manager import group_manager
|
|
try:
|
|
import ujson as json
|
|
except ModuleNotFoundError:
|
|
import json
|
|
|
|
|
|
time_data = {}
|
|
|
|
|
|
async def init():
|
|
global time_data
|
|
bot = get_bot()
|
|
gl = await bot.get_group_list()
|
|
gl = [g["group_id"] for g in gl]
|
|
data = read_data("group_last_chat_time.json")
|
|
for g in gl:
|
|
if not data.get(g):
|
|
time_data[g] = time.time()
|
|
if not time_data.get("check_time"):
|
|
time_data["check_time"] = time.time()
|
|
if not time_data.get("_group"):
|
|
time_data["_group"] = []
|
|
save_data()
|
|
return time_data
|
|
|
|
|
|
def read_data(file_name: str):
|
|
try:
|
|
with open(DATA_PATH + file_name, "r", encoding="utf8") as f:
|
|
return json.load(f)
|
|
except (ValueError, FileNotFoundError):
|
|
return {}
|
|
|
|
|
|
def save_data():
|
|
with open(DATA_PATH + "group_last_chat_time.json", "w") as f:
|
|
json.dump(time_data, f, indent=4)
|
|
logger.info(
|
|
f'自动存储 group_last_chat_time.json 时间：{str(datetime.now()).split(".")[0]}'
|
|
)
|
|
|
|
|
|
# 取消全部通知
|
|
async def cancel_all_notice(group_id):
|
|
group_id = int(group_id)
|
|
for command in group_manager.get_task_data():
|
|
if await group_manager.check_group_task_status(group_id, command):
|
|
await group_manager.close_group_task(group_id, command)
|
|
logger.info(f"关闭了 {group_id} 群的全部通知")
|
|
|
|
|
|
async def get_data():
|
|
global time_data
|
|
if not time_data:
|
|
time_data = await init()
|
|
return time_data
|
|
|
|
|
|
def set_data_value(key, value):
|
|
global time_data
|
|
time_data[key] = value
|