import adsk.core, traceback
import codecs
def run(context):
ui = None
try:
app: adsk.core.Application = adsk.core.Application.get()
ui = app.userInterface
fileDialog = ui.createFileDialog()
fileDialog.isMultiSelectEnabled = False
fileDialog.title = "Specify result filename"
fileDialog.filter = 'XML files (*.xml)'
fileDialog.filterIndex = 0
dialogResult = fileDialog.showSave()
if dialogResult == adsk.core.DialogResults.DialogOK:
filename = fileDialog.filename
else:
return
result = '<UserInterface>\n'
result += f'{TabSpace(1)}<Workspaces count="{ui.workspaces.count}">\n'
for wsIndex in range(ui.workspaces.count):
try:
ws: adsk.core.Workspace = ui.workspaces.item(wsIndex)
except:
ws = None
if ws:
result += f'{TabSpace(2)}<Workspace name="{ws.name}" id="{ws.id}">\n'
try:
tabs = ws.toolbarTabs
except:
tabs = None
if tabs:
result += f'{TabSpace(3)}<ToolbarTabs count="{tabs.count}">\n'
for tab in tabs:
result += f'{TabSpace(4)}<ToolbarTab name="{tab.name}" id="{tab.id}">\n'
result += GetPanelsXML(tab.toolbarPanels, 5)
result += f'{TabSpace(4)}</ToolbarTab>\n'
result += f'{TabSpace(3)}</ToolbarTabs>\n'
else:
result += f'{TabSpace(3)}<ToolbarTabs error="Failed to get toolbar tabs.">\n'
result += f'{TabSpace(3)}</ToolbarTabs>\n'
result += f'{TabSpace(2)}</Workspace>\n'
result += f'{TabSpace(1)}</Workspaces>\n'
result += f'{TabSpace(1)}<Toolbars count="{ui.toolbars.count}">\n'
toolbar: adsk.core.Toolbar
for toolbar in ui.toolbars:
result += f'{TabSpace(2)}<Toolbar id="{toolbar.id}">\n'
result += f'{TabSpace(3)}<ToolbarControls count="{toolbar.controls.count}">\n'
result += GetControls(toolbar.controls, 1, False)
result += f'{TabSpace(3)}</ToolbarControls>\n'
result += f'{TabSpace(2)}</Toolbar>\n'
result += f'{TabSpace(1)}</Toolbars>\n'
result += '</UserInterface>'
f = open(filename, 'w', -1, 'utf-8-sig')
f.write(result)
f.close()
ui.messageBox(f'Finished writing to:\n{filename}')
except:
if ui:
ui.messageBox('Failed:\n{}'.format(traceback.format_exc()))
# Builds XML data for all of the panel information from the ToolbarPanels collection passed in.
def GetPanelsXML(panels: adsk.core.ToolbarPanels, tabs: int) -> str:
result = f'{TabSpace(tabs)}<ToolbarPanels count="{panels.count}">\n'
for panelIndex in range(panels.count):
try:
panel: adsk.core.ToolbarPanel = panels.item(panelIndex)
except:
panel = None
if panel:
result += f'{TabSpace(tabs + 1)}<ToolbarPanel name="{panel.name}" id="{panel.id}">\n'
result += f'{TabSpace(tabs + 2)}<ToolbarControls count="{panel.controls.count}">\n'
result += GetControls(panel.controls, tabs, True)
result += f'{TabSpace(tabs + 2)}</ToolbarControls>\n'
result += f'{TabSpace(tabs + 1)}</ToolbarPanel>\n'
result += f'{TabSpace(tabs + 1)}</ToolbarPanels>\n'
return result
# Builds XML data for all of the controls in the ToolbarControls collection passed in.
def GetControls(controls: adsk.core.ToolbarControls, tabs: int, isPanel: bool) -> str:
result = ''
for control in controls:
if control.objectType == adsk.core.DropDownControl.classType():
dropControl: adsk.core.DropDownControl = control
if isPanel:
try:
dropName = dropControl.name
except:
dropName = "**** Error getting name."
result += f'{TabSpace(tabs + 3)}<DropDownControl name="{dropName}" id="{dropControl.id}" count="{dropControl.controls.count}">\n'
else:
result += f'{TabSpace(tabs + 3)}<DropDownControl id="{dropControl.id}" count="{dropControl.controls.count}">\n'
result += GetControls(dropControl.controls, tabs + 1, isPanel)
result += f'{TabSpace(tabs + 3)}</DropDownControl>\n'
elif control.objectType == adsk.core.SplitButtonControl.classType():
splitControl: adsk.core.SplitButtonControl = control
result += f'{TabSpace(tabs + 3)}<SplitButtonControl>\n'
try:
defaultCmdDef = splitControl.defaultCommandDefinition
except:
defaultCmdDef = None
if defaultCmdDef:
result += f'{TabSpace(tabs + 4)}<defaultCommandDefinition name="{defaultCmdDef.name}" id="{defaultCmdDef.id}"/>\n'
additionalDefs = splitControl.additionalDefinitions
result += f'{TabSpace(tabs + 4)}<additionalDefinitions count="{len(additionalDefs)}">\n'
for additionalDef in additionalDefs:
result += f'{TabSpace(tabs + 5)}<{ObjectName(additionalDef)} name="{additionalDef.name}" id="{additionalDef.id}"/>\n'
result += f'{TabSpace(tabs + 4)}</additionalDefinitions>\n'
else:
result += f'{TabSpace(tabs + 4)}<defaultCommandDefinition error="**** Failed to get CommandDefinition"/>\n'
result += f'{TabSpace(tabs + 3)}</SplitButtonControl>\n'
else:
if control.objectType == adsk.core.SeparatorControl.classType():
result += f'{TabSpace(tabs + 3)}<SeparatorControl id="{control.id}" />\n'
else:
cmdDef: adsk.core.CommandDefinition = None
try:
cmdDef = control.commandDefinition
except:
cmdDef = None
if cmdDef:
try:
commandType = ObjectName(cmdDef.controlDefinition)
except:
commandType = '**** Failed to get associated control.'
isPromotedOK = True
try:
isPromoted = control.isPromoted
except:
isPromotedOK = False
if isPanel and isPromotedOK:
result += f'{TabSpace(tabs + 3)}<{ObjectName(control)} name="{cmdDef.name}" id="{cmdDef.id}" commandType="{commandType}" isPromoted="{isPromoted}" />\n'
else:
result += f'{TabSpace(tabs + 3)}<{ObjectName(control)} name="{cmdDef.name}" id="{cmdDef.id}" commandType="{commandType}" />\n'
else:
result += f'{TabSpace(tabs + 3)}<{ObjectName(control)} error="**** Failed to get CommandDefinition for {control.id}" />\n'
return result
# Return a string of spaces that can be used to prepend to a string to
# represent the specified number of tabs.
def TabSpace(tabs: int) -> str:
spacesPerTab = 4
return ' ' * (spacesPerTab * tabs)
# Splits out the object name from the full object name passed in.
def ObjectName(object: adsk.core.Base) -> str:
parts = object.objectType.split('::')
return parts[len(parts)-1]