profitroll/TelegramPoster
Archived
1
0
Fork 0
Code Issues Pull Requests Releases 3 Wiki Activity

Updated user messages

Browse Source
This commit is contained in:
Profitroll
2023-03-20 14:50:57 +01:00
parent 56adb16a3e
commit 399fc5050d
3 changed files with 79 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
plugins/commands/photos.py
View File

@@ -14,7 +14,7 @@ from classes.poster_client import PosterClient
from modules.api_client import upload_pic
from modules.app import app, users_with_context
from modules.logger import logWrite
from modules.utils import configGet, extract_and_save
from modules.utils import configGet, extract_and_save, locale
@app.on_message(~filters.scheduled & filters.command(["import"], prefixes=["", "/"]))
@@ -26,30 +26,47 @@ async def cmd_import(app: PosterClient, msg: Message):
else:
return
await msg.reply_text(
f"Alright, please send me a zip archive with your media to be imported. Use /cancel if you want to abort this operation."
locale("import_request", "message", locale=msg.from_user.language_code)
)
answer = await listen_message(app, msg.chat.id, timeout=600)
users_with_context.remove(msg.from_user.id)
if answer is None:
await msg.reply_text("No response, aborting import.", quote=True)
await msg.reply_text(
locale("import_ignored", "message", locale=msg.from_user.language_code),
quote=True,
)
return
if answer.text == "/cancel":
await answer.reply_text("Okay, aborting.")
await answer.reply_text(
locale("import_abort", "message", locale=msg.from_user.language_code)
)
return
if answer.document is None:
await answer.reply_text(
"File to import must be a zip archive. Aborting.", quote=True
locale(
"import_invalid_media",
"message",
locale=msg.from_user.language_code,
),
quote=True,
)
return
if answer.document.mime_type != "application/zip":
await answer.reply_text(
"Provided file is not supported. Please send `application/zip`. Aborting.",
locale(
"import_invalid_mime", "message", locale=msg.from_user.language_code
),
quote=True,
)
return
if disk_usage(getcwd())[2] < (answer.document.file_size) * 3:
await msg.reply_text(
f"You archive is `{answer.document.file_size//(2**30)} GiB` big, but system has only `{disk_usage(getcwd())[2]//(2**30)} GiB` free. Unpacking may take even more space. Aborting."
locale(
"import_too_big", "message", locale=msg.from_user.language_code
).format(
answer.document.file_size // (2**30),
disk_usage(getcwd())[2] // (2**30),
)
)
return
tmp_dir = str(uuid4())
@@ -58,9 +75,14 @@ async def cmd_import(app: PosterClient, msg: Message):
)
makedirs(path.join(configGet("tmp", "locations"), tmp_dir), exist_ok=True)
tmp_path = path.join(configGet("tmp", "locations"), answer.document.file_id)
downloading = await answer.reply_text("Okay, downloading...", quote=True)
downloading = await answer.reply_text(
locale("import_downloading", "message", locale=msg.from_user.language_code),
quote=True,
)
await app.download_media(answer, file_name=tmp_path)
await downloading.edit("Downloaded, unpacking...")
await downloading.edit(
locale("import_unpacking", "message", locale=msg.from_user.language_code)
)
try:
with ZipFile(tmp_path, "r") as handle:
tasks = [
@@ -75,11 +97,15 @@ async def cmd_import(app: PosterClient, msg: Message):
f"Could not import '{answer.document.file_name}' due to {exp}: {format_exc}"
)
await answer.reply_text(
f"Could not unpack the archive\n\nException: {exp}\n\nTraceback:\n```python\n{format_exc}\n```"
locale(
"import_unpack_error", "message", locale=msg.from_user.language_code
).format(exp, format_exc())
)
return
logWrite(f"Downloaded '{answer.document.file_name}' - awaiting upload")
await downloading.edit("Unpacked, uploading...")
await downloading.edit(
locale("import_uploading", "message", locale=msg.from_user.language_code)
)
remove(tmp_path)
for filename in iglob(
@@ -96,12 +122,20 @@ async def cmd_import(app: PosterClient, msg: Message):
)
if len(uploaded[1]) > 0:
await msg.reply_text(
f"Could not upload `{path.basename(filename)}` because there're duplicates on server.",
locale(
"import_upload_error_duplicate",
"message",
locale=msg.from_user.language_code,
).format(path.basename(filename)),
disable_notification=True,
)
else:
await msg.reply_text(
f"Could not upload `{path.basename(filename)}`. Probably disallowed filetype",
locale(
"import_upload_error_other",
"message",
locale=msg.from_user.language_code,
).format(path.basename(filename)),
disable_notification=True,
)
else:
@@ -116,7 +150,10 @@ async def cmd_import(app: PosterClient, msg: Message):
debug=True,
)
rmtree(path.join(configGet("tmp", "locations"), tmp_dir), ignore_errors=True)
await answer.reply_text("Done.", quote=True)
await answer.reply_text(
locale("import_finished", "message", locale=msg.from_user.language_code),
quote=True,
)
return