PySideの小ネタ Mixin編+α

サンプルコード ソースを展開 ソースを折りたたむ

from PySide2 import QtWidgets, QtCore, QtGui
from maya.app.general import mayaMixin
 
 
class LogWidget(mayaMixin.MayaQWidgetBaseMixin, QtWidgets.QWidget):
    def __init__(self, dlg_name='', parent=None):
        super(LogWidget, self).__init__(parent)
 
        self.setWindowTitle(dlg_name or 'Log')
        self.setAttribute(QtCore.Qt.WA_DeleteOnClose)
 
        self.__log = QtWidgets.QTextEdit(self)
        self.__log.setLineWrapMode(QtWidgets.QTextEdit.NoWrap)
        # self.__log.setReadOnly(True)
 
        font = QtGui.QFont(u'ＭＳ ゴシック')
        font.setPointSize(10)
        self.__log.setFont(font)
 
        layout = QtWidgets.QVBoxLayout(self)
        layout.addWidget(self.__log)
 
        self.setLayout(layout)
        self.resize(300, 200)
 
    @property
    def log(self):
        return self.__log
 
 
dlg = LogWidget()
dlg.log.setText('output log.')
dlg.show()

サンプルコード ソースを展開 ソースを折りたたむ

import os
import tempfile
 
 
class ShareSettingMixin(object):
    """QSettingsを使用したパラメータ保存
    """
    _SETTING_FILE_NAME = 'hoge'
 
    def __init__(self, *args, **kwargs):
        super(ShareSettingMixin, self).__init__(*args, **kwargs)
 
    @property
    def setting(self):
        setting_file = os.path.join(tempfile.gettempdir(), self._SETTING_FILE_NAME + '.ini')
        return QtCore.QSettings(setting_file, QtCore.QSettings.IniFormat)
 
 
class ShareSettingWidget(ShareSettingMixin, LogWidget):
    _SETTING_FILE_NAME = '__test_setting'
 
    def __init__(self, parent=None):
        super(ShareSettingWidget, self).__init__(parent)
        self.log.setText(self.setting.value('text'))
 
    def closeEvent(self, evt):
        super(ShareSettingWidget, self).closeEvent(evt)
        self.setting.setValue('text', self.log.toPlainText())
 
 
dlg = ShareSettingWidget()
dlg.show()

サンプルコード ソースを展開 ソースを折りたたむ

class RestoreGeometryMixin(ShareSettingMixin):
    """ShareSettingMixinを使用したツールの位置・サイズの保存・復元
    """
    _SETTING_FILE_NAME = 'file_name'
 
    def __init__(self, parent=None):
        super(RestoreGeometryMixin, self).__init__(parent)
 
    def save_geo(self):
        self.setting.setValue('win_pos', self.pos())
        self.setting.setValue('win_size', self.size())
 
    def load_geo(self):
        p = self.setting.value('win_pos')
        s = self.setting.value('win_size')
        if p:
            self.move(p)
        if s:
            self.resize(s)
 
 
class RestoreGeometryWidget(RestoreGeometryMixin, LogWidget):
    _SETTING_FILE_NAME = '__test_setting'
 
    def __init__(self, parent=None):
        super(RestoreGeometryWidget, self).__init__(parent)
        self.load_geo()
 
    def closeEvent(self, evt):
        super(RestoreGeometryWidget, self).closeEvent(evt)
        self.save_geo()
 
 
dlg = RestoreGeometryWidget()
dlg.show()

サンプルコード ソースを展開 ソースを折りたたむ

class RunOnlyMixin(object):
    """ツールが2つ以上起動しないように既に起動しているツールを閉じる
    """
    #: :type: dict
    __RUNNING = {}
 
    def __new__(cls, *args, **kwargs):
        print('run_only: ', cls)
        if cls in RunOnlyMixin.__RUNNING.keys():
            w = RunOnlyMixin.__RUNNING.pop(cls)
            try:
                w.close()
            except RuntimeError as e:
                pass
 
        ins = super(RunOnlyMixin, cls).__new__(cls, args, kwargs)
        RunOnlyMixin.__RUNNING[cls] = ins
 
        return ins
 
 
