notifications-admin/app/notify_client/billing_api_client.py

from app.notify_client import NotifyAdminAPIClient


class BillingAPIClient(NotifyAdminAPIClient):
    # Fudge assert in the super __init__ so
    # we can set those variables later.
    def __init__(self):
        super().__init__("a" * 73, "b")

    def get_billable_units_ft(self, service_id, year):
        return self.get(
            '/service/{0}/billing/ft-monthly-usage'.format(service_id),
            params=dict(year=year)
        )

    def get_service_usage_ft(self, service_id, year=None):
        return self.get(
            '/service/{0}/billing/ft-yearly-usage-summary'.format(service_id),
            params=dict(year=year)
        )

    def get_free_sms_fragment_limit_for_year(self, service_id, year=None):
        result = self.get(
            '/service/{0}/billing/free-sms-fragment-limit'.format(service_id),
            params=dict(financial_year_start=year)
        )
        return result['free_sms_fragment_limit']

    def get_free_sms_fragment_limit_for_all_years(self, service_id, year=None):
        return self.get(
            '/service/{0}/billing/free-sms-fragment-limit'.format(service_id))

    def create_or_update_free_sms_fragment_limit(self, service_id, free_sms_fragment_limit, year=None):
        # year = None will update current and future year in the API
        data = {
            "financial_year_start": year,
            "free_sms_fragment_limit": free_sms_fragment_limit
        }

        return self.post(
            url='/service/{0}/billing/free-sms-fragment-limit'.format(service_id),
            data=data
        )