Blame view

BOm/bom_csv_grouped_by_value.py 5.03 KB
f8ffc2c5   mahmoudrabia   first commit
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
  #
  # 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()