154 lines
5.0 KiB
Python
154 lines
5.0 KiB
Python
#
|
|
# Example python script to generate a BOM from a KiCad generic netlist
|
|
#
|
|
# Example: Sorted and Grouped CSV BOM
|
|
#
|
|
|
|
"""
|
|
@package
|
|
Output: CSV (comma-separated)
|
|
Grouped By: Value, Symbol Name, Footprint, DNP
|
|
Sorted By: Ref
|
|
Fields: Item, Qty, Reference(s), Value, LibPart, Footprint, Datasheet, DNP, all additional symbol fields
|
|
|
|
Outputs ungrouped components first, then outputs grouped components.
|
|
|
|
Command line:
|
|
python "pathToFile/bom_csv_grouped_by_value.py" "%I" "%O.csv"
|
|
"""
|
|
|
|
from __future__ import print_function
|
|
|
|
# Import the KiCad python helper module and the csv formatter
|
|
import kicad_netlist_reader
|
|
import kicad_utils
|
|
import csv
|
|
import sys
|
|
|
|
# A helper function to filter/convert a string read in netlist
|
|
#currently: do nothing
|
|
def fromNetlistText( aText ):
|
|
return aText
|
|
|
|
def myEqu(self, other):
|
|
"""myEqu is a more advanced equivalence function for components which is
|
|
used by component grouping. Normal operation is to group components based
|
|
on their value and footprint.
|
|
|
|
In this example of a custom equivalency operator we compare the
|
|
value, the part name and the footprint.
|
|
"""
|
|
result = True
|
|
if self.getValue() != other.getValue():
|
|
result = False
|
|
elif self.getPartName() != other.getPartName():
|
|
result = False
|
|
elif self.getFootprint() != other.getFootprint():
|
|
result = False
|
|
elif self.getDNP() != other.getDNP():
|
|
result = False
|
|
|
|
return result
|
|
|
|
# Override the component equivalence operator - it is important to do this
|
|
# before loading the netlist, otherwise all components will have the original
|
|
# equivalency operator.
|
|
kicad_netlist_reader.comp.__eq__ = myEqu
|
|
|
|
if len(sys.argv) != 3:
|
|
print("Usage ", __file__, "<generic_netlist.xml> <output.csv>", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
|
|
# Generate an instance of a generic netlist, and load the netlist tree from
|
|
# the command line option. If the file doesn't exist, execution will stop
|
|
net = kicad_netlist_reader.netlist(sys.argv[1])
|
|
|
|
# Open a file to write to, if the file cannot be opened output to stdout
|
|
# instead
|
|
try:
|
|
f = kicad_utils.open_file_writeUTF8(sys.argv[2], 'w')
|
|
except IOError:
|
|
e = "Can't open output file for writing: " + sys.argv[2]
|
|
print( __file__, ":", e, sys.stderr )
|
|
f = sys.stdout
|
|
|
|
# subset the components to those wanted in the BOM, controlled
|
|
# by <configure> block in kicad_netlist_reader.py
|
|
components = net.getInterestingComponents( excludeBOM=True )
|
|
|
|
compfields = net.gatherComponentFieldUnion(components)
|
|
partfields = net.gatherLibPartFieldUnion()
|
|
|
|
# remove Reference, Value, Datasheet, and Footprint, they will come from 'columns' below
|
|
partfields -= set( ['Reference', 'Value', 'Datasheet', 'Footprint'] )
|
|
|
|
columnset = compfields | partfields # union
|
|
|
|
# prepend an initial 'hard coded' list and put the enchillada into list 'columns'
|
|
hardcoded_columns = ['Item', 'Qty', 'Reference(s)', 'Value', 'LibPart', 'Footprint', 'Datasheet', 'DNP']
|
|
columns = hardcoded_columns + sorted(list(columnset))
|
|
|
|
# Create a new csv writer object to use as the output formatter
|
|
out = csv.writer( f, lineterminator='\n', delimiter=',', quotechar='\"', quoting=csv.QUOTE_ALL )
|
|
|
|
# override csv.writer's writerow() to support encoding conversion (initial encoding is utf8):
|
|
def writerow( acsvwriter, columns ):
|
|
utf8row = []
|
|
for col in columns:
|
|
utf8row.append( fromNetlistText( str(col) ) )
|
|
acsvwriter.writerow( utf8row )
|
|
|
|
# Output a set of rows as a header providing general information
|
|
writerow( out, ['Source:', net.getSource()] )
|
|
writerow( out, ['Date:', net.getDate()] )
|
|
writerow( out, ['Tool:', net.getTool()] )
|
|
writerow( out, ['Generator:', sys.argv[0]] )
|
|
writerow( out, ['Component Count:', len(components)] )
|
|
writerow( out, [] )
|
|
|
|
writerow( out, ['Collated Components:'] )
|
|
writerow( out, [] ) # blank line
|
|
writerow( out, columns ) # reuse same columns
|
|
|
|
row=[]
|
|
|
|
# Get all of the components in groups of matching parts + values
|
|
# (see kicad_netlist_reader.py)
|
|
grouped = net.groupComponents(components)
|
|
|
|
|
|
# Output component information organized by group, aka as collated:
|
|
item = 0
|
|
for group in grouped:
|
|
del row[:]
|
|
refs = ""
|
|
|
|
# Add the reference of every component in the group and keep a reference
|
|
# to the component so that the other data can be filled in once per group
|
|
for component in group:
|
|
if len(refs) > 0:
|
|
refs += ", "
|
|
refs += component.getRef()
|
|
c = component
|
|
|
|
# Fill in the component groups common data
|
|
# columns = ['Item', 'Qty', 'Reference(s)', 'Value', 'LibPart', 'Footprint', 'Datasheet', 'DNP'] + sorted(list(columnset))
|
|
item += 1
|
|
row.append( item )
|
|
row.append( len(group) )
|
|
row.append( refs );
|
|
row.append( c.getValue() )
|
|
row.append( c.getLibName() + ":" + c.getPartName() )
|
|
row.append( net.getGroupFootprint(group) )
|
|
row.append( net.getGroupDatasheet(group) )
|
|
row.append( c.getDNPString() )
|
|
|
|
# add user fields by name
|
|
for field in columns[len(hardcoded_columns):]:
|
|
row.append( net.getGroupField(group, field) );
|
|
|
|
writerow( out, row )
|
|
|
|
f.close()
|