Spaces:
Running
Running
File size: 1,556 Bytes
b00d2c6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
#Thanks @DeletedFromEarth for helping in this journey
import jinja2
from info import *
from lazybot import LazyPrincessBot
from util.human_readable import humanbytes
from util.file_properties import get_file_ids
from server.exceptions import InvalidHash
import urllib.parse
import logging
import aiohttp
async def render_page(id, secure_hash, src=None):
file = await LazyPrincessBot.get_messages(int(LOG_CHANNEL), int(id))
file_data = await get_file_ids(LazyPrincessBot, int(LOG_CHANNEL), int(id))
if file_data.unique_id[:6] != secure_hash:
logging.debug(f"link hash: {secure_hash} - {file_data.unique_id[:6]}")
logging.debug(f"Invalid hash for message with - ID {id}")
raise InvalidHash
src = urllib.parse.urljoin(
URL,
f"{id}/{urllib.parse.quote_plus(file_data.file_name)}?hash={secure_hash}",
)
tag = file_data.mime_type.split("/")[0].strip()
file_size = humanbytes(file_data.file_size)
if tag in ["video", "audio"]:
template_file = "template/req.html"
else:
template_file = "template/dl.html"
async with aiohttp.ClientSession() as s:
async with s.get(src) as u:
file_size = humanbytes(int(u.headers.get("Content-Length")))
with open(template_file) as f:
template = jinja2.Template(f.read())
file_name = file_data.file_name.replace("_", " ")
return template.render(
file_name=file_name,
file_url=src,
file_size=file_size,
file_unique_id=file_data.unique_id,
)
|