From fd227ad04ba739e3cdcda5ee19232b82b6cdebc8 Mon Sep 17 00:00:00 2001 From: Natalia Borovskikh Date: Fri, 12 Dec 2025 12:44:11 +0000 Subject: [PATCH] Sort utilities by customer count (Close #12) --- tariff_fetch/cli.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tariff_fetch/cli.py b/tariff_fetch/cli.py index c62358c..7788e08 100644 --- a/tariff_fetch/cli.py +++ b/tariff_fetch/cli.py @@ -53,7 +53,7 @@ def prompt_utility(state: str) -> Utility: pl.col("customers").filter(pl.col("customer_class") == "residential").sum().alias("customers"), pl.col("entity_type").last().alias("entity_type"), ) - .sort(["entity_type", "utility_name"]) + .sort(["entity_type", "customers", "utility_name"], descending=(False, True, False)) ) rows = list(yearly_sales_df.iter_rows(named=True)) @@ -62,6 +62,7 @@ def prompt_utility(state: str) -> Utility: ENTITY_TYPES_SORTORDER.index(_["entity_type"]) if _["entity_type"] in ENTITY_TYPES_SORTORDER else abs(hash(_["entity_type"])) + 4, + -_["customers"], _["utility_name"], ) )