Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import hashlib
6 
7app = FastAPI()
8 
9users = {}
10recruiters = {}
11tokens = {}
12user_id_counter = 1
13recruiter_id_counter = 1
14 
15class SignupRequest(BaseModel):
16    email: str
17    password: str
18    name: str
19 
20class LoginRequest(BaseModel):
21    email: str
22    password: str
23 
24class RecruiterCreate(BaseModel):
25    company_name: str
26    contact: str
27    job_posting_limits: int = 10
28    plan_tier: str = "basic"
29    is_admin: bool = False
30 
31class RecruiterUpdate(BaseModel):
32    company_name: Optional[str] = None
33    contact: Optional[str] = None
34    job_posting_limits: Optional[int] = None
35    plan_tier: Optional[str] = None
36    is_admin: Optional[bool] = None
37 
38def get_current_user(authorization: str = Header(None)):
39    if not authorization:
40        raise HTTPException(status_code=401, detail="No auth header")
41    token = authorization.replace("Bearer ", "")
42    if token not in tokens:
43        raise HTTPException(status_code=401, detail="Invalid token")
44    return tokens[token]
45 
46@app.post("/signup")
47def signup(req: SignupRequest):
48    global user_id_counter
49    for u in users.values():
50        if u["email"] == req.email:
51            raise HTTPException(status_code=400, detail="Email already exists")
52    user_id = user_id_counter
53    user_id_counter += 1
54    users[user_id] = {
55        "id": user_id,
56        "email": req.email,
57        "password": hashlib.sha256(req.password.encode()).hexdigest(),
58        "name": req.name
59    }
60    return {"id": user_id, "email": req.email, "name": req.name}
61 
62@app.post("/login")
63def login(req: LoginRequest):
64    for u in users.values():
65        if u["email"] == req.email and u["password"] == hashlib.sha256(req.password.encode()).hexdigest():
66            token = secrets.token_hex(32)
67            tokens[token] = u["id"]
68            return {"token": token, "user_id": u["id"]}
69    raise HTTPException(status_code=401, detail="Invalid credentials")
70 
71@app.get("/recruiters/{recruiter_id}")
72def get_recruiter(recruiter_id: int, authorization: str = Header(None)):
73    user_id = get_current_user(authorization)
74    if recruiter_id not in recruiters:
75        raise HTTPException(status_code=404, detail="Recruiter not found")
76    return recruiters[recruiter_id]
77 
78@app.post("/recruiters")
79def create_recruiter(req: RecruiterCreate, authorization: str = Header(None)):
80    global recruiter_id_counter
81    user_id = get_current_user(authorization)
82    recruiter_id = recruiter_id_counter
83    recruiter_id_counter += 1
84    recruiters[recruiter_id] = {
85        "id": recruiter_id,
86        "user_id": user_id,
87        "company_name": req.company_name,
88        "contact": req.contact,
89        "job_posting_limits": req.job_posting_limits,
90        "plan_tier": req.plan_tier,
91        "is_admin": req.is_admin
92    }
93    return recruiters[recruiter_id]
94 
95@app.put("/recruiters/{recruiter_id}")
96def update_recruiter(recruiter_id: int, req: RecruiterUpdate, authorization: str = Header(None)):
97    user_id = get_current_user(authorization)
98    if recruiter_id not in recruiters:
99        raise HTTPException(status_code=404, detail="Recruiter not found")
100    recruiter = recruiters[recruiter_id]
101    if req.company_name is not None:
102        recruiter["company_name"] = req.company_name
103    if req.contact is not None:
104        recruiter["contact"] = req.contact
105    if req.job_posting_limits is not None:
106        recruiter["job_posting_limits"] = req.job_posting_limits
107    if req.plan_tier is not None:
108        recruiter["plan_tier"] = req.plan_tier
109    if req.is_admin is not None:
110        recruiter["is_admin"] = req.is_admin
111    return recruiter

1fastapi
2uvicorn