Spaces:
Sleeping
Sleeping
File size: 2,360 Bytes
f02c5b9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 | """Folder management use cases."""
from typing import List, Optional
from sqlalchemy.orm import Session
from app.models import Folder
class FolderService:
"""Manages folder operations."""
def __init__(self, db: Session):
self.db = db
def create_folder(
self,
name: str,
org_id: str,
parent_id: Optional[str] = None
) -> dict:
"""Create a new folder."""
folder = Folder(
name=name,
org_id=org_id,
parent_id=parent_id
)
self.db.add(folder)
self.db.commit()
self.db.refresh(folder)
return {
"id": folder.id,
"name": folder.name,
"parent_id": folder.parent_id,
"created_at": folder.created_at
}
def list_folders(
self,
org_id: str,
parent_id: Optional[str] = None
) -> List[dict]:
"""List folders for organization."""
query = self.db.query(Folder).filter(Folder.org_id == org_id)
if parent_id:
query = query.filter(Folder.parent_id == parent_id)
else:
query = query.filter(Folder.parent_id == None)
folders = query.order_by(Folder.created_at.desc()).all()
return [
{
"id": f.id,
"name": f.name,
"parent_id": f.parent_id,
"created_at": f.created_at
}
for f in folders
]
def delete_folder(self, folder_id: str, org_id: str) -> None:
"""Delete a folder (must be empty)."""
folder = self.db.query(Folder).filter(
Folder.id == folder_id,
Folder.org_id == org_id
).first()
if not folder:
raise ValueError("Folder not found")
# Check if folder has documents
if folder.documents:
raise ValueError("Folder must be empty before deletion (contains documents)")
# Check if folder has subfolders
subfolders = self.db.query(Folder).filter(Folder.parent_id == folder_id).count()
if subfolders > 0:
raise ValueError("Folder must be empty before deletion (contains subfolders)")
self.db.delete(folder)
self.db.commit()
|