class RunOnlyWidget(RunOnlyMixin, LogWidget):
    pass
 
 
dlg = RunOnlyWidget()
dlg.show()

サンプルコード ソースを展開 ソースを折りたたむ

import uuid
import shiboken2
 
from maya import cmds, mel
import maya.OpenMaya as om
import maya.OpenMayaUI as omui
import pymel.core as pm
 
 
class QMayaWidget(QtWidgets.QWidget):
    """MayaUIをQWidgetに追加
    """
    def __init__(self, maya_ui_path, parent=None):
        super(QMayaWidget, self).__init__(parent)
 
        ptr = omui.MQtUtil.findControl(maya_ui_path)
        self.__core = shiboken2.wrapInstance(long(ptr), QtWidgets.QWidget)
 
        layout = QtWidgets.QVBoxLayout()
        layout.setContentsMargins(0, 0, 0, 0)
 
        self.setLayout(layout)
        self.layout().addWidget(self.__core)
 
 
class CustomShelfMixin(object):
    """ツール毎のShelf作成
    """
    def __init__(self, parent=None):
        super(CustomShelfMixin, self).__init__(parent)
 
        # PySideで作成されたツールのfullNameを取得
        win_path = omui.MQtUtil.fullName(long(shiboken2.getCppPointer(self)[0]))
        # Shelfに登録されるIconサイズの調整用にMayaのUIScalingを取得
        scaling = cmds.mayaDpiSetting(rsv=True, q=True)
 
        # shelfLayoutの名前はUniqueでなければならない
        layout_name = str(uuid.uuid4())
        self.__tab = pm.shelfTabLayout('ShortcutShelf', parent=win_path, height=65)
        self.__layout = pm.shelfLayout(layout_name, cellWidthHeight=[34 / scaling, 34 / scaling])
        pm.shelfTabLayout(self.__tab, e=True, tabLabel=[self.__layout, 'Shortcut'])
 
        # 終了時のイベント設定
        self.__evt = om.MEventMessage.addEventCallback('quitApplication', self.__finalize)
 
        # QtにMayaUIの追加・設定
        self.__widget = QMayaWidget(self.__tab, self)
        self.__widget.setSizePolicy(QtWidgets.QSizePolicy.Preferred, QtWidgets.QSizePolicy.Fixed)
        self.__widget.mouseDoubleClickEvent = self.__change_proc
        self.__enable_add = False
 
        self.__top_shelf = ''
        self.layout().insertWidget(0, self.__widget)
 
    @property
    def __shelf_path(self):
        """Shelfの情報が保存されているファイルパス(拡張子.mel無し)
        """
        return cmds.internalVar(userTmpDir=True) + self.__shelf_command
 
    @property
    def __shelf_command(self):
        """sourceされたShelf情報MELの実行コマンド
        """
        return 'shelf_' + self.objectName().replace(' ', '')
 
    def init_shelf(self):
        """Shelfの初期設定
        """
        pm.setParent(self.__layout)
 
        if os.path.exists(self.__shelf_path + '.mel'):
            # Shelf構築MELがあったら再構築
            print('load shelf file from: {0}'.format(self.__shelf_path))
            mel.eval('source "{0}";'.format(self.__shelf_path))
            mel.eval('{0}();'.format(self.__shelf_command))
        else:
            # Shelf構築MELが無かったら初期のボタン設定を適用
            print('shelf file not found.')
            self._default_shelf()
 
        pm.setParent('..')
 
    def reset_shelf(self):
        if os.path.exists(self.__shelf_path + '.mel'):
            os.remove(self.__shelf_path + '.mel')
 
        btns = pm.shelfLayout(self.__layout, q=True, ca=True) or []
        if any(btns):
            pm.deleteUI(btns)
        self.init_shelf()
 
    def _default_shelf(self):
        """初期のボタン設定を実装
        """
        pass
 
    def __change_proc(self, event):
        """登録対象のShelfを変更
        """
        self.__enable_add = not self.__enable_add
 
        if self.__enable_add:
            self.__widget.setStyleSheet('background-color:rgb(82, 133, 166, 255);')
            self.__top_shelf = mel.eval('$temp_var = $gShelfTopLevel;')
            mel.eval('$gShelfTopLevel = "{0}";'.format(self.__tab))
        else:
            self.__widget.setStyleSheet('')
            if self.__top_shelf:
                mel.eval('$gShelfTopLevel = "{0}";'.format(self.__top_shelf))
        pass
 
    def __finalize(self, *args, **kwargs):
        """終了時処理
        """
        if self.__evt is not None:
            om.MMessage.removeCallback(self.__evt)
            self.__evt = None
 
        pm.saveShelf(self.__layout, self.__shelf_path)
        if self.__top_shelf:
            mel.eval('$gShelfTopLevel = "{0}";'.format(self.__top_shelf))
 
        self.deleteLater()
 
    def closeEvent(self, *args, **kwargs):
        super(CustomShelfMixin, self).closeEvent(*args, **kwargs)
        self.__finalize()
 
    def dockCloseEventTriggered(self, *args, **kwargs):
        super(CustomShelfMixin, self).dockCloseEventTriggered(*args, **kwargs)
        self.__finalize()
 
 
