Add manufacturer to API query

action.yml

@@ -15,6 +15,10 @@ inputs:
       The name of the part number column in the BOM file. Defaults to 'Part
       Number'.
     default: Part Number
+  bom_manufacturer_column:
+    description: >
+      The name of the manufacturer column in the BOM file. Defaults to 'Manufacturer'.
+    default: Manufacturer
   bom_quantity_column:
     description: >
       The name of the quantity column in the BOM file. Defaults to 'Quantity'.
@@ -43,6 +47,8 @@ runs:
     - "${{ inputs.quantities }}"
     - "--bom-part-number-column"
     - ${{ inputs.bom_part_number_column }}
+    - "--bom-manufacturer-column"
+    - ${{ inputs.bom_manufacturer_column }}
     - "--bom-quantity-column"
     - ${{ inputs.bom_quantity_column }}
     - "--search-strategy"

entrypoint.py

@@ -11,10 +11,13 @@ from contextlib import ExitStack
 import csv
 import os
 import sys
+
 import requests
 
 
-def fetch_price_for_part(part_number: str, search_strategy: str) -> dict[int, float]:
+def fetch_price_for_part(
+    part_number: str, manufacturer: str, search_strategy: str
+) -> dict[int, float]:
     """
     Get the price of a component per n units.
 
@@ -53,6 +56,10 @@ def fetch_price_for_part(part_number: str, search_strategy: str) -> dict[int, fl
             "Please set the COFACTR_API_KEY and COFACTR_CLIENT_ID environment variables"
         )
 
+    query = part_number
+    if manufacturer:
+        query += f" {manufacturer}"
+
     search_response = requests.get(
         "https://graph.cofactr.com/products/",
         headers={
@@ -60,7 +67,7 @@ def fetch_price_for_part(part_number: str, search_strategy: str) -> dict[int, fl
             "X-CLIENT-ID": client_id,
         },
         params={
-            "q": part_number,
+            "q": query,
             "search_strategy": search_strategy,
             "schema": "product-offers-v0",
             "external": "true",
@@ -110,6 +117,11 @@ if __name__ == "__main__":
         help="The name of the part number column in the BOM file. Defaults to '%(default)s'.",
         default="Part Number",
     )
+    parser.add_argument(
+        "--bom-manufacturer-column",
+        help="The name of the manufacturer column in the BOM file. Defaults to '%(default)s'.",
+        default="Manufacturer",
+    )
     parser.add_argument(
         "--bom-quantity-column",
         help="The name of the quantity column in the BOM file. Defaults to '%(default)s'.",
@@ -131,6 +143,7 @@ if __name__ == "__main__":
     quantities = [int(quantity) for quantity in args.quantities.split(",")]
 
     part_number_column = args.bom_part_number_column
+    manufacturer_column = args.bom_manufacturer_column
     quantity_column = args.bom_quantity_column
 
     with open(args.bom_file, "r") as bom_file:
@@ -147,7 +160,8 @@ if __name__ == "__main__":
 
     for part in parts:
         part_number = part[part_number_column]
-        prices = fetch_price_for_part(part_number, args.search_strategy)
+        manufacturer = part[manufacturer_column]
+        prices = fetch_price_for_part(part_number, manufacturer, args.search_strategy)
         if prices and len(prices) > 0:
             prices_for_parts[part_number] = prices