Files
notifications-api/app/billing/billing_schemas.py
2022-04-26 13:07:48 +01:00

39 lines
1.1 KiB
Python

from datetime import datetime
create_or_update_free_sms_fragment_limit_schema = {
"$schema": "http://json-schema.org/draft-07/schema#",
"description": "POST annual billing schema",
"type": "object",
"title": "Create",
"properties": {
"free_sms_fragment_limit": {"type": "integer", "minimum": 0},
},
"required": ["free_sms_fragment_limit"]
}
def serialize_ft_billing_remove_emails(rows):
return [
{
"month": (datetime.strftime(row.month, "%B")),
"notification_type": row.notification_type,
"billing_units": row.billable_units,
"rate": float(row.rate),
"postage": row.postage,
}
for row in rows
if row.notification_type != 'email'
]
def serialize_ft_billing_yearly_totals(rows):
return [
{
"notification_type": row.notification_type,
"billing_units": row.billable_units,
"rate": float(row.rate),
"letter_total": float(row.billable_units * row.rate) if row.notification_type == 'letter' else 0,
}
for row in rows
]