Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3from typing import Optional
4import uuid
5import random
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11videos = {}
12comments = {}
13video_id_counter = 1
14user_id_counter = 1
15comment_id_counter = 1
16 
17def get_current_user(authorization: Optional[str] = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="No auth token")
20    token = authorization.replace("Bearer ", "")
21    if token not in tokens:
22        raise HTTPException(status_code=401, detail="Invalid token")
23    return tokens[token]
24 
25@app.post("/signup")
26def signup(username: str, password: str):
27    global user_id_counter
28    if username in users:
29        raise HTTPException(status_code=400, detail="Username exists")
30    user_id = user_id_counter
31    user_id_counter += 1
32    users[username] = {"id": user_id, "username": username, "password": password}
33    return {"id": user_id, "username": username}
34 
35@app.post("/login")
36def login(username: str, password: str):
37    if username not in users or users[username]["password"] != password:
38        raise HTTPException(status_code=401, detail="Invalid credentials")
39    token = str(uuid.uuid4())
40    tokens[token] = users[username]["id"]
41    return {"token": token}
42 
43@app.post("/videos")
44def create_video(title: str, duration: int, authorization: Optional[str] = Header(None)):
45    global video_id_counter
46    user_id = get_current_user(authorization)
47    video_id = video_id_counter
48    video_id_counter += 1
49    videos[video_id] = {
50        "id": video_id,
51        "title": title,
52        "duration": duration,
53        "view_count": 0,
54        "likes": 0,
55        "dislikes": 0,
56        "comments": [],
57        "user_id": user_id,
58        "created_at": datetime.now()
59    }
60    return videos[video_id]
61 
62@app.get("/videos/{video_id}")
63def get_video(video_id: int):
64    if video_id not in videos:
65        raise HTTPException(status_code=404, detail="Video not found")
66    videos[video_id]["view_count"] += 1
67    return {
68        "id": video_id,
69        "title": videos[video_id]["title"],
70        "duration": videos[video_id]["duration"],
71        "view_count": videos[video_id]["view_count"]
72    }
73 
74@app.get("/videos")
75def list_videos():
76    return [
77        {
78            "id": vid["id"],
79            "title": vid["title"],
80            "duration": vid["duration"],
81            "view_count": vid["view_count"]
82        }
83        for vid in videos.values()
84    ]
85 
86@app.post("/comment")
87def add_comment(video_id: int, text: str, authorization: Optional[str] = Header(None)):
88    global comment_id_counter
89    user_id = get_current_user(authorization)
90    if video_id not in videos:
91        raise HTTPException(status_code=404, detail="Video not found")
92    comment_id = comment_id_counter
93    comment_id_counter += 1
94    comment = {
95        "id": comment_id,
96        "video_id": video_id,
97        "user_id": user_id,
98        "text": text,
99        "timestamp": datetime.now().isoformat()
100    }
101    comments[comment_id] = comment
102    videos[video_id]["comments"].append(comment_id)
103    return comment
104 
105@app.get("/trending")
106def get_trending():
107    one_week_ago = datetime.now() - timedelta(days=7)
108    trending = []
109    for vid in videos.values():
110        if vid["created_at"] < one_week_ago:
111            continue
112        total_engagement = vid["likes"] + vid["dislikes"] + len(vid["comments"])
113        if vid["view_count"] > 0:
114            engagement_rate = (total_engagement / vid["view_count"]) * 100
115            if engagement_rate > 10:
116                trending.append({
117                    "id": vid["id"],
118                    "title": vid["title"],
119                    "duration": vid["duration"],
120                    "view_count": vid["view_count"],
121                    "engagement_rate": round(engagement_rate, 2)
122                })
123    return trending

1fastapi
2uvicorn