import maya.cmds as cmds


def SubdivTaggerUI():
    windowSize = (300, 50)
    if (cmds.window("SubdivTaggerUI", exists=True)):
            cmds.deleteUI("SubdivTaggerUI")
    cmds.window("SubdivTaggerUI", widthHeight=(windowSize[0], windowSize[1]), sizeable=False, title='Arnold Subdivisions Tagger' )
    cmds.rowLayout("Row2", numberOfColumns = 3, width = 280, columnWidth3=[140,50,50], columnAlign3=['center','center','center'])
    Subdivisions = cmds.optionMenu( "Subdivisions", label='  Subdivisions' )
    cmds.menuItem( '0', label='0' )
    cmds.menuItem( '1', label='1' )
    cmds.menuItem( '2', label='2' )
    cmds.menuItem( '3', label='3' )
    cmds.menuItem( '4', label='4' )
    cmds.menuItem( '5', label='5' )
    cmds.button("Apply Render Subdivisions to Selection", label = "Set Subdivisions", width = 150, height = 20, command='SubdivisonsTag()' )
    cmds.showWindow('SubdivTaggerUI')
    cmds.window( 'SubdivTaggerUI', edit=True, widthHeight=(windowSize[0], windowSize[1]) )



def SubdivisonsTag():
    SubdivQuery = `cmds.optionMenu("Subdivisions", query = True, value = True)`
    Subdivisions = SubdivQuery[2:-1]
    sel = cmds.ls(sl=True)
    for obj in sel:
        try:
            cmds.setAttr((obj + ".aiSubdivType"), 1)
            cmds.setAttr((obj+ ".aiSubdivIterations"), float(Subdivisions))
        except:
            print ("Skipping " + obj)


SubdivTaggerUI()