mirror of
https://github.com/TeamPGM/PagerMaid_Plugins.git
synced 2024-11-21 14:38:31 +00:00
79 lines
2.9 KiB
Python
79 lines
2.9 KiB
Python
import asyncio, zipfile, os
|
|
from io import BytesIO
|
|
from os.path import exists, isfile
|
|
from pagermaid import bot, version
|
|
from pagermaid.listener import listener
|
|
from pagermaid.utils import alias_command
|
|
|
|
|
|
async def make_zip(source_dir, output_filename):
|
|
zipf = zipfile.ZipFile(output_filename, "w")
|
|
pre_len = len(os.path.dirname(source_dir))
|
|
for parent, dirnames, filenames in os.walk(source_dir):
|
|
for filename in filenames:
|
|
pathfile = os.path.join(parent, filename)
|
|
arcname = pathfile[pre_len:].strip(os.path.sep)
|
|
zipf.write(pathfile, arcname)
|
|
zipf.close()
|
|
|
|
|
|
async def del_msg(context, t_lim):
|
|
await asyncio.sleep(t_lim)
|
|
try:
|
|
await context.delete()
|
|
except:
|
|
pass
|
|
|
|
|
|
@listener(is_plugin=True, outgoing=True, command=alias_command("transfer"),
|
|
description="上传 / 下载文件",
|
|
parameters="upload <filepath>` 或 `download <filepath>")
|
|
async def transfer(context):
|
|
params = context.parameter
|
|
if len(params) < 2:
|
|
await context.edit("参数缺失")
|
|
await del_msg(context, 3)
|
|
return
|
|
params[1] = " ".join(params[1:])
|
|
file_list = params[1].split("\n")
|
|
chat_id = context.chat_id
|
|
if params[0] == "upload":
|
|
index = 1
|
|
for file_path in file_list:
|
|
await context.edit(f"正在上传第 {index} 个文件")
|
|
if exists(file_path):
|
|
if isfile(file_path):
|
|
await bot.send_file(chat_id, file_path, force_document=True)
|
|
else:
|
|
token = file_path.split("/")
|
|
token = token[len(token) - 1]
|
|
await make_zip(file_path, f"/tmp/{token}.zip")
|
|
await bot.send_file(chat_id, f"/tmp/{token}.zip", force_document=True)
|
|
os.remove(f"/tmp/{token}.zip")
|
|
index += 1
|
|
await context.edit("上传完毕")
|
|
await del_msg(context, 3)
|
|
elif params[0] == "download":
|
|
message = await context.get_reply_message()
|
|
if message and message.media:
|
|
_file = BytesIO()
|
|
try:
|
|
await bot.download_file(message.media.document, _file)
|
|
except AttributeError:
|
|
await context.edit('无法下载此类型的文件。')
|
|
return
|
|
if not exists(file_list[0]):
|
|
with open(file_list[0], "wb") as f:
|
|
f.write(_file.getvalue())
|
|
await context.edit(f"保存成功, 保存路径 {file_list[0]}")
|
|
await del_msg(context, 5)
|
|
else:
|
|
await context.edit("路径已存在文件")
|
|
await del_msg(context, 3)
|
|
else:
|
|
await context.edit("未回复消息或回复消息中不包含文件")
|
|
await del_msg(context, 3)
|
|
else:
|
|
await context.edit("未知命令")
|
|
await del_msg(context, 3)
|