Refactor Plex interaction into separate module
This commit is contained in:
+5
-1
@@ -1,3 +1,7 @@
|
||||
{
|
||||
"theme": "light"
|
||||
"theme": "light",
|
||||
"token": "",
|
||||
"server_url": "",
|
||||
"server_port": "",
|
||||
"library": ""
|
||||
}
|
||||
+25
-5
@@ -4,6 +4,7 @@ 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 connect_plex
|
||||
|
||||
app = FastAPI()
|
||||
templates = Jinja2Templates(directory=os.path.join(os.path.dirname(__file__), "templates"))
|
||||
@@ -36,11 +37,30 @@ async def login(
|
||||
):
|
||||
config = load_config()
|
||||
theme = config.get("theme", "auto")
|
||||
# demo:假装连接成功
|
||||
if user == "admin":
|
||||
return templates.TemplateResponse("login.html", {"request": request, "message": "连接成功", "success": True, "theme": theme, "path": "/login"})
|
||||
else:
|
||||
return templates.TemplateResponse("login.html", {"request": request, "message": "连接失败:用户名错误", "success": False, "theme": theme, "path": "/login"})
|
||||
try:
|
||||
connect_plex(config, user, pw, url, port, library)
|
||||
save_config(config)
|
||||
return templates.TemplateResponse(
|
||||
"login.html",
|
||||
{
|
||||
"request": request,
|
||||
"message": "连接成功",
|
||||
"success": True,
|
||||
"theme": theme,
|
||||
"path": "/login",
|
||||
},
|
||||
)
|
||||
except Exception as e:
|
||||
return templates.TemplateResponse(
|
||||
"login.html",
|
||||
{
|
||||
"request": request,
|
||||
"message": f"连接失败:{str(e)}",
|
||||
"success": False,
|
||||
"theme": theme,
|
||||
"path": "/login",
|
||||
},
|
||||
)
|
||||
|
||||
@app.get("/playlist", response_class=HTMLResponse)
|
||||
async def get_playlist(request: Request):
|
||||
|
||||
@@ -0,0 +1,16 @@
|
||||
from plexapi.myplex import MyPlexAccount
|
||||
from plexapi.server import PlexServer
|
||||
|
||||
|
||||
def connect_plex(config, username, password, url, port, library):
|
||||
"""Return a connected PlexServer instance and update config with token and server info."""
|
||||
token = config.get("token")
|
||||
if not token:
|
||||
account = MyPlexAccount(username, password)
|
||||
token = account.authenticationToken
|
||||
config["token"] = token
|
||||
|
||||
base_url = f"http://{url}:{port}"
|
||||
server = PlexServer(base_url, token)
|
||||
config.update({"server_url": url, "server_port": port, "library": library})
|
||||
return server
|
||||
Reference in New Issue
Block a user