def remove_category(self, db: Session, db_obj: Product, category: Category): db_obj.categories = list( filter(lambda c: c.id != category.id, db_obj.categories)) db.add(db_obj) db.commit() db.refresh(db_obj)
def add_categories(self, db: Session, db_obj: Product, categories: List[Category]) -> List[Category]: temp = [*db_obj.categories, *categories] db_obj.categories = temp db.add(db_obj) db.commit() db.refresh(db_obj) return temp