fix #175 订阅更新总集数后缺失集数未更新的问题

fix #174
2023-08-19 09:33:46 +08:00
parent 0151b1d4f0
commit 5e06444ccc
2 changed files with 14 additions and 17 deletions
--- a/app/api/endpoints/subscribe.py
+++ b/app/api/endpoints/subscribe.py
@ -99,7 +99,14 @@ def update_subscribe(
    # 避免更新缺失集数
    subscribe_dict = subscribe_in.dict()
    if not subscribe_in.lack_episode:
+        # 没有缺失集数时，缺失集数清空，避免更新为0
        subscribe_dict.pop("lack_episode")
+    elif subscribe_in.total_episode:
+        # 总集数增加时，缺失集数也要增加
+        if subscribe_in.total_episode > (subscribe.total_episode or 0):
+            subscribe_dict["lack_episode"] = (subscribe.lack_episode
+                                              + (subscribe_in.total_episode
+                                                 - (subscribe.total_episode or 0)))
    subscribe.update(db, subscribe_dict)
    return schemas.Response(success=True)