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)