mirror of
https://github.com/anonymousx97/social-dl.git
synced 2025-02-20 11:13:19 +08:00
31d19bc48b
What's New: - Sort Plugins into their respective folders. - download, upload, rename commands. - Help Docstrings. - Proper Logging. - rename .exec cmd to .py
75 lines
1.9 KiB
Python
75 lines
1.9 KiB
Python
import time
|
|
|
|
from pyrogram.types import Message, User
|
|
from telegraph.aio import Telegraph
|
|
|
|
from app import LOGGER, Config
|
|
from app.utils.media_helper import bytes_to_mb
|
|
|
|
TELEGRAPH: None | Telegraph = None
|
|
|
|
PROGRESS_DICT = {}
|
|
|
|
|
|
async def init_task():
|
|
global TELEGRAPH
|
|
TELEGRAPH = Telegraph()
|
|
try:
|
|
await TELEGRAPH.create_account(
|
|
short_name="Social-DL",
|
|
author_name="Social-DL",
|
|
author_url="https://github.com/anonymousx97/social-dl",
|
|
)
|
|
except Exception:
|
|
LOGGER.error("Failed to Create Telegraph Account.")
|
|
|
|
|
|
async def post_to_telegraph(title: str, text: str):
|
|
telegraph = await TELEGRAPH.create_page(
|
|
title=title,
|
|
html_content=f"<p>{text}</p>",
|
|
author_name="Plain-UB",
|
|
author_url=Config.UPSTREAM_REPO,
|
|
)
|
|
return telegraph["url"]
|
|
|
|
|
|
def get_name(user: User) -> str:
|
|
first = user.first_name or ""
|
|
last = user.last_name or ""
|
|
return f"{first} {last}".strip()
|
|
|
|
|
|
def extract_user_data(user: User) -> dict:
|
|
return dict(name=get_name(user), username=user.username, mention=user.mention)
|
|
|
|
|
|
async def progress(
|
|
current: int,
|
|
total: int,
|
|
response: Message | None = None,
|
|
action: str = "",
|
|
file_name: str = "",
|
|
file_path: str = "",
|
|
):
|
|
if not response:
|
|
return
|
|
if current == total:
|
|
PROGRESS_DICT.pop(file_path, "")
|
|
return
|
|
current_time = time.time()
|
|
if file_path not in PROGRESS_DICT or (current_time - PROGRESS_DICT[file_path]) > 5:
|
|
PROGRESS_DICT[file_path] = current_time
|
|
if total:
|
|
percentage = round((current * 100 / total), 1)
|
|
else:
|
|
percentage = 0
|
|
await response.edit(
|
|
f"<b>{action}</b>"
|
|
f"\n<pre language=bash>"
|
|
f"\nfile={file_name}"
|
|
f"\npath={file_path}"
|
|
f"\nsize={bytes_to_mb(total)}mb"
|
|
f"\ncompleted={bytes_to_mb(current)}mb | {percentage}%</pre>"
|
|
)
|