def get_add_new_country(current_user, user_country):
if request.method == 'GET':
countries = Country.query.all()
return make_response(
jsonify({"countries": countries_schema.dump(countries)})), 200
if request.method == 'POST':
name = request.data.get('name', '')
code = request.data.get('code', '')
if name and code:
new_country = Country(name, code)
new_country.save()
return make_response(
jsonify({
"message": "Country successfully saved!",
"country": country_schema.dump(new_country)
})), 201
else:
return make_response(
jsonify({"message": "Add a country name and code"})), 400
