Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import hashlib
5import uuid
6 
7app = FastAPI()
8 
9users = {}
10proposals = {}
11proposal_id_counter = 1
12tokens = {}
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class CreateProposalRequest(BaseModel):
23    title: str
24    description: str
25    budget: float
26    status: str = "draft"
27    is_urgent: bool = False
28 
29class UpdateProposalRequest(BaseModel):
30    title: Optional[str] = None
31    description: Optional[str] = None
32    budget: Optional[float] = None
33    status: Optional[str] = None
34    is_urgent: Optional[bool] = None
35 
36def get_user_id_from_token(authorization: str = Header(...)):
37    if not authorization.startswith("Bearer "):
38        raise HTTPException(status_code=401, detail="Invalid auth header")
39    token = authorization[7:]
40    user_id = tokens.get(token)
41    if user_id is None:
42        raise HTTPException(status_code=401, detail="Invalid token")
43    return user_id
44 
45@app.post("/signup")
46def signup(req: SignupRequest):
47    if req.username in users:
48        raise HTTPException(status_code=400, detail="Username already exists")
49    user_id = len(users) + 1
50    users[req.username] = {"id": user_id, "password": hashlib.sha256(req.password.encode()).hexdigest()}
51    return {"id": user_id, "username": req.username}
52 
53@app.post("/login")
54def login(req: LoginRequest):
55    user = users.get(req.username)
56    if not user or user["password"] != hashlib.sha256(req.password.encode()).hexdigest():
57        raise HTTPException(status_code=401, detail="Invalid credentials")
58    token = str(uuid.uuid4())
59    tokens[token] = user["id"]
60    return {"token": token}
61 
62@app.get("/proposals/{proposal_id}")
63def get_proposal(proposal_id: int, authorization: str = Header(...)):
64    user_id = get_user_id_from_token(authorization)
65    proposal = proposals.get(proposal_id)
66    if not proposal:
67        raise HTTPException(status_code=404, detail="Proposal not found")
68    return proposal
69 
70@app.post("/proposals")
71def create_proposal(req: CreateProposalRequest, authorization: str = Header(...)):
72    global proposal_id_counter
73    user_id = get_user_id_from_token(authorization)
74    proposal = {
75        "id": proposal_id_counter,
76        "user_id": user_id,
77        "title": req.title,
78        "description": req.description,
79        "budget": req.budget,
80        "status": req.status,
81        "is_urgent": req.is_urgent
82    }
83    proposals[proposal_id_counter] = proposal
84    proposal_id_counter += 1
85    return proposal
86 
87@app.patch("/proposals/{proposal_id}")
88def update_proposal(proposal_id: int, req: UpdateProposalRequest, authorization: str = Header(...)):
89    user_id = get_user_id_from_token(authorization)
90    proposal = proposals.get(proposal_id)
91    if not proposal:
92        raise HTTPException(status_code=404, detail="Proposal not found")
93    if proposal["user_id"] != user_id:
94        raise HTTPException(status_code=403, detail="Not authorized to update this proposal")
95    if req.title is not None:
96        proposal["title"] = req.title
97    if req.description is not None:
98        proposal["description"] = req.description
99    if req.budget is not None:
100        proposal["budget"] = req.budget
101    if req.status is not None:
102        proposal["status"] = req.status
103    if req.is_urgent is not None:
104        proposal["is_urgent"] = req.is_urgent
105    return proposal

1fastapi
2uvicorn