|
|
from fastapi import FastAPI, File, UploadFile, HTTPException |
|
|
from firebase_admin import credentials, storage |
|
|
import firebase_admin |
|
|
from uuid import uuid4 |
|
|
from fastapi import APIRouter |
|
|
from config.firebase_cfg import init_firebase |
|
|
|
|
|
init_firebase() |
|
|
|
|
|
upload_router = APIRouter() |
|
|
|
|
|
@upload_router.post("/upload_pdf") |
|
|
async def upload_pdf(file: UploadFile = File(...)): |
|
|
if file.content_type != "application/pdf": |
|
|
raise HTTPException(status_code=400, detail="must be PDF file") |
|
|
|
|
|
file_name = f"{uuid4()}.pdf" |
|
|
|
|
|
bucket = storage.bucket() |
|
|
blob = bucket.blob(file_name) |
|
|
|
|
|
blob.upload_from_file(file.file, content_type="application/pdf") |
|
|
blob.make_public() |
|
|
|
|
|
file_url = blob.public_url |
|
|
return {"file_url": file_url} |
|
|
|
|
|
@upload_router.post("/upload_docs") |
|
|
async def upload_docs(file: UploadFile = File(...)): |
|
|
if file.content_type != "application/docs": |
|
|
raise HTTPException(status_code=400, detail="must be DOCS file") |
|
|
|
|
|
file_name = f"{uuid4()}.docs" |
|
|
|
|
|
bucket = storage.bucket() |
|
|
blob = bucket.blob(file_name) |
|
|
|
|
|
blob.upload_from_file(file.file, content_type="application/docs") |
|
|
blob.make_public() |
|
|
|
|
|
file_url = blob.public_url |
|
|
return {"file_url": file_url} |
|
|
|
|
|
@upload_router.post("/delete_file") |
|
|
async def delete_file(file_name: str): |
|
|
try: |
|
|
bucket = storage.bucket() |
|
|
blob = bucket.blob(file_name) |
|
|
|
|
|
if not blob.exists(): |
|
|
raise HTTPException(status_code=404, detail="File not found") |
|
|
|
|
|
blob.delete() |
|
|
|
|
|
return {"message": f"File {file_name} deleted successfully"} |
|
|
except Exception as e: |
|
|
raise HTTPException(status_code=500, detail=f"An error occurred: {str(e)}") |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
