Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3import hashlib
4import random
5import string
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11documents = {}
12document_versions = {}
13next_user_id = 1
14next_doc_id = 1
15 
16def generate_token():
17    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
18 
19def get_current_user(authorization: str = Header(None)):
20    if not authorization or not authorization.startswith("Bearer "):
21        raise HTTPException(status_code=401, detail="Invalid auth")
22    token = authorization[7:]
23    if token not in tokens:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return tokens[token]
26 
27class SignupRequest(BaseModel):
28    username: str
29    password: str
30 
31class LoginRequest(BaseModel):
32    username: str
33    password: str
34 
35class DocumentCreate(BaseModel):
36    content: str
37 
38@app.post("/signup")
39def signup(req: SignupRequest):
40    global next_user_id
41    if req.username in users:
42        raise HTTPException(status_code=400, detail="User exists")
43    user_id = next_user_id
44    next_user_id += 1
45    users[req.username] = {"id": user_id, "password": hashlib.sha256(req.password.encode()).hexdigest()}
46    return {"id": user_id, "username": req.username}
47 
48@app.post("/login")
49def login(req: LoginRequest):
50    if req.username not in users:
51        raise HTTPException(status_code=400, detail="Invalid credentials")
52    if users[req.username]["password"] != hashlib.sha256(req.password.encode()).hexdigest():
53        raise HTTPException(status_code=400, detail="Invalid credentials")
54    token = generate_token()
55    tokens[token] = req.username
56    return {"token": token}
57 
58@app.post("/documents")
59def create_document(doc: DocumentCreate, authorization: str = Header(None)):
60    global next_doc_id
61    user = get_current_user(authorization)
62    doc_id = next_doc_id
63    next_doc_id += 1
64    documents[doc_id] = {"id": doc_id, "content": doc.content, "version": 1, "created_by": user}
65    document_versions[doc_id] = {1: doc.content}
66    return documents[doc_id]
67 
68@app.get("/documents/{doc_id}")
69def get_document(doc_id: int, authorization: str = Header(None)):
70    user = get_current_user(authorization)
71    if doc_id not in documents:
72        raise HTTPException(status_code=404, detail="Document not found")
73    return documents[doc_id]
74 
75@app.post("/documents/{doc_id}/versions")
76def create_version(doc_id: int, doc: DocumentCreate, authorization: str = Header(None)):
77    user = get_current_user(authorization)
78    if doc_id not in documents:
79        raise HTTPException(status_code=404, detail="Document not found")
80    current = documents[doc_id]
81    new_version = current["version"] + 1
82    current["content"] = doc.content
83    current["version"] = new_version
84    document_versions[doc_id][new_version] = doc.content
85    return current
86 
87@app.get("/documents/{doc_id}/versions/{version}")
88def get_document_version(doc_id: int, version: int, authorization: str = Header(None)):
89    user = get_current_user(authorization)
90    if doc_id not in documents:
91        raise HTTPException(status_code=404, detail="Document not found")
92    if version not in document_versions[doc_id]:
93        raise HTTPException(status_code=404, detail="Version not found")
94    return {"doc_id": doc_id, "version": version, "content": document_versions[doc_id][version]}
95 
96@app.get("/documents/{doc_id}/diff")
97def diff_document(doc_id: int, version1: int, version2: int, authorization: str = Header(None)):
98    user = get_current_user(authorization)
99    if doc_id not in documents:
100        raise HTTPException(status_code=404, detail="Document not found")
101    if version1 not in document_versions[doc_id] or version2 not in document_versions[doc_id]:
102        raise HTTPException(status_code=404, detail="Version not found")
103    lines1 = document_versions[doc_id][version1].splitlines()
104    lines2 = document_versions[doc_id][version2].splitlines()
105    diff_lines = []
106    import difflib
107    for line in difflib.unified_diff(lines1, lines2, lineterm='', fromfile=f'v{version1}', tofile=f'v{version2}'):
108        diff_lines.append(line)
109    return {"doc_id": doc_id, "version1": version1, "version2": version2, "diff": diff_lines}

1fastapi
2uvicorn