149 lines
4.1 KiB
Python
149 lines
4.1 KiB
Python
from typing import Annotated
|
||
|
||
from fastapi import APIRouter, Depends, HTTPException
|
||
from pydantic import BaseModel
|
||
|
||
from middleware.auth import AuthRequired
|
||
from middleware.dependencies import SessionDep
|
||
from models import Object, ObjectType, User, response
|
||
|
||
directory_router = APIRouter(
|
||
prefix="/directory",
|
||
tags=["directory"]
|
||
)
|
||
|
||
|
||
class DirectoryCreateRequest(BaseModel):
|
||
"""创建目录请求"""
|
||
|
||
path: str
|
||
"""目录路径，如 /docs/images"""
|
||
|
||
policy_id: int | None = None
|
||
"""存储策略ID，不指定则继承父目录"""
|
||
|
||
@directory_router.get(
|
||
path="/{path:path}",
|
||
summary="获取目录内容",
|
||
)
|
||
async def router_directory_get(
|
||
session: SessionDep,
|
||
user: Annotated[User, Depends(AuthRequired)],
|
||
path: str = ""
|
||
) -> response.ResponseModel:
|
||
"""
|
||
获取目录内容
|
||
|
||
:param session: 数据库会话
|
||
:param user: 当前登录用户
|
||
:param path: 目录路径，空或 "/" 表示根目录
|
||
:return: 目录内容
|
||
"""
|
||
folder = await Object.get_by_path(session, user.id, path or "/")
|
||
|
||
if not folder:
|
||
raise HTTPException(status_code=404, detail="目录不存在")
|
||
|
||
if not folder.is_folder:
|
||
raise HTTPException(status_code=400, detail="指定路径不是目录")
|
||
|
||
children = await Object.get_children(session, user.id, folder.id)
|
||
policy = await folder.awaitable_attrs.policy
|
||
|
||
objects = [
|
||
response.ObjectModel(
|
||
id=str(child.id),
|
||
name=child.name,
|
||
path=f"/{child.name}", # TODO: 完整路径
|
||
thumb=False,
|
||
size=child.size,
|
||
type="folder" if child.is_folder else "file",
|
||
date=child.updated_at,
|
||
create_date=child.created_at,
|
||
source_enabled=False,
|
||
)
|
||
for child in children
|
||
]
|
||
|
||
return response.ResponseModel(
|
||
data=response.DirectoryModel(
|
||
parent=str(folder.parent_id) if folder.parent_id else None,
|
||
objects=objects,
|
||
policy=response.PolicyModel(
|
||
id=str(policy.id),
|
||
name=policy.name,
|
||
type=policy.type.value,
|
||
max_size=policy.max_size,
|
||
file_type=[],
|
||
)
|
||
)
|
||
)
|
||
|
||
|
||
@directory_router.put(
|
||
path="/",
|
||
summary="创建目录",
|
||
)
|
||
async def router_directory_create(
|
||
session: SessionDep,
|
||
user: Annotated[User, Depends(AuthRequired)],
|
||
request: DirectoryCreateRequest
|
||
) -> response.ResponseModel:
|
||
"""
|
||
创建目录
|
||
|
||
:param session: 数据库会话
|
||
:param user: 当前登录用户
|
||
:param request: 创建请求
|
||
:return: 创建结果
|
||
"""
|
||
path = request.path.strip()
|
||
if not path or path == "/":
|
||
raise HTTPException(status_code=400, detail="路径不能为空或根目录")
|
||
|
||
# 解析路径
|
||
if path.startswith("/"):
|
||
path = path[1:]
|
||
parts = [p for p in path.split("/") if p]
|
||
|
||
if not parts:
|
||
raise HTTPException(status_code=400, detail="无效的目录路径")
|
||
|
||
new_folder_name = parts[-1]
|
||
parent_path = "/" + "/".join(parts[:-1]) if len(parts) > 1 else "/"
|
||
|
||
parent = await Object.get_by_path(session, user.id, parent_path)
|
||
if not parent:
|
||
raise HTTPException(status_code=404, detail="父目录不存在")
|
||
|
||
if not parent.is_folder:
|
||
raise HTTPException(status_code=400, detail="父路径不是目录")
|
||
|
||
# 检查是否已存在同名对象
|
||
existing = await Object.get(
|
||
session,
|
||
(Object.owner_id == user.id) &
|
||
(Object.parent_id == parent.id) &
|
||
(Object.name == new_folder_name)
|
||
)
|
||
if existing:
|
||
raise HTTPException(status_code=409, detail="同名文件或目录已存在")
|
||
|
||
policy_id = request.policy_id if request.policy_id else parent.policy_id
|
||
|
||
new_folder = await Object(
|
||
name=new_folder_name,
|
||
type=ObjectType.FOLDER,
|
||
owner_id=user.id,
|
||
parent_id=parent.id,
|
||
policy_id=policy_id,
|
||
).save(session)
|
||
|
||
return response.ResponseModel(
|
||
data={
|
||
"id": new_folder.id,
|
||
"name": new_folder.name,
|
||
"path": f"{parent_path.rstrip('/')}/{new_folder_name}",
|
||
}
|
||
)
|