Browse the data

1import os
2import shutil
3from fastapi import FastAPI, HTTPException, Header, Body
4from fastapi.responses import FileResponse
5from typing import Optional
6import uuid
7import hashlib
8import time
9 
10app = FastAPI()
11 
12users = {}
13tokens = {}
14blueprints = {}
15blueprint_versions = {}
16next_user_id = 1
17next_blueprint_id = 1
18 
19BLUEPRINT_STORAGE = "/blueprint_storage"
20 
21def verify_token(authorization: Optional[str] = Header(None)):
22    if not authorization:
23        raise HTTPException(status_code=401, detail="Missing auth header")
24    token = authorization.replace("Bearer ", "") if authorization.startswith("Bearer ") else authorization
25    if token not in tokens:
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return tokens[token]
28 
29@app.post("/signup")
30def signup(username: str = Body(...), password: str = Body(...)):
31    global next_user_id
32    for user in users.values():
33        if user["username"] == username:
34            raise HTTPException(status_code=400, detail="Username already exists")
35    user_id = next_user_id
36    next_user_id += 1
37    users[user_id] = {"id": user_id, "username": username, "password": password}
38    return {"id": user_id, "username": username}
39 
40@app.post("/login")
41def login(username: str = Body(...), password: str = Body(...)):
42    for user in users.values():
43        if user["username"] == username and user["password"] == password:
44            token = str(uuid.uuid4())
45            tokens[token] = user["id"]
46            return {"token": token}
47    raise HTTPException(status_code=401, detail="Invalid credentials")
48 
49@app.get("/blueprints/{blueprint_id}")
50def get_blueprint(blueprint_id: int, authorization: Optional[str] = Header(None)):
51    verify_token(authorization)
52    if blueprint_id not in blueprints:
53        raise HTTPException(status_code=404, detail="Blueprint not found")
54    return blueprints[blueprint_id]
55 
56@app.post("/blueprints")
57def create_blueprint(name: str = Body(...), project_id: int = Body(...), authorization: Optional[str] = Header(None)):
58    user_id = verify_token(authorization)
59    global next_blueprint_id
60    blueprint_id = next_blueprint_id
61    next_blueprint_id += 1
62    blueprints[blueprint_id] = {
63        "id": blueprint_id,
64        "name": name,
65        "project_id": project_id,
66        "created_by": user_id,
67        "current_version": 1
68    }
69    blueprint_versions[blueprint_id] = {1: {"version": 1, "created_at": time.time()}}
70    return blueprints[blueprint_id]
71 
72@app.get("/blueprints/{project_id}/{filename}")
73def download_blueprint(project_id: int, filename: str, authorization: Optional[str] = Header(None)):
74    verify_token(authorization)
75    file_path = os.path.join(BLUEPRINT_STORAGE, str(project_id), filename)
76    if not os.path.exists(file_path):
77        raise HTTPException(status_code=404, detail="File not found")
78    return FileResponse(file_path, media_type="application/octet-stream", filename=filename)
79 
80@app.get("/blueprints/{project_id}/{filename}/versions")
81def get_blueprint_versions(project_id: int, filename: str, authorization: Optional[str] = Header(None)):
82    verify_token(authorization)
83    for bp_id, bp in blueprints.items():
84        if bp["project_id"] == project_id and bp["name"] == filename:
85            return blueprint_versions.get(bp_id, {})
86    raise HTTPException(status_code=404, detail="Blueprint not found")
87 
88@app.get("/blueprints/{project_id}/{filename}/version/{version}")
89def download_blueprint_version(project_id: int, filename: str, version: int, authorization: Optional[str] = Header(None)):
90    verify_token(authorization)
91    for bp_id, bp in blueprints.items():
92        if bp["project_id"] == project_id and bp["name"] == filename:
93            version_path = os.path.join(BLUEPRINT_STORAGE, str(project_id), f"v{version}", filename)
94            if os.path.exists(version_path):
95                return FileResponse(version_path, media_type="application/octet-stream", filename=filename)
96            raise HTTPException(status_code=404, detail="Version not found")
97    raise HTTPException(status_code=404, detail="Blueprint not found")

1fastapi
2uvicorn