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 = {}
12milestones = {}
13signatures = {}
14contract_versions = {}
15next_user_id = 1
16next_contract_id = 1
17next_milestone_id = 1
18next_signature_id = 1
19next_version_id = 1
20 
21class SignupRequest(BaseModel):
22    email: str
23    password: str
24    name: str
25 
26class LoginRequest(BaseModel):
27    email: str
28    password: str
29 
30class ContractCreate(BaseModel):
31    title: str
32    description: Optional[str] = ""
33    freelancer_id: int
34    client_id: int
35    milestones: Optional[list] = []
36 
37class MilestoneCreate(BaseModel):
38    contract_id: int
39    title: str
40    description: Optional[str] = ""
41    amount: float
42    due_date: Optional[str] = None
43 
44class SignatureCreate(BaseModel):
45    contract_id: int
46    user_id: int
47 
48def get_current_user(authorization: str = Header(...)):
49    token = authorization.replace("Bearer ", "")
50    if token not in tokens:
51        raise HTTPException(status_code=401, detail="Invalid token")
52    return tokens[token]
53 
54@app.post("/signup")
55def signup(req: SignupRequest):
56    global next_user_id
57    for u in users.values():
58        if u["email"] == req.email:
59            raise HTTPException(status_code=400, detail="Email already exists")
60    user_id = next_user_id
61    next_user_id += 1
62    users[user_id] = {
63        "id": user_id,
64        "email": req.email,
65        "password": req.password,
66        "name": req.name
67    }
68    return {"id": user_id, "email": req.email, "name": req.name}
69 
70@app.post("/login")
71def login(req: LoginRequest):
72    for u in users.values():
73        if u["email"] == req.email and u["password"] == req.password:
74            token = secrets.token_hex(16)
75            tokens[token] = u["id"]
76            return {"token": token, "user_id": u["id"]}
77    raise HTTPException(status_code=401, detail="Invalid credentials")
78 
79@app.get("/users/{user_id}")
80def get_user(user_id: int, authorization: str = Header(...)):
81    get_current_user(authorization)
82    if user_id not in users:
83        raise HTTPException(status_code=404, detail="User not found")
84    return users[user_id]
85 
86@app.get("/contracts/{contract_id}")
87def get_contract(contract_id: int, authorization: str = Header(...)):
88    get_current_user(authorization)
89    if contract_id not in contracts:
90        raise HTTPException(status_code=404, detail="Contract not found")
91    return contracts[contract_id]
92 
93@app.get("/milestones/{milestone_id}")
94def get_milestone(milestone_id: int, authorization: str = Header(...)):
95    get_current_user(authorization)
96    if milestone_id not in milestones:
97        raise HTTPException(status_code=404, detail="Milestone not found")
98    return milestones[milestone_id]
99 
100@app.get("/signatures/{signature_id}")
101def get_signature(signature_id: int, authorization: str = Header(...)):
102    get_current_user(authorization)
103    if signature_id not in signatures:
104        raise HTTPException(status_code=404, detail="Signature not found")
105    return signatures[signature_id]
106 
107@app.get("/contracts/{contract_id}/versions")
108def get_contract_versions(contract_id: int, authorization: str = Header(...)):
109    get_current_user(authorization)
110    if contract_id not in contracts:
111        raise HTTPException(status_code=404, detail="Contract not found")
112    return [v for v in contract_versions.values() if v["contract_id"] == contract_id]
113 
114@app.get("/contracts/{contract_id}/versions/{version_id}")
115def get_contract_version(contract_id: int, version_id: int, authorization: str = Header(...)):
116    get_current_user(authorization)
117    if contract_id not in contracts:
118        raise HTTPException(status_code=404, detail="Contract not found")
119    if version_id not in contract_versions:
120        raise HTTPException(status_code=404, detail="Version not found")
121    v = contract_versions[version_id]
122    if v["contract_id"] != contract_id:
123        raise HTTPException(status_code=404, detail="Version not found for this contract")
124    return v
125 
126@app.post("/contracts")
127def create_contract(req: ContractCreate, authorization: str = Header(...)):
128    global next_contract_id, next_version_id
129    user_id = get_current_user(authorization)
130    contract_id = next_contract_id
131    next_contract_id += 1
132    now = datetime.datetime.utcnow().isoformat()
133    contract = {
134        "id": contract_id,
135        "title": req.title,
136        "description": req.description,
137        "freelancer_id": req.freelancer_id,
138        "client_id": req.client_id,
139        "status": "draft",
140        "created_at": now,
141        "updated_at": now
142    }
143    contracts[contract_id] = contract
144 
145    version_id = next_version_id
146    next_version_id += 1
147    contract_versions[version_id] = {
148        "id": version_id,
149        "contract_id": contract_id,
150        "version_number": 1,
151        "data": contract.copy(),
152        "created_at": now
153    }
154 
155    if req.milestones:
156        for m in req.milestones:
157            mid = next_milestone_id
158            next_milestone_id += 1
159            milestones[mid] = {
160                "id": mid,
161                "contract_id": contract_id,
162                "title": m.get("title", ""),
163                "description": m.get("description", ""),
164                "amount": m.get("amount", 0.0),
165                "due_date": m.get("due_date"),
166                "status": "pending"
167            }
168 
169    return contract
170 
171@app.post("/milestones")
172def create_milestone(req: MilestoneCreate, authorization: str = Header(...)):
173    global next_milestone_id
174    get_current_user(authorization)
175    if req.contract_id not in contracts:
176        raise HTTPException(status_code=404, detail="Contract not found")
177    milestone_id = next_milestone_id
178    next_milestone_id += 1
179    milestones[milestone_id] = {
180        "id": milestone_id,
181        "contract_id": req.contract_id,
182        "title": req.title,
183        "description": req.description,
184        "amount": req.amount,
185        "due_date": req.due_date,
186        "status": "pending"
187    }
188    return milestones[milestone_id]
189 
190@app.post("/signatures")
191def create_signature(req: SignatureCreate, authorization: str = Header(...)):
192    global next_signature_id, next_version_id
193    user_id = get_current_user(authorization)
194    if req.contract_id not in contracts:
195        raise HTTPException(status_code=404, detail="Contract not found")
196    if req.user_id not in users:
197        raise HTTPException(status_code=404, detail="User not found")
198    signature_id = next_signature_id
199    next_signature_id += 1
200    now = datetime.datetime.utcnow().isoformat()
201    sig = {
202        "id": signature_id,
203        "contract_id": req.contract_id,
204        "user_id": req.user_id,
205        "signed_at": now
206    }
207    signatures[signature_id] = sig
208 
209    contract = contracts[req.contract_id]
210    contract["status"] = "signed"
211    contract["updated_at"] = now
212 
213    version_id = next_version_id
214    next_version_id += 1
215    contract_versions[version_id] = {
216        "id": version_id,
217        "contract_id": req.contract_id,
218        "version_number": len([v for v in contract_versions.values() if v["contract_id"] == req.contract_id]) + 1,
219        "data": contract.copy(),
220        "created_at": now
221    }
222 
223    return sig

1fastapi
2uvicorn