def get_authors():
id = request.args.get('id')
author = request.args.get('author')
authors = []
if id:
obj_id = ObjectId(id)
authors = authors_collection.find_one({"_id": obj_id})
if author:
authors = authors_collection.find_one({"name": author})
if not authors:
authors = [author for author in authors_collection.find()]
return make_response(
jsonify({"success": PyMongoParser.parse_json(authors)}), 200)
def update_news():
news = None
if not request.json:
return make_response(jsonify({"error": "empty json body"}), 400)
id = request.json.get("id")
title = request.json.get("title")
content = request.json.get("content")
author_id = request.json.get("author_id")
if id:
news_obj_id = ObjectId(id)
news = news_collection.find_one({"_id": news_obj_id})
if not news:
return make_response(
jsonify({"error": "News id {} not found".format(id)}), 400)
title, errors = NewsValidator.title_validators(title)
content, errors = NewsValidator.content_validators(content)
author_id, errors = AuthorValidator.author_key_validators(author_id)
if len(errors) > 0:
print(errors)
return make_response(jsonify({"error": errors}), 400)
obj_id = ObjectId(author_id)
author = authors_collection.find_one({"_id": obj_id})
if not author:
return make_response(
jsonify({
"error":
"author with id {} does not exist".format(author_id)
}), 400)
data = {
"title": title,
"content": content,
"updated_at": datetime.now(),
"author": author,
}
try:
news.update(data)
news_collection.save(news)
return make_response(
jsonify({"success": PyMongoParser.parse_json(news)}), 201)
except Exception as error:
print(error)
return make_response(
jsonify({"error": "error on updating: {}".format(error)}), 400)
def add_news():
if not request.json:
return make_response(jsonify({"error": "empty json body"}), 400)
title = request.json.get('title')
content = request.json.get('content')
author_id = request.json.get('author_id')
title, errors = NewsValidator.title_validators(title)
content, errors = NewsValidator.content_validators(content)
author_id, errors = AuthorValidator.author_key_validators(author_id)
if len(errors) > 0:
print(errors)
return make_response(jsonify({"error": errors}), 400)
obj_id = ObjectId(author_id)
author = authors_collection.find_one({"_id": obj_id})
if not author:
return make_response(
jsonify({
"error":
"author with id {} does not exist".format(author_id)
}), 400)
if news_collection.find_one({"title": title}):
return make_response(
jsonify(
{"error": "News with title {} already exists".format(title)}),
400)
data = {
"title": title,
"content": content,
"created_at": datetime.now(),
"author": author
}
added_news = news_collection.insert_one(data)
return make_response(
jsonify({"success": PyMongoParser.parse_json(added_news.inserted_id)}),
201)
def get_news():
id = request.args.get("id")
title = request.args.get("title")
author_name = request.args.get("author")
author_id = request.args.get("author_id")
news = []
if id:
news_obj_id = ObjectId(id)
news = news_collection.find_one({"_id": news_obj_id})
if author_id:
author_obj_id = ObjectId(author_id)
found_author = authors_collection.find_one({"_id": author_obj_id})
news = [
result for result in news_collection.find(
{"author": {
"name": found_author.get("name")
}})
]
if author_name:
news = [
result for result in news_collection.find(
{"author": {
"name": author_name
}})
]
print(news)
if title:
news = news_collection.find_one({"title": title})
if not news:
news = [result for result in news_collection.find()]
return make_response(jsonify({"success": PyMongoParser.parse_json(news)}),
200)
def add_author():
name = request.json['name']
city = request.json['city']
author_name, errors = AuthorValidator.author_name_validators(name)
if len(errors) > 0:
print(errors)
return make_response(jsonify({'result': errors}), 400)
data = {"name": name, "city": city, "created_at": datetime.now()}
existing_author = authors_collection.find_one({"name": name})
if existing_author:
print("Author {} já existe".format(existing_author))
return make_response(
jsonify({"error": "Author {} already exists".format(name)}), 400)
new_author = authors_collection.insert_one(data)
return make_response(
jsonify({"success": PyMongoParser.parse_json(new_author.inserted_id)}),
201)