xbackend/crud/curd_space.py
2021-04-30 09:36:12 +08:00

24 lines
675 B
Python

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)