32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
from typing import Any, Dict, Optional, Union
|
|
|
|
from fastapi import HTTPException
|
|
from fastapi.encoders import jsonable_encoder
|
|
from sqlalchemy.orm import Session
|
|
|
|
from crud.base import CRUDBase
|
|
from models.folders import Folder
|
|
from models.kanban import KanBan
|
|
from models.space import Space
|
|
from schemas import FolderUpdate, FolderCreate
|
|
|
|
|
|
class CRUDFolder(CRUDBase[Folder, FolderCreate, FolderUpdate]):
|
|
def create_with_folder(
|
|
self, db: Session, *, obj_in: FolderCreate, user_id: int, project_id: int
|
|
) -> Folder:
|
|
|
|
if (not (obj_in.space_id and db.query(Space).get(obj_in.space_id))) ^ \
|
|
(not (obj_in.kanban_id and db.query(KanBan).get(obj_in.kanban_id))):
|
|
obj_in_data = jsonable_encoder(obj_in)
|
|
db_obj = self.model(**obj_in_data, user_id=user_id, project_id=project_id)
|
|
db.add(db_obj)
|
|
db.commit()
|
|
db.refresh(db_obj)
|
|
return db_obj
|
|
else:
|
|
raise HTTPException(status_code=404, detail="找不到上级")
|
|
|
|
|
|
folder = CRUDFolder(Folder)
|