def test_fetch_letter_line_items_for_all_service(notify_db_session): org_1, org_2, service_1, service_2, service_3, service_sms_only = set_up_usage_data(datetime(2019, 6, 1)) results = fetch_letter_line_items_for_all_services(datetime(2019, 6, 1), datetime(2019, 9, 30)) assert len(results) == 5 assert results[0] == (org_1.name, org_1.id, service_1.name, service_1.id, Decimal('0.45'), 'second', 6) assert results[1] == (org_1.name, org_1.id, service_1.name, service_1.id, Decimal("0.35"), 'first', 2) assert results[2] == (org_2.name, org_2.id, service_2.name, service_2.id, Decimal("0.65"), 'second', 20) assert results[3] == (org_2.name, org_2.id, service_2.name, service_2.id, Decimal("0.50"), 'first', 2) assert results[4] == (None, None, service_3.name, service_3.id, Decimal("0.55"), 'second', 15)
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_line_items_for_all_service(notify_db_session): fixtures = set_up_usage_data(datetime(2019, 6, 1)) results = fetch_letter_line_items_for_all_services(datetime(2019, 6, 1), datetime(2019, 9, 30)) assert len(results) == 7 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('0.45'), 'second', 6) assert results[1] == (fixtures["org_1"].name, fixtures["org_1"].id, fixtures["service_1_sms_and_letter"].name, fixtures["service_1_sms_and_letter"].id, Decimal("0.35"), 'first', 2) assert results[2] == (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("0.65"), 'second', 20) assert results[3] == (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("0.50"), 'first', 2) assert results[4] == (None, None, fixtures["service_with_letters_without_org"].name, fixtures["service_with_letters_without_org"].id, Decimal("0.35"), 'second', 2) assert results[5] == (None, None, fixtures["service_with_letters_without_org"].name, fixtures["service_with_letters_without_org"].id, Decimal("0.50"), 'first', 1) assert results[6] == (None, None, fixtures["service_with_letters_without_org"].name, fixtures["service_with_letters_without_org"].id, Decimal("1.55"), 'international', 15)
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)