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