63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
UI Utilities Module for Plugin
|
||
UI工具模块 - 提供UI相关的通用函数
|
||
"""
|
||
|
||
#===================================== IMPORT MODULES =====================================
|
||
from Qt import QtWidgets, QtCore, QtGui
|
||
from Qt.QtCompat import wrapInstance
|
||
from maya import OpenMayaUI as omui
|
||
import sys
|
||
import os
|
||
import config
|
||
#===================================== UI UTILITY FUNCTIONS ===================================
|
||
|
||
TOOL_NAME = config.TOOL_NAME
|
||
|
||
def get_parent_widget(widget_name):
|
||
"""
|
||
根据控件名称查找父容器控件
|
||
|
||
Args:
|
||
widget_name (str): 控件名称
|
||
|
||
Returns:
|
||
QWidget: 找到的父容器控件，如果未找到则返回None
|
||
"""
|
||
# 查找主窗口中的所有控件
|
||
main_window = None
|
||
for widget in QtWidgets.QApplication.topLevelWidgets():
|
||
if widget.objectName() == f"{TOOL_NAME}MainWindow" or widget.objectName().endswith("MainWindow"):
|
||
main_window = widget
|
||
break
|
||
|
||
if not main_window:
|
||
print(f"无法找到主窗口，无法获取父容器: {widget_name}")
|
||
return None
|
||
|
||
# 在主窗口中查找指定名称的控件
|
||
found_widget = main_window.findChild(QtWidgets.QWidget, widget_name)
|
||
if found_widget:
|
||
return found_widget
|
||
|
||
# 如果未找到精确匹配，尝试模糊匹配
|
||
for child in main_window.findChildren(QtWidgets.QWidget):
|
||
if widget_name.lower() in child.objectName().lower():
|
||
return child
|
||
|
||
print(f"无法找到控件: {widget_name}")
|
||
return None
|
||
|
||
def get_maya_main_window():
|
||
"""
|
||
获取Maya主窗口
|
||
|
||
Returns:
|
||
QWidget: Maya主窗口控件
|
||
"""
|
||
main_window_ptr = omui.MQtUtil.mainWindow()
|
||
return wrapInstance(int(main_window_ptr), QtWidgets.QWidget)
|