Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5import uvicorn
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11projects = {}
12proposals = {}
13project_id_counter = 1
14proposal_id_counter = 1
15user_id_counter = 1
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20 
21class LoginRequest(BaseModel):
22    username: str
23    password: str
24 
25class ProjectCreate(BaseModel):
26    budget_min: float
27    budget_max: float
28    skills: List[str]
29    description: str
30 
31class ProposalCreate(BaseModel):
32    project_id: int
33    price: float
34    timeline: int
35 
36def get_current_user(authorization: Optional[str] = Header(None)):
37    if not authorization:
38        raise HTTPException(status_code=401, detail="Missing auth header")
39    token = authorization.replace("Bearer ", "")
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="Username already exists")
49    user_id = user_id_counter
50    user_id_counter += 1
51    users[req.username] = {"id": user_id, "username": req.username, "password": req.password}
52    token = secrets.token_hex(16)
53    tokens[token] = req.username
54    return {"token": token, "user_id": user_id}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    if req.username not in users or users[req.username]["password"] != req.password:
59        raise HTTPException(status_code=401, detail="Invalid credentials")
60    token = secrets.token_hex(16)
61    tokens[token] = req.username
62    return {"token": token}
63 
64@app.get("/users/{user_id}")
65def get_user(user_id: int):
66    for u in users.values():
67        if u["id"] == user_id:
68            return u
69    raise HTTPException(status_code=404, detail="User not found")
70 
71@app.post("/projects")
72def create_project(project: ProjectCreate, authorization: Optional[str] = Header(None)):
73    get_current_user(authorization)
74    global project_id_counter
75    pid = project_id_counter
76    project_id_counter += 1
77    projects[pid] = {
78        "id": pid,
79        "budget_min": project.budget_min,
80        "budget_max": project.budget_max,
81        "skills": project.skills,
82        "description": project.description,
83        "status": "open"
84    }
85    return projects[pid]
86 
87@app.get("/projects")
88def get_projects(authorization: Optional[str] = Header(None)):
89    get_current_user(authorization)
90    result = []
91    for pid, proj in projects.items():
92        if proj["status"] == "open":
93            count = sum(1 for p in proposals.values() if p["project_id"] == pid)
94            result.append({**proj, "applied_count": count})
95    return result
96 
97@app.get("/projects/{project_id}")
98def get_project(project_id: int, authorization: Optional[str] = Header(None)):
99    get_current_user(authorization)
100    if project_id not in projects:
101        raise HTTPException(status_code=404, detail="Project not found")
102    proj = projects[project_id]
103    count = sum(1 for p in proposals.values() if p["project_id"] == project_id)
104    return {**proj, "applied_count": count}
105 
106@app.post("/proposals")
107def create_proposal(proposal: ProposalCreate, authorization: Optional[str] = Header(None)):
108    current_user = get_current_user(authorization)
109    global proposal_id_counter
110    if proposal.project_id not in projects:
111        raise HTTPException(status_code=404, detail="Project not found")
112    pid = proposal_id_counter
113    proposal_id_counter += 1
114    proposals[pid] = {
115        "id": pid,
116        "project_id": proposal.project_id,
117        "freelancer": current_user,
118        "price": proposal.price,
119        "timeline": proposal.timeline
120    }
121    return proposals[pid]
122 
123@app.get("/proposals/{proposal_id}")
124def get_proposal(proposal_id: int, authorization: Optional[str] = Header(None)):
125    get_current_user(authorization)
126    if proposal_id not in proposals:
127        raise HTTPException(status_code=404, detail="Proposal not found")
128    return proposals[proposal_id]

1fastapi
2uvicorn