FileStream/utils/render_template.py

import aiohttp
import aiofiles
import urllib.parse
from FileStream.config import Telegram, Server
from FileStream.utils.database import Database
from FileStream.utils.human_readable import humanbytes
db = Database(Telegram.DATABASE_URL, Telegram.SESSION_NAME)

async def render_page(db_id):
    file_data=await db.get_file(db_id)
    src = urllib.parse.urljoin(Server.URL, f'dl/{file_data["_id"]}')

    if str((file_data['mime_type']).split('/')[0].strip()) == 'video':
        async with aiofiles.open('FileStream/template/stream.html') as r:
            heading = 'Watch {}'.format(file_data['file_name'])
            html_template = await r.read()
            html = html_template.replace('streamMediaLink', src).replace('streamHeading', heading)
    else:
        async with aiofiles.open('FileStream/template/dl.html') as r:
            async with aiohttp.ClientSession() as s:
                async with s.get(src) as u:
                    heading = 'Download {}'.format(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