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

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