145 lines
4.2 KiB
Python
145 lines
4.2 KiB
Python
import os
|
|
from app.utils.config import server_config
|
|
from fastapi import FastAPI, Request, Form
|
|
from fastapi.responses import HTMLResponse, RedirectResponse
|
|
from fastapi.templating import Jinja2Templates
|
|
from fastapi.staticfiles import StaticFiles
|
|
from app.utils.plex_client import plex_client
|
|
from app.utils.logger import logger
|
|
|
|
app = FastAPI()
|
|
templates = Jinja2Templates(
|
|
directory=os.path.join(os.path.dirname(__file__), "templates")
|
|
)
|
|
|
|
# mount static files
|
|
# 这里的路径是相对于 main.py 文件所在的目录
|
|
app.mount(
|
|
"/static",
|
|
StaticFiles(directory=os.path.join(os.path.dirname(__file__), "static")),
|
|
name="static",
|
|
)
|
|
|
|
|
|
# 显示主页
|
|
@app.get("/", response_class=HTMLResponse)
|
|
async def home(request: Request):
|
|
server_config.load()
|
|
return templates.TemplateResponse(
|
|
"login.html",
|
|
{
|
|
"request": request,
|
|
"theme": server_config.theme,
|
|
"path": "/login",
|
|
"scheme": server_config.scheme,
|
|
"token": server_config.token,
|
|
"server_url": server_config.url,
|
|
"port": server_config.port,
|
|
},
|
|
)
|
|
|
|
|
|
# 登录页面和处理
|
|
@app.get("/login", response_class=HTMLResponse)
|
|
async def login_page(request: Request):
|
|
server_config.load()
|
|
return templates.TemplateResponse(
|
|
"login.html",
|
|
{
|
|
"request": request,
|
|
"theme": server_config.theme,
|
|
"path": "/login",
|
|
"token": server_config.token,
|
|
"scheme": server_config.scheme,
|
|
"server_url": server_config.url,
|
|
"port": server_config.port,
|
|
},
|
|
)
|
|
|
|
|
|
@app.post("/login", response_class=HTMLResponse)
|
|
async def login(
|
|
request: Request,
|
|
user: str = Form(...),
|
|
pw: str = Form(...),
|
|
token: str = Form(...),
|
|
scheme: str = Form("https"),
|
|
url: str = Form(...),
|
|
port: str = Form("32400"),
|
|
):
|
|
# 尝试连接到 Plex 服务器
|
|
try:
|
|
# 优先使用 token 连接，如果 token 为空则使用用户名和密码连接
|
|
_, token_success = plex_client.connect(
|
|
username=user,
|
|
password=pw,
|
|
token=token,
|
|
scheme=scheme,
|
|
url=url,
|
|
port=port,
|
|
)
|
|
# 成功连接后保存配置到配置文件
|
|
if plex_client.connected:
|
|
server_config.set_and_save_config(
|
|
token=token_success, scheme=scheme, url=url, port=port
|
|
)
|
|
return templates.TemplateResponse(
|
|
"login.html",
|
|
{
|
|
"request": request,
|
|
"message": "连接成功",
|
|
"message_type": "success",
|
|
"theme": server_config.theme,
|
|
"path": "/login",
|
|
"token": server_config.token,
|
|
"scheme": server_config.scheme,
|
|
"server_url": server_config.url,
|
|
"port": server_config.port,
|
|
},
|
|
)
|
|
except Exception as e:
|
|
return templates.TemplateResponse(
|
|
"login.html",
|
|
{
|
|
"request": request,
|
|
"message": f"连接失败：{str(e)}",
|
|
"message_type": "danger",
|
|
"theme": server_config.theme,
|
|
"path": "/login",
|
|
"scheme": scheme,
|
|
"server_url": url,
|
|
"port": port,
|
|
},
|
|
)
|
|
|
|
|
|
@app.get("/playlist", response_class=HTMLResponse)
|
|
async def get_playlist(request: Request):
|
|
return templates.TemplateResponse(
|
|
"playlist.html",
|
|
{"request": request, "theme": server_config.theme, "path": "/playlist"},
|
|
)
|
|
|
|
|
|
@app.post("/playlist", response_class=HTMLResponse)
|
|
async def set_playlist(
|
|
request: Request, address: str = Form(...), interval: str = Form(...)
|
|
):
|
|
# demo：返回提交的设置
|
|
return templates.TemplateResponse(
|
|
"playlist.html",
|
|
{
|
|
"request": request,
|
|
"message": f"设置成功：地址 {address}，间隔 {interval} 分钟",
|
|
"message_type": "info",
|
|
"theme": server_config.theme,
|
|
"path": "/playlist",
|
|
},
|
|
)
|
|
|
|
|
|
@app.post("/set-theme")
|
|
async def set_theme(theme: str = Form(...)):
|
|
server_config.set_and_save_config(theme=theme)
|
|
return RedirectResponse("/login", status_code=303)
|