def put(self, name):
data = Product.parser.parse_args()
product = ProductModel.find_by_name(name)
if product is None:
product = ProductModel(name, **data)
else:
product.price = data['price']
product.save_to_db()
return product.json()
def test_save_product(self):
products = len(ProductModel.list())
new_product = ProductModel()
new_product.name = "Produto 02"
new_product.provider = ProviderModel.get_by_cnpj("04790618000153")
new_product.quantity = 100
new_product.price = 10.0
new_product.available = True
new_product.save()
assert products + 1 == len(ProductModel.list())
def first_product():
if ProviderModel.get_by_cnpj("04790618000153") == None:
first_provider()
return
new_product = ProductModel()
new_product.name = "Produto 01"
new_product.provider = ProviderModel.get_by_cnpj("04790618000153")
new_product.quantity = 100
new_product.price = 10.0
new_product.available = True
new_product.save()
return
def put(self, name):
data = Product.parser.parse_args()
product = ProductModel.find_by_name(name)
if product is None:
product = ProductModel(name, data['price'], data['category_name'],
data['imageUrl'])
else:
product.price = data['price']
product.category = data['category_name']
product.imageUrl = data['imageUrl']
try:
product.save_to_db()
except:
return {"message": " unable to save /update product in DB"}, 500
return product.json(), 200
