product_routing_service.py 876 B

1234567891011121314151617181920212223
  1. # app/services/product_routing_service.py
  2. from sqlalchemy.orm import Session
  3. from app.core.biz_exception import NotFoundError, PermissionDeniedError, BizLogicError
  4. from app.models.product_routing import VasProductRouting
  5. from app.schemas.product_routing import VasProductRoutingCreate
  6. class ProductRoutingService:
  7. def create(db: Session, data: VasProductRoutingCreate):
  8. rec = VasProductRouting(**data.dict())
  9. db.add(rec)
  10. db.commit()
  11. db.refresh(rec)
  12. return rec
  13. def list_by_product(db: Session, product_id:int):
  14. return db.query(VasProductRouting).filter_by(product_id=product_id).all()
  15. def delete(db: Session, id: int):
  16. obj = db.query(VasProductRouting).filter_by(id=id).first()
  17. if not obj:
  18. raise NotFoundError("Product routing not exist")
  19. db.delete(obj)
  20. db.commit()