Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9companies = {}
10jobs = {}
11applications = {}
12tokens = {}
13next_user_id = 1
14next_company_id = 1
15next_job_id = 1
16next_application_id = 1
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21    role: str = "applicant"
22 
23class LoginRequest(BaseModel):
24    username: str
25    password: str
26 
27class CompanyCreate(BaseModel):
28    name: str
29    description: str = ""
30 
31class JobCreate(BaseModel):
32    company_id: int
33    title: str
34    description: str = ""
35 
36class ApplicationCreate(BaseModel):
37    job_id: int
38 
39def get_current_user(authorization: Optional[str] = Header(None)):
40    if not authorization:
41        raise HTTPException(status_code=401, detail="Missing auth header")
42    token = authorization.replace("Bearer ", "")
43    user_id = tokens.get(token)
44    if user_id is None:
45        raise HTTPException(status_code=401, detail="Invalid token")
46    return user_id
47 
48@app.post("/signup")
49def signup(req: SignupRequest):
50    global next_user_id
51    user_id = next_user_id
52    next_user_id += 1
53    users[user_id] = {"id": user_id, "username": req.username, "password": req.password, "role": req.role}
54    return {"id": user_id, "username": req.username, "role": req.role}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for uid, u in users.items():
59        if u["username"] == req.username and u["password"] == req.password:
60            token = secrets.token_hex(16)
61            tokens[token] = uid
62            return {"token": token}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65@app.post("/companies")
66def create_company(req: CompanyCreate, authorization: Optional[str] = Header(None)):
67    user_id = get_current_user(authorization)
68    global next_company_id
69    cid = next_company_id
70    next_company_id += 1
71    companies[cid] = {"id": cid, "name": req.name, "description": req.description, "owner_id": user_id}
72    return companies[cid]
73 
74@app.get("/companies/{company_id}")
75def get_company(company_id: int):
76    comp = companies.get(company_id)
77    if not comp:
78        raise HTTPException(status_code=404, detail="Company not found")
79    return comp
80 
81@app.post("/jobs")
82def create_job(req: JobCreate, authorization: Optional[str] = Header(None)):
83    user_id = get_current_user(authorization)
84    if req.company_id not in companies:
85        raise HTTPException(status_code=404, detail="Company not found")
86    if companies[req.company_id]["owner_id"] != user_id:
87        raise HTTPException(status_code=403, detail="Not your company")
88    global next_job_id
89    jid = next_job_id
90    next_job_id += 1
91    jobs[jid] = {"id": jid, "company_id": req.company_id, "title": req.title, "description": req.description}
92    return jobs[jid]
93 
94@app.get("/jobs/{job_id}")
95def get_job(job_id: int):
96    job = jobs.get(job_id)
97    if not job:
98        raise HTTPException(status_code=404, detail="Job not found")
99    return job
100 
101@app.get("/jobs")
102def list_jobs():
103    return list(jobs.values())
104 
105@app.post("/applications")
106def create_application(req: ApplicationCreate, authorization: Optional[str] = Header(None)):
107    user_id = get_current_user(authorization)
108    if req.job_id not in jobs:
109        raise HTTPException(status_code=404, detail="Job not found")
110    global next_application_id
111    aid = next_application_id
112    next_application_id += 1
113    applications[aid] = {"id": aid, "job_id": req.job_id, "applicant_id": user_id}
114    return applications[aid]
115 
116@app.get("/applications/{application_id}")
117def get_application(application_id: int):
118    app = applications.get(application_id)
119    if not app:
120        raise HTTPException(status_code=404, detail="Application not found")
121    return app
122 
123@app.get("/applications")
124def list_applications(authorization: Optional[str] = Header(None)):
125    user_id = get_current_user(authorization)
126    return [a for a in applications.values() if a["applicant_id"] == user_id]

1fastapi
2uvicorn