def test_fetch_letter_costs_for_all_services(notify_db_session): org, org_2, service, service_2, service_3, service_sms_only = set_up_usage_data(datetime(2019, 6, 1)) results = fetch_letter_costs_for_all_services(datetime(2019, 6, 1), datetime(2019, 9, 30)) assert len(results) == 3 assert results[0] == (org.name, org.id, service.name, service.id, Decimal('3.40')) assert results[1] == (org_2.name, org_2.id, service_2.name, service_2.id, Decimal('14.00')) assert results[2] == (None, None, service_3.name, service_3.id, Decimal('8.25'))
def get_usage_for_all_services(): # TODO: Add defeaults or request validation start_date = request.args.get('start_date') end_date = request.args.get('end_date') start_date, end_date = validate_date_range_is_within_a_financial_year(start_date, end_date) sms_costs = fetch_sms_billing_for_all_services(start_date, end_date) letter_costs = fetch_letter_costs_for_all_services(start_date, end_date) letter_breakdown = fetch_letter_line_items_for_all_services(start_date, end_date) lb_by_service = [ (lb.service_id, "{} {} class letters at {}p".format(lb.letters_sent, lb.postage, int(lb.letter_rate * 100))) for lb in letter_breakdown ] combined = {} # TODO: Add email costs? for s in sms_costs: entry = { "organisation_id": str(s.organisation_id) if s.organisation_id else "", "organisation_name": s.organisation_name or "", "service_id": str(s.service_id), "service_name": s.service_name, "sms_cost": float(s.sms_cost), "sms_fragments": s.chargeable_billable_sms, "letter_cost": 0, "letter_breakdown": "" } combined[s.service_id] = entry for l in letter_costs: if l.service_id in combined: combined[l.service_id].update({'letter_cost': float(l.letter_cost)}) else: letter_entry = { "organisation_id": str(l.organisation_id) if l.organisation_id else "", "organisation_name": l.organisation_name or "", "service_id": str(l.service_id), "service_name": l.service_name, "sms_cost": 0, "sms_fragments": 0, "letter_cost": float(l.letter_cost), "letter_breakdown": "" } combined[l.service_id] = letter_entry for service_id, breakdown in lb_by_service: combined[service_id]['letter_breakdown'] += (breakdown + '\n') # sorting first by name == '' means that blank orgs will be sorted last. return jsonify(sorted(combined.values(), key=lambda x: ( x['organisation_name'] == '', x['organisation_name'], x['service_name'] )))
def test_fetch_letter_costs_for_all_services(notify_db_session): fixtures = set_up_usage_data(datetime(2019, 6, 1)) results = fetch_letter_costs_for_all_services(datetime(2019, 6, 1), datetime(2019, 9, 30)) assert len(results) == 3 assert results[0] == (fixtures["org_1"].name, fixtures["org_1"].id, fixtures["service_1_sms_and_letter"].name, fixtures["service_1_sms_and_letter"].id, Decimal('3.40')) assert results[1] == (fixtures["org_for_service_with_letters"].name, fixtures["org_for_service_with_letters"].id, fixtures["service_with_letters"].name, fixtures["service_with_letters"].id, Decimal('14.00')) assert results[2] == (None, None, fixtures["service_with_letters_without_org"].name, fixtures["service_with_letters_without_org"].id, Decimal('24.45'))
def get_usage_for_all_services(): start_date = request.args.get("start_date") end_date = request.args.get("end_date") start_date, end_date = validate_date_range_is_within_a_financial_year( start_date, end_date) sms_costs = fetch_sms_billing_for_all_services(start_date, end_date) letter_costs = fetch_letter_costs_for_all_services(start_date, end_date) letter_breakdown = fetch_letter_line_items_for_all_services( start_date, end_date) lb_by_service = [( lb.service_id, "{} {} class letters at {}p".format(lb.letters_sent, lb.postage, int(lb.letter_rate * 100)), ) for lb in letter_breakdown] combined = {} for s in sms_costs: entry = { "organisation_id": str(s.organisation_id) if s.organisation_id else "", "organisation_name": s.organisation_name or "", "service_id": str(s.service_id), "service_name": s.service_name, "sms_cost": float(s.sms_cost), "sms_fragments": s.chargeable_billable_sms, "letter_cost": 0, "letter_breakdown": "", } combined[s.service_id] = entry for letter_cost in letter_costs: if letter_cost.service_id in combined: combined[letter_cost.service_id].update( {"letter_cost": float(letter_cost.letter_cost)}) else: letter_entry = { "organisation_id": str(letter_cost.organisation_id) if letter_cost.organisation_id else "", "organisation_name": letter_cost.organisation_name or "", "service_id": str(letter_cost.service_id), "service_name": letter_cost.service_name, "sms_cost": 0, "sms_fragments": 0, "letter_cost": float(letter_cost.letter_cost), "letter_breakdown": "", } combined[letter_cost.service_id] = letter_entry for service_id, breakdown in lb_by_service: combined[service_id]["letter_breakdown"] += breakdown + "\n" # sorting first by name == '' means that blank orgs will be sorted last. return jsonify( sorted( combined.values(), key=lambda x: ( x["organisation_name"] == "", x["organisation_name"], x["service_name"], ), ))
def get_data_for_billing_report(): start_date = request.args.get('start_date') end_date = request.args.get('end_date') start_date, end_date = validate_date_range_is_within_a_financial_year(start_date, end_date) sms_costs = fetch_sms_billing_for_all_services(start_date, end_date) letter_costs = fetch_letter_costs_for_all_services(start_date, end_date) letter_breakdown = fetch_letter_line_items_for_all_services(start_date, end_date) lb_by_service = [ (lb.service_id, f"{lb.letters_sent} {postage_description(lb.postage)} letters at {format_letter_rate(lb.letter_rate)}") for lb in letter_breakdown ] combined = {} for s in sms_costs: if float(s.sms_cost) > 0: entry = { "organisation_id": str(s.organisation_id) if s.organisation_id else "", "organisation_name": s.organisation_name or "", "service_id": str(s.service_id), "service_name": s.service_name, "sms_cost": float(s.sms_cost), "sms_fragments": s.chargeable_billable_sms, "letter_cost": 0, "letter_breakdown": "" } combined[s.service_id] = entry for letter_cost in letter_costs: if letter_cost.service_id in combined: combined[letter_cost.service_id].update({'letter_cost': float(letter_cost.letter_cost)}) else: letter_entry = { "organisation_id": str(letter_cost.organisation_id) if letter_cost.organisation_id else "", "organisation_name": letter_cost.organisation_name or "", "service_id": str(letter_cost.service_id), "service_name": letter_cost.service_name, "sms_cost": 0, "sms_fragments": 0, "letter_cost": float(letter_cost.letter_cost), "letter_breakdown": "" } combined[letter_cost.service_id] = letter_entry for service_id, breakdown in lb_by_service: combined[service_id]['letter_breakdown'] += (breakdown + '\n') billing_details = fetch_billing_details_for_all_services() for service in billing_details: if service.service_id in combined: combined[service.service_id].update({ 'purchase_order_number': service.purchase_order_number, 'contact_names': service.billing_contact_names, 'contact_email_addresses': service.billing_contact_email_addresses, 'billing_reference': service.billing_reference }) # sorting first by name == '' means that blank orgs will be sorted last. result = sorted(combined.values(), key=lambda x: ( x['organisation_name'] == '', x['organisation_name'], x['service_name'] )) return jsonify(result)