Browse the data

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

1fastapi
2uvicorn