Browse the data

1import os
2import time
3from datetime import datetime, timedelta
4from pathlib import Path
5 
6from fastapi import FastAPI, HTTPException, Header
7from pydantic import BaseModel
8import uvicorn
9 
10app = FastAPI()
11 
12# In-memory user store
13users = {}
14user_id_counter = 1
15# In-memory token store: token -> user_id
16tokens = {}
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21 
22class LoginRequest(BaseModel):
23    username: str
24    password: str
25 
26class ResourceCreate(BaseModel):
27    pass
28 
29resources = {}
30resource_type_counters = {}
31 
32@app.post("/signup")
33def signup(req: SignupRequest):
34    global user_id_counter
35    if req.username in users:
36        raise HTTPException(status_code=400, detail="Username already exists")
37    users[req.username] = {"password": req.password, "id": user_id_counter}
38    user_id_counter += 1
39    return {"message": "User created", "user_id": users[req.username]["id"]}
40 
41@app.post("/login")
42def login(req: LoginRequest, authorization: str = Header(None)):
43    if req.username not in users or users[req.username]["password"] != req.password:
44        raise HTTPException(status_code=401, detail="Invalid credentials")
45    token = f"tok_{req.username}_{int(time.time())}"
46    tokens[token] = users[req.username]["id"]
47    return {"token": token}
48 
49def get_user_id(token: str):
50    if token.startswith("Bearer "):
51        token = token[7:]
52    if token not in tokens:
53        raise HTTPException(status_code=401, detail="Invalid token")
54    return tokens[token]
55 
56@app.get("/{resource_type}/{resource_id}")
57def get_resource(resource_type: str, resource_id: int, authorization: str = Header(None)):
58    user_id = get_user_id(authorization)
59    key = f"{resource_type}:{resource_id}"
60    if key not in resources:
61        raise HTTPException(status_code=404, detail="Resource not found")
62    return resources[key]
63 
64@app.post("/{resource_type}")
65def create_resource(resource_type: str, req: ResourceCreate, authorization: str = Header(None)):
66    user_id = get_user_id(authorization)
67    if resource_type not in resource_type_counters:
68        resource_type_counters[resource_type] = 1
69    else:
70        resource_type_counters[resource_type] += 1
71    resource_id = resource_type_counters[resource_type]
72    key = f"{resource_type}:{resource_id}"
73    resources[key] = {"id": resource_id, "data": req.dict()}
74    return {"id": resource_id}
75 
76# Image serving endpoint
77@app.get("/article/{date}/{file}")
78def serve_article_image(date: str, file: str, if_modified_since: str = Header(None)):
79    base_path = Path("/var/news") / date
80    file_path = base_path / file
81    
82    if not file_path.exists() or not file_path.is_file():
83        raise HTTPException(status_code=404, detail="File not found")
84    
85    stat = file_path.stat()
86    last_modified = datetime.fromtimestamp(stat.st_mtime)
87    expires = datetime.utcnow() + timedelta(days=1)
88    
89    if if_modified_since:
90        try:
91            ims = datetime.strptime(if_modified_since, "%a, %d %b %Y %H:%M:%S %Z")
92            if last_modified <= ims:
93                return "", 304, {
94                    "Cache-Control": "public, max-age=86400",
95                    "Expires": expires.strftime("%a, %d %b %Y %H:%M:%S GMT"),
96                    "Last-Modified": last_modified.strftime("%a, %d %b %Y %H:%M:%S GMT"),
97                }
98        except:
99            pass
100    
101    with open(file_path, "rb") as f:
102        content = f.read()
103    
104    import mimetypes
105    content_type, _ = mimetypes.guess_type(file)
106    if content_type is None:
107        content_type = "application/octet-stream"
108    
109    return content, 200, {
110        "Content-Type": content_type,
111        "Cache-Control": "public, max-age=86400",
112        "Expires": expires.strftime("%a, %d %b %Y %H:%M:%S GMT"),
113        "Last-Modified": last_modified.strftime("%a, %d %b %Y %H:%M:%S GMT"),
114    }
115 
116if __name__ == "__main__":
117    uvicorn.run(app, host="127.0.0.1", port=8000)

1fastapi
2uvicorn
3python-multipart