Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import hashlib
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10lenders = {}
11borrowers = {}
12loans = {}
13payments = {}
14next_user_id = 1
15next_lender_id = 1
16next_borrower_id = 1
17next_loan_id = 1
18next_payment_id = 1
19 
20def get_current_user(authorization: Optional[str] = Header(None)):
21    if not authorization:
22        raise HTTPException(status_code=401, detail="Missing auth header")
23    token = authorization.replace("Bearer ", "")
24    for uid, t in tokens.items():
25        if t == token:
26            return uid
27    raise HTTPException(status_code=401, detail="Invalid token")
28 
29@app.post("/signup")
30def signup(username: str, password: str):
31    global next_user_id
32    if any(u["username"] == username for u in users.values()):
33        raise HTTPException(status_code=400, detail="Username taken")
34    uid = next_user_id
35    next_user_id += 1
36    users[uid] = {"id": uid, "username": username, "password": hashlib.sha256(password.encode()).hexdigest()}
37    return {"id": uid, "username": username}
38 
39@app.post("/login")
40def login(username: str, password: str):
41    for uid, u in users.items():
42        if u["username"] == username and u["password"] == hashlib.sha256(password.encode()).hexdigest():
43            token = secrets.token_hex(32)
44            tokens[uid] = token
45            return {"token": token}
46    raise HTTPException(status_code=401, detail="Invalid credentials")
47 
48@app.get("/lenders/{lender_id}")
49def get_lender(lender_id: int, authorization: Optional[str] = Header(None)):
50    get_current_user(authorization)
51    if lender_id not in lenders:
52        raise HTTPException(status_code=404, detail="Lender not found")
53    return lenders[lender_id]
54 
55@app.post("/lenders")
56def create_lender(name: str, balance: float = 0.0, authorization: Optional[str] = Header(None)):
57    get_current_user(authorization)
58    global next_lender_id
59    lid = next_lender_id
60    next_lender_id += 1
61    lenders[lid] = {"id": lid, "name": name, "balance": balance}
62    return lenders[lid]
63 
64@app.get("/borrowers/{borrower_id}")
65def get_borrower(borrower_id: int, authorization: Optional[str] = Header(None)):
66    get_current_user(authorization)
67    if borrower_id not in borrowers:
68        raise HTTPException(status_code=404, detail="Borrower not found")
69    return borrowers[borrower_id]
70 
71@app.post("/borrowers")
72def create_borrower(name: str, credit_score: int = 0, authorization: Optional[str] = Header(None)):
73    get_current_user(authorization)
74    global next_borrower_id
75    bid = next_borrower_id
76    next_borrower_id += 1
77    borrowers[bid] = {"id": bid, "name": name, "credit_score": credit_score}
78    return borrowers[bid]
79 
80@app.get("/loans/{loan_id}")
81def get_loan(loan_id: int, authorization: Optional[str] = Header(None)):
82    get_current_user(authorization)
83    if loan_id not in loans:
84        raise HTTPException(status_code=404, detail="Loan not found")
85    return loans[loan_id]
86 
87@app.post("/loans")
88def create_loan(lender_id: int, borrower_id: int, amount: float, interest_rate: float, term_months: int, authorization: Optional[str] = Header(None)):
89    get_current_user(authorization)
90    global next_loan_id
91    if lender_id not in lenders:
92        raise HTTPException(status_code=400, detail="Lender not found")
93    if borrower_id not in borrowers:
94        raise HTTPException(status_code=400, detail="Borrower not found")
95    lid = next_loan_id
96    next_loan_id += 1
97    loans[lid] = {
98        "id": lid,
99        "lender_id": lender_id,
100        "borrower_id": borrower_id,
101        "amount": amount,
102        "interest_rate": interest_rate,
103        "term_months": term_months,
104        "status": "active"
105    }
106    return loans[lid]
107 
108@app.get("/payments/{payment_id}")
109def get_payment(payment_id: int, authorization: Optional[str] = Header(None)):
110    get_current_user(authorization)
111    if payment_id not in payments:
112        raise HTTPException(status_code=404, detail="Payment not found")
113    return payments[payment_id]
114 
115@app.post("/payments")
116def create_payment(loan_id: int, amount: float, authorization: Optional[str] = Header(None)):
117    get_current_user(authorization)
118    global next_payment_id
119    if loan_id not in loans:
120        raise HTTPException(status_code=400, detail="Loan not found")
121    pid = next_payment_id
122    next_payment_id += 1
123    payments[pid] = {
124        "id": pid,
125        "loan_id": loan_id,
126        "amount": amount,
127        "status": "completed"
128    }
129    return payments[pid]

1fastapi
2uvicorn