def delete(self):
data = self.parser.parse_args()
companyName = data['companyName']
company = CompanyModel.find_by_name(companyName)
if company:
company.delete_from_db()
return {'message': 'Company deleted.'}
return {'message': 'Company not found.'}, 404
def post(self, name):
if CompanyModel.find_by_name(name):
return {'message': "A store with name '{}' already exists.".format(name)}, 400
company = CompanyModel(name)
try:
company.save_to_db()
except:
return {"message": "An error occurred creating the store."}, 500
return company.json(), 201
def test_crud(self):
with self.app_context():
competition = CompanyModel('AVIANCA')
self.assertIsNone(
CompanyModel.find_by_name('AVIANCA').first(),
"Found an competition with name 'AVIANCA' before save_to_db")
competition.save_to_db()
self.assertIsNotNone(
CompanyModel.find_by_name('AVIANCA').first(),
"Did not find a competition with name 'AVIANCA' after save_to_db"
)
competition.delete_from_db()
self.assertIsNone(
CompanyModel.find_by_name('AVIANCA').first(),
"Found an competition with name 'AVIANCA' after delete_from_db"
)
def test_create_company(self):
with self.app() as c:
with self.app_context():
r = c.post('/company/Azul')
self.assertEqual(r.status_code, 201)
self.assertIsNotNone(CompanyModel.find_by_name('Azul'))
self.assertDictEqual(d1={
'id': 1,
'name': 'Azul'
},
d2=json.loads(r.data.decode('utf-8')))
def put(self):
data = self.parser.parse_args()
companyName = data['companyName']
company = CompanyModel.find_by_name(companyName)
# if company:
# company.price = data['price']
# else:
# company = CompanyModel(companyName, **data)
company.save_to_db()
return company.json()
def post(self, name):
company = CompanyModel.find_by_name(name)
if company:
data = Purchase.parser.parse_args()
purchase_date = datetime.strptime(data['purchase_date'],
"%Y-%m-%d")
purchase = PurchaseModel(data['price'], data['quantity'],
company.id, purchase_date)
try:
purchase.save()
except:
return {
"message": "An error occurred inserting the purchase."
}, 500
else:
return {"message": "Company not found."}, 204
return purchase.json(), 201
def post(self, company_name):
company = CompanyModel.find_by_name(company_name)
if company:
data = Income.parser.parse_args()
date = datetime.strptime(data['receive_date'], "%Y-%m-%d")
income = IncomeModel(data['value'], data['income_type'],
company.id, date)
try:
income.save()
except:
return {
"message": "An error occurred inserting the income."
}, 500
else:
return {"message": "Company not found"}, 204
return income.json()
def post(self):
data = self.parser.parse_args()
companyName = data['companyName']
if CompanyModel.find_by_name(companyName):
return {
'message':
"An item with name '{}' already exists.".format(companyName)
}, 400
company = CompanyModel(companyName, **data)
try:
company.save_to_db()
except:
return {
"message": "An error occurred while inserting the item."
}, 500
return company.json(), 201
def get(self,name):
company = CompanyModel.find_by_name(name)
if company:
return company.json()
return {'message':'company does not exist'},400
def delete(self, name):
company = CompanyModel.find_by_name(name)
if company:
company.delete_from_db()
return {'message': 'Store deleted'}
def get(self, name):
company = CompanyModel.find_by_name(name)
if company:
return company.json()
return {'message': 'Company not found'}, 404
