File size: 1,599 Bytes
b0bc1a0 | 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 | # This file is a part of TG-FileStreamBot
import urllib.parse
import aiohttp
import aiofiles
from WebStreamer.vars import Var
from WebStreamer.utils.database import Database
from WebStreamer.utils.human_readable import humanbytes
db = Database(Var.DATABASE_URL, Var.SESSION_NAME)
async def render_page(db_id):
html: str = ""
file_data=await db.get_file(db_id)
src = urllib.parse.urljoin(Var.URL, f'dl/{file_data["_id"]}')
if str((file_data['mime_type']).split('/')[0].strip()) == 'video':
async with aiofiles.open('WebStreamer/template/req.html') as r:
heading = f"Watch {file_data['file_name']}"
tag = (file_data['mime_type']).split('/')[0].strip()
html = (await r.read()).replace('tag', tag) % (heading, file_data['file_name'], src)
elif str((file_data['mime_type']).split('/')[0].strip()) == 'audio':
async with aiofiles.open('WebStreamer/template/req.html') as r:
heading = f"Listen {file_data['file_name']}"
tag = (file_data['mime_type']).split('/')[0].strip()
html = (await r.read()).replace('tag', tag) % (heading, file_data['file_name'], src)
else:
async with aiofiles.open('WebStreamer/template/dl.html') as r:
async with aiohttp.ClientSession() as s:
async with s.get(src) as u:
heading = f"Download {file_data['file_name']}"
file_size = humanbytes(int(u.headers.get('Content-Length')))
html = (await r.read()) % (heading, file_data['file_name'], src, file_size)
return html
|