1、QFileDialog.getOpenFileName()
2、QMessageBox.StandardButton.Yes
# -*- coding: utf-8 -*-
"""
Author: administrator
Date: 2020/5/15 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 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_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.apply_bttn = QtWidgets.QPushButton("Apply")
self.close_bttn = QtWidgets.QPushButton("Close")
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.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.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.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 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()