mirror of
https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git
synced 2024-11-16 14:21:10 +00:00
458 lines
20 KiB
Python
458 lines
20 KiB
Python
""" PagerMaid module to handle sticker collection. """
|
||
|
||
from PIL import Image
|
||
from os.path import exists
|
||
from os import sep
|
||
from random import randint
|
||
|
||
from pyrogram import Client
|
||
from pyrogram.errors import PeerIdInvalid, UsernameNotOccupied
|
||
from pyrogram.types import User, Chat
|
||
|
||
from pagermaid.single_utils import sqlite, safe_remove
|
||
from pagermaid.listener import listener
|
||
from pagermaid.utils import alias_command, client, Message, lang
|
||
|
||
from collections import defaultdict
|
||
import json
|
||
|
||
git_source = "https://gitlab.com/Xtao-Labs/PagerMaid_Plugins/-/raw/v2/"
|
||
positions = {
|
||
"1": [297, 288],
|
||
"2": [85, 368],
|
||
"3": [127, 105],
|
||
"4": [76, 325],
|
||
"5": [256, 160],
|
||
"6": [298, 22],
|
||
}
|
||
notifyStrArr = {
|
||
"6": "踢人",
|
||
}
|
||
extensionConfig = {}
|
||
max_number = len(positions)
|
||
configFilePath = f"plugins{sep}eat{sep}config.json"
|
||
configFileRemoteUrlKey = "eat.configFileRemoteUrl"
|
||
|
||
|
||
async def eat_it(context, user, base, mask, photo, number, layer=0):
|
||
mask_size = mask.size
|
||
photo_size = photo.size
|
||
if mask_size[0] < photo_size[0] and mask_size[1] < photo_size[1]:
|
||
scale = photo_size[1] / mask_size[1]
|
||
photo = photo.resize((int(photo_size[0] / scale), int(photo_size[1] / scale)), Image.LANCZOS)
|
||
photo = photo.crop((0, 0, mask_size[0], mask_size[1]))
|
||
mask1 = Image.new('RGBA', mask_size)
|
||
mask1.paste(photo, mask=mask)
|
||
numberPosition = positions[str(number)]
|
||
isSwap = False
|
||
# 处理头像，放到和背景同样大小画布的特定位置
|
||
try:
|
||
isSwap = extensionConfig[str(number)]["isSwap"]
|
||
except:
|
||
pass
|
||
if isSwap:
|
||
photoBg = Image.new('RGBA', base.size)
|
||
photoBg.paste(mask1, (numberPosition[0], numberPosition[1]), mask1)
|
||
photoBg.paste(base, (0, 0), base)
|
||
base = photoBg
|
||
else:
|
||
base.paste(mask1, (numberPosition[0], numberPosition[1]), mask1)
|
||
|
||
# 增加判断是否有第二个头像孔
|
||
isContinue = len(numberPosition) > 2 and layer == 0
|
||
if isContinue:
|
||
await context._client.download_media(
|
||
user.photo.big_file_id, f"plugins{sep}eat{sep}{str(user.id)}.jpg"
|
||
)
|
||
|
||
try:
|
||
markImg = Image.open(f"plugins{sep}eat{sep}{str(user.id)}.jpg")
|
||
maskImg = Image.open(f"plugins{sep}eat{sep}mask{str(numberPosition[2])}.png").convert("RGBA")
|
||
except:
|
||
await context.edit(f"图片模版加载出错，请检查并更新配置：mask{str(numberPosition[2])}.png")
|
||
return base
|
||
base = await eat_it(context, user, base, maskImg, markImg, numberPosition[2], layer + 1)
|
||
|
||
temp = base.size[0] if base.size[0] > base.size[1] else base.size[1]
|
||
if temp != 512:
|
||
scale = 512 / temp
|
||
base = base.resize((int(base.size[0] * scale), int(base.size[1] * scale)), Image.LANCZOS)
|
||
|
||
return base
|
||
|
||
|
||
async def updateConfig(context):
|
||
if configFileRemoteUrl := sqlite.get(configFileRemoteUrlKey, ""):
|
||
if downloadFileFromUrl(configFileRemoteUrl, configFilePath) == 0:
|
||
return await loadConfigFile(context, True)
|
||
sqlite[configFileRemoteUrlKey] = configFileRemoteUrl
|
||
return -1
|
||
return 0
|
||
|
||
|
||
async def downloadFileFromUrl(url, filepath):
|
||
try:
|
||
re = await client.get(url)
|
||
with open(filepath, 'wb') as ms:
|
||
ms.write(re.content)
|
||
except:
|
||
return -1
|
||
return 0
|
||
|
||
|
||
async def loadConfigFile(context, forceDownload=False):
|
||
global positions, notifyStrArr, extensionConfig
|
||
try:
|
||
with open(configFilePath, 'r', encoding='utf8') as cf:
|
||
# 读取已下载的配置文件
|
||
remoteConfigJson = json.load(cf)
|
||
# positionsStr = json.dumps(positions)
|
||
# positions = json.loads(positionsStr)
|
||
|
||
# 读取配置文件中的positions
|
||
positionsStr = json.dumps(remoteConfigJson["positions"])
|
||
data = json.loads(positionsStr)
|
||
# 与预设positions合并
|
||
positions = mergeDict(positions, data)
|
||
|
||
# 读取配置文件中的notifies
|
||
data = json.loads(json.dumps(remoteConfigJson["notifies"]))
|
||
# 与预设positions合并
|
||
notifyStrArr = mergeDict(notifyStrArr, data)
|
||
|
||
# 读取配置文件中的extensionConfig
|
||
try:
|
||
data = json.loads(json.dumps(remoteConfigJson["extensionConfig"]))
|
||
# 与预设extensionConfig合并
|
||
extensionConfig = mergeDict(extensionConfig, data)
|
||
except:
|
||
# 新增扩展配置，为了兼容旧的配置文件更新不出错，无视异常
|
||
pass
|
||
|
||
# 读取配置文件中的needDownloadFileList
|
||
data = json.loads(json.dumps(remoteConfigJson["needDownloadFileList"]))
|
||
# 下载列表中的文件
|
||
for file_url in data:
|
||
try:
|
||
fsplit = file_url.split("/")
|
||
filePath = f"plugins{sep}eat{sep}{fsplit[len(fsplit) - 1]}"
|
||
if not exists(filePath) or forceDownload:
|
||
await downloadFileFromUrl(file_url, filePath)
|
||
|
||
except:
|
||
await context.edit(f"下载文件异常，url：{file_url}")
|
||
return -1
|
||
except:
|
||
return -1
|
||
return 0
|
||
|
||
|
||
def mergeDict(d1, d2):
|
||
dd = defaultdict(list)
|
||
|
||
for d in (d1, d2):
|
||
for key, value in d.items():
|
||
dd[key] = value
|
||
return dict(dd)
|
||
|
||
|
||
async def downloadFileByIds(ids, context):
|
||
idsStr = f',{",".join(ids)},'
|
||
try:
|
||
with open(configFilePath, 'r', encoding='utf8') as cf:
|
||
# 读取已下载的配置文件
|
||
remoteConfigJson = json.load(cf)
|
||
data = json.loads(json.dumps(remoteConfigJson["needDownloadFileList"]))
|
||
# 下载列表中的文件
|
||
sucSet = set()
|
||
failSet = set()
|
||
for file_url in data:
|
||
try:
|
||
fsplit = file_url.split("/")
|
||
fileFullName = fsplit[len(fsplit) - 1]
|
||
fileName = fileFullName.split(".")[0].replace("eat", "").replace("mask", "")
|
||
if f',{fileName},' in idsStr:
|
||
filePath = f"plugins{sep}eat{sep}{fileFullName}"
|
||
if (await downloadFileFromUrl(file_url, filePath)) == 0:
|
||
sucSet.add(fileName)
|
||
else:
|
||
failSet.add(fileName)
|
||
except:
|
||
failSet.add(fileName)
|
||
await context.edit(f"下载文件异常，url：{file_url}")
|
||
notifyStr = "更新模版完成"
|
||
if sucSet:
|
||
notifyStr = f'{notifyStr}\n成功模版如下：{"，".join(sucSet)}'
|
||
if failSet:
|
||
notifyStr = f'{notifyStr}\n失败模版如下：{"，".join(failSet)}'
|
||
await context.edit(notifyStr)
|
||
except:
|
||
await context.edit("更新下载模版图片失败，请确认配置文件是否正确")
|
||
|
||
|
||
@listener(is_plugin=True, outgoing=True, command=alias_command("eat"),
|
||
description="生成一张 吃头像 图片\n"
|
||
"可选：当第二个参数是数字时，读取预存的配置；\n\n"
|
||
"当第二个参数是.开头时，头像旋转180°，并且判断r后面是数字则读取对应的配置生成\n\n"
|
||
"当第二个参数是/开头时，在/后面加url则从url下载配置文件保存到本地，如果就一个/，则直接更新配置文件，删除则是/delete；或者/后面加模版id可以手动更新指定模版配置\n\n"
|
||
"当第二个参数是-开头时，在-后面加上模版id，即可设置默认模版-eat直接使用该模版，删除默认模版是-eat -\n\n"
|
||
"当第二个参数是!或者！开头时，列出当前可用模版",
|
||
parameters="<username/uid> [随意内容]")
|
||
async def eat(client_: Client, context: Message):
|
||
if len(context.parameter) > 2:
|
||
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
||
return
|
||
diu_round = False
|
||
if context.from_user:
|
||
from_user_id = context.from_user.id
|
||
else:
|
||
from_user_id = context.sender_chat.id
|
||
if context.reply_to_message:
|
||
if context.reply_to_message.from_user:
|
||
user = context.reply_to_message.from_user
|
||
else:
|
||
user = context.reply_to_message.sender_chat
|
||
if not user:
|
||
return await context.edit(f"{lang('error_prefix')}{lang('profile_e_no')}")
|
||
else:
|
||
if len(context.parameter) == 1:
|
||
user = context.parameter[0]
|
||
if user.isdigit():
|
||
user = int(user)
|
||
else:
|
||
if context.from_user:
|
||
user = context.from_user
|
||
else:
|
||
user = context.sender_chat
|
||
if context.entities is not None:
|
||
if context.entities[0].type == "text_mention":
|
||
user = context.entities[0].user
|
||
elif context.entities[0].type == "phone_number":
|
||
user = int(context.parameter[0])
|
||
elif context.entities[0].type == "bot_command":
|
||
if context.from_user:
|
||
user = context.from_user
|
||
else:
|
||
user = context.sender_chat
|
||
else:
|
||
return await context.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
||
if not (isinstance(user, User) or isinstance(user, Chat)):
|
||
if user[:1] in [".", "/", "-", "!"]:
|
||
if context.from_user:
|
||
user = context.from_user
|
||
else:
|
||
user = context.sender_chat
|
||
else:
|
||
try:
|
||
try:
|
||
user = await client_.get_users(user)
|
||
except IndexError:
|
||
user = await client_.get_chat(user) # noqa
|
||
except PeerIdInvalid:
|
||
return await context.edit(f"{lang('error_prefix')}{lang('profile_e_nof')}")
|
||
except UsernameNotOccupied:
|
||
return await context.edit(f"{lang('error_prefix')}{lang('profile_e_nou')}")
|
||
except OverflowError:
|
||
return await context.edit(f"{lang('error_prefix')}{lang('profile_e_long')}")
|
||
except Exception as exception:
|
||
raise exception
|
||
target_user_id = user.id
|
||
photo = await client_.download_media(
|
||
user.photo.big_file_id,
|
||
f"plugins{sep}eat{sep}" + str(target_user_id) + ".jpg",)
|
||
|
||
reply_to = context.reply_to_message.id if context.reply_to_message else None
|
||
if exists(f"plugins{sep}eat{sep}" + str(target_user_id) + ".jpg"):
|
||
for num in range(1, max_number + 1):
|
||
if not exists(f"plugins{sep}eat{sep}eat" + str(num) + ".png"):
|
||
re = await client.get(f"{git_source}eat/eat" + str(num) + ".png")
|
||
with open(f"plugins{sep}eat{sep}eat" + str(num) + ".png", "wb") as bg:
|
||
bg.write(re.content)
|
||
if not exists(f"plugins{sep}eat{sep}mask" + str(num) + ".png"):
|
||
re = await client.get(f"{git_source}eat/mask" + str(num) + ".png")
|
||
with open(f"plugins{sep}eat{sep}mask" + str(num) + ".png", "wb") as ms:
|
||
ms.write(re.content)
|
||
number = randint(1, max_number)
|
||
try:
|
||
p1 = 0
|
||
p2 = 0
|
||
if len(context.parameter) == 1:
|
||
p1 = context.parameter[0]
|
||
if p1[0] == ".":
|
||
diu_round = True
|
||
if len(p1) > 1:
|
||
try:
|
||
p2 = int("".join(p1[1:]))
|
||
except:
|
||
# 可能也有字母的参数
|
||
p2 = "".join(p1[1:])
|
||
elif p1[0] == "-":
|
||
if len(p1) > 1:
|
||
try:
|
||
p2 = int("".join(p1[1:]))
|
||
except:
|
||
# 可能也有字母的参数
|
||
p2 = "".join(p1[1:])
|
||
if p2:
|
||
sqlite["eat.default-config"] = p2
|
||
await context.edit(f"已经设置默认配置为：{p2}")
|
||
else:
|
||
del sqlite["eat.default-config"]
|
||
await context.edit(f"已经清空默认配置")
|
||
return
|
||
elif p1[0] == "/":
|
||
await context.edit(f"正在更新远程配置文件")
|
||
if len(p1) > 1:
|
||
# 获取参数中的url
|
||
p2 = "".join(p1[1:])
|
||
if p2 == "delete":
|
||
del sqlite[configFileRemoteUrlKey]
|
||
await context.edit(f"已清空远程配置文件url")
|
||
return
|
||
if p2.startswith("http"):
|
||
# 下载文件
|
||
if (await downloadFileFromUrl(p2, configFilePath)) != 0:
|
||
await context.edit(f"下载配置文件异常，请确认url是否正确")
|
||
return
|
||
else:
|
||
# 下载成功，加载配置文件
|
||
sqlite[configFileRemoteUrlKey] = p2
|
||
if await loadConfigFile(context, True) != 0:
|
||
await context.edit(f"加载配置文件异常，请确认从远程下载的配置文件格式是否正确")
|
||
return
|
||
else:
|
||
await context.edit(f"下载并加载配置文件成功")
|
||
else:
|
||
# 根据传入模版id更新模版配置，多个用"，"或者","隔开
|
||
# 判断redis是否有保存配置url
|
||
|
||
splitStr = "，"
|
||
if "," in p2:
|
||
splitStr = ","
|
||
ids = p2.split(splitStr)
|
||
if len(ids) > 0:
|
||
# 下载文件
|
||
configFileRemoteUrl = sqlite.get(configFileRemoteUrlKey, "")
|
||
if configFileRemoteUrl:
|
||
if (await downloadFileFromUrl(configFileRemoteUrl, configFilePath)) != 0:
|
||
await context.edit(f"下载配置文件异常，请确认url是否正确")
|
||
return
|
||
else:
|
||
# 下载成功，更新对应配置
|
||
if await loadConfigFile(context) != 0:
|
||
await context.edit(f"加载配置文件异常，请确认从远程下载的配置文件格式是否正确")
|
||
return
|
||
else:
|
||
await downloadFileByIds(ids, context)
|
||
else:
|
||
await context.edit(f"你没有订阅远程配置文件，更新个🔨")
|
||
else:
|
||
# 没传url直接更新
|
||
if await updateConfig(context) != 0:
|
||
await context.edit(f"更新配置文件异常，请确认是否订阅远程配置文件，或从远程下载的配置文件格式是否正确")
|
||
return
|
||
else:
|
||
await context.edit(f"从远程更新配置文件成功")
|
||
return
|
||
elif p1[0] == "！" or p1[0] == "!":
|
||
# 加载配置
|
||
if exists(configFilePath):
|
||
if await loadConfigFile(context) != 0:
|
||
await context.edit(f"加载配置文件异常，请确认从远程下载的配置文件格式是否正确")
|
||
return
|
||
txt = ""
|
||
if len(positions) > 0:
|
||
noShowList = []
|
||
for key in positions:
|
||
txt = f"{txt}，{key}"
|
||
if len(positions[key]) > 2:
|
||
noShowList.append(positions[key][2])
|
||
for key in noShowList:
|
||
txt = txt.replace(f"，{key}", "")
|
||
if txt != "":
|
||
txt = txt[1:]
|
||
await context.edit(f"目前已有的模版列表如下：\n{txt}")
|
||
return
|
||
defaultConfig = sqlite.get("eat.default-config", "")
|
||
if isinstance(p2, str):
|
||
number = p2
|
||
elif isinstance(p2, int) and p2 > 0:
|
||
number = int(p2)
|
||
elif not diu_round and ((isinstance(p1, int) and int(p1) > 0) or isinstance(p1, str)):
|
||
try:
|
||
number = int(p1)
|
||
except:
|
||
number = p1
|
||
elif defaultConfig:
|
||
try:
|
||
defaultConfig = defaultConfig.decode()
|
||
number = int(defaultConfig)
|
||
except:
|
||
number = str(defaultConfig)
|
||
# 支持配置默认是倒立的头像
|
||
if number.startswith("."):
|
||
diu_round = True
|
||
number = number[1:]
|
||
|
||
except:
|
||
number = randint(1, max_number)
|
||
|
||
# 加载配置
|
||
if exists(configFilePath):
|
||
if await loadConfigFile(context) != 0:
|
||
await context.edit(f"加载配置文件异常，请确认从远程下载的配置文件格式是否正确")
|
||
return
|
||
|
||
try:
|
||
notifyStr = notifyStrArr[str(number)]
|
||
except:
|
||
notifyStr = "吃头像"
|
||
final_msg = await context.edit(f"正在生成 {notifyStr} 图片中 . . .")
|
||
markImg = Image.open(f"plugins{sep}eat{sep}" + str(target_user_id) + ".jpg")
|
||
try:
|
||
eatImg = Image.open(f"plugins{sep}eat{sep}eat" + str(number) + ".png")
|
||
maskImg = Image.open(f"plugins{sep}eat{sep}mask" + str(number) + ".png").convert("RGBA")
|
||
except:
|
||
await context.edit(f"图片模版加载出错，请检查并更新配置：{str(number)}")
|
||
return
|
||
|
||
if diu_round:
|
||
markImg = markImg.rotate(180) # 对图片进行旋转
|
||
try:
|
||
number = str(number)
|
||
except:
|
||
pass
|
||
result = await eat_it(context, context.from_user, eatImg, maskImg, markImg, number)
|
||
result.save(f"plugins{sep}eat{sep}eat.webp")
|
||
safe_remove(f"plugins{sep}eat{sep}" + str(target_user_id) + ".jpg")
|
||
safe_remove(f"plugins{sep}eat{sep}" + str(target_user_id) + ".png")
|
||
safe_remove(f"plugins{sep}eat{sep}" + str(from_user_id) + ".jpg")
|
||
safe_remove(f"plugins{sep}eat{sep}" + str(from_user_id) + ".png")
|
||
else:
|
||
return await context.edit("此用户未设置头像或头像对您不可见。")
|
||
if reply_to:
|
||
try:
|
||
await client_.send_document(
|
||
context.chat.id,
|
||
f"plugins{sep}eat{sep}eat.webp",
|
||
reply_to_message_id=reply_to
|
||
)
|
||
await final_msg.safe_delete()
|
||
except TypeError:
|
||
await final_msg.edit("此用户未设置头像或头像对您不可见。")
|
||
except:
|
||
await final_msg.edit("此群组无法发送贴纸。")
|
||
else:
|
||
try:
|
||
await client_.send_document(
|
||
context.chat.id,
|
||
f"plugins{sep}eat{sep}eat.webp",
|
||
)
|
||
await final_msg.safe_delete()
|
||
except TypeError:
|
||
await final_msg.edit("此用户未设置头像或头像对您不可见。")
|
||
except:
|
||
await final_msg.edit("此群组无法发送贴纸。")
|
||
safe_remove(f"plugins{sep}eat{sep}eat.webp")
|
||
safe_remove(photo)
|