Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10sessions = {}
11session_id_counter = 1
12user_id_counter = 1
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class SessionCreate(BaseModel):
23    time: str
24    therapist: str
25    notes: str = ""
26    type: str = "standard"
27    priority: str = "normal"
28 
29class SessionUpdate(BaseModel):
30    time: Optional[str] = None
31    therapist: Optional[str] = None
32    notes: Optional[str] = None
33    type: Optional[str] = None
34    priority: Optional[str] = None
35 
36def get_current_user(authorization: str = Header(None)):
37    if not authorization or not authorization.startswith("Bearer "):
38        raise HTTPException(status_code=401, detail="Invalid auth header")
39    token = authorization.split(" ")[1]
40    if token not in tokens:
41        raise HTTPException(status_code=401, detail="Invalid token")
42    return tokens[token]
43 
44@app.post("/signup")
45def signup(req: SignupRequest):
46    global user_id_counter
47    if req.username in users:
48        raise HTTPException(status_code=400, detail="User exists")
49    user_id = user_id_counter
50    user_id_counter += 1
51    users[req.username] = {"id": user_id, "password": req.password}
52    return {"id": user_id, "username": req.username}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    if req.username not in users or users[req.username]["password"] != req.password:
57        raise HTTPException(status_code=401, detail="Invalid credentials")
58    token = secrets.token_hex(16)
59    tokens[token] = req.username
60    return {"token": token}
61 
62@app.get("/sessions/{session_id}")
63def get_session(session_id: int, authorization: str = Header(None)):
64    get_current_user(authorization)
65    if session_id not in sessions:
66        raise HTTPException(status_code=404, detail="Session not found")
67    return sessions[session_id]
68 
69@app.post("/sessions")
70def create_session(req: SessionCreate, authorization: str = Header(None)):
71    global session_id_counter
72    get_current_user(authorization)
73    session_id = session_id_counter
74    session_id_counter += 1
75    sessions[session_id] = {
76        "id": session_id,
77        "time": req.time,
78        "therapist": req.therapist,
79        "notes": req.notes,
80        "type": req.type,
81        "priority": req.priority
82    }
83    return sessions[session_id]
84 
85@app.patch("/sessions/{session_id}")
86def update_session(session_id: int, req: SessionUpdate, authorization: str = Header(None)):
87    get_current_user(authorization)
88    if session_id not in sessions:
89        raise HTTPException(status_code=404, detail="Session not found")
90    session = sessions[session_id]
91    if req.time is not None:
92        session["time"] = req.time
93    if req.therapist is not None:
94        session["therapist"] = req.therapist
95    if req.notes is not None:
96        session["notes"] = req.notes
97    if req.type is not None:
98        session["type"] = req.type
99    if req.priority is not None:
100        session["priority"] = req.priority
101    return session

1fastapi
2uvicorn