Browse the data

1import time
2import hashlib
3import secrets
4from fastapi import FastAPI, HTTPException, Header
5from pydantic import BaseModel
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11next_user_id = 1
12 
13class SignupRequest(BaseModel):
14    username: str
15    password: str
16 
17class LoginRequest(BaseModel):
18    username: str
19    password: str
20 
21class UserCreate(BaseModel):
22    username: str
23    password: str
24 
25class ResourceCreate(BaseModel):
26    data: dict = {}
27 
28resources = {}
29next_resource_id = 1
30 
31def get_current_user(authorization: str = Header(None)):
32    if not authorization:
33        raise HTTPException(status_code=401, detail="Missing token")
34    token = authorization.replace("Bearer ", "")
35    user_id = tokens.get(token)
36    if user_id is None:
37        raise HTTPException(status_code=401, detail="Invalid token")
38    return user_id
39 
40@app.post("/signup")
41def signup(req: SignupRequest):
42    global next_user_id
43    if req.username in [u["username"] for u in users.values()]:
44        raise HTTPException(status_code=400, detail="Username taken")
45    user_id = next_user_id
46    next_user_id += 1
47    now = int(time.time())
48    users[user_id] = {
49        "id": user_id,
50        "username": req.username,
51        "password": req.password,  # plaintext, as requested
52        "created_at": now,
53        "last_active": now
54    }
55    return {"id": user_id, "username": req.username}
56 
57@app.post("/login")
58def login(req: LoginRequest):
59    for uid, u in users.items():
60        if u["username"] == req.username and u["password"] == req.password:
61            token = secrets.token_hex(16)
62            tokens[token] = uid
63            return {"token": token}
64    raise HTTPException(status_code=401, detail="Invalid credentials")
65 
66@app.get("/users")
67def search_users(username: str = "", min_account_age_days: float = None, max_account_age_days: float = None, min_last_active_days: float = None, max_last_active_days: float = None, authorization: str = Header(None)):
68    get_current_user(authorization)
69    now = int(time.time())
70    results = []
71    for uid, u in users.items():
72        if username and username.lower() not in u["username"].lower():
73            continue
74        if min_account_age_days is not None:
75            if (now - u["created_at"]) < min_account_age_days * 86400:
76                continue
77        if max_account_age_days is not None:
78            if (now - u["created_at"]) > max_account_age_days * 86400:
79                continue
80        if min_last_active_days is not None:
81            if (now - u["last_active"]) < min_last_active_days * 86400:
82                continue
83        if max_last_active_days is not None:
84            if (now - u["last_active"]) > max_last_active_days * 86400:
85                continue
86        results.append(u)
87    return results
88 
89@app.get("/{resource}/{resource_id}")
90def get_resource(resource: str, resource_id: int, authorization: str = Header(None)):
91    get_current_user(authorization)
92    if resource == "users":
93        if resource_id not in users:
94            raise HTTPException(status_code=404, detail="User not found")
95        return users[resource_id]
96    elif resource == "resources":
97        if resource_id not in resources:
98            raise HTTPException(status_code=404, detail="Resource not found")
99        return resources[resource_id]
100    else:
101        raise HTTPException(status_code=404, detail="Unknown resource type")
102 
103@app.post("/{resource}")
104def create_resource(resource: str, req: ResourceCreate, authorization: str = Header(None)):
105    global next_resource_id
106    get_current_user(authorization)
107    if resource == "users":
108        return signup(SignupRequest(username=req.data.get("username", ""), password=req.data.get("password", "")))
109    elif resource == "resources":
110        rid = next_resource_id
111        next_resource_id += 1
112        now = int(time.time())
113        resources[rid] = {"id": rid, "data": req.data, "created_at": now}
114        return resources[rid]
115    else:
116        raise HTTPException(status_code=404, detail="Unknown resource type")

1fastapi
2uvicorn