Browse the data

1import uuid
2import hashlib
3import os
4from fastapi import FastAPI, HTTPException, Header
5from pydantic import BaseModel
6from typing import Optional
7import zipfile
8from pathlib import Path
9import xml.etree.ElementTree as ET
10import shutil
11 
12app = FastAPI()
13 
14users = {}
15tokens = {}
16books = {}
17orders = {}
18 
19def get_user_from_token(authorization: str = Header(None)):
20    if not authorization:
21        raise HTTPException(status_code=401, detail="No auth header")
22    token = authorization.replace("Bearer ", "")
23    user_id = tokens.get(token)
24    if not user_id:
25        raise HTTPException(status_code=401, detail="Invalid token")
26    return user_id
27 
28class SignupRequest(BaseModel):
29    username: str
30    password: str
31 
32class LoginRequest(BaseModel):
33    username: str
34    password: str
35 
36class BookCreate(BaseModel):
37    title: str
38    author: str
39 
40class OrderCreate(BaseModel):
41    book_id: int
42    user_id: int
43 
44@app.post("/signup")
45def signup(req: SignupRequest):
46    if req.username in users:
47        raise HTTPException(status_code=400, detail="User exists")
48    salt = os.urandom(16).hex()
49    hashed = hashlib.sha256((req.password + salt).encode()).hexdigest()
50    user_id = len(users) + 1
51    users[req.username] = {"id": user_id, "password": hashed, "salt": salt}
52    return {"id": user_id, "username": req.username}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    user = users.get(req.username)
57    if not user:
58        raise HTTPException(status_code=401, detail="Invalid credentials")
59    hashed = hashlib.sha256((req.password + user["salt"]).encode()).hexdigest()
60    if hashed != user["password"]:
61        raise HTTPException(status_code=401, detail="Invalid credentials")
62    token = str(uuid.uuid4())
63    tokens[token] = user["id"]
64    return {"token": token}
65 
66@app.get("/books/{book_id}")
67def get_book(book_id: int, authorization: str = Header(None)):
68    user_id = get_user_from_token(authorization)
69    book = books.get(book_id)
70    if not book:
71        raise HTTPException(status_code=404, detail="Book not found")
72    return book
73 
74@app.post("/books")
75def create_book(book: BookCreate, authorization: str = Header(None)):
76    user_id = get_user_from_token(authorization)
77    book_id = len(books) + 1
78    books[book_id] = {"id": book_id, "title": book.title, "author": book.author, "user_id": user_id}
79    return books[book_id]
80 
81@app.get("/orders/{order_id}")
82def get_order(order_id: int, authorization: str = Header(None)):
83    user_id = get_user_from_token(authorization)
84    order = orders.get(order_id)
85    if not order:
86        raise HTTPException(status_code=404, detail="Order not found")
87    return order
88 
89@app.post("/orders")
90def create_order(order: OrderCreate, authorization: str = Header(None)):
91    user_id = get_user_from_token(authorization)
92    if order.book_id not in books:
93        raise HTTPException(status_code=404, detail="Book not found")
94    order_id = len(orders) + 1
95    orders[order_id] = {"id": order_id, "book_id": order.book_id, "user_id": order.user_id}
96    return orders[order_id]
97 
98@app.get("/books/{book_id}/chapters/{chapter_number}")
99def get_chapter(book_id: int, chapter_number: int, authorization: str = Header(None)):
100    user_id = get_user_from_token(authorization)
101    epub_path = Path(f"/ebooks/{book_id}")
102    if not epub_path.exists():
103        raise HTTPException(status_code=404, detail="Book not found")
104    
105    with zipfile.ZipFile(epub_path, 'r') as z:
106        # Find the OPF file
107        opf_files = [f for f in z.namelist() if f.endswith('.opf')]
108        if not opf_files:
109            raise HTTPException(status_code=500, detail="Invalid EPUB")
110        
111        # Parse OPF to get manifest and spine
112        opf_content = z.read(opf_files[0])
113        root = ET.fromstring(opf_content)
114        ns = {'opf': 'http://www.idpf.org/2007/opf'}
115        
116        # Get spine order
117        spine = root.find('.//opf:spine', ns)
118        if spine is None:
119            raise HTTPException(status_code=500, detail="No spine found")
120        
121        itemrefs = spine.findall('opf:itemref', ns)
122        
123        if chapter_number > len(itemrefs) or chapter_number < 1:
124            raise HTTPException(status_code=404, detail="Chapter not found")
125        
126        # Get the item id from spine
127        itemref = itemrefs[chapter_number - 1]
128        idref = itemref.get('idref')
129        
130        # Find the file path in manifest
131        manifest = root.find('.//opf:manifest', ns)
132        items = manifest.findall('opf:item', ns)
133        
134        for item in items:
135            if item.get('id') == idref:
136                file_path = item.get('href')
137                break
138        else:
139            raise HTTPException(status_code=500, detail="Chapter file not found in manifest")
140        
141        # Read the chapter content
142        content = z.read(file_path)
143        return {"chapter_number": chapter_number, "content": content.decode('utf-8')}

1fastapi
2uvicorn
3python-multipart