Browse the data

1import os
2import random
3import string
4from fastapi import FastAPI, HTTPException, Header
5from fastapi.responses import FileResponse, Response
6from pydantic import BaseModel
7 
8app = FastAPI()
9 
10# In-memory stores
11users = {}
12tokens = {}
13orders = {}
14products = {}
15invoices = {}
16next_user_id = 1
17next_order_id = 1
18next_product_id = 1
19next_invoice_id = 1
20 
21# Auth helper
22def get_user_from_token(authorization: str = Header(None)):
23    if not authorization:
24        raise HTTPException(status_code=401, detail="Missing Authorization header")
25    token = authorization.replace("Bearer ", "")
26    if token not in tokens:
27        raise HTTPException(status_code=401, detail="Invalid token")
28    return tokens[token]
29 
30# Signup
31class SignupRequest(BaseModel):
32    username: str
33    password: str
34 
35@app.post("/signup")
36def signup(req: SignupRequest):
37    global next_user_id
38    user_id = next_user_id
39    next_user_id += 1
40    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
41    return {"id": user_id, "username": req.username}
42 
43# Login
44class LoginRequest(BaseModel):
45    username: str
46    password: str
47 
48@app.post("/login")
49def login(req: LoginRequest):
50    for uid, u in users.items():
51        if u["username"] == req.username and u["password"] == req.password:
52            token = ''.join(random.choices(string.ascii_letters + string.digits, k=32))
53            tokens[token] = uid
54            return {"token": token}
55    raise HTTPException(status_code=401, detail="Invalid credentials")
56 
57# Orders
58class OrderCreate(BaseModel):
59    description: str = ""
60 
61@app.post("/orders")
62def create_order(order: OrderCreate, authorization: str = Header(None)):
63    user_id = get_user_from_token(authorization)
64    global next_order_id
65    oid = next_order_id
66    next_order_id += 1
67    orders[oid] = {"id": oid, "user_id": user_id, "description": order.description}
68    return orders[oid]
69 
70@app.get("/orders/{order_id}")
71def get_order(order_id: int, authorization: str = Header(None)):
72    user_id = get_user_from_token(authorization)
73    if order_id not in orders:
74        raise HTTPException(status_code=404, detail="Order not found")
75    return orders[order_id]
76 
77# Products
78class ProductCreate(BaseModel):
79    name: str = ""
80 
81@app.post("/products")
82def create_product(product: ProductCreate, authorization: str = Header(None)):
83    user_id = get_user_from_token(authorization)
84    global next_product_id
85    pid = next_product_id
86    next_product_id += 1
87    products[pid] = {"id": pid, "user_id": user_id, "name": product.name}
88    return products[pid]
89 
90@app.get("/products/{product_id}")
91def get_product(product_id: int, authorization: str = Header(None)):
92    user_id = get_user_from_token(authorization)
93    if product_id not in products:
94        raise HTTPException(status_code=404, detail="Product not found")
95    return products[product_id]
96 
97# Invoices
98class InvoiceCreate(BaseModel):
99    amount: float = 0.0
100 
101@app.post("/invoices")
102def create_invoice(invoice: InvoiceCreate, authorization: str = Header(None)):
103    user_id = get_user_from_token(authorization)
104    global next_invoice_id
105    iid = next_invoice_id
106    next_invoice_id += 1
107    invoices[iid] = {"id": iid, "user_id": user_id, "amount": invoice.amount}
108    return invoices[iid]
109 
110@app.get("/invoices/{invoice_id}")
111def get_invoice(invoice_id: int, authorization: str = Header(None)):
112    user_id = get_user_from_token(authorization)
113    if invoice_id not in invoices:
114        raise HTTPException(status_code=404, detail="Invoice not found")
115    return invoices[invoice_id]
116 
117# Stock chart endpoint
118@app.get("/charts/{symbol}/{timeframe}")
119def get_chart(symbol: str, timeframe: str):
120    filepath = f"stock_charts/{symbol}_{timeframe}.png"
121    if not os.path.exists(filepath):
122        raise HTTPException(status_code=404, detail="Chart not found")
123    return FileResponse(filepath, media_type="image/png")

1fastapi
2uvicorn