Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10surveys = {}
11responses = {}
12 
13user_id_counter = 1
14survey_id_counter = 1
15response_id_counter = 1
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20 
21class LoginRequest(BaseModel):
22    username: str
23    password: str
24 
25class Question(BaseModel):
26    type: str
27    text: str
28    options: Optional[list[str]] = None
29 
30class SurveyCreate(BaseModel):
31    title: str
32    questions: list[Question]
33 
34class SurveyResponse(BaseModel):
35    answers: dict
36 
37@app.post("/signup")
38def signup(req: SignupRequest):
39    global user_id_counter
40    if req.username in users:
41        raise HTTPException(400, "User exists")
42    users[req.username] = {"id": user_id_counter, "password": req.password}
43    user_id_counter += 1
44    return {"id": users[req.username]["id"], "username": req.username}
45 
46@app.post("/login")
47def login(req: LoginRequest):
48    user = users.get(req.username)
49    if not user or user["password"] != req.password:
50        raise HTTPException(401, "Invalid credentials")
51    token = secrets.token_hex(16)
52    tokens[token] = user["id"]
53    return {"token": token}
54 
55def get_user_id(authorization: str = Header(None)):
56    if not authorization or not authorization.startswith("Bearer "):
57        raise HTTPException(401, "Missing or invalid token")
58    token = authorization[7:]
59    user_id = tokens.get(token)
60    if not user_id:
61        raise HTTPException(401, "Invalid token")
62    return user_id
63 
64@app.post("/survey")
65def create_survey(survey: SurveyCreate, authorization: str = Header(None)):
66    global survey_id_counter
67    user_id = get_user_id(authorization)
68    survey_id = survey_id_counter
69    survey_id_counter += 1
70    surveys[survey_id] = {
71        "id": survey_id,
72        "title": survey.title,
73        "questions": [q.dict() for q in survey.questions],
74        "created_by": user_id
75    }
76    responses[survey_id] = []
77    return surveys[survey_id]
78 
79@app.get("/survey/{survey_id}")
80def get_survey(survey_id: int, authorization: str = Header(None)):
81    get_user_id(authorization)
82    survey = surveys.get(survey_id)
83    if not survey:
84        raise HTTPException(404, "Survey not found")
85    return survey
86 
87@app.post("/survey/{survey_id}/response")
88def submit_response(survey_id: int, response: SurveyResponse, authorization: str = Header(None)):
89    user_id = get_user_id(authorization)
90    if survey_id not in surveys:
91        raise HTTPException(404, "Survey not found")
92    global response_id_counter
93    resp_id = response_id_counter
94    response_id_counter += 1
95    entry = {
96        "id": resp_id,
97        "survey_id": survey_id,
98        "user_id": user_id,
99        "answers": response.answers
100    }
101    responses[survey_id].append(entry)
102    return entry
103 
104@app.get("/survey/{survey_id}/responses")
105def get_responses(survey_id: int, authorization: str = Header(None)):
106    get_user_id(authorization)
107    if survey_id not in surveys:
108        raise HTTPException(404, "Survey not found")
109    return responses.get(survey_id, [])
110 
111@app.get("/survey/{survey_id}/response-counts")
112def get_response_counts(survey_id: int, authorization: str = Header(None)):
113    get_user_id(authorization)
114    if survey_id not in surveys:
115        raise HTTPException(404, "Survey not found")
116    survey = surveys[survey_id]
117    counts = {}
118    for q in survey["questions"]:
119        qid = q["text"]
120        if q["type"] == "multiple_choice":
121            counts[qid] = {opt: 0 for opt in q["options"]}
122        elif q["type"] == "text":
123            counts[qid] = 0
124    for resp in responses.get(survey_id, []):
125        for qid, ans in resp["answers"].items():
126            if qid in counts:
127                if isinstance(counts[qid], dict):
128                    if ans in counts[qid]:
129                        counts[qid][ans] += 1
130                else:
131                    counts[qid] += 1
132    return counts

1fastapi
2uvicorn