build_purchases rev1

build_purchases.py

@@ -7,11 +7,7 @@ import build_canonical_layer
 import build_observed_products
 import validate_cross_retailer_flow
 from enrich_giant import format_decimal, to_decimal
-<<<<<<< HEAD
 from layer_helpers import read_csv_rows, stable_id, write_csv_rows
-=======
-from layer_helpers import read_csv_rows, write_csv_rows
->>>>>>> be1bf63 (Build pivot-ready purchase log)
 
 
 PURCHASE_FIELDS = [
@@ -22,11 +18,8 @@ PURCHASE_FIELDS = [
     "observed_item_key",
     "observed_product_id",
     "canonical_product_id",
-<<<<<<< HEAD
     "review_status",
     "resolution_action",
-=======
->>>>>>> be1bf63 (Build pivot-ready purchase log)
     "raw_item_name",
     "normalized_item_name",
     "retailer_item_id",
@@ -69,7 +62,6 @@ EXAMPLE_FIELDS = [
     "notes",
 ]
 
-<<<<<<< HEAD
 CATALOG_FIELDS = [
     "canonical_product_id",
     "canonical_name",
@@ -95,8 +87,6 @@ RESOLUTION_FIELDS = [
     "reviewed_at",
 ]
 
-=======
->>>>>>> be1bf63 (Build pivot-ready purchase log)
 
 def decimal_or_zero(value):
     return to_decimal(value) or Decimal("0")
@@ -175,7 +165,6 @@ def order_lookup(rows, retailer):
     }
 
 
-<<<<<<< HEAD
 def read_optional_csv_rows(path):
     path = Path(path)
     if not path.exists():
@@ -220,9 +209,6 @@ def catalog_row_from_canonical(row):
 
 
 def build_link_state(enriched_rows):
-=======
-def build_link_lookup(enriched_rows):
->>>>>>> be1bf63 (Build pivot-ready purchase log)
     observed_rows = build_observed_products.build_observed_products(enriched_rows)
     canonical_rows, link_rows = build_canonical_layer.build_canonical_layer(observed_rows)
     giant_row, costco_row = validate_cross_retailer_flow.find_proof_pair(observed_rows)
@@ -239,7 +225,6 @@ def build_link_lookup(enriched_rows):
     canonical_id_by_observed = {
         row["observed_product_id"]: row["canonical_product_id"] for row in link_rows
     }
-<<<<<<< HEAD
     return observed_rows, canonical_rows, link_rows, observed_id_by_key, canonical_id_by_observed
 
 
@@ -268,14 +253,6 @@ def build_purchase_rows(
             canonical_id_by_observed[observed_product_id] = resolution["canonical_product_id"]
         elif action == "exclude":
             canonical_id_by_observed[observed_product_id] = ""
-=======
-    return observed_id_by_key, canonical_id_by_observed
-
-
-def build_purchase_rows(giant_enriched_rows, costco_enriched_rows, giant_orders, costco_orders):
-    all_enriched_rows = giant_enriched_rows + costco_enriched_rows
-    observed_id_by_key, canonical_id_by_observed = build_link_lookup(all_enriched_rows)
->>>>>>> be1bf63 (Build pivot-ready purchase log)
     orders_by_id = {}
     orders_by_id.update(order_lookup(giant_orders, "giant"))
     orders_by_id.update(order_lookup(costco_orders, "costco"))
@@ -289,10 +266,7 @@ def build_purchase_rows(giant_enriched_rows, costco_enriched_rows, giant_orders,
         observed_product_id = observed_id_by_key.get(observed_key, "")
         order_row = orders_by_id.get((row["retailer"], row["order_id"]), {})
         metrics = derive_metrics(row)
-<<<<<<< HEAD
         resolution = resolution_lookup.get(observed_product_id, {})
-=======
->>>>>>> be1bf63 (Build pivot-ready purchase log)
         purchase_rows.append(
             {
                 "purchase_date": row["order_date"],
@@ -302,11 +276,8 @@ def build_purchase_rows(giant_enriched_rows, costco_enriched_rows, giant_orders,
                 "observed_item_key": row["observed_item_key"],
                 "observed_product_id": observed_product_id,
                 "canonical_product_id": canonical_id_by_observed.get(observed_product_id, ""),
-<<<<<<< HEAD
                 "review_status": resolution.get("status", ""),
                 "resolution_action": resolution.get("resolution_action", ""),
-=======
->>>>>>> be1bf63 (Build pivot-ready purchase log)
                 "raw_item_name": row["item_name"],
                 "normalized_item_name": row["item_name_norm"],
                 "retailer_item_id": row["retailer_item_id"],
@@ -330,7 +301,6 @@ def build_purchase_rows(giant_enriched_rows, costco_enriched_rows, giant_orders,
                 **metrics,
             }
         )
-<<<<<<< HEAD
     return purchase_rows, observed_rows, canonical_rows, link_rows
 
 
@@ -358,9 +328,6 @@ def apply_manual_resolutions_to_links(link_rows, resolution_rows):
                 "link_notes": resolution.get("resolution_notes", ""),
             }
     return sorted(link_by_observed.values(), key=lambda row: row["observed_product_id"])
-=======
-    return purchase_rows
->>>>>>> be1bf63 (Build pivot-ready purchase log)
 
 
 def build_comparison_examples(purchase_rows):
@@ -399,12 +366,9 @@ def build_comparison_examples(purchase_rows):
 @click.option("--costco-items-enriched-csv", default="costco_output/items_enriched.csv", show_default=True)
 @click.option("--giant-orders-csv", default="giant_output/orders.csv", show_default=True)
 @click.option("--costco-orders-csv", default="costco_output/orders.csv", show_default=True)
-<<<<<<< HEAD
 @click.option("--resolutions-csv", default="combined_output/review_resolutions.csv", show_default=True)
 @click.option("--catalog-csv", default="combined_output/canonical_catalog.csv", show_default=True)
 @click.option("--links-csv", default="combined_output/product_links.csv", show_default=True)
-=======
->>>>>>> be1bf63 (Build pivot-ready purchase log)
 @click.option("--output-csv", default="combined_output/purchases.csv", show_default=True)
 @click.option("--examples-csv", default="combined_output/comparison_examples.csv", show_default=True)
 def main(
@@ -412,7 +376,6 @@ def main(
     costco_items_enriched_csv,
     giant_orders_csv,
     costco_orders_csv,
-<<<<<<< HEAD
     resolutions_csv,
     catalog_csv,
     links_csv,
@@ -421,17 +384,10 @@ def main(
 ):
     resolution_rows = read_optional_csv_rows(resolutions_csv)
     purchase_rows, _observed_rows, canonical_rows, link_rows = build_purchase_rows(
-=======
-    output_csv,
-    examples_csv,
-):
-    purchase_rows = build_purchase_rows(
->>>>>>> be1bf63 (Build pivot-ready purchase log)
         read_csv_rows(giant_items_enriched_csv),
         read_csv_rows(costco_items_enriched_csv),
         read_csv_rows(giant_orders_csv),
         read_csv_rows(costco_orders_csv),
-<<<<<<< HEAD
         resolution_rows,
     )
     existing_catalog_rows = read_optional_csv_rows(catalog_csv)
@@ -448,14 +404,6 @@ def main(
     click.echo(
         f"wrote {len(purchase_rows)} purchase rows to {output_csv}, "
         f"{len(merged_catalog_rows)} catalog rows to {catalog_csv}, "
-=======
-    )
-    example_rows = build_comparison_examples(purchase_rows)
-    write_csv_rows(output_csv, purchase_rows, PURCHASE_FIELDS)
-    write_csv_rows(examples_csv, example_rows, EXAMPLE_FIELDS)
-    click.echo(
-        f"wrote {len(purchase_rows)} purchase rows to {output_csv} "
->>>>>>> be1bf63 (Build pivot-ready purchase log)
         f"and {len(example_rows)} comparison examples to {examples_csv}"
     )