notifications-api/app/dao/stats_template_usage_by_month_dao.py

from app import db
from app.statsd_decorators import statsd
from app.dao.dao_utils import transactional
from app.models import StatsTemplateUsageByMonth, Template


@transactional
@statsd(namespace="dao")
def insert_or_update_stats_for_template(template_id, month, year, count):
    result = db.session.query(
        StatsTemplateUsageByMonth
    ).filter(
        StatsTemplateUsageByMonth.template_id == template_id,
        StatsTemplateUsageByMonth.month == month,
        StatsTemplateUsageByMonth.year == year
    ).update(
        {
            'count': count
        }
    )
    if result == 0:
        monthly_stats = StatsTemplateUsageByMonth(
            template_id=template_id,
            month=month,
            year=year,
            count=count
        )

        db.session.add(monthly_stats)


@statsd(namespace="dao")
def dao_get_template_usage_stats_by_service(service_id, year):
    return db.session.query(
        StatsTemplateUsageByMonth.template_id,
        Template.name,
        StatsTemplateUsageByMonth.month,
        StatsTemplateUsageByMonth.year,
        StatsTemplateUsageByMonth.count
    ).join(
        Template, StatsTemplateUsageByMonth.template_id == Template.id
    ).filter(
        Template.service_id == service_id,
        StatsTemplateUsageByMonth.year == year
    ).all()