explorigin/image_server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Move / to /random

Browse Source
This commit is contained in:
Timothy Farrell 2026-03-09 08:28:40 +00:00
parent e11a45ab70
commit bab4e68d4b
1 changed files with 20 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
main.py
View File

@ -8,7 +8,7 @@ import secrets
import zipfile import zipfile
from pathlib import Path from pathlib import Path
from fastapi import FastAPI, HTTPException from fastapi import FastAPI, HTTPException
from fastapi.responses import FileResponse, StreamingResponse from fastapi.responses import FileResponse, StreamingResponse, RedirectResponse
import mimetypes import mimetypes
app = FastAPI() app = FastAPI()
@ -49,7 +49,7 @@ class FileIndexer:
return None return None
file_path = self.file_mapping[file_hash] file_path = self.file_mapping[file_hash]
with open(file_path, 'rb') as f: with open(file_path, "rb") as f:
yield from f yield from f
def get_filename_by_hash(self, file_hash: str) -> str: def get_filename_by_hash(self, file_hash: str) -> str:
@ -62,7 +62,7 @@ class FileIndexer:
class ZipFileIndexer(FileIndexer): class ZipFileIndexer(FileIndexer):
def _index(self): def _index(self):
"""Index all files in the zip file""" """Index all files in the zip file"""
with zipfile.ZipFile(self.path, 'r') as zip_file: with zipfile.ZipFile(self.path, "r") as zip_file:
self.file_mapping = { self.file_mapping = {
self._hash_path(file_info.filename): file_info self._hash_path(file_info.filename): file_info
for file_info in zip_file.infolist() for file_info in zip_file.infolist()
@ -76,7 +76,7 @@ class ZipFileIndexer(FileIndexer):
file_info = self.file_mapping[file_hash] file_info = self.file_mapping[file_hash]
with zipfile.ZipFile(self.path, 'r') as zip_file: with zipfile.ZipFile(self.path, "r") as zip_file:
yield from BytesIO(zip_file.read(file_info.filename)) yield from BytesIO(zip_file.read(file_info.filename))
def get_filename_by_hash(self, file_hash: str) -> str: def get_filename_by_hash(self, file_hash: str) -> str:
@ -85,9 +85,8 @@ class ZipFileIndexer(FileIndexer):
return None return None
return self.file_mapping[file_hash].filename return self.file_mapping[file_hash].filename
INDEXER_MAP = {
".zip": ZipFileIndexer INDEXER_MAP = {".zip": ZipFileIndexer}
}
def initialize_server(args: argparse.Namespace): def initialize_server(args: argparse.Namespace):
@ -105,7 +104,14 @@ def initialize_server(args: argparse.Namespace):
print(f"Indexed {len(indexer.file_mapping)} files") print(f"Indexed {len(indexer.file_mapping)} files")
@app.get("/") @app.get("/")
async def root():
"""Redirect to /random"""
return RedirectResponse(url="/random")
@app.get("/random")
async def get_random_file(): async def get_random_file():
"""Serve a random file from the mapping""" """Serve a random file from the mapping"""
if not indexer.file_mapping: if not indexer.file_mapping:
@ -125,11 +131,12 @@ async def get_random_file():
"Content-Disposition": f"inline; filename={os.path.basename(filename)}", "Content-Disposition": f"inline; filename={os.path.basename(filename)}",
"Cache-Control": "no-cache, no-store, must-revalidate", "Cache-Control": "no-cache, no-store, must-revalidate",
"Pragma": "no-cache", "Pragma": "no-cache",
"Expires": "0" "Expires": "0",
} },
) )
return response return response
@app.get("/{file_hash}") @app.get("/{file_hash}")
async def get_file_by_hash(file_hash: str): async def get_file_by_hash(file_hash: str):
"""Serve a specific file by its hash""" """Serve a specific file by its hash"""
@ -143,14 +150,16 @@ async def get_file_by_hash(file_hash: str):
media_type=content_type, media_type=content_type,
headers={ headers={
"Content-Disposition": f"inline; filename={os.path.basename(file_path)}", "Content-Disposition": f"inline; filename={os.path.basename(file_path)}",
} },
) )
# Optional: Add a health check endpoint # Optional: Add a health check endpoint
@app.get("/health") @app.get("/health")
async def health_check(): async def health_check():
return {"status": "healthy", "file_count": len(file_mapping)} return {"status": "healthy", "file_count": len(file_mapping)}
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Run the file server") parser = argparse.ArgumentParser(description="Run the file server")
parser.add_argument("source", type=str, help="Path to directory or ZIP archive") parser.add_argument("source", type=str, help="Path to directory or ZIP archive")
@ -161,4 +170,5 @@ if __name__ == "__main__":
initialize_server(args) initialize_server(args)
import uvicorn import uvicorn
uvicorn.run(app, host=args.host, port=args.port) uvicorn.run(app, host=args.host, port=args.port)