mirror of
https://github.com/GSA/notifications-api.git
synced 2026-01-01 05:11:52 -05:00
7f4b828aff
* Added unit tests for the get_complaint_count endpoint * Updated the schema to use 'date' format instead of 'datetime' * Updated the complaint endpoint to convert start_date and end_date to be dates instead of strings
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
from datetime import datetime
|
|
|
|
from flask import Blueprint, jsonify, request
|
|
from sqlalchemy import desc
|
|
|
|
from app.complaint.complaint_schema import complaint_count_request
|
|
from app.dao.complaint_dao import fetch_count_of_complaints
|
|
from app.errors import register_errors
|
|
from app.models import Complaint
|
|
from app.schema_validation import validate
|
|
|
|
complaint_blueprint = Blueprint('complaint', __name__, url_prefix='/complaint')
|
|
|
|
register_errors(complaint_blueprint)
|
|
|
|
|
|
@complaint_blueprint.route('', methods=['GET'])
|
|
def get_all_complaints():
|
|
complaints = Complaint.query.order_by(desc(Complaint.created_at)).all()
|
|
|
|
return jsonify([x.serialize() for x in complaints]), 200
|
|
|
|
|
|
@complaint_blueprint.route('/count-by-date-range', methods=['GET'])
|
|
def get_complaint_count():
|
|
if request.args:
|
|
validate(request.args, complaint_count_request)
|
|
|
|
# If start and end date are not set, we are expecting today's stats.
|
|
today = str(datetime.utcnow().date())
|
|
|
|
start_date = datetime.strptime(request.args.get('start_date', today), '%Y-%m-%d').date()
|
|
end_date = datetime.strptime(request.args.get('end_date', today), '%Y-%m-%d').date()
|
|
count_of_complaints = fetch_count_of_complaints(start_date=start_date, end_date=end_date)
|
|
|
|
return jsonify(count_of_complaints), 200
|