TeamPGM/PagerMaid-Modify
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

✨ apt 支持批量安装抽奖 (#50)

Browse Source 
* 新增 apt 批量安装
This commit is contained in:
c3b2a 2021-02-07 18:55:35 +08:00 committed by GitHub
parent 0687b4c051
commit c4eb7d92d1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 83 additions and 80 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

117
pagermaid/modules/plugin.py
View File

@ -13,11 +13,33 @@ from pagermaid.utils import upload_attachment
from pagermaid.modules import plugin_list as active_plugins, __list_plugins from pagermaid.modules import plugin_list as active_plugins, __list_plugins
def move_plugin(file_path):
plugin_directory = f"{working_dir}/plugins/"
if exists(f"{plugin_directory}{file_path}"):
remove(f"{plugin_directory}{file_path}")
move(file_path, plugin_directory)
elif exists(f"{plugin_directory}{file_path}.disabled"):
remove(f"{plugin_directory}{file_path}.disabled")
move(file_path, f"{plugin_directory}{file_path}.disabled")
else:
move(file_path, plugin_directory)
def update_version(file_path, plugin_content, plugin_name, version):
with open(file_path, 'wb') as f:
f.write(plugin_content)
with open(f"{plugin_directory}version.json", 'r', encoding="utf-8") as f:
version_json = json.load(f)
version_json[plugin_name] = version
with open(f"{plugin_directory}version.json", 'w') as f:
json.dump(version_json, f)
@listener(is_plugin=False, outgoing=True, command="apt", diagnostics=False, @listener(is_plugin=False, outgoing=True, command="apt", diagnostics=False,
description="用于管理安装到 PagerMaid-Modify 的插件。", description="用于管理安装到 PagerMaid-Modify 的插件。",
parameters="{update|search|show|status|install|remove|enable|disable|upload} <插件名称/文件>") parameters="{update|search|show|status|install|remove|enable|disable|upload} <插件名称/文件>")
async def plugin(context): async def plugin(context):
if len(context.parameter) > 2 or len(context.parameter) == 0: if len(context.parameter) == 0:
await context.edit("出错了呜呜呜 ~ 无效的参数。") await context.edit("出错了呜呜呜 ~ 无效的参数。")
return return
reply = await context.get_reply_message() reply = await context.get_reply_message()
@ -36,19 +58,17 @@ async def plugin(context):
except FileNotFoundError: except FileNotFoundError:
pass pass
return return
if exists(f"{plugin_directory}{file_path}"): move_plugin(file_path)
remove(f"{plugin_directory}{file_path}")
move(file_path, plugin_directory)
elif exists(f"{plugin_directory}{file_path}.disabled"):
remove(f"{plugin_directory}{file_path}.disabled")
move(file_path, f"{plugin_directory}{file_path}.disabled")
else:
move(file_path, plugin_directory)
await context.edit(f"插件 {path.basename(file_path)[:-3]} 已安装，PagerMaid-Modify 正在重新启动。") await context.edit(f"插件 {path.basename(file_path)[:-3]} 已安装，PagerMaid-Modify 正在重新启动。")
await log(f"成功安装插件 {path.basename(file_path)[:-3]}.") await log(f"成功安装插件 {path.basename(file_path)[:-3]}.")
await context.client.disconnect() await context.client.disconnect()
elif len(context.parameter) == 2: elif len(context.parameter) >= 2:
plugin_name = context.parameter[1] await context.edit("安装插件中 . . .")
success_list = []
failed_list = []
noneed_list = []
for x in range(len(context.parameter) - 1):
plugin_name = context.parameter[1 + x]
plugin_online = \ plugin_online = \
json.loads(get("https://raw.githubusercontent.com/xtaodada/PagerMaid_Plugins/master/list.json").content)[ json.loads(get("https://raw.githubusercontent.com/xtaodada/PagerMaid_Plugins/master/list.json").content)[
'list'] 'list']
@ -64,57 +84,46 @@ async def plugin(context):
with open(f"{plugin_directory}version.json", 'w') as f: with open(f"{plugin_directory}version.json", 'w') as f:
json.dump(temp_dict, f) json.dump(temp_dict, f)
plugin_version = False plugin_version = False
flag = False
for i in plugin_online: for i in plugin_online:
if i['name'] == plugin_name: if i['name'] == plugin_name:
flag = True
if plugin_version: if plugin_version:
if (float(i['version']) - float(plugin_version)) <= 0: if (float(i['version']) - float(plugin_version)) <= 0:
await context.edit(f"插件 {plugin_name} 为最新版本，无需更新。") noneed_list.append(plugin_name)
return break
else: else:
file_path = plugin_name + ".py" file_path = plugin_name + ".py"
plugin_content = get( plugin_content = get(
f"https://raw.githubusercontent.com/xtaodada/PagerMaid_Plugins/master/{plugin_name}.py").content f"https://raw.githubusercontent.com/xtaodada/PagerMaid_Plugins/master/{plugin_name}.py").content
with open(file_path, 'wb') as f: update_version(file_path, plugin_content, plugin_name, i['version'])
f.write(plugin_content) move_plugin(file_path)
with open(f"{plugin_directory}version.json", 'r', encoding="utf-8") as f: success_list.append(path.basename(file_path)[:-3])
version_json = json.load(f) break
version_json[plugin_name] = i['version']
with open(f"{plugin_directory}version.json", 'w') as f:
json.dump(version_json, f)
if exists(f"{plugin_directory}{file_path}"):
remove(f"{plugin_directory}{file_path}")
move(file_path, plugin_directory)
elif exists(f"{plugin_directory}{file_path}.disabled"):
remove(f"{plugin_directory}{file_path}.disabled")
move(file_path, f"{plugin_directory}{file_path}.disabled")
else:
move(file_path, plugin_directory)
await context.edit(f"插件 {path.basename(file_path)[:-3]} 已安装，PagerMaid-Modify 正在重新启动。")
await log(f"成功安装插件 {path.basename(file_path)[:-3]}.")
await context.client.disconnect()
else: else:
file_path = plugin_name + ".py" file_path = plugin_name + ".py"
plugin_content = get( plugin_content = get(
f"https://raw.githubusercontent.com/xtaodada/PagerMaid_Plugins/master/{plugin_name}.py").content f"https://raw.githubusercontent.com/xtaodada/PagerMaid_Plugins/master/{plugin_name}.py").content
with open(file_path, 'wb') as f: update_version(file_path, plugin_content, plugin_name, i['version'])
f.write(plugin_content) move_plugin(file_path)
with open(f"{plugin_directory}version.json", 'r', encoding="utf-8") as f: success_list.append(path.basename(file_path)[:-3])
version_json = json.load(f) if not flag:
version_json[plugin_name] = i['version'] now_message += f"错误：没有找到插件 {plugin_name} 。\n"
with open(f"{plugin_directory}version.json", 'w') as f: failed_list.append(plugin_name)
json.dump(version_json, f) message = ""
if exists(f"{plugin_directory}{file_path}"): if len(success_list) > 0:
remove(f"{plugin_directory}{file_path}") message += "安装成功 : %s\n" % ", ".join(success_list)
move(file_path, plugin_directory) if len(failed_list) > 0:
elif exists(f"{plugin_directory}{file_path}.disabled"): message += "安装失败 : %s\n" % ", ".join(failed_list)
remove(f"{plugin_directory}{file_path}.disabled") if len(noneed_list) > 0:
move(file_path, f"{plugin_directory}{file_path}.disabled") message += "无需更新 : %s\n" % ", ".join(noneed_list)
else: await log(message)
move(file_path, plugin_directory) restart = len(success_list) > 0
await context.edit(f"插件 {path.basename(file_path)[:-3]} 已安装，PagerMaid-Modify 正在重新启动。") if restart:
await log(f"成功安装插件 {path.basename(file_path)[:-3]}.") message += "PagerMaid-Modify 正在重启。"
await context.edit(message)
if restart:
await context.client.disconnect() await context.client.disconnect()
await context.edit(f"错误：没有找到插件 {plugin_name} 。")
else: else:
await context.edit("出错了呜呜呜 ~ 无效的参数。") await context.edit("出错了呜呜呜 ~ 无效的参数。")
elif context.parameter[0] == "remove": elif context.parameter[0] == "remove":
@ -273,15 +282,9 @@ async def plugin(context):
version_json[i] = m['version'] version_json[i] = m['version']
with open(f"{plugin_directory}version.json", 'w') as f: with open(f"{plugin_directory}version.json", 'w') as f:
json.dump(version_json, f) json.dump(version_json, f)
if exists(f"{plugin_directory}{file_path}"): move_plugin(file_path)
remove(f"{plugin_directory}{file_path}")
move(file_path, plugin_directory)
elif exists(f"{plugin_directory}{file_path}.disabled"):
remove(f"{plugin_directory}{file_path}.disabled")
move(file_path, f"{plugin_directory}{file_path}.disabled")
else:
move(file_path, plugin_directory)
await context.edit('正在读取云端插件列表...完成\n正在读取本地插件版本信息...完成\n' + need_update) await context.edit('正在读取云端插件列表...完成\n正在读取本地插件版本信息...完成\n' + need_update)
await context.client.disconnect()
elif context.parameter[0] == "search": elif context.parameter[0] == "search":
if len(context.parameter) == 1: if len(context.parameter) == 1:
await context.edit("没插件名我怎么搜索？") await context.edit("没插件名我怎么搜索？")