From 4aaf5997dfae35754362bc1b882842d2e565a0b8 Mon Sep 17 00:00:00 2001
From: thsrite <thsrite@gmail.com>
Date: Tue, 24 Oct 2023 11:17:32 +0800
Subject: [PATCH] =?UTF-8?q?fix=20=E7=99=BB=E5=BD=95=E9=A1=B5=E6=B5=B7?=
 =?UTF-8?q?=E6=8A=A5=E6=94=AF=E6=8C=81=E8=87=AA=E5=AE=9A=E4=B9=89tmdb/bing?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 README.md                  |  1 +
 app/api/endpoints/login.py | 20 ++++++++++++++++++++
 app/core/config.py         |  2 ++
 config/app.env             |  2 ++
 4 files changed, 25 insertions(+)

diff --git a/README.md b/README.md
index dca9755e..2f3fcf51 100644
--- a/README.md
+++ b/README.md
@@ -60,6 +60,7 @@ MoviePilot需要配套下载器和媒体服务器配合使用。
 - **PUID**：运行程序用户的`uid`，默认`0`（仅支持环境变量配置）
 - **PGID**：运行程序用户的`gid`，默认`0`（仅支持环境变量配置）
 - **UMASK**：掩码权限，默认`000`，可以考虑设置为`022`（仅支持环境变量配置）
+- **WALLPAPER：** 登录首页电影海报，`tmdb`/`bing`，默认`tmdb`
 - **PROXY_HOST：** 网络代理，访问themoviedb或者重启更新需要使用代理访问，格式为`http(s)://ip:port`、`socks5://user:pass@host:port`（仅支持环境变量配置）
 - **MOVIEPILOT_AUTO_UPDATE**：重启更新，`true`/`false`，默认`true` **注意：如果出现网络问题可以配置`PROXY_HOST`**（仅支持环境变量配置）
 - **MOVIEPILOT_AUTO_UPDATE_DEV**：重启时更新到未发布的开发版本代码，`true`/`false`，默认`false`（仅支持环境变量配置）
diff --git a/app/api/endpoints/login.py b/app/api/endpoints/login.py
index d9f99546..14f7bf08 100644
--- a/app/api/endpoints/login.py
+++ b/app/api/endpoints/login.py
@@ -61,6 +61,26 @@ async def login_access_token(
     )
 
 
+@router.get("/wallpaper", summary="登录页面电影海报", response_model=schemas.Response)
+def wallpaper() -> Any:
+    """
+    获取登录页面电影海报
+    """
+    if settings.WALLPAPER == "tmdb":
+        wallpager = TmdbChain().get_random_wallpager()
+        if wallpager:
+            return schemas.Response(
+                success=True,
+                message=wallpager
+            )
+    elif settings.WALLPAPER == "bing":
+        url = WebUtils.get_bing_wallpaper()
+        if url:
+            return schemas.Response(success=False,
+                                    message=url)
+    return schemas.Response(success=False)
+
+
 @router.get("/bing", summary="Bing每日壁纸", response_model=schemas.Response)
 def bing_wallpaper() -> Any:
     """
diff --git a/app/core/config.py b/app/core/config.py
index 02af7e67..4f657ffc 100644
--- a/app/core/config.py
+++ b/app/core/config.py
@@ -39,6 +39,8 @@ class Settings(BaseSettings):
     SUPERUSER_PASSWORD: str = "password"
     # API密钥，需要更换
     API_TOKEN: str = "moviepilot"
+    # 登录页面电影海报,tmdb/bing
+    WALLPAPER: str = "tmdb"
     # 网络代理 IP:PORT
     PROXY_HOST: str = None
     # 媒体信息搜索来源
diff --git a/config/app.env b/config/app.env
index c7118a2c..9904b0fe 100644
--- a/config/app.env
+++ b/config/app.env
@@ -17,6 +17,8 @@ SUPERUSER=admin
 SUPERUSER_PASSWORD=password
 # 【*】API密钥，建议更换复杂字符串
 API_TOKEN=moviepilot
+# 登录页面电影海报,tmdb/bing
+WALLPAPER=tmdb
 # TMDB图片地址，无需修改需保留默认值
 TMDB_IMAGE_DOMAIN=image.tmdb.org
 # TMDB API地址，无需修改需保留默认值