Spaces:
Running
Running
#FileStream/server/routes_api | |
import os | |
import json | |
import time | |
import math | |
import logging | |
import asyncio | |
import traceback | |
import aiohttp_cors | |
from aiohttp import web | |
from pyrogram import raw | |
from bson import ObjectId | |
from bson.json_util import dumps | |
from aiohttp.http_exceptions import BadStatusLine | |
#---------------------Local Imports----------------------------------# | |
from FileStream.bot import req_client | |
from FileStream.config import Telegram,Server | |
from FileStream.Database import Database | |
from FileStream.TMDB.Endpoint import search_tmdb_any,search_tmdb_tv,search_tmdb_movies | |
from FileStream.server.exceptions import FIleNotFound, InvalidHash | |
from .Functions.downloader import media_streamer | |
CORS_HEADERS = { | |
"Access-Control-Allow-Origin": "*", | |
"Access-Control-Allow-Headers": "*" | |
} | |
async def handle_v2(request): | |
return web.Response(text="Hello from app api!") | |
#api.router.add_get('/10/files', list_10_all_files_db) | |
async def list_10_all_files_db(request): | |
#file_range = [file_list_no * 10 - 10 + 1, file_list_no * 10] | |
db = Database(Telegram.DATABASE_URL, Telegram.SESSION_NAME) | |
files, total_files = await db.find_all_files([1, 10]) | |
print(files) | |
return web.json_response([ dict(file) async for file in files], headers=CORS_HEADERS) | |
#return file_list, total_files | |
#api.router.add_get('/files', list_all_files_db) | |
async def list_all_files_db(request): | |
db = Database(Telegram.DATABASE_URL, Telegram.SESSION_NAME) | |
files= await db.get_all_files_api() | |
#print(files, type(files)) | |
return web.json_response(json.loads(dumps(files)), headers=CORS_HEADERS) | |
#api.router.add_get('/tmdb/mix', list_all_files_tmdb) | |
async def list_all_tmdb_movies_from_db(request): | |
db = Database(Telegram.DATABASE_URL, Telegram.SESSION_NAME) | |
files= await db.get_all_files() | |
#print(files) | |
response=[] | |
async for row in files: | |
#print(row['file']['caption']) | |
try : | |
#print("* Response",search_tmdb(row['file']['caption'] if row['file']['caption'] else row['file']['file_name'])) | |
resp = search_tmdb_movies( str(row['file']['caption']) if str(row['file']['caption']) else str(row['file']['file_name'])) | |
if resp != None : | |
#resp= dict(resp) | |
#print("TMDB Response :",resp) | |
response.append(resp) | |
else: | |
print("\n * Skipped:",row['file']['caption'],str(row['file']['file_name'])) | |
continue | |
except Exception as e: | |
print("Error ",e) | |
break | |
return web.json_response(json.loads(dumps(response)),headers=CORS_HEADERS) | |
#api.router.add_get('/tmdb/tv', list_all_tmdb_tv_from_db) | |
async def list_all_tmdb_tv_from_db(request): | |
db = Database(Telegram.DATABASE_URL, Telegram.SESSION_NAME) | |
files= await db.get_all_files() | |
#print(files) | |
response=[] | |
async for row in files: | |
#print(row['file']['caption']) | |
try : | |
#print("* Response",search_tmdb(row['file']['caption'] if row['file']['caption'] else row['file']['file_name'])) | |
resp = search_tmdb_tv( str(row['file']['caption']) if str(row['file']['caption']) else str(row['file']['file_name'])) | |
if resp != None : | |
#resp= dict(resp) | |
#print("TMDB Response :",resp) | |
response.append(resp) | |
else: | |
print("\n * Skipped:",row['file']['caption'],str(row['file']['file_name'])) | |
continue | |
except Exception as e: | |
print("Error ",e) | |
break | |
return web.json_response(json.loads(dumps(response)), headers=CORS_HEADERS) | |
#api.router.add_get('/tmdb/list', list_all_files_tmdb) | |
async def list_all_files_tmdb(request): | |
db = Database(Telegram.DATABASE_URL, Telegram.SESSION_NAME) | |
files= await db.get_all_files() | |
async def data_generator(): | |
async for row in files: | |
#print(row['file']['caption']) | |
try : | |
#print("* Response",search_tmdb(row['file']['caption'] if row['file']['caption'] else row['file']['file_name'])) | |
resp = search_tmdb_any( str(row['file']['caption']) if str(row['file']['caption']) else str(row['file']['file_name'])) | |
if resp != None : | |
#resp= dict(resp) | |
#print("TMDB Response :",resp) | |
#response.append(resp) | |
yield json.dumps(resp) + + '\n' | |
else: | |
print("\n * Skipped:",row['file']['caption'],str(row['file']['file_name'])) | |
continue | |
except Exception as e: | |
print("Error ",e) | |
break | |
return web.Response(body=data_generator(), content_type='application/json', headers=CORS_HEADERS) | |
#api.router.add_get('/tmdb/files', list_all_files) | |
async def list_all_files(request): | |
db = Database(Telegram.DATABASE_URL, Telegram.SESSION_NAME) | |
files=await db.get_all_files_api() | |
resp=[{ | |
"adult": False, | |
"backdrop_path": "/c1bz69r0v065TGFA5nqBiKzPDys.jpg", | |
"genre_ids": [ | |
35, | |
10751, | |
10402 | |
], | |
"id": f"{row['_id']}", | |
"original_language": "en-hi", | |
"original_title": f"{str(row['file']['caption'])}", | |
"overview": "XRepo Movies", | |
"popularity": 1710.176, | |
"poster_path": "/irIS5Tn3TXjNi1R9BpWvGAN4CZ1.jpg", | |
"release_date": "2022-10-07", | |
"title": f"{str(row['file']['caption'])}", | |
"link": f"{Server.URL}api/dl/{row['_id']}", | |
"vote_average": 7.8, | |
"vote_count": 122, | |
} | |
for row in files] | |
return web.json_response(json.loads(dumps(resp)), headers=CORS_HEADERS) | |
#----------------------------------------Upload----------------------------------------------# | |
#@routes.post("/upload") | |
async def upload_file(request: web.Request): | |
data = await request.post() | |
file = data.get('file').file | |
chunk = file.read() | |
""" | |
user_id : | |
file_id :"BAACAgUAAxkBAAJBHGYI_aJSvyL_ijKwrVHQVzFgC1YZAAItDQACOnNAVMIpWwl6b63EHg…" | |
file_unique_id :"AgADLQ0AAjpzQFQ" | |
file_name :"Dadur_Kirti_S01_COMBINED_720p_HOICHOI_WEB_DL_Bengali@COOL_MOVIES.mp4" | |
file_size : 1354816011 | |
mime_type : "video/mp4" | |
time : 1711865251.1016757 | |
user_type: "TELEGRAM"/"WEB" | |
privacy_type: "PUBLIC"/"PRIVATE" | |
file_ids : | |
""" | |
file_details = dict(user_id="thebinary1", | |
dropzone_id=str(data["dzuuid"]), | |
file=dict(file_id=str(data["dzuuid"]), | |
file_unique_id=str(data["dzuuid"]), | |
file_name=str(data.get('file').filename), | |
file_size=int(data["dztotalfilesize"]), | |
mime_type=mime_identifier(str(data.get('file').filename)), | |
part_size=int(data["dzchunksize"]), | |
file_part=int(data["dzchunkindex"]), | |
total_parts=int(data["dztotalchunkcount"])), | |
time=Time_ISTKolNow(), | |
user_type="WEB", | |
privacy_type="PRIVATE") | |
print(file_details) if (file_details["file"]["file_part"] == 0) else None | |
client_req = await req_client() | |
#faster_client = client_req["client"] | |
#index = client_req["index"] | |
print("using :", client_req["index"]) | |
tg_connect = TeleUploader(client_req["client"]) | |
main = await tg_connect.upload_web_file(file_details, chunk) | |
#print("Response:", main) | |
return web.json_response({ | |
"status": main.get("status"), | |
"message": main.get("message") | |
}) | |
#-------------Routes to Downloada File Witha Path-----------------# | |
#@routes.get("/dl/{path}", allow_head=True) | |
async def stream_handler(request: web.Request): | |
try: | |
path = request.match_info["path"] | |
return await media_streamer(request, path, "FAST") | |
except InvalidHash as e: | |
raise web.HTTPForbidden(text=e.message) | |
except FIleNotFound as e: | |
raise web.HTTPNotFound(text=e.message) | |
except (AttributeError, BadStatusLine, ConnectionResetError): | |
pass | |
except Exception as e: | |
traceback.print_exc() | |
logging.critical(e.with_traceback(None)) | |
logging.debug(traceback.format_exc()) | |
raise web.HTTPInternalServerError(text=str(e)) | |
api = web.Application() | |
cors = aiohttp_cors.setup(api, defaults={ | |
"*": aiohttp_cors.ResourceOptions( | |
allow_credentials=False, | |
expose_headers="*", | |
allow_headers="*", | |
allow_methods="*" #["GET"] Allowing specific methods | |
) | |
}) | |
cors.add(api.router.add_get('/', handle_v2)) | |
api.router.add_get('/files', list_all_files_db) | |
api.router.add_get('/files/mix', list_all_files) | |
api.router.add_get('/tmdb/mix', list_all_files_tmdb) | |
api.router.add_get('/10/files', list_10_all_files_db) | |
api.router.add_get('/tmdb/tv', list_all_tmdb_tv_from_db) | |
api.router.add_get('/tmdb/movies', list_all_tmdb_movies_from_db) | |
api.router.add_get('/upload', upload_file) | |
api.router.add_get('/dl/{path}', stream_handler) |