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:
Richard Chapman
2017-11-15 15:55:00 +00:00
parent 618c7676e6
commit 1e4850b70f
7 changed files with 157 additions and 31 deletions

View File

@@ -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'

View File

@@ -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

View File

@@ -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()

View File

@@ -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)