mirror of
https://github.com/GSA/notifications-api.git
synced 2026-08-05 14:10:47 -04:00
Add Template name to response and filter by year
The template name should be returned for the response and the user will pick a year, so ths adds those two features to the notifications/templates_usage/monthly endpoint and added some tests to test the functionality.
This commit is contained in:
@@ -307,7 +307,7 @@ class Config(object):
|
||||
|
||||
class Development(Config):
|
||||
NOTIFY_LOG_PATH = 'application.log'
|
||||
SQLALCHEMY_ECHO = False
|
||||
SQLALCHEMY_ECHO = True
|
||||
NOTIFY_EMAIL_DOMAIN = 'notify.tools'
|
||||
CSV_UPLOAD_BUCKET_NAME = 'development-notifications-csv-upload'
|
||||
DVLA_RESPONSE_BUCKET_NAME = 'notify.tools-ftp'
|
||||
|
||||
@@ -547,20 +547,19 @@ def dao_fetch_monthly_historical_stats_by_template():
|
||||
).all()
|
||||
|
||||
|
||||
@transactional
|
||||
@statsd(namespace="dao")
|
||||
def dao_fetch_monthly_historical_usage_by_template_for_service(service_id):
|
||||
def dao_fetch_monthly_historical_usage_by_template_for_service(service_id, year):
|
||||
|
||||
results = dao_get_template_usage_stats_by_service(service_id)
|
||||
results = dao_get_template_usage_stats_by_service(service_id, year)
|
||||
|
||||
stats = list()
|
||||
for result in results:
|
||||
stat = StatsTemplateUsageByMonth(
|
||||
template_id=result.template_id,
|
||||
month=result.month,
|
||||
year=result.year,
|
||||
count=result.count
|
||||
)
|
||||
stat = type("", (), {})()
|
||||
stat.template_id = result.template_id
|
||||
stat.name = str(result.name)
|
||||
stat.month = result.month
|
||||
stat.year = result.year
|
||||
stat.count = result.count
|
||||
stats.append(stat)
|
||||
|
||||
month = get_london_month_from_utc_column(Notification.created_at)
|
||||
@@ -569,13 +568,17 @@ def dao_fetch_monthly_historical_usage_by_template_for_service(service_id):
|
||||
|
||||
today_results = db.session.query(
|
||||
Notification.template_id,
|
||||
Template.name,
|
||||
extract('month', month).label('month'),
|
||||
extract('year', year).label('year'),
|
||||
func.count().label('count')
|
||||
).join(
|
||||
Template, Notification.template_id == Template.id
|
||||
).filter(
|
||||
Notification.created_at >= start_date
|
||||
).group_by(
|
||||
Notification.template_id,
|
||||
Template.name,
|
||||
month,
|
||||
year
|
||||
).order_by(
|
||||
@@ -590,13 +593,12 @@ def dao_fetch_monthly_historical_usage_by_template_for_service(service_id):
|
||||
add_to_stats = False
|
||||
|
||||
if add_to_stats:
|
||||
new_stat = StatsTemplateUsageByMonth(
|
||||
template_id=today_result.template_id,
|
||||
month=today_result.month,
|
||||
year=today_result.year,
|
||||
count=today_result.count
|
||||
)
|
||||
|
||||
new_stat = type("", (), {})()
|
||||
new_stat.template_id = today_result.template_id
|
||||
new_stat.name = today_result.name
|
||||
new_stat.month = today_result.month
|
||||
new_stat.year = today_result.year
|
||||
new_stat.count = today_result.count
|
||||
stats.append(new_stat)
|
||||
|
||||
return stats
|
||||
|
||||
@@ -30,11 +30,16 @@ def insert_or_update_stats_for_template(template_id, month, year, count):
|
||||
|
||||
|
||||
@statsd(namespace="dao")
|
||||
def dao_get_template_usage_stats_by_service(service_id):
|
||||
def dao_get_template_usage_stats_by_service(service_id, year):
|
||||
return db.session.query(
|
||||
StatsTemplateUsageByMonth
|
||||
StatsTemplateUsageByMonth.template_id,
|
||||
Template.name,
|
||||
StatsTemplateUsageByMonth.month,
|
||||
StatsTemplateUsageByMonth.year,
|
||||
StatsTemplateUsageByMonth.count
|
||||
).join(
|
||||
Template, StatsTemplateUsageByMonth.template_id == Template.id
|
||||
).filter(
|
||||
Template.service_id == service_id
|
||||
Template.service_id == service_id,
|
||||
StatsTemplateUsageByMonth.year == year
|
||||
).all()
|
||||
|
||||
@@ -527,8 +527,24 @@ def resume_service(service_id):
|
||||
@service_blueprint.route('/<uuid:service_id>/notifications/templates_usage/monthly', methods=['GET'])
|
||||
def get_monthly_template_usage(service_id):
|
||||
try:
|
||||
data = dao_fetch_monthly_historical_usage_by_template_for_service(service_id)
|
||||
return jsonify(stats=[i.serialize() for i in data]), 200
|
||||
data = dao_fetch_monthly_historical_usage_by_template_for_service(
|
||||
service_id,
|
||||
int(request.args.get('year', 'NaN'))
|
||||
)
|
||||
|
||||
stats = list()
|
||||
for i in data:
|
||||
stats.append(
|
||||
{
|
||||
'template_id': str(i.template_id),
|
||||
'name': i.name,
|
||||
'month': i.month,
|
||||
'year': i.year,
|
||||
'count': i.count
|
||||
}
|
||||
)
|
||||
|
||||
return jsonify(stats=stats), 200
|
||||
except ValueError:
|
||||
raise InvalidRequest('Year must be a number', status_code=400)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user