def createProductsForOrder(id):
#parse json
requestJson = request.get_json(force=True)
#validate params
lines = requestJson.get('lines')
if not lines:
return jsonify(error="Required data 'lines' is not set"), 400
if len(lines)<1:
return jsonify(error="Required data not set, specify at least 1 product sku"), 400
order = models.Order.query.get(id)
if order:
#get all product types for this order
productTypes = db.session.query(models.ProductType).join(models.Product).filter(models.Product.order_id == order.id)
#get a list of all the skus, lambdas are so fancy :)
skus = map((lambda x: x.sku), productTypes)
#add each line item
for line in lines:
#validate line params
sku = line.get('sku')
quantity = line.get('quantity')
if (not sku) or (not quantity):
return jsonify(error="Required data on line item not set, specify both sku and quantity"), 400
quantity = int(quantity)
#check if product sku exists
productType = models.ProductType.query.filter_by(sku=sku).first()
#inventory check and update
if not productType:
return jsonify(error="Product sku not found for sku: "+sku), 400
elif not productType.verifyAndUpdateInventory(quantity):
return jsonify(error="Not enough inventory for sku: "+sku), 400
#add product to this order if it is not already on the order
if sku not in skus:
product = models.Product(quantity, productType, order)
db.session.add(product)
#add up total
lineTotal = productType.price * quantity
order.increaseTotal(lineTotal)
#else, order already has product type, ignore
try:
db.session.commit()
return jsonify(order_api.formatOrderForGetResponse(order)), 200
except exc.SQLAlchemyError as err:
return jsonify(error="Failed to add line item to Order"), 400
else:
return jsonify(error="Failed to find given Order"), 400
def updateProductsFromOrder(id):
#parse json
requestJson = request.get_json(force=True)
#validate params
lines = requestJson.get('lines')
if not lines:
return jsonify(error="Required data 'lines' is not set"), 400
if len(lines)<1:
return jsonify(error="Required data not set, specify at least 1 product sku"), 400
order = models.Order.query.get(id)
if order:
for line in lines:
#validate line params
sku = line.get('sku')
quantity = line.get('quantity')
if (not sku) or (not quantity):
return jsonify(error="Required data on line item not set, specify both sku and quantity"), 400
quantity = int(quantity)
if quantity < 1:
#ignore, and continue
continue
#check if product type sku exists
productType = models.ProductType.query.filter_by(sku=sku).first()
#check and update inventory
if not productType:
return jsonify(error="Product sku not found for sku: "+sku), 400
elif not productType.verifyAndUpdateInventory(quantity):
return jsonify(error="Not enough inventory for sku: "+sku), 400
#get the product
product = models.Product.query.filter_by(order_id=order.id, product_type_id=productType.id ).first()
#update total, reduce or increase price based on quantity
order.adjustTotalForQuantity(product.quantity, quantity, productType.price)
#update the product quantity
product.quantity = quantity
try:
db.session.commit()
return jsonify(order_api.formatOrderForGetResponse(order)), 200
except exc.SQLAlchemyError as err:
return jsonify(error="Failed to update line items on Order"), 400
else:
return jsonify(error="Failed to find given Order"), 400
def deleteProductsFromOrder(id):
#parse json
requestJson = request.get_json(force=True)
#validate params
lines = requestJson.get('lines')
if not lines:
return jsonify(error="Required data 'lines' is not set"), 400
order = models.Order.query.get(id)
if order:
#validate line params
skusToRemove = validateLineItemsToRemove(lines)
if skusToRemove == None:
return jsonify(error="Required data on line item not set, specify a sku"), 400
#get all product types for this order
productTypes = db.session.query(models.ProductType).join(models.Product).filter(models.Product.order_id == order.id)
for productType in productTypes:
if productType.sku in skusToRemove:
#get product
product = models.Product.query.filter_by(order_id=order.id, product_type_id=productType.id ).first()
#reduce total
lineTotal = productType.price * product.quantity
order.decreaseTotal(lineTotal)
#remove product
db.session.delete(product)
try:
db.session.commit()
return jsonify(order_api.formatOrderForGetResponse(order)), 200
except exc.SQLAlchemyError as err:
return jsonify(error="Failed to remove line items from Order"), 400
else:
return jsonify(error="Failed to find given Order"), 400