102 lines
3.5 KiB
Python
102 lines
3.5 KiB
Python
#!/usr/bin/env python
|
|
############################################################################################
|
|
# Hellen-One: A BOM processing script.
|
|
# (c) andreika <prometheus.pcb@gmail.com>
|
|
############################################################################################
|
|
|
|
from collections import OrderedDict
|
|
import csv, sys, re
|
|
|
|
if len(sys.argv) < 2:
|
|
print ("Error! Please specify a BOM file name.")
|
|
sys.exit(1)
|
|
fileName = sys.argv[1]
|
|
|
|
if len(sys.argv) > 2:
|
|
repl_csv = sys.argv[2]
|
|
|
|
print ("Opening BOM file " + fileName + "...")
|
|
|
|
rows = OrderedDict()
|
|
emptyId = 1
|
|
|
|
with open(fileName, 'rb') as f:
|
|
reader = csv.reader(f, delimiter=',')
|
|
print ("Searching for duplicates...")
|
|
for row in reader:
|
|
rowName = row[3]
|
|
# all empty names should be saved separately
|
|
if not rowName:
|
|
rows["_" + str(emptyId)] = row
|
|
emptyId += 1
|
|
continue
|
|
row[1] = row[1].split(", ")
|
|
if rowName in rows:
|
|
oldRow = rows[rowName]
|
|
if oldRow[0] != row[0]:
|
|
print ("* Error! Comment mismatch for the part #" + rowName + ": " + oldRow[0] + " != " + row[0])
|
|
sys.exit(2)
|
|
if oldRow[2] != row[2]:
|
|
print ("* Warning! Footprint mismatch for the part #" + rowName + ": " + oldRow[2] + " != " + row[2])
|
|
#sys.exit(3)
|
|
print ("* Duplicates found for " + rowName + " (" + row[0] + ")! Merging...")
|
|
row[1] = oldRow[1] + row[1]
|
|
rows[rowName] = row
|
|
#for idx,item in enumerate(row):
|
|
# print idx , ": ", item
|
|
|
|
replList = list()
|
|
print ("Reading replacement list from the CSV file " + repl_csv + "...")
|
|
with open(repl_csv, 'rb') as f:
|
|
reader = csv.reader(f, delimiter=',')
|
|
for row in reader:
|
|
# skip empty lines and comments (this is not strictly CSV-compliant, but useful for our purposes)
|
|
if (len(row) < 1 or row[0].startswith("#")):
|
|
continue
|
|
replList.append(row)
|
|
|
|
print ("Processing the board replacements...")
|
|
for r in replList:
|
|
reDesignator = r[0]
|
|
for rowName in rows:
|
|
row = rows[rowName]
|
|
if reDesignator in row[1]:
|
|
print ("* Removing " + reDesignator + " from the old row...")
|
|
row[1].remove(reDesignator)
|
|
if not row[1]:
|
|
print ("* Deleting an empty row...")
|
|
del rows[rowName]
|
|
if len(r) < 4:
|
|
continue
|
|
reComment = r[1]
|
|
reFootprint = r[2]
|
|
rePartNumber = r[3]
|
|
# find the matching row by partnumber (if set)
|
|
if rePartNumber:
|
|
if rePartNumber in rows:
|
|
print ("* Adding " + reDesignator + " to another existing row...")
|
|
rows[rePartNumber][1] += [reDesignator]
|
|
else:
|
|
print ("* Appending a new row for " + reDesignator + "...")
|
|
rows[rePartNumber] = [reComment, [reDesignator], reFootprint, rePartNumber]
|
|
|
|
print ("Saving...")
|
|
with open (fileName, 'wb') as new_f:
|
|
rowIdx = 0
|
|
for rowName in rows:
|
|
#for idx,item in enumerate(rows[rowName]):
|
|
# print idx , ": ", item
|
|
if rowIdx == 0:
|
|
writer = csv.writer(new_f, quoting=csv.QUOTE_NONE, quotechar='"', delimiter=',', lineterminator='\n')
|
|
elif rowIdx == 1:
|
|
writer = csv.writer(new_f, quoting=csv.QUOTE_ALL, quotechar='"', delimiter=',', lineterminator='\n')
|
|
row = rows[rowName]
|
|
# restore empty names
|
|
if rowName[0] == '_':
|
|
row[3] = ""
|
|
if type(row[1]) == list:
|
|
row[1] = ", ".join(row[1])
|
|
writer.writerow(row)
|
|
rowIdx += 1
|
|
print ("Done!")
|