mirror of
https://github.com/bellingcat/auto-archiver-api.git
synced 2026-06-08 03:28:35 +03:00
78 lines
3.2 KiB
Python
78 lines
3.2 KiB
Python
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from fastapi.responses import JSONResponse
|
|
from datetime import datetime
|
|
from loguru import logger
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.shared.config import ALLOW_ANY_EMAIL
|
|
from app.shared import schemas
|
|
from app.shared.task_messaging import get_celery
|
|
from app.web.security import get_token_or_user_auth, get_user_state
|
|
from app.shared.db import crud
|
|
from app.web.db.user_state import UserState
|
|
from app.shared.db.database import get_db_dependency
|
|
|
|
from urllib.parse import urlparse
|
|
|
|
url_router = APIRouter(prefix="/url", tags=["Single URL operations"])
|
|
|
|
celery = get_celery()
|
|
|
|
@url_router.post("/archive", status_code=201, summary="Submit a single URL archive request, starts an archiving task.", response_description="task_id for the archiving task, will match the archive id.")
|
|
def archive_url(
|
|
archive: schemas.ArchiveTrigger,
|
|
email=Depends(get_token_or_user_auth),
|
|
db: Session = Depends(get_db_dependency)
|
|
) -> schemas.Task:
|
|
archive.author_id = email
|
|
logger.info(f"new {archive.public=} task for {email=} and {archive.group_id=}: {archive.url}")
|
|
|
|
parsed_url = urlparse(archive.url)
|
|
if not all([parsed_url.scheme, parsed_url.netloc]):
|
|
raise HTTPException(status_code=400, detail="Invalid URL received.")
|
|
|
|
if email != ALLOW_ANY_EMAIL:
|
|
user = UserState(db, email)
|
|
if archive.group_id and not user.in_group(archive.group_id):
|
|
raise HTTPException(status_code=403, detail="User does not have access to this group.")
|
|
if not user.has_quota_max_monthly_urls(archive.group_id):
|
|
raise HTTPException(status_code=429, detail="User has reached their monthly URL quota.")
|
|
if not user.has_quota_max_monthly_mbs(archive.group_id):
|
|
raise HTTPException(status_code=429, detail="User has reached their monthly MB quota.")
|
|
|
|
archive_create = schemas.ArchiveCreate(**archive.model_dump())
|
|
|
|
task = celery.signature("create_archive_task", args=[archive_create.model_dump_json()]).delay()
|
|
task_response = schemas.Task(id=task.id)
|
|
return JSONResponse(task_response.model_dump(), status_code=201)
|
|
|
|
|
|
@url_router.get("/search", summary="Search for archive entries by URL.")
|
|
def search_by_url(
|
|
url: str, skip: int = 0, limit: int = 25,
|
|
archived_after: datetime = None, archived_before: datetime = None,
|
|
db: Session = Depends(get_db_dependency),
|
|
email: str = Depends(get_token_or_user_auth)
|
|
) -> list[schemas.ArchiveResult]:
|
|
|
|
if email != ALLOW_ANY_EMAIL:
|
|
user = UserState(db, email)
|
|
if not user.read and not user.read_public:
|
|
raise HTTPException(status_code=403, detail="User does not have read access.")
|
|
|
|
return crud.search_archives_by_url(db, url.strip(), email, skip=skip, limit=limit, archived_after=archived_after, archived_before=archived_before)
|
|
|
|
|
|
@url_router.delete("/{id}", summary="Delete a single URL archive by id.")
|
|
def delete_task(
|
|
id:str,
|
|
user: UserState = Depends(get_user_state),
|
|
db: Session = Depends(get_db_dependency)
|
|
) -> schemas.TaskDelete:
|
|
logger.info(f"deleting url archive task {id} request by {user.email}")
|
|
return JSONResponse({
|
|
"id": id,
|
|
"deleted": crud.soft_delete_task(db, id, user.email)
|
|
})
|