Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import datetime
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11contracts = {}
12contracts_counter = 0
13 
14class UserCreate(BaseModel):
15    username: str
16    password: str
17 
18class UserLogin(BaseModel):
19    username: str
20    password: str
21 
22class ContractCreate(BaseModel):
23    title: str
24    terms: str = ""
25    payment_amount: float = 0.0
26    deadlines: str = ""
27    milestones: str = ""
28 
29class ContractUpdate(BaseModel):
30    terms: Optional[str] = None
31    payment_amount: Optional[float] = None
32    deadlines: Optional[str] = None
33    milestones: Optional[str] = None
34    status: Optional[str] = None
35    signature: Optional[str] = None
36 
37def get_current_user(authorization: str = Header(None)):
38    if not authorization:
39        raise HTTPException(status_code=401, detail="Missing auth header")
40    token = authorization.replace("Bearer ", "")
41    if token not in tokens:
42        raise HTTPException(status_code=401, detail="Invalid token")
43    return tokens[token]
44 
45@app.post("/signup")
46def signup(user: UserCreate):
47    if user.username in users:
48        raise HTTPException(status_code=400, detail="User already exists")
49    users[user.username] = {"username": user.username, "password": user.password}
50    return {"message": "User created"}
51 
52@app.post("/login")
53def login(user: UserLogin):
54    if user.username not in users or users[user.username]["password"] != user.password:
55        raise HTTPException(status_code=401, detail="Invalid credentials")
56    token = secrets.token_hex(16)
57    tokens[token] = user.username
58    return {"token": token}
59 
60@app.get("/contracts/{contract_id}")
61def get_contract(contract_id: int, authorization: str = Header(None)):
62    get_current_user(authorization)
63    if contract_id not in contracts:
64        raise HTTPException(status_code=404, detail="Contract not found")
65    return contracts[contract_id]
66 
67@app.post("/contracts")
68def create_contract(contract: ContractCreate, authorization: str = Header(None)):
69    get_current_user(authorization)
70    global contracts_counter
71    contracts_counter += 1
72    contracts[contracts_counter] = {
73        "id": contracts_counter,
74        "title": contract.title,
75        "terms": contract.terms,
76        "payment_amount": contract.payment_amount,
77        "deadlines": contract.deadlines,
78        "milestones": contract.milestones,
79        "status": "draft",
80        "signature": None,
81        "created_at": datetime.datetime.utcnow().isoformat(),
82        "updated_at": datetime.datetime.utcnow().isoformat()
83    }
84    return contracts[contracts_counter]
85 
86@app.patch("/contracts/{contract_id}")
87def update_contract(contract_id: int, updates: ContractUpdate, authorization: str = Header(None)):
88    get_current_user(authorization)
89    if contract_id not in contracts:
90        raise HTTPException(status_code=404, detail="Contract not found")
91    contract = contracts[contract_id]
92    update_data = updates.dict(exclude_unset=True)
93    for field, value in update_data.items():
94        if value is not None:
95            contract[field] = value
96    contract["updated_at"] = datetime.datetime.utcnow().isoformat()
97    contracts[contract_id] = contract
98    return contract

1fastapi
2uvicorn