1、paintEvent,顺序有影响
def paintEvent(self, event):
painter = QtGui.QPainter(self)
painter.fillRect(0, 0, self.width(), self.height(), self.background_color)
painter.drawPixmap(self.rect(), self.pixmap)
2、及时刷新:self.update()
def set_background_color(self, color):
self.background_color = color
self.update()
# -*- coding: utf-8 -*-
"""
Author: administrator
Date: 2020/5/16 11:52
"""
from PySide2 import QtCore
from PySide2 import QtWidgets
from PySide2 import QtGui
from shiboken2 import wrapInstance
import maya.OpenMayaUI as omui
import maya.OpenMaya as om
import maya.cmds as cmds
import maya.mel as mel
def maya_main_window():
"""
Return the Maya main window widget as a Python object
"""
main_window_ptr = omui.MQtUtil.mainWindow()
return wrapInstance(long(main_window_ptr), QtWidgets.QWidget)
class CustomImageWidget(QtWidgets.QWidget):
def __init__(self, width, height, image_path, parent=None):
super(CustomImageWidget, self).__init__(parent)
self.set_size(width, height)
self.set_image(image_path)
self.set_background_color(QtCore.Qt.black)
def set_size(self, width, height):
self.setFixedSize(width, height)
def set_image(self, image_path):
image = QtGui.QImage(image_path)
image = image.scaled(self.width(), self.height(), QtCore.Qt.IgnoreAspectRatio, QtCore.Qt.SmoothTransformation)
self.pixmap = QtGui.QPixmap()
self.pixmap.convertFromImage(image)
def set_background_color(self, color):
self.background_color = color
self.update()
def paintEvent(self, event):
painter = QtGui.QPainter(self)
painter.fillRect(0, 0, self.width(), self.height(), self.background_color)
painter.drawPixmap(self.rect(), self.pixmap)
class OpenImportDialog(QtWidgets.QDialog):
WINDOW_TITLE = "Open/Import/Reference"
FILE_FILTERS = "Maya (*.ma *.mb);;Maya ASCII (*.ma);;Maya Binary (*.mb);;All Files(*.*)"
selected_filter = "Maya (*.ma *.mb)"
def __init__(self, parent=maya_main_window()):
super(OpenImportDialog, self).__init__(parent)
self.setWindowTitle(self.WINDOW_TITLE)
self.setMinimumSize(300, 80)
self.setWindowFlags(self.windowFlags() ^ QtCore.Qt.WindowContextHelpButtonHint)
self.create_widgets()
self.create_title_label()
self.create_layout()
self.create_connections()
def create_widgets(self):
self.filePath_le = QtWidgets.QLineEdit()
self.select_file_path_btn = QtWidgets.QPushButton()
self.select_file_path_btn.setIcon(QtGui.QIcon(":fileOpen.png"))
self.select_file_path_btn.setToolTip("Select File")
self.select_file_path_btn.setFlat(True)
self.open_rb = QtWidgets.QRadioButton("Open")
self.open_rb.setChecked(True)
self.import_rb = QtWidgets.QRadioButton("Import")
self.reference_rb = QtWidgets.QRadioButton("Reference")
self.force_cb = QtWidgets.QCheckBox("Force")
self.green_bttn = QtWidgets.QPushButton("Green")
self.apply_bttn = QtWidgets.QPushButton("Apply")
self.close_bttn = QtWidgets.QPushButton("Close")
def create_title_label(self):
image_path = "D:/Workspace/PyCharm/PySide2 for Maya/image.png"
self.title_label = CustomImageWidget(280, 70, image_path)
self.title_label.set_background_color(QtCore.Qt.red)
def create_layout(self):
file_path_layout = QtWidgets.QHBoxLayout()
file_path_layout.addWidget(self.filePath_le)
file_path_layout.addWidget(self.select_file_path_btn)
radio_bttn_layout = QtWidgets.QHBoxLayout()
radio_bttn_layout.addWidget(self.open_rb)
radio_bttn_layout.addWidget(self.import_rb)
radio_bttn_layout.addWidget(self.reference_rb)
form_layout = QtWidgets.QFormLayout()
form_layout.addRow("File", file_path_layout)
form_layout.addRow("", radio_bttn_layout)
form_layout.addRow("", self.force_cb)
bttn_layout = QtWidgets.QHBoxLayout()
bttn_layout.addWidget(self.green_bttn)
bttn_layout.addStretch()
bttn_layout.addWidget(self.apply_bttn)
bttn_layout.addWidget(self.close_bttn)
main_layout = QtWidgets.QVBoxLayout(self)
main_layout.setContentsMargins(5, 5, 5, 5)
main_layout.addWidget(self.title_label)
main_layout.addLayout(form_layout)
main_layout.addLayout(bttn_layout)
def create_connections(self):
self.select_file_path_btn.clicked.connect(self.show_file_select_dialog)
self.open_rb.toggled.connect(self.update_force_visibility)
self.green_bttn.clicked.connect(self.set_background_green)
self.apply_bttn.clicked.connect(self.load_file)
self.close_bttn.clicked.connect(self.close)
def show_file_select_dialog(self):
file_path, self.selected_filter = QtWidgets.QFileDialog.getOpenFileName(self, "Select File", "",
self.FILE_FILTERS, self.selected_filter)
if file_path:
self.filePath_le.setText(file_path)
def update_force_visibility(self, checked):
self.force_cb.setVisible(checked)
def load_file(self):
file_path = self.filePath_le.text()
if not file_path:
return
file_info = QtCore.QFileInfo(file_path)
if not file_info.exists():
om.MGlobal.displayError("File does not exist: {0}".format(file_path))
return
if self.open_rb.isChecked():
self.open_file(file_path)
elif self.import_rb.isChecked():
self.import_file(file_path)
else:
self.reference_file(file_path)
def open_file(self, path):
force = self.force_cb.isChecked()
if not force and cmds.file(query=True, modified=True):
result = QtWidgets.QMessageBox.question(self, "Modified", "Current scene has unsaved changes. Continue?")
if result == QtWidgets.QMessageBox.StandardButton.Yes:
force = True
else:
return
cmds.file(path, open=True, ignoreVersion=True, force=force)
def import_file(self, path):
cmds.file(path, i=True, ignoreVersion=True)
def reference_file(self, path):
cmds.file(path, reference=True, ignoreVersion=True)
def set_background_green(self):
self.title_label.set_background_color(QtCore.Qt.green)
def showWindow():
global dialog
try:
dialog.close()
dialog.deleteLater()
except:
pass
dialog = OpenImportDialog()
dialog.show()
if __name__ == "__main__":
try:
dialog.close()
dialog.deleteLater()
except:
pass
dialog = OpenImportDialog()
dialog.show()