mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-15 01:32:20 -05:00
Move the serialize method to the billing_schema Update variable names Improve tests Fix bug in command
27 lines
904 B
Python
27 lines
904 B
Python
from datetime import datetime
|
|
|
|
create_or_update_free_sms_fragment_limit_schema = {
|
|
"$schema": "http://json-schema.org/draft-04/schema#",
|
|
"description": "POST annual billing schema",
|
|
"type": "object",
|
|
"title": "Create",
|
|
"properties": {
|
|
"free_sms_fragment_limit": {"type": "integer", "minimum": 1},
|
|
},
|
|
"required": ["free_sms_fragment_limit"]
|
|
}
|
|
|
|
|
|
def serialize_ft_billing_remove_emails(data):
|
|
results = []
|
|
billed_notifications = [x for x in data if x.notification_type != 'email']
|
|
for notifications in billed_notifications:
|
|
json_result = {
|
|
"month": (datetime.strftime(notifications.month, "%B")),
|
|
"notification_type": notifications.notification_type,
|
|
"billing_units": int(notifications.billable_units),
|
|
"rate": float(notifications.rate),
|
|
}
|
|
results.append(json_result)
|
|
return results
|