Browse the data

1import os
2import uuid
3import shutil
4import subprocess
5from pathlib import Path
6from typing import List
7 
8from fastapi import FastAPI, UploadFile, File, HTTPException, Header
9from fastapi.responses import FileResponse
10 
11app = FastAPI()
12 
13users = {}
14tokens = {}
15user_id_counter = 1
16media_id_counter = 1
17media_store = {}  # media_id -> {user_id, filename, original_path, compressed_path}
18 
19BASE_DIR = Path("dating_profiles")
20BASE_DIR.mkdir(exist_ok=True)
21 
22def get_current_user(authorization: str = Header(None)):
23    if not authorization or not authorization.startswith("Bearer "):
24        raise HTTPException(status_code=401, detail="Unauthorized")
25    token = authorization.split(" ")[1]
26    for uid, t in tokens.items():
27        if t == token:
28            return uid
29    raise HTTPException(status_code=401, detail="Invalid token")
30 
31@app.post("/signup")
32def signup(username: str, password: str):
33    global user_id_counter
34    uid = user_id_counter
35    user_id_counter += 1
36    users[uid] = {"username": username, "password": password}
37    token = str(uuid.uuid4())
38    tokens[uid] = token
39    return {"user_id": uid, "token": token}
40 
41@app.post("/login")
42def login(username: str, password: str):
43    for uid, u in users.items():
44        if u["username"] == username and u["password"] == password:
45            token = str(uuid.uuid4())
46            tokens[uid] = token
47            return {"user_id": uid, "token": token}
48    raise HTTPException(status_code=401, detail="Invalid credentials")
49 
50@app.get("/profiles/{user_id}/media/{media_id}.mp4")
51def stream_media(user_id: int, media_id: int, authorization: str = Header(None)):
52    get_current_user(authorization)
53    if media_id not in media_store:
54        raise HTTPException(status_code=404, detail="Media not found")
55    m = media_store[media_id]
56    if m["user_id"] != user_id:
57        raise HTTPException(status_code=403, detail="Forbidden")
58    path = m["compressed_path"] or m["original_path"]
59    if not path or not os.path.exists(path):
60        raise HTTPException(status_code=404, detail="File not found")
61    return FileResponse(path, media_type="video/mp4")
62 
63@app.post("/profiles/{user_id}/media")
64async def upload_media(user_id: int, files: List[UploadFile] = File(...), authorization: str = Header(None)):
65    get_current_user(authorization)
66    if len(files) > 3:
67        raise HTTPException(status_code=400, detail="Max 3 files")
68    user_dir = BASE_DIR / str(user_id)
69    user_dir.mkdir(exist_ok=True)
70    ids = []
71    for f in files:
72        global media_id_counter
73        mid = media_id_counter
74        media_id_counter += 1
75        ext = os.path.splitext(f.filename)[1] or ".mp4"
76        original_path = user_dir / f"{mid}_original{ext}"
77        compressed_path = user_dir / f"{mid}.mp4"
78        content = await f.read()
79        with open(original_path, "wb") as out:
80            out.write(content)
81        # auto-compress using ffmpeg
82        try:
83            subprocess.run(
84                ["ffmpeg", "-i", str(original_path), "-vcodec", "libx264", "-crf", "28", str(compressed_path)],
85                capture_output=True, check=True
86            )
87        except (subprocess.CalledProcessError, FileNotFoundError):
88            compressed_path = original_path
89        media_store[mid] = {
90            "user_id": user_id,
91            "filename": f.filename,
92            "original_path": str(original_path),
93            "compressed_path": str(compressed_path) if compressed_path != original_path else None
94        }
95        ids.append(mid)
96    return {"media_ids": ids}

1fastapi
2uvicorn
3python-multipart