Add Homepage watchlist summary endpoint

This commit is contained in:
Dymas
2026-06-07 02:15:13 +02:00
parent 18d82a8678
commit bff36940ac
3 changed files with 37 additions and 0 deletions
+23
View File
@@ -1780,6 +1780,23 @@ class WatchlistStore:
}, },
} }
def homepage_summary(self):
with self.lock, self._connect() as conn:
rows = conn.execute(
"SELECT category, COUNT(*) AS count FROM watchlist GROUP BY category"
).fetchall()
counts = {key: 0 for key in WATCHLIST_CATEGORY_CHOICES}
for row in rows:
counts[normalize_watchlist_category(row["category"])] = int(row["count"] or 0)
total = sum(counts.values())
return {
"watchlist": counts["watching"],
"planned": counts["planned"],
"finished": counts["finished"],
"dropped": counts["dropped"],
"total": total,
}
def get(self, show_id): def get(self, show_id):
with self.lock, self._connect() as conn: with self.lock, self._connect() as conn:
row = conn.execute("SELECT * FROM watchlist WHERE show_id = ?", (str(show_id).strip(),)).fetchone() row = conn.execute("SELECT * FROM watchlist WHERE show_id = ?", (str(show_id).strip(),)).fetchone()
@@ -2407,6 +2424,11 @@ def get_watchlist(page=1, per_page=30, category=None):
return WATCHLIST.list(page=page, per_page=per_page, category=category) return WATCHLIST.list(page=page, per_page=per_page, category=category)
def get_watchlist_homepage_summary():
ensure_runtime()
return WATCHLIST.homepage_summary()
def update_watchlist_status(show_id, _mode=None): def update_watchlist_status(show_id, _mode=None):
ensure_runtime() ensure_runtime()
item = WATCHLIST.queue_refresh(show_id, source="manual") item = WATCHLIST.queue_refresh(show_id, source="manual")
@@ -2857,6 +2879,7 @@ Handler = build_handler_class(
ensure_runtime=ensure_runtime, ensure_runtime=ensure_runtime,
episode_list=episode_list, episode_list=episode_list,
get_config_snapshot=get_config_snapshot, get_config_snapshot=get_config_snapshot,
get_watchlist_homepage_summary=get_watchlist_homepage_summary,
get_watchlist=get_watchlist, get_watchlist=get_watchlist,
get_watchlist_refresh_status=get_watchlist_refresh_status, get_watchlist_refresh_status=get_watchlist_refresh_status,
http_error=HttpError, http_error=HttpError,
+4
View File
@@ -52,6 +52,7 @@ class HandlerContext:
ensure_runtime: object ensure_runtime: object
episode_list: object episode_list: object
get_config_snapshot: object get_config_snapshot: object
get_watchlist_homepage_summary: object
get_watchlist: object get_watchlist: object
get_watchlist_refresh_status: object get_watchlist_refresh_status: object
http_error: object http_error: object
@@ -602,6 +603,9 @@ class Handler(BaseHTTPRequestHandler):
return return
debug_log("thumbnail.route.hit", show_id=show_id, resolved_path=path) debug_log("thumbnail.route.hit", show_id=show_id, resolved_path=path)
self.file(path) self.file(path)
elif parsed.path == "/api/watchlist/homepage":
Handler._runtime(self)
self.json(Handler._context(self).get_watchlist_homepage_summary())
elif parsed.path in {"/api/watchlist", "/api/watchlist/get"}: elif parsed.path in {"/api/watchlist", "/api/watchlist/get"}:
Handler._runtime(self) Handler._runtime(self)
params = parse_qs(parsed.query) params = parse_qs(parsed.query)
+10
View File
@@ -3052,6 +3052,16 @@ class HandlerRouteTests(unittest.TestCase):
self.assertEqual(handler.json_payload, payload) self.assertEqual(handler.json_payload, payload)
self.assertEqual(handler.json_status, HTTPStatus.OK) self.assertEqual(handler.json_status, HTTPStatus.OK)
def test_watchlist_homepage_route_returns_summary_counts(self):
handler = DummyHandler("/api/watchlist/homepage")
payload = {"watchlist": 1, "planned": 0, "finished": 2, "dropped": 0, "total": 3}
handler.handler_context = mock.Mock(get_watchlist_homepage_summary=mock.Mock(return_value=payload))
APP.Handler.do_GET(handler)
self.assertEqual(handler.json_payload, payload)
self.assertEqual(handler.json_status, HTTPStatus.OK)
def test_update_all_post_returns_accepted_for_new_job(self): def test_update_all_post_returns_accepted_for_new_job(self):
handler = DummyHandler("/api/watchlist/update-all", body=b"{}", content_length=2) handler = DummyHandler("/api/watchlist/update-all", body=b"{}", content_length=2)
payload = {"message": "Watchlist refresh started.", "job": {"id": "refresh-1"}, "created": True} payload = {"message": "Watchlist refresh started.", "job": {"id": "refresh-1"}, "created": True}