def edit_post(post_id): if not post_exists(post_id): return error('Post does not exist.', 404) data = request.get_json() post_data = { field: data[field] for field in Post._meta.allowed_fields if data.get(field) is not None } Post.update(post_data).where(Post.id == post_id).execute() return jsonify(Post.get_by_id(post_id).to_dict()), 200
def edit_users_post(user_id, post_id): user = user_exists(user_id, return_user=True) if not user: return error('User does not exist.', 404) if user.posts.where(Post.id == post_id).first() is None: return error('Selected user does not have the post.', 404) data = request.get_json() post_data = {field: data[field] for field in Post._meta.allowed_fields if data.get(field) is not None} Post.update(post_data).where(Post.id == post_id).execute() post = Post.get_by_id(post_id) return jsonify(post.to_dict()), 200