mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-11 07:42:20 -05:00
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
from datetime import datetime
|
|
|
|
from flask import Blueprint, jsonify, request
|
|
|
|
from app.complaint.complaint_schema import complaint_count_request
|
|
from app.dao.complaint_dao import fetch_count_of_complaints, fetch_paginated_complaints
|
|
from app.errors import register_errors
|
|
from app.schema_validation import validate
|
|
from app.utils import pagination_links, utc_now
|
|
|
|
complaint_blueprint = Blueprint("complaint", __name__, url_prefix="/complaint")
|
|
|
|
register_errors(complaint_blueprint)
|
|
|
|
|
|
@complaint_blueprint.route("", methods=["GET"])
|
|
def get_all_complaints():
|
|
page = int(request.args.get("page", 1))
|
|
pagination = fetch_paginated_complaints(page=page)
|
|
|
|
return (
|
|
jsonify(
|
|
complaints=[x.serialize() for x in pagination.items],
|
|
links=pagination_links(
|
|
pagination, ".get_all_complaints", **request.args.to_dict()
|
|
),
|
|
),
|
|
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(utc_now().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
|