class CustomShelfWidget(CustomShelfMixin, LogWidget):
    def __init__(self, parent=None):
        super(CustomShelfWidget, self).__init__(parent)
 
        # ObjectNameが必要
        self.setObjectName('CustomShelfDialog')
        # Shelfの初期化
        self.init_shelf()
        # QTextEditダブルクリックでShelfリセット
        self.log.mouseDoubleClickEvent = lambda x: self.reset_shelf()
 
    def _default_shelf(self):
        """初期のボタン設定を実装
 
        %TEMP%\shelf_CustomShelfWidget.melから初期設定にしたい段階でコピー
        """
        mel.eval("""
        shelfButton
            -enableCommandRepeat 1
            -flexibleWidthType 3
            -flexibleWidthValue 32
            -enable 1
            -width 35
            -height 34
            -manage 1
            -visible 1
            -preventOverride 0
            -annotation "Create a polygonal sphere on the grid"
            -enableBackground 0
            -backgroundColor 0 0 0
            -highlightColor 0.321569 0.521569 0.65098
            -align "center"
            -label "Sphere"
            -labelOffset 0
            -rotation 0
            -flipX 0
            -flipY 0
            -useAlpha 1
            -font "plainLabelFont"
            -overlayLabelColor 0.8 0.8 0.8
            -overlayLabelBackColor 0 0 0 0.5
            -image "polySphere.png"
            -image1 "polySphere.png"
            -style "iconOnly"
            -marginWidth 1
            -marginHeight 1
            -command "polySphere -r 1 -sx 20 -sy 20 -ax 0 1 0 -cuv 2 -ch 1; objectMoveCommand;"
            -sourceType "mel"
            -doubleClickCommand "CreatePolygonSphereOptions"
            -commandRepeatable 1
            -flat 1
        ;
        """)
 
 
dlg = CustomShelfWidget()
dlg.show()

サンプルコード ソースを展開 ソースを折りたたむ

import sys
import traceback
 
 
def exception_dlg(func):
    """エラー発生時に内容をダイアログに表示
    """
    def wrapper(*args, **kwargs):
        try:
            func(*args, **kwargs)
        except Exception as e:
            err_items = []
            for s in traceback.format_exception(*sys.exc_info()):
                if not any(s):
                    continue
 
                try:
                    err_items.append(s.decode('unicode-escape').rstrip())
                except Exception as e:
                    err_items.append(s.rstrip())
 
            dlg = LogWidget('Error')
            dlg.resize(900, 500)
            dlg.log.setText('\n'.join(err_items))
            dlg.show()
    return wrapper
 
 
class ExceptionWidget(LogWidget):
    def __init__(self, parent=None):
        super(ExceptionWidget, self).__init__(parent)
 
        # QTextEditダブルクリックで例外発生
        self.log.mouseDoubleClickEvent = self.__on_double_click
 
    @exception_dlg
    def __on_double_click(self, *args, **kwargs):
        raise RuntimeError(u'エラー発生')
 
 
dlg = ExceptionWidget()
dlg.show()