darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-23 17:01:35 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add and test new DAO method that counts the billable units multiplied by rate multiplier for a given service for a given time period.

Browse Source 
Currently this is SMS only.

Used by the dashboard for a headline figure.
This commit is contained in:
Martyn Inglis
2017-05-19 16:42:33 +01:00
parent f00dbfa34f
commit 2a0669636d
2 changed files with 212 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
app/dao/notification_usage_dao.py
View File

@@ -153,3 +153,21 @@ def rate_multiplier():
(NotificationHistory.rate_multiplier == None, literal_column("'1'")), # noqa
(NotificationHistory.rate_multiplier != None, NotificationHistory.rate_multiplier), # noqa
]), Integer())
@statsd(namespace="dao")
def get_total_billable_units_for_sent_sms_notifications_in_date_range(start_date, end_date, service_id):
result = db.session.query(
func.sum(
NotificationHistory.billable_units * func.coalesce(NotificationHistory.rate_multiplier, 1)
).label('billable_units')
).filter(
NotificationHistory.service_id == service_id,
NotificationHistory.notification_type == 'sms',
NotificationHistory.created_at >= start_date,
NotificationHistory.created_at <= end_date,
NotificationHistory.status.in_(NOTIFICATION_STATUS_TYPES_BILLABLE)
)
if result.scalar():
return int(result.scalar())
return 0