SolidWorks BOM to database parent-child part number conversion

sw-struct.py

"""
Python script to generate SolidWorks BOM lists into a database-usable parent
child relationship.

CSV data of the first column:

    "<parent>.<child>.<subchild>.<subsubchild>"

Usage:

    python sw-struct.py input.csv output.csv

"""

import sys
import csv

with open(sys.argv[1], 'r') as csvfile:
    data = {}
    tree = {}
    for row in csv.reader(csvfile, delimiter=',', quotechar='"'):
        _id = row[0].strip()
        _parent = row[0].split('.')
        _parent = '.'.join(_parent[0:len(_parent)-1]).strip()
        _parent = _parent if _parent else 'ROOT'
        tree.setdefault(_parent, []).append(_id)
        data.setdefault(_parent, row)
        data.setdefault(_id, row)

out = csv.writer(open(sys.argv[2], 'w'), delimiter=',', quoting=csv.QUOTE_ALL)
for _parent in tree:
    for _id in tree[_parent]:
        out.writerow([data[_parent][1]] + data[_id])