Spaces:
Running
Running
File size: 1,249 Bytes
e566133 12e4bed e566133 |
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 |
import jinja2
import aiohttp
import urllib.parse
from FileStream.config import Telegram, Server
from FileStream.Database import Database
from FileStream.Tools.file 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'api/dl/{file_data["_id"]}')
file_size = humanbytes(file_data['file']['file_size'])
file_name = file_data['file']['file_name'].replace("_", " ")
if str((file_data['file']['mime_type']).split('/')[0].strip()) == 'video':
template_file = "FileStream/server/template/play.html"
else:
template_file = "FileStream/server/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())
return template.render(file_name=file_name,
file_url=src,
file_size=file_size)
async def render_upload():
template_file = "FileStream/server/template/upload.html"
with open(template_file) as f:
template = jinja2.Template(f.read())
return template.render()
|