rbox/mywebdav/routers/users.py

from fastapi import APIRouter, Depends
from ..auth import get_current_user
from ..models import User_Pydantic, User, File, Folder
from typing import Dict, Any

router = APIRouter(
    prefix="/users",
    tags=["users"],
)


@router.get("/me", response_model=User_Pydantic)
async def read_users_me(current_user: User = Depends(get_current_user)):
    return await User_Pydantic.from_tortoise_orm(current_user)


@router.get("/me/export", response_model=Dict[str, Any])
async def export_my_data(current_user: User = Depends(get_current_user)):
    """
    Exports all personal data associated with the current user.
    Includes user profile, and metadata for all owned files and folders.
    """
    user_data = await User_Pydantic.from_tortoise_orm(current_user)

    files = await File.filter(owner=current_user).values(
        "id", "name", "size", "created_at", "modified_at", "file_type", "parent_id"
    )
    folders = await Folder.filter(owner=current_user).values(
        "id", "name", "created_at", "modified_at", "parent_id"
    )

    return {
        "user_profile": user_data.dict(),
        "files_metadata": files,
        "folders_metadata": folders,
        # In a more complete implementation, other data like activity logs,
        # share information, etc., would also be included.
    }


@router.delete("/me", status_code=204)
async def delete_my_account(current_user: User = Depends(get_current_user)):
    """
    Deletes the current user's account and all associated data.
    This includes all files and folders owned by the user.
    """
    # Delete all files and folders owned by the user
    await File.filter(owner=current_user).delete()
    await Folder.filter(owner=current_user).delete()

    # Finally, delete the user account
    await current_user.delete()
    return {}