58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
import argparse
|
|
import csv
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Generates a Digikey BOM from Joost's KiCAD BOM")
|
|
parser.add_argument("input", help="input BOM")
|
|
parser.add_argument("output", help="output BOM")
|
|
parser.add_argument("--m", help="quantity multiplier", default="1")
|
|
|
|
args = parser.parse_args()
|
|
multiplier = 1 if args.m is None else int(args.m)
|
|
|
|
print("multiplier: {}".format(multiplier))
|
|
|
|
digikey_idx = None
|
|
count_idx = None
|
|
components = []
|
|
# TODO skip until it sees "Collated Components"
|
|
with open(args.input, 'r') as f:
|
|
reader = csv.reader(f)
|
|
found_start = False
|
|
for line_number, line in enumerate(reader):
|
|
if not found_start:
|
|
if any(["Collated" in p for p in line]):
|
|
#print("header at {}".format(line_number + 1))
|
|
found_start = True
|
|
continue
|
|
if digikey_idx is None or count_idx is None:
|
|
for i, part in enumerate(line):
|
|
if 'Digikey' == part:
|
|
digikey_idx = i
|
|
elif 'Qty' == part:
|
|
count_idx = i
|
|
continue
|
|
if len(line) < max(digikey_idx, count_idx):
|
|
print("[WARN] line {} is missing fields".format(line_number + 1))
|
|
continue
|
|
part_num = line[digikey_idx]
|
|
count = line[count_idx]
|
|
if len(part_num) == 0:
|
|
print("[WARN] line {} is missing part number".format(line_number + 1))
|
|
print(parts)
|
|
continue
|
|
if count == 0:
|
|
print("[WARN] line {} has zero quantity".format(line_number + 1))
|
|
if part_num == 'NoPart':
|
|
continue
|
|
components.append((part_num, count))
|
|
|
|
with open(args.output, 'w') as f:
|
|
# header is necessary if you don't want to miss the first part
|
|
f.write("Digikey,Quantity\n")
|
|
for c in components:
|
|
f.write("\"{}\",{}\n".format(c[0], multiplier*int(c[1])))
|
|
|
|
if __name__ == "__main__":
|
|
main()
|