Update main.py

This commit is contained in:
TylerCG 2025-04-26 00:23:02 -04:00
parent 26dc647fb8
commit fd3b5dfb4a

View File

@ -1,4 +1,4 @@
from fastapi import FastAPI, Request from fastapi import FastAPI, Request, Form
from fastapi.responses import HTMLResponse, JSONResponse from fastapi.responses import HTMLResponse, JSONResponse
from fastapi.templating import Jinja2Templates from fastapi.templating import Jinja2Templates
# from fastapi.staticfiles import StaticFiles # from fastapi.staticfiles import StaticFiles
@ -6,6 +6,7 @@ from fastapi.templating import Jinja2Templates
from pathlib import Path from pathlib import Path
from functools import partial from functools import partial
import json, download, asyncio import json, download, asyncio
from typing import Optional
app = FastAPI() app = FastAPI()
@ -33,7 +34,11 @@ async def dropoutSeries():
return JSONResponse(content={"error": "File not found"}, status_code=404) return JSONResponse(content={"error": "File not found"}, status_code=404)
@app.post("/dropout/download") @app.post("/dropout/download")
async def dropoutDownload(show: str, season: str, episode: str = None): async def dropoutDownload(
show: str = Form(...),
season: int = Form(...),
episode: Optional[int] = Form(None)
):
try: try:
await loop.run_in_executor(None, partial(download.dropout.show,show,season,episode)) await loop.run_in_executor(None, partial(download.dropout.show,show,season,episode))
# download.dropout.show(show,season,episode) # download.dropout.show(show,season,episode)
@ -42,7 +47,11 @@ async def dropoutDownload(show: str, season: str, episode: str = None):
return JSONResponse(status_code=500, content={"status": "error", "message": str(e)}) return JSONResponse(status_code=500, content={"status": "error", "message": str(e)})
@app.post("/dropout/download/specials") @app.post("/dropout/download/specials")
async def dropoutDownload(show: str, season: str, episode: str = None): async def dropoutDownload(
show: str = Form(...),
season: int = Form(...),
episode: Optional[int] = Form(None)
):
try: try:
await loop.run_in_executor(None, partial(download.dropout.specials,show,season,episode)) await loop.run_in_executor(None, partial(download.dropout.specials,show,season,episode))
# download.dropout.show(show,season,episode) # download.dropout.show(show,season,episode)