111 lines
3 KiB
Python
111 lines
3 KiB
Python
|
from helper.is_solid import is_object_solid
|
||
|
|
import FreeCAD as App
|
||
|
|
import Spreadsheet
|
||
|
|
|
||
|
|
|
||
|
|
def createSpreadsheet():
|
||
|
|
|
||
|
|
if App.ActiveDocument.getObject("BoM_List") == None:
|
||
|
|
|
||
|
|
sheet = App.activeDocument().addObject('Spreadsheet::Sheet', 'BoM_List')
|
||
|
|
sheet.set('A1', 'п.п.')
|
||
|
|
sheet.set('B1', 'Наименование детали')
|
||
|
|
sheet.set('C1', 'Количество')
|
||
|
|
else:
|
||
|
|
|
||
|
|
sheet = App.ActiveDocument.getObject("BoM_List")
|
||
|
|
App.ActiveDocument.BoM_List.clear('A1:ZZ16384')
|
||
|
|
sheet.set('A1', 'п.п.')
|
||
|
|
sheet.set('B1', 'Наименование детали')
|
||
|
|
sheet.set('C1', 'Количество')
|
||
|
|
|
||
|
|
return (sheet)
|
||
|
|
|
||
|
|
|
||
|
|
class SolidBodiesParcer:
|
||
|
|
_asmThere = []
|
||
|
|
|
||
|
|
def __init__(self) -> None:
|
||
|
|
if (self._asmThere.__len__() == 0):
|
||
|
|
|
||
|
|
self.initParse()
|
||
|
|
pass
|
||
|
|
|
||
|
|
def initParse(self):
|
||
|
|
for el in App.ActiveDocument.RootObjects:
|
||
|
|
if (is_object_solid(el) and hasattr(el, 'Group')):
|
||
|
|
self.getSubPartsLink(el.Group, el.Label)
|
||
|
|
|
||
|
|
def getSubPartsLink(self, group, label):
|
||
|
|
groupLink = {label: []}
|
||
|
|
for el in group:
|
||
|
|
if (is_object_solid(el)):
|
||
|
|
groupLink[label].append(
|
||
|
|
{'label': el.Label, 'isGroup': hasattr(el, 'Group'), 'solid': el})
|
||
|
|
|
||
|
|
for el in groupLink[label]:
|
||
|
|
if ('isGroup' in el):
|
||
|
|
if (el['isGroup'] == False):
|
||
|
|
self._asmThere.append(el['solid'].Label)
|
||
|
|
if (el['isGroup']):
|
||
|
|
self.getSubPartsLink(el['solid'].Group, el['label']),
|
||
|
|
|
||
|
|
return groupLink
|
||
|
|
|
||
|
|
|
||
|
|
def uniquePartsSort(labelParts):
|
||
|
|
|
||
|
|
uniquePartsLabels = {}
|
||
|
|
|
||
|
|
for el in labelParts:
|
||
|
|
for k in labelParts:
|
||
|
|
if (App.ActiveDocument.getObjectsByLabel(str(el))[0].Shape.isPartner(App.ActiveDocument.getObjectsByLabel(str(k))[0].Shape)):
|
||
|
|
|
||
|
|
if uniquePartsLabels.get(el) == None:
|
||
|
|
uniquePartsLabels[el] = k
|
||
|
|
|
||
|
|
sortedParts = {}
|
||
|
|
|
||
|
|
for k, v in uniquePartsLabels.items():
|
||
|
|
|
||
|
|
if sortedParts.get(v) == None:
|
||
|
|
sortedParts[v] = [k]
|
||
|
|
else:
|
||
|
|
sortedParts[v].append(k)
|
||
|
|
|
||
|
|
return sortedParts
|
||
|
|
|
||
|
|
|
||
|
|
def countForUniques(sortedParts):
|
||
|
|
countedParts = {}
|
||
|
|
for k in sortedParts:
|
||
|
|
countedParts[k] = len(sortedParts[k])
|
||
|
|
return countedParts
|
||
|
|
|
||
|
|
|
||
|
|
def fillInBoMList(sheet, countedParts):
|
||
|
|
|
||
|
|
a = 1
|
||
|
|
|
||
|
|
for label, count in countedParts.items():
|
||
|
|
a += 1
|
||
|
|
b = label
|
||
|
|
c = count
|
||
|
|
sheet.set('A' + str(a), str(a-1))
|
||
|
|
sheet.set('B' + str(a), str(b))
|
||
|
|
sheet.set('C' + str(a), str(c))
|
||
|
|
|
||
|
|
total_count = sum(countedParts.values())
|
||
|
|
sheet.set('B'+str(a+1), 'Итого')
|
||
|
|
|
||
|
|
sheet.set('C' + str(a+1), str(total_count))
|
||
|
|
|
||
|
|
|
||
|
|
def run_BoM_list():
|
||
|
|
createSpreadsheet()
|
||
|
|
sheet = App.ActiveDocument.getObject("BoM_List")
|
||
|
|
labelParts = SolidBodiesParcer()._asmThere
|
||
|
|
sortedParts = uniquePartsSort(labelParts)
|
||
|
|
countedParts = countForUniques(sortedParts)
|
||
|
|
fillInBoMList(sheet, countedParts)
|