mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-14 13:42:56 +08:00
211 lines
8.8 KiB
Python
211 lines
8.8 KiB
Python
from models.group_remind import GroupRemind
|
||
from services.log import logger
|
||
from configs.path_config import DATA_PATH
|
||
import os
|
||
import aiofiles
|
||
import aiohttp
|
||
from util.init_result import image
|
||
from util.utils import get_local_proxy, get_bot
|
||
from pathlib import Path
|
||
from nonebot import require
|
||
from configs.config import plugins2name_dict
|
||
from models.group_member_info import GroupInfoUser
|
||
import time
|
||
from datetime import datetime
|
||
from services.db_context import db
|
||
from models.level_user import LevelUser
|
||
from configs.config import ADMIN_DEFAULT_AUTH
|
||
try:
|
||
import ujson as json
|
||
except ModuleNotFoundError:
|
||
import json
|
||
|
||
|
||
export = require("nonebot_plugin_manager")
|
||
|
||
|
||
command_dict = {
|
||
'早晚安': 'zwa',
|
||
'进群欢迎': 'hy',
|
||
'每日开箱重置提醒': 'kxcz',
|
||
'b站转发解析': 'blpar',
|
||
'epic': 'epic',
|
||
'丢人爬': 'pa',
|
||
'原神黄历提醒': 'almanac',
|
||
}
|
||
command_list = list(command_dict.values())
|
||
command_info_dt = {
|
||
'早晚安': '将会在每晚11:59晚安，在6:01早安哦',
|
||
'每日开箱重置提醒': '将会在每日00:01提示开箱重置！',
|
||
'epic': '将会在每日中午12:01发送可白嫖的epic游戏',
|
||
'原神黄历提醒': '将会在每日8:00发送当日的原神黄历',
|
||
}
|
||
|
||
|
||
async def remind_status(group: int, name: str, flag: bool) -> str:
|
||
_name = ''
|
||
if name in command_dict.values():
|
||
_name = list(command_dict.keys())[list(command_dict.values()).index(name)]
|
||
if flag:
|
||
rst = '开启'
|
||
if await GroupRemind.get_status(group, name):
|
||
return f'该群已经{rst}过 {_name}，请勿重复开启！'
|
||
else:
|
||
rst = '关闭'
|
||
if not await GroupRemind.get_status(group, name):
|
||
return f'该群已经{rst}过 {_name}，请勿重复开启！'
|
||
if await GroupRemind.set_status(group, name, flag):
|
||
info = command_info_dt[_name] if command_info_dt.get(_name) else ''
|
||
if info:
|
||
info = '\n' + info
|
||
return f'成功{rst} {_name}！0v0 {info}'
|
||
else:
|
||
return f'{rst} {_name} 失败了...'
|
||
|
||
|
||
async def set_group_status(name: str, group_id: int):
|
||
flag = None
|
||
if name[:2] == '开启':
|
||
flag = True
|
||
elif name[:2] == '关闭':
|
||
flag = False
|
||
cmd = name[2:]
|
||
if cmd in ['全部通知', '所有通知']:
|
||
for command in command_list:
|
||
await remind_status(group_id, command, flag)
|
||
return f'已{name[:2]}所有通知！'
|
||
return await remind_status(group_id, command_dict[cmd], flag)
|
||
|
||
|
||
async def group_current_status(group_id: int):
|
||
result = f'（被动技能）\n早晚安通知：{"√" if await GroupRemind.get_status(group_id, "zwa") else "×"}\n' \
|
||
f'进群欢迎：{"√" if await GroupRemind.get_status(group_id, "hy") else "×"}\n' \
|
||
f'每日开箱重置通知：{"√" if await GroupRemind.get_status(group_id, "kxcz") else "×"}\n' \
|
||
f'b站转发解析：{"√" if await GroupRemind.get_status(group_id, "blpar") else "×"}\n' \
|
||
f'丢人爬：{"√" if await GroupRemind.get_status(group_id, "pa") else "×"}\n' \
|
||
f'epic免费游戏：{"√" if await GroupRemind.get_status(group_id, "epic") else "×"}\n' \
|
||
f'原神黄历提醒：{"√" if await GroupRemind.get_status(group_id, "almanac") else "×"}'
|
||
return result
|
||
|
||
|
||
custom_welcome_msg_json = Path() / "data" / "custom_welcome_msg" / "custom_welcome_msg.json"
|
||
|
||
|
||
async def custom_group_welcome(msg, imgs, user_id, group_id):
|
||
img_result = ''
|
||
img = imgs[0] if imgs else ''
|
||
result = ''
|
||
if os.path.exists(DATA_PATH + f'custom_welcome_msg/{group_id}.jpg'):
|
||
os.remove(DATA_PATH + f'custom_welcome_msg/{group_id}.jpg')
|
||
# print(custom_welcome_msg_json)
|
||
if not custom_welcome_msg_json.exists():
|
||
custom_welcome_msg_json.parent.mkdir(parents=True, exist_ok=True)
|
||
data = {}
|
||
else:
|
||
try:
|
||
data = json.load(open(custom_welcome_msg_json, 'r'))
|
||
except FileNotFoundError:
|
||
data = {}
|
||
try:
|
||
if msg:
|
||
data[str(group_id)] = str(msg)
|
||
json.dump(data, open(custom_welcome_msg_json, 'w'), indent=4, ensure_ascii=False)
|
||
logger.info(f'USER {user_id} GROUP {group_id} 更换群欢迎消息 {msg}')
|
||
result += msg
|
||
if img:
|
||
async with aiohttp.ClientSession() as session:
|
||
async with session.get(img, proxy=get_local_proxy()) as response:
|
||
async with aiofiles.open(DATA_PATH + f'custom_welcome_msg/{group_id}.jpg', 'wb') as f:
|
||
await f.write(await response.read())
|
||
img_result = image(abspath=DATA_PATH + f'custom_welcome_msg/{group_id}.jpg')
|
||
logger.info(f'USER {user_id} GROUP {group_id} 更换群欢迎消息图片')
|
||
except Exception as e:
|
||
logger.error(f'GROUP {group_id} 替换群消息失败 e:{e}')
|
||
return '替换群消息失败..'
|
||
return f'替换群欢迎消息成功：\n{result}' + img_result
|
||
|
||
|
||
async def change_group_switch(cmd: str, group_id: int):
|
||
group_id = str(group_id)
|
||
status = cmd[:2]
|
||
cmd = cmd[2:]
|
||
try:
|
||
with open(DATA_PATH + 'manager/plugin_list.json', 'r', encoding='utf8') as f:
|
||
plugin_list = json.load(f)
|
||
except ValueError:
|
||
pass
|
||
except FileNotFoundError:
|
||
pass
|
||
for plugin_cmd in plugins2name_dict.keys():
|
||
if cmd in plugins2name_dict[plugin_cmd]:
|
||
# print(plugin_list[plugin_cmd])
|
||
if status == '开启':
|
||
# if group_id in plugin_list[plugin_cmd]:
|
||
if group_id not in plugin_list[plugin_cmd] or plugin_list[plugin_cmd][group_id]:
|
||
return f'功能 {cmd} 正处于开启状态！不要重复开启.'
|
||
export.unblock_plugin(group_id, plugin_cmd)
|
||
else:
|
||
if group_id in plugin_list[plugin_cmd]:
|
||
if not plugin_list[plugin_cmd][group_id]:
|
||
return f'功能 {cmd} 正处于关闭状态！不要重复关闭.'
|
||
export.block_plugin(group_id, plugin_cmd)
|
||
if os.path.exists(DATA_PATH + f'group_help/{group_id}.png'):
|
||
os.remove(DATA_PATH + f'group_help/{group_id}.png')
|
||
|
||
return f'{status} {cmd} 功能！'
|
||
return f'没有找到 {cmd} 功能喔'
|
||
|
||
|
||
async def update_member_info(group_id: int) -> bool:
|
||
bot = get_bot()
|
||
_group_user_list = await bot.get_group_member_list(group_id=group_id)
|
||
_error_member_list = []
|
||
_exist_member_list = []
|
||
# try:
|
||
for user_info in _group_user_list:
|
||
if user_info['card'] == "":
|
||
nickname = user_info['nickname']
|
||
else:
|
||
nickname = user_info['card']
|
||
async with db.transaction():
|
||
# 更新权限
|
||
if user_info['role'] in ['owner', 'admin'] and not await LevelUser.is_group_flag(user_info['user_id'], group_id):
|
||
await LevelUser.set_level(user_info['user_id'], user_info['group_id'], ADMIN_DEFAULT_AUTH)
|
||
if str(user_info['user_id']) in bot.config.superusers:
|
||
await LevelUser.set_level(user_info['user_id'], user_info['group_id'], 9)
|
||
user = await GroupInfoUser.select_member_info(user_info['user_id'], user_info['group_id'])
|
||
if user:
|
||
if user.user_name != nickname:
|
||
await user.update(user_name=nickname).apply()
|
||
logger.info(f"用户{user_info['user_id']} 所属{user_info['group_id']} 更新群昵称成功")
|
||
_exist_member_list.append(int(user_info['user_id']))
|
||
continue
|
||
join_time = datetime.strptime(
|
||
time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(user_info['join_time'])), "%Y-%m-%d %H:%M:%S")
|
||
if await GroupInfoUser.insert(
|
||
user_info['user_id'],
|
||
user_info['group_id'],
|
||
nickname,
|
||
join_time,):
|
||
_exist_member_list.append(int(user_info['user_id']))
|
||
logger.info(f"用户{user_info['user_id']} 所属{user_info['group_id']} 更新成功")
|
||
else:
|
||
_error_member_list.append(f"用户{user_info['user_id']} 所属{user_info['group_id']} 更新失败\n")
|
||
_del_member_list = list(set(_exist_member_list).difference(set(await GroupInfoUser.query_group_member_list(group_id))))
|
||
if _del_member_list:
|
||
for del_user in _del_member_list:
|
||
if await GroupInfoUser.delete_member_info(del_user, group_id):
|
||
logger.info(f"退群用户{del_user} 所属{group_id} 已删除")
|
||
else:
|
||
logger.info(f"退群用户{del_user} 所属{group_id} 删除失败")
|
||
if _error_member_list:
|
||
result = ""
|
||
for error_user in _error_member_list:
|
||
result += error_user
|
||
await bot.send_private_msg(user_id=int(list(bot.config.superusers)[0]), message=result[:-1])
|
||
return True
|
||
|
||
|
||
|
||
|