Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8companies = {}
9jobs = {}
10applications = {}
11next_user_id = 1
12next_company_id = 1
13next_job_id = 1
14next_application_id = 1
15tokens = {}
16 
17def get_current_user(authorization: Optional[str] = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="Missing Authorization header")
20    token = authorization.replace("Bearer ", "")
21    user_id = tokens.get(token)
22    if not user_id:
23        raise HTTPException(status_code=401, detail="Invalid token")
24    return user_id
25 
26@app.post("/signup")
27def signup(username: str, password: str, role: str = "applicant"):
28    global next_user_id
29    user_id = next_user_id
30    users[user_id] = {"id": user_id, "username": username, "password": password, "role": role}
31    next_user_id += 1
32    return {"id": user_id}
33 
34@app.post("/login")
35def login(username: str, password: str):
36    for user_id, user in users.items():
37        if user["username"] == username and user["password"] == password:
38            token = secrets.token_hex(16)
39            tokens[token] = user_id
40            return {"token": token}
41    raise HTTPException(status_code=401, detail="Invalid credentials")
42 
43@app.get("/companies/{company_id}")
44def get_company(company_id: int):
45    company = companies.get(company_id)
46    if not company:
47        raise HTTPException(status_code=404, detail="Company not found")
48    return company
49 
50@app.post("/companies")
51def create_company(name: str, description: str = "", authorization: Optional[str] = Header(None)):
52    user_id = get_current_user(authorization)
53    global next_company_id
54    company_id = next_company_id
55    companies[company_id] = {"id": company_id, "name": name, "description": description, "owner_id": user_id}
56    next_company_id += 1
57    return {"id": company_id}
58 
59@app.get("/jobs/{job_id}")
60def get_job(job_id: int):
61    job = jobs.get(job_id)
62    if not job:
63        raise HTTPException(status_code=404, detail="Job not found")
64    return job
65 
66@app.post("/jobs")
67def create_job(company_id: int, title: str, description: str = "", authorization: Optional[str] = Header(None)):
68    user_id = get_current_user(authorization)
69    if company_id not in companies:
70        raise HTTPException(status_code=404, detail="Company not found")
71    if companies[company_id]["owner_id"] != user_id:
72        raise HTTPException(status_code=403, detail="Not your company")
73    global next_job_id
74    job_id = next_job_id
75    jobs[job_id] = {"id": job_id, "company_id": company_id, "title": title, "description": description}
76    next_job_id += 1
77    return {"id": job_id}
78 
79@app.get("/applications/{application_id}")
80def get_application(application_id: int):
81    application = applications.get(application_id)
82    if not application:
83        raise HTTPException(status_code=404, detail="Application not found")
84    return application
85 
86@app.post("/applications")
87def create_application(job_id: int, cover_letter: str = "", authorization: Optional[str] = Header(None)):
88    user_id = get_current_user(authorization)
89    if job_id not in jobs:
90        raise HTTPException(status_code=404, detail="Job not found")
91    global next_application_id
92    application_id = next_application_id
93    applications[application_id] = {"id": application_id, "job_id": job_id, "applicant_id": user_id, "cover_letter": cover_letter}
94    next_application_id += 1
95    return {"id": application_id}

1fastapi
2uvicorn