PySide + Python: Gets the selection level and items for a QTreeWidget

QTreeWidget Selection Level | .py

""" 
- return[0] true if all items selected are the same level in heirarchy tree
- return[1] unique list of all levels containing a item selected
- return[2] select items 
""" 
def get_treewidget_selection_level(treewidget):

    items = treewidget.selectedItems()
    levels = set()

    for item in items:
        level = 0
        index = treewidget.indexFromItem(item, 0)
        
        while index.parent().isValid():
            index = index.parent()
            level += 1
    
        levels.add(level)

    if len(levels) == 1:
        return ( True, list(levels)[0], items )
    else:
        return ( False, list(levels), items )