return all zeros when NotificationStatistics dont exist

app/dao/notifications_dao.py

@@ -1,4 +1,4 @@
-from sqlalchemy import (desc, func, Integer, and_, asc)
+from sqlalchemy import (desc, func, Integer, and_, or_, asc)
 from sqlalchemy.sql.expression import cast
 
 from datetime import (
@@ -12,6 +12,7 @@ from werkzeug.datastructures import MultiDict
 
 from app import db
 from app.models import (
+    Service,
     Notification,
     Job,
     NotificationStatistics,
@@ -57,6 +58,51 @@ def dao_get_notification_statistics_for_day(day):
     ).all()
 
 
+def dao_get_potential_notification_statistics_for_day(day):
+    all_services = db.session.query(
+        Service.id,
+        NotificationStatistics
+    ).outerjoin(
+        Service.service_notification_stats
+    ).filter(
+        or_(
+            NotificationStatistics.day == day,
+            NotificationStatistics.day == None  # noqa
+        )
+    ).order_by(
+        asc(Service.created_at)
+    )
+
+    notification_statistics = []
+    for service_notification_stats_pair in all_services:
+        if service_notification_stats_pair.NotificationStatistics:
+            notification_statistics.append(
+                service_notification_stats_pair.NotificationStatistics
+            )
+        else:
+            notification_statistics.append(
+                create_notification_statistics_dict(
+                    service_notification_stats_pair,
+                    day
+                )
+            )
+    return notification_statistics
+
+
+def create_notification_statistics_dict(service_id, day):
+    return {
+        'id': None,
+        'emails_requested': 0,
+        'emails_delivered': 0,
+        'emails_failed': 0,
+        'sms_requested': 0,
+        'sms_delivered': 0,
+        'sms_failed': 0,
+        'day': day.isoformat(),
+        'service': service_id
+    }
+
+
 def dao_get_7_day_agg_notification_statistics_for_service(service_id,
                                                           date_from,
                                                           week_count=52):

app/notifications/rest.py

@@ -28,7 +28,8 @@ from app.schemas import (
     sms_template_notification_schema,
     notification_status_schema,
     notifications_filter_schema,
-    notifications_statistics_schema
+    notifications_statistics_schema,
+    day_schema,
 )
 from app.celery.tasks import send_sms, send_email
 
@@ -389,14 +390,12 @@ def send_notification(notification_type):
 
 @notifications.route('/notifications/statistics')
 def get_notification_statistics_for_day():
-    data, errors = notifications_statistics_schema.load(request.args)
+    data, errors = day_schema.load(request.args)
     if errors:
         return jsonify(result='error', message=errors), 400
-    if not data.day:
-        return jsonify(result='error', message='Please provide day as query parameter.'), 400
 
-    statistics = notifications_dao.dao_get_notification_statistics_for_day(
-        day=data.day
+    statistics = notifications_dao.dao_get_potential_notification_statistics_for_day(
+        day=data['day']
     )
 
     data, errors = notifications_statistics_schema.dump(statistics, many=True)

app/schemas.py

@@ -168,7 +168,6 @@ class RequestVerifyCodeSchema(ma.Schema):
 
 class NotificationSchema(ma.Schema):
     personalisation = fields.Dict(required=False)
-    pass
 
 
 class SmsNotificationSchema(NotificationSchema):
@@ -360,6 +359,14 @@ class FromToDateSchema(ma.Schema):
             raise ValidationError("date_from needs to be greater than date_to")
 
 
+class DaySchema(ma.Schema):
+    day = fields.Date(required=True)
+
+    @validates('day')
+    def validate_day(self, value):
+        _validate_not_in_future(value)
+
+
 class WeekAggregateNotificationStatisticsSchema(ma.Schema):
 
     date_from = fields.Date()
@@ -405,3 +412,4 @@ event_schema = EventSchema()
 from_to_date_schema = FromToDateSchema()
 provider_details_schema = ProviderDetailsSchema()
 week_aggregate_notification_statistics_schema = WeekAggregateNotificationStatisticsSchema()
+day_schema = DaySchema()