Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4import datetime
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10articles = {}
11promotions = {}
12curators = {}
13featured_posts = {}
14topics = {}
15next_user_id = 1
16next_article_id = 1
17next_promotion_id = 1
18next_topic_id = 1
19 
20def get_current_user(authorization: Optional[str] = Header(None)):
21    if not authorization:
22        raise HTTPException(status_code=401, detail="No auth token")
23    token = authorization.replace("Bearer ", "")
24    user_id = tokens.get(token)
25    if not user_id:
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return user_id
28 
29@app.post("/signup")
30def signup(email: str, password: str, name: str = ""):
31    global next_user_id
32    if email in [u["email"] for u in users.values()]:
33        raise HTTPException(status_code=400, detail="Email already exists")
34    user_id = next_user_id
35    next_user_id += 1
36    users[user_id] = {"id": user_id, "email": email, "password": password, "name": name, "role": "submitter"}
37    return {"user_id": user_id, "email": email}
38 
39@app.post("/login")
40def login(email: str, password: str):
41    for user_id, user in users.items():
42        if user["email"] == email and user["password"] == password:
43            token = secrets.token_hex(16)
44            tokens[token] = user_id
45            return {"token": token, "user_id": user_id}
46    raise HTTPException(status_code=401, detail="Invalid credentials")
47 
48@app.get("/users/{user_id}")
49def get_user(user_id: int):
50    user = users.get(user_id)
51    if not user:
52        raise HTTPException(status_code=404, detail="User not found")
53    return user
54 
55@app.post("/articles")
56def create_article(title: str, url: str, summary: str = "", authorization: Optional[str] = Header(None)):
57    submitter_id = get_current_user(authorization)
58    global next_article_id
59    article_id = next_article_id
60    next_article_id += 1
61    articles[article_id] = {
62        "id": article_id,
63        "title": title,
64        "url": url,
65        "summary": summary,
66        "submitter_id": submitter_id,
67        "featured": False,
68        "topics": [],
69        "created_at": datetime.datetime.now().isoformat()
70    }
71    return articles[article_id]
72 
73@app.get("/articles/{article_id}")
74def get_article(article_id: int):
75    article = articles.get(article_id)
76    if not article:
77        raise HTTPException(status_code=404, detail="Article not found")
78    return article
79 
80@app.post("/promote")
81def promote_reader(target_user_id: int, authorization: Optional[str] = Header(None)):
82    submitter_id = get_current_user(authorization)
83    submitter = users.get(submitter_id)
84    if not submitter or submitter["role"] != "submitter":
85        raise HTTPException(status_code=403, detail="Only submitters can promote")
86    target_user = users.get(target_user_id)
87    if not target_user:
88        raise HTTPException(status_code=404, detail="User not found")
89    target_user["role"] = "curator"
90    curators[target_user_id] = target_user
91    return {"message": f"User {target_user_id} promoted to curator", "user": target_user}
92 
93@app.post("/feature")
94def feature_post(article_id: int, topic: str = "", authorization: Optional[str] = Header(None)):
95    curator_id = get_current_user(authorization)
96    curator = users.get(curator_id)
97    if not curator or curator["role"] != "curator":
98        raise HTTPException(status_code=403, detail="Only curators can feature")
99    article = articles.get(article_id)
100    if not article:
101        raise HTTPException(status_code=404, detail="Article not found")
102    article["featured"] = True
103    if topic:
104        global next_topic_id
105        topic_id = next_topic_id
106        next_topic_id += 1
107        topics[topic_id] = {"id": topic_id, "name": topic, "article_id": article_id}
108        article["topics"].append(topic)
109    featured_posts[article_id] = article
110    return {"message": "Article featured", "article": article}
111 
112@app.post("/tag")
113def tag_article(article_id: int, topic: str, authorization: Optional[str] = Header(None)):
114    curator_id = get_current_user(authorization)
115    curator = users.get(curator_id)
116    if not curator or curator["role"] != "curator":
117        raise HTTPException(status_code=403, detail="Only curators can tag")
118    article = articles.get(article_id)
119    if not article:
120        raise HTTPException(status_code=404, detail="Article not found")
121    global next_topic_id
122    topic_id = next_topic_id
123    next_topic_id += 1
124    topics[topic_id] = {"id": topic_id, "name": topic, "article_id": article_id}
125    article["topics"].append(topic)
126    return {"message": "Topic added", "article": article}

1fastapi
2uvicorn