Add a script to interpose powerboards on itkpd.

This script is located in the database directory and independent of other scripts in the directory. To run it, use linux terminal and type python pbv3_add_interposer.py -p [PWB serial number(s)] or pbv3_add_interposer.py -c [panel serial number(s)]. The next step could be integrate it with the test.html so that we can use GUI to access this feature.

database/pbv3_add_interposer.py

+import os
+import itkdb
+import csv
+import argparse
+import sys
+
+from database import pwbdbtools
+
+'''
+Run this script to manually add an interposer component to existing powerboards,
+given their parent panel ID.
+Assume that the user has configurated their itkpd account.
+'''
+
+# ------------------------------------------
+# Function: read_PWB_from_csv
+# Description: temp function for reading CSV files
+# Arguments: path to the csv file
+# Returns: a list of PWB ID
+# ------------------------------------------
+def read_PWB_from_csv(csv_file_path):
+    loaded_PWB = []
+
+    with open(csv_file_path, mode='r', newline='') as file:
+        reader = csv.reader(file)
+
+        # Skip the first row (header)
+        next(reader)
+
+        for row in reader:
+            # Extract the fourth to thirteenth columns (index 3 to 12)
+            selected_columns = row[3:13]
+            # Filter out blank entries
+            filtered_columns = [f'20USBP0{value}' for value in selected_columns if value.strip()]
+            # Only append rows that have non-blank entries
+            if filtered_columns:
+                loaded_PWB.extend(filtered_columns)
+
+    return loaded_PWB
+
+
+# ------------------------------------------
+# Function: read_panel
+# Description: use panel ID to search for PWB
+# Arguments: a list of panel ID
+# Returns: a list of child PWB ID
+# ------------------------------------------
+def read_panel(client, panel_ls):
+    PWB_ls = []
+
+    for panel_ID in panel_ls:
+        panel = client.get("getComponent", json={"component": panel_ID})
+
+        child_PWB = [
+            child["component"]["id"]
+            for child in panel.get("children", [])
+            if (child["component"] and child['componentType']['code']=="PWB")
+        ]
+
+        PWB_ls.extend(child_PWB)
+        print(f"Panel {panel_ID} has been processed and {child_PWB} are added to the list.")
+
+    return PWB_ls
+
+# ------------------------------------------
+# Function: load_interposer
+# Description: create and load an interpsoer for each of the PWB
+# Arguments: a list of PWB ID
+# Returns: NA
+# ------------------------------------------
+def load_interposer(client, PWB_ls, check_interposer=True):
+
+    component_template = client.get('generateComponentTypeDtoSample', json={'project': 'S', 'code': 'PWB_INTERPOSER'})
+    for PWB_ID in PWB_ls:
+
+        PWB = client.get("getComponent", json={"component": PWB_ID})
+
+        if check_interposer:
+            if any((PWBchild['componentType']['code'] == 'PWB_INTERPOSER' and PWBchild['component']) for PWBchild in PWB.get("children", [])):
+             print(f"PWB {PWB['serialNumber']} has already been loaded with an interposer.")
+             continue
+
+        if PWB.get('inTransit', False) == True:
+            print(f"PWB {PWB['serialNumber']} is in transit and cannot be loaded with an interposer.")
+            continue
+
+        try:
+            new_interposer = client.post(
+                "registerComponent",
+                json={
+                    **component_template,
+                    'subproject': 'SB',
+                    'institution': 'LBNL_STRIP_POWERBOARDS',
+                }
+            )
+
+            # Hard coded parameters.
+            output = client.post(
+                "assembleComponent",
+                json={
+                    "parent": PWB['serialNumber'],
+                    "child": new_interposer['component']['code'],
+                    "properties": {
+                        "GLASS_BEADS_USED": True,
+                        'INTERPOSER_MATERIAL': 'Kapton',
+                        'INTERPOSER_THICKNESS': 50,
+                        'INTERPOSER_ADHESIVE': 'SE4445',
+                        'ADHESIVE_THICKNESS': 110,
+                        'INTERPOSER_SURFACE_PREP': 'IPA',
+                    },
+                }
+            )
+            print(f"Interposer {new_interposer['component']['code']} has been created and mounted on PWB {output['serialNumber']}.")
+
+        except Exception as e:
+            print(f"An error occurred while processing PWB {PWB['serialNumber']}: {e}")
+
+# ------------------------------------------
+# Function: load_interposer_for_panel
+# Description: use panel number to add interposer for all pwb's on this panel 
+# Arguments: panel id
+# Returns: NA
+# ------------------------------------------
+def load_interposer_for_panel(client, panel_id, check_interposer_in=True):
+    pwb_ls_in = read_panel(client, [panel_id])
+    load_interposer(client, pwb_ls_in, check_interposer=check_interposer_in)
+
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Process either PWB number or Panel number, but not neither.")
+
+    # Take in PWB and Panel numbers
+    parser.add_argument('-p', '--pwbSN', nargs='+', help="PWB serial number(s)")
+    parser.add_argument('-c', '--panel', nargs='+', help="Panel serial number(s)")
+
+    args = parser.parse_args()
+
+    # Check if either PWB number or Panel number is provided (but not both or neither)
+    if not args.pwbSN and not args.panel:
+        print("Error: You must provide either a PWB number or a Panel number.")
+        parser.print_help()
+        sys.exit(1)
+    elif args.pwbSN and args.panel:
+        print("Error: You cannot provide both a PWB number and a Panel number.")
+        parser.print_help()
+        sys.exit(1)
+
+    c = pwbdbtools.get_db_client()
+    
+    pwb_ls = []
+
+    if args.pwbSN:
+        print(f"PWB number(s) provided: {args.pwbSN}")
+        pwb_ls = args.pwbSN
+
+    if args.panel:
+        print(f"Panel number(s) provided: {args.panel}")
+        pwb_ls = read_panel(c, args.panel)
+
+    load_interposer(c, pwb_ls)