Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import uuid
5import hashlib
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11campaigns = {}
12campaign_id_counter = 1
13 
14def get_current_user(authorization: str = Header(None)):
15    if not authorization:
16        raise HTTPException(status_code=401, detail="Missing auth header")
17    token = authorization.replace("Bearer ", "")
18    if token not in tokens:
19        raise HTTPException(status_code=401, detail="Invalid token")
20    return tokens[token]
21 
22class SignupRequest(BaseModel):
23    username: str
24    password: str
25 
26class LoginRequest(BaseModel):
27    username: str
28    password: str
29 
30class CampaignCreate(BaseModel):
31    headline: str
32    target_audience: str
33    budget: float
34    publisher_tier: str
35 
36class CampaignUpdate(BaseModel):
37    headline: Optional[str] = None
38    target_audience: Optional[str] = None
39    budget: Optional[float] = None
40    publisher_tier: Optional[str] = None
41 
42@app.post("/signup")
43def signup(req: SignupRequest):
44    if req.username in users:
45        raise HTTPException(status_code=400, detail="User already exists")
46    users[req.username] = hashlib.sha256(req.password.encode()).hexdigest()
47    return {"message": "User created"}
48 
49@app.post("/login")
50def login(req: LoginRequest):
51    if req.username not in users or users[req.username] != hashlib.sha256(req.password.encode()).hexdigest():
52        raise HTTPException(status_code=401, detail="Invalid credentials")
53    token = str(uuid.uuid4())
54    tokens[token] = req.username
55    return {"token": token}
56 
57@app.get("/campaigns/{id}")
58def get_campaign(id: int, authorization: str = Header(None)):
59    user = get_current_user(authorization)
60    if id not in campaigns:
61        raise HTTPException(status_code=404, detail="Campaign not found")
62    return campaigns[id]
63 
64@app.post("/campaigns")
65def create_campaign(campaign: CampaignCreate, authorization: str = Header(None)):
66    global campaign_id_counter
67    user = get_current_user(authorization)
68    cid = campaign_id_counter
69    campaign_id_counter += 1
70    campaigns[cid] = {
71        "id": cid,
72        "headline": campaign.headline,
73        "target_audience": campaign.target_audience,
74        "budget": campaign.budget,
75        "publisher_tier": campaign.publisher_tier,
76        "owner": user
77    }
78    return campaigns[cid]
79 
80@app.patch("/campaigns/{id}")
81def update_campaign(id: int, update: CampaignUpdate, authorization: str = Header(None)):
82    user = get_current_user(authorization)
83    if id not in campaigns:
84        raise HTTPException(status_code=404, detail="Campaign not found")
85    if campaigns[id]["owner"] != user:
86        raise HTTPException(status_code=403, detail="Not your campaign")
87    if update.headline is not None:
88        campaigns[id]["headline"] = update.headline
89    if update.target_audience is not None:
90        campaigns[id]["target_audience"] = update.target_audience
91    if update.budget is not None:
92        campaigns[id]["budget"] = update.budget
93    if update.publisher_tier is not None:
94        campaigns[id]["publisher_tier"] = update.publisher_tier
95    return campaigns[id]

1fastapi
2uvicorn