Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3import hashlib
4import random
5import string
6from datetime import datetime
7 
8app = FastAPI()
9 
10# In-memory stores
11users = {}
12tokens = {}
13news_articles = {}
14article_id_counter = 1
15topics = {}
16 
17# Simple auth token generation
18def generate_token():
19    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
20 
21def get_user_from_token(authorization: str = Header(None)):
22    if not authorization:
23        raise HTTPException(status_code=401, detail="Missing auth token")
24    token = authorization.replace("Bearer ", "")
25    user_id = tokens.get(token)
26    if not user_id:
27        raise HTTPException(status_code=401, detail="Invalid token")
28    return user_id
29 
30class SignupRequest(BaseModel):
31    username: str
32    password: str
33 
34class LoginRequest(BaseModel):
35    username: str
36    password: str
37 
38class NewsArticleCreate(BaseModel):
39    title: str
40    source: str
41    summary: str
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    if req.username in users:
46        raise HTTPException(status_code=400, detail="User already exists")
47    users[req.username] = hashlib.sha256(req.password.encode()).hexdigest()
48    return {"message": "User created"}
49 
50@app.post("/login")
51def login(req: LoginRequest):
52    if req.username not in users:
53        raise HTTPException(status_code=400, detail="Invalid credentials")
54    if users[req.username] != hashlib.sha256(req.password.encode()).hexdigest():
55        raise HTTPException(status_code=400, detail="Invalid credentials")
56    token = generate_token()
57    tokens[token] = req.username
58    return {"token": token}
59 
60@app.get("/news/headlines")
61def get_headlines(authorization: str = Header(None)):
62    user_id = get_user_from_token(authorization)
63    headlines = []
64    for aid, article in news_articles.items():
65        headlines.append({
66            "id": aid,
67            "title": article["title"],
68            "source": article["source"],
69            "publish_time": article["publish_time"],
70            "summary": article["summary"]
71        })
72    return {"headlines": headlines}
73 
74@app.get("/news/topics")
75def get_topics(authorization: str = Header(None)):
76    user_id = get_user_from_token(authorization)
77    topic_list = []
78    for topic, count in topics.items():
79        topic_list.append({"topic": topic, "article_count": count})
80    return {"topics": topic_list}
81 
82@app.post("/news/article")
83def create_article(article: NewsArticleCreate, authorization: str = Header(None)):
84    global article_id_counter
85    user_id = get_user_from_token(authorization)
86    aid = article_id_counter
87    article_id_counter += 1
88    news_articles[aid] = {
89        "title": article.title,
90        "source": article.source,
91        "summary": article.summary,
92        "publish_time": datetime.now().isoformat()
93    }
94    # Update topics
95    words = article.title.split() + article.summary.split()
96    for word in words:
97        word_lower = word.lower().strip(".,!?")
98        if word_lower in topics:
99            topics[word_lower] += 1
100        else:
101            topics[word_lower] = 1
102    return {"id": aid}
103 
104@app.get("/news/article/{article_id}")
105def get_article(article_id: int, authorization: str = Header(None)):
106    user_id = get_user_from_token(authorization)
107    article = news_articles.get(article_id)
108    if not article:
109        raise HTTPException(status_code=404, detail="Article not found")
110    return article

1fastapi
2uvicorn