Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict
4import secrets
5import uvicorn
6 
7app = FastAPI()
8 
9# In-memory stores
10users: Dict[int, dict] = {}
11jobs: Dict[int, dict] = {}
12applications: Dict[int, dict] = {}
13tokens: Dict[str, int] = {}  # token -> user_id
14 
15# ID counters
16user_id_counter = 1
17job_id_counter = 1
18application_id_counter = 1
19 
20class SignupRequest(BaseModel):
21    username: str
22    password: str
23 
24class LoginRequest(BaseModel):
25    username: str
26    password: str
27 
28class JobCreate(BaseModel):
29    title: str
30    company: str
31    location: str
32    salary_range: str
33    remote: bool
34 
35class ApplicationCreate(BaseModel):
36    name: str
37    resume_url: str
38 
39def get_current_user(authorization: Optional[str] = Header(None)):
40    if not authorization:
41        raise HTTPException(status_code=401, detail="Missing auth token")
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 user_id_counter
51    user_id = user_id_counter
52    user_id_counter += 1
53    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
54    return {"id": user_id, "username": req.username}
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.get("/jobs/{job_id}")
66def get_job(job_id: int):
67    job = jobs.get(job_id)
68    if not job:
69        raise HTTPException(status_code=404, detail="Job not found")
70    return job
71 
72@app.get("/jobs")
73def list_jobs():
74    return [j for j in jobs.values() if j.get("status", "open") == "open"]
75 
76@app.post("/jobs")
77def create_job(job: JobCreate, authorization: Optional[str] = Header(None)):
78    get_current_user(authorization)
79    global job_id_counter
80    job_id = job_id_counter
81    job_id_counter += 1
82    jobs[job_id] = {
83        "id": job_id,
84        "title": job.title,
85        "company": job.company,
86        "location": job.location,
87        "salary_range": job.salary_range,
88        "remote": job.remote,
89        "status": "open"
90    }
91    return jobs[job_id]
92 
93@app.post("/jobs/apply")
94def apply_for_job(app: ApplicationCreate, authorization: Optional[str] = Header(None)):
95    user_id = get_current_user(authorization)
96    global application_id_counter
97    app_id = application_id_counter
98    application_id_counter += 1
99    applications[app_id] = {
100        "id": app_id,
101        "user_id": user_id,
102        "name": app.name,
103        "resume_url": app.resume_url
104    }
105    return applications[app_id]
106 
107@app.get("/applications/{app_id}")
108def get_application(app_id: int, authorization: Optional[str] = Header(None)):
109    get_current_user(authorization)
110    app = applications.get(app_id)
111    if not app:
112        raise HTTPException(status_code=404, detail="Application not found")
113    return app

1fastapi
2uvicorn