from typing import Any, Dict, Optional, Union

from fastapi.encoders import jsonable_encoder
from sqlalchemy.orm import Session

from crud.base import CRUDBase
from models.space import Space
from schemas import SpaceCreate, SpaceUpdate


class CRUDSpace(CRUDBase[Space, SpaceCreate, SpaceUpdate]):
    def create_with_space(
            self, db: Session, *, obj_in: SpaceCreate, user_id: int, project_id: int
    ) -> Space:
        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


space = CRUDSpace(Space)