Initial commit — YT Hub
Self-hosted personal YouTube management app. FastAPI + SQLite backend, React + Vite + Tailwind frontend. Dockerfiles and compose included for Portainer deployment. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
144
backend/routers/stats.py
Normal file
144
backend/routers/stats.py
Normal file
@@ -0,0 +1,144 @@
|
||||
from fastapi import APIRouter, Depends
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy import text
|
||||
|
||||
from ..auth_utils import get_current_user
|
||||
from ..database import get_db
|
||||
from ..models import User
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
@router.get("")
|
||||
def get_stats(
|
||||
db: Session = Depends(get_db),
|
||||
current_user: User = Depends(get_current_user),
|
||||
):
|
||||
uid = current_user.id
|
||||
|
||||
totals = db.execute(
|
||||
text("""
|
||||
SELECT
|
||||
COUNT(*) AS total_watched,
|
||||
SUM(uv.watch_progress_seconds) AS total_watch_seconds
|
||||
FROM user_videos uv
|
||||
WHERE uv.user_id = :uid AND uv.watched = 1
|
||||
"""),
|
||||
{"uid": uid},
|
||||
).mappings().first()
|
||||
|
||||
top_channels = db.execute(
|
||||
text("""
|
||||
SELECT c.id, c.name,
|
||||
COUNT(*) AS watch_count,
|
||||
SUM(uv.watch_progress_seconds) AS watch_seconds
|
||||
FROM user_videos uv
|
||||
JOIN videos v ON uv.video_id = v.id
|
||||
JOIN channels c ON v.channel_id = c.id
|
||||
WHERE uv.user_id = :uid AND uv.watched = 1
|
||||
GROUP BY c.id, c.name
|
||||
ORDER BY watch_seconds DESC
|
||||
LIMIT 10
|
||||
"""),
|
||||
{"uid": uid},
|
||||
).mappings().all()
|
||||
|
||||
daily = db.execute(
|
||||
text("""
|
||||
SELECT date(uv.last_watched_at) AS date,
|
||||
COUNT(*) AS count,
|
||||
SUM(uv.watch_progress_seconds) AS seconds
|
||||
FROM user_videos uv
|
||||
WHERE uv.user_id = :uid
|
||||
AND uv.watched = 1
|
||||
AND uv.last_watched_at >= datetime('now', '-30 days')
|
||||
GROUP BY date(uv.last_watched_at)
|
||||
ORDER BY date ASC
|
||||
"""),
|
||||
{"uid": uid},
|
||||
).mappings().all()
|
||||
|
||||
this_week = db.execute(
|
||||
text("""
|
||||
SELECT COUNT(*) AS count, SUM(uv.watch_progress_seconds) AS seconds
|
||||
FROM user_videos uv
|
||||
WHERE uv.user_id = :uid AND uv.watched = 1
|
||||
AND uv.last_watched_at >= datetime('now', '-7 days')
|
||||
"""),
|
||||
{"uid": uid},
|
||||
).mappings().first()
|
||||
|
||||
this_month = db.execute(
|
||||
text("""
|
||||
SELECT COUNT(*) AS count, SUM(uv.watch_progress_seconds) AS seconds
|
||||
FROM user_videos uv
|
||||
WHERE uv.user_id = :uid AND uv.watched = 1
|
||||
AND uv.last_watched_at >= datetime('now', '-30 days')
|
||||
"""),
|
||||
{"uid": uid},
|
||||
).mappings().first()
|
||||
|
||||
avg_completion = db.execute(
|
||||
text("""
|
||||
SELECT AVG(uv.completion_percent) AS avg_pct,
|
||||
COUNT(CASE WHEN uv.completion_percent >= 90 THEN 1 END) AS finished_count,
|
||||
COUNT(CASE WHEN uv.completion_percent < 20 AND uv.completion_percent IS NOT NULL THEN 1 END) AS bailed_count,
|
||||
SUM(uv.rewatch_count) AS total_rewatches,
|
||||
COUNT(CASE WHEN uv.rewatch_count > 0 THEN 1 END) AS rewatched_videos
|
||||
FROM user_videos uv
|
||||
WHERE uv.user_id = :uid AND uv.watched = 1
|
||||
"""),
|
||||
{"uid": uid},
|
||||
).mappings().first()
|
||||
|
||||
top_categories = db.execute(
|
||||
text("""
|
||||
SELECT v.category, COUNT(*) AS watch_count,
|
||||
AVG(uv.completion_percent) AS avg_completion
|
||||
FROM user_videos uv
|
||||
JOIN videos v ON uv.video_id = v.id
|
||||
WHERE uv.user_id = :uid AND uv.watched = 1 AND v.category IS NOT NULL
|
||||
GROUP BY v.category
|
||||
ORDER BY watch_count DESC
|
||||
LIMIT 8
|
||||
"""),
|
||||
{"uid": uid},
|
||||
).mappings().all()
|
||||
|
||||
taste_profile = db.execute(
|
||||
text("""
|
||||
SELECT tag, score FROM user_tag_affinity
|
||||
WHERE user_id = :uid AND score > 0
|
||||
ORDER BY score DESC
|
||||
LIMIT 20
|
||||
"""),
|
||||
{"uid": uid},
|
||||
).mappings().all()
|
||||
|
||||
liked_count = db.execute(
|
||||
text("SELECT COUNT(*) AS n FROM user_videos WHERE user_id = :uid AND liked = 1"),
|
||||
{"uid": uid},
|
||||
).mappings().first()
|
||||
|
||||
return {
|
||||
"total_watched": totals["total_watched"] or 0,
|
||||
"total_watch_seconds": totals["total_watch_seconds"] or 0,
|
||||
"top_channels": [dict(r) for r in top_channels],
|
||||
"daily": [dict(r) for r in daily],
|
||||
"this_week": {
|
||||
"count": this_week["count"] or 0,
|
||||
"seconds": this_week["seconds"] or 0,
|
||||
},
|
||||
"this_month": {
|
||||
"count": this_month["count"] or 0,
|
||||
"seconds": this_month["seconds"] or 0,
|
||||
},
|
||||
"avg_completion_percent": round(avg_completion["avg_pct"] or 0, 1),
|
||||
"finished_count": avg_completion["finished_count"] or 0,
|
||||
"bailed_count": avg_completion["bailed_count"] or 0,
|
||||
"total_rewatches": avg_completion["total_rewatches"] or 0,
|
||||
"rewatched_videos": avg_completion["rewatched_videos"] or 0,
|
||||
"total_liked": liked_count["n"] or 0,
|
||||
"top_categories": [dict(r) for r in top_categories],
|
||||
"taste_profile": [dict(r) for r in taste_profile],
|
||||
}
|
||||
Reference in New Issue
Block a user