def put(self, name):
data = Store.parser.parse_args()
store = StoreModel.get_item_by_name(name)
if store:
store.street = data['street']
store.city = data['city']
store.province = data['province']
else:
data['name'] = name
store = StoreModel(**data)
store.save_to_db()
return store.convert_to_dict(), 201
def post(self, name):
data = Store.parser.parse_args()
if StoreModel.get_item_by_name(name):
return {'message': f'A store with "{name}" already exists.'}, 400
data['name'] = name
store = StoreModel(**data)
try:
store.save_to_db()
except:
return {
'message': "An error occurred while creating the store."
}, 500
return store.convert_to_dict(), 201
