Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5import datetime
6 
7app = FastAPI()
8 
9users = {}
10articles = {}
11edit_history = {}
12tokens = {}
13next_user_id = 1
14next_article_id = 1
15next_history_id = 1
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20    role: str = "editor"
21 
22class LoginRequest(BaseModel):
23    username: str
24    password: str
25 
26class ArticleCreate(BaseModel):
27    title: str
28    content: str
29    author_name: str
30    category: str
31    status: str = "draft"
32 
33class ArticleUpdate(BaseModel):
34    status: Optional[str] = None
35    author_name: Optional[str] = None
36    category: Optional[str] = None
37    editor_role: Optional[str] = None
38 
39def get_current_user(authorization: str = Header(None)):
40    if not authorization:
41        raise HTTPException(status_code=401, detail="Missing auth token")
42    token = authorization.replace("Bearer ", "")
43    if token not in tokens:
44        raise HTTPException(status_code=401, detail="Invalid token")
45    return tokens[token]
46 
47@app.post("/signup")
48def signup(req: SignupRequest):
49    global next_user_id
50    for u in users.values():
51        if u["username"] == req.username:
52            raise HTTPException(status_code=400, detail="User exists")
53    user_id = next_user_id
54    next_user_id += 1
55    users[user_id] = {
56        "id": user_id,
57        "username": req.username,
58        "password": req.password,
59        "role": req.role
60    }
61    return {"id": user_id, "username": req.username, "role": req.role}
62 
63@app.post("/login")
64def login(req: LoginRequest):
65    for user in users.values():
66        if user["username"] == req.username and user["password"] == req.password:
67            token = secrets.token_hex(16)
68            tokens[token] = user
69            return {"token": token}
70    raise HTTPException(status_code=401, detail="Invalid credentials")
71 
72@app.post("/articles")
73def create_article(article: ArticleCreate, authorization: str = Header(None)):
74    user = get_current_user(authorization)
75    global next_article_id, next_history_id
76    article_id = next_article_id
77    next_article_id += 1
78    now = datetime.datetime.utcnow().isoformat()
79    articles[article_id] = {
80        "id": article_id,
81        "title": article.title,
82        "content": article.content,
83        "author_name": article.author_name,
84        "category": article.category,
85        "status": article.status,
86        "created_by": user["id"],
87        "created_at": now,
88        "updated_at": now
89    }
90    edit_history[article_id] = []
91    history_entry = {
92        "id": next_history_id,
93        "article_id": article_id,
94        "edited_by": user["id"],
95        "timestamp": now,
96        "changes": {
97            "status": article.status,
98            "author_name": article.author_name,
99            "category": article.category,
100            "editor_role": user["role"]
101        }
102    }
103    next_history_id += 1
104    edit_history[article_id].append(history_entry)
105    return articles[article_id]
106 
107@app.get("/articles/{article_id}")
108def get_article(article_id: int, authorization: str = Header(None)):
109    user = get_current_user(authorization)
110    if article_id not in articles:
111        raise HTTPException(status_code=404, detail="Article not found")
112    return articles[article_id]
113 
114@app.patch("/articles/{article_id}")
115def update_article(article_id: int, update: ArticleUpdate, authorization: str = Header(None)):
116    user = get_current_user(authorization)
117    if article_id not in articles:
118        raise HTTPException(status_code=404, detail="Article not found")
119    global next_history_id
120    article = articles[article_id]
121    now = datetime.datetime.utcnow().isoformat()
122    changes = {}
123    if update.status is not None:
124        changes["status"] = update.status
125        article["status"] = update.status
126    if update.author_name is not None:
127        changes["author_name"] = update.author_name
128        article["author_name"] = update.author_name
129    if update.category is not None:
130        changes["category"] = update.category
131        article["category"] = update.category
132    if update.editor_role is not None:
133        changes["editor_role"] = update.editor_role
134    changes["editor_role"] = user["role"]
135    article["updated_at"] = now
136    history_entry = {
137        "id": next_history_id,
138        "article_id": article_id,
139        "edited_by": user["id"],
140        "timestamp": now,
141        "changes": changes
142    }
143    next_history_id += 1
144    edit_history[article_id].append(history_entry)
145    return article
146 
147@app.get("/articles/{article_id}/history")
148def get_article_history(article_id: int, authorization: str = Header(None)):
149    user = get_current_user(authorization)
150    if article_id not in articles:
151        raise HTTPException(status_code=404, detail="Article not found")
152    return edit_history.get(article_id, [])

1fastapi
2uvicorn