Ejemplo n.º 1
0
 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)
Ejemplo n.º 2
0
 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