class BaseButtonGroup(base.BaseWidget, object): def __init__(self, orientation=Qt.Horizontal, parent=None): self._orientation = 'horizontal' if orientation == Qt.Horizontal else 'vertical' super(BaseButtonGroup, self).__init__(parent=parent) def get_main_layout(self): main_layout = QBoxLayout(QBoxLayout.LeftToRight if self._orientation == 'horizontal' else QBoxLayout.TopToBottom) main_layout.setContentsMargins(0, 0, 0, 0) return main_layout def ui(self): super(BaseButtonGroup, self).ui() self.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Minimum) self._button_group = QButtonGroup() @decorators.abstractmethod def create_button(self, data_dict): """ Must be implemented in custom button groups Creates a new button for this group :param data_dict: dict :return: new button instance """ raise NotImplementedError( 'Function create_button for class "{}" is not implemented!'.format( self.__class__.__name__)) def get_button_group(self): """ Returns button group internal object :return: QButtonGroup """ return self._button_group def clear(self): """ Clears all buttons contained in this group """ for btn in self._button_group.buttons(): self._button_group.removeButton(btn) self.main_layout.removeWidget(btn) btn.setVisible(False) btn.deleteLater() def add_button(self, data_dict, index=None): """ Adds a new button to this group :param data_dict: dict :param index: int or None :return: new added button """ if python.is_string(data_dict): data_dict = {'text': data_dict} elif isinstance(data_dict, QIcon): data_dict = {'icon': data_dict} new_btn = self.create_button(data_dict) new_btn.setProperty('combine', self._orientation) if data_dict.get('text'): new_btn.setProperty('text', data_dict.get('text')) if data_dict.get('icon'): new_btn.setProperty('icon', data_dict.get('icon')) if data_dict.get('data'): new_btn.setProperty('data', data_dict.get('data')) if data_dict.get('checked'): new_btn.setProperty('checked', data_dict.get('checked')) if data_dict.get('shortcut'): new_btn.setProperty('shortcut', data_dict.get('shortcut')) if data_dict.get('tooltip'): new_btn.setProperty('toolTip', data_dict.get('tooltip')) if data_dict.get('clicked'): new_btn.clicked.connect(data_dict.get('clicked')) if data_dict.get('toggled'): new_btn.toggled.connect(data_dict.get('toggled')) if index is None: self._button_group.addButton(new_btn) else: self._button_group.addButton(new_btn, index) if self.main_layout.count() == 0: new_btn.setChecked(True) self.main_layout.insertWidget(self.main_layout.count(), new_btn) return new_btn def set_button_list(self, button_list): """ Empties group and add all buttons given in the list of buttons :param button_list: list(dict) """ self.clear() for index, data_dict in enumerate(button_list): new_btn = self.add_button(data_dict=data_dict, index=index) if index == 0: new_btn.setProperty('position', 'left') elif index == len(button_list) - 1: new_btn.setProperty('position', 'right') else: new_btn.setProperty('position', 'center')
class BreadcrumbWidget(base.BaseWidget, object): """ Widget that display current location withing a hierarchy It allows going back/forward inside a hierarchy """ def __init__(self, separator=None, parent=None): super(BreadcrumbWidget, self).__init__(parent=parent) current_theme = self.theme() separator_color = current_theme.accent_color if current_theme else '#E2AC2C' self._separator = separator or "<span style='color:{}'> ▸ </span>".format( separator_color) self._separators = list() self.setObjectName('BreadcrumbWidget') self.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Minimum) def get_main_layout(self): main_layout = layouts.HorizontalLayout(spacing=0, margins=(0, 0, 0, 0)) main_layout.addStretch() return main_layout def ui(self): super(BreadcrumbWidget, self).ui() self._button_group = QButtonGroup() def set_items(self, data_list): """ Sets the the items of the breadcrumb cleaning old ones :param data_list: :return: """ for btn in self._button_group.buttons(): self._button_group.removeButton(btn) self.main_layout.removeWidget(btn) btn.setVisible(False) btn.deleteLater() for sep in self._separators: self.main_layout.removeWidget(sep) sep.setVisible(False) sep.deleteLater() python.clear_list(self._separators) for index, data_dict in enumerate(data_list): self.add_item(data_dict, index) def add_item(self, data_dict, index=None): """ Adds a new item to the breadcrumb :param data_dict: dict :param index: int """ btn = buttons.BaseToolButton() btn.setText(data_dict.get('text')) if data_dict.get('image'): btn.image(data_dict.get('image')) if data_dict.get('tooltip'): btn.setProperty('toolTip', data_dict.get('tooltip')) if data_dict.get('clicked'): btn.clicked.connect(data_dict.get('clicked')) if data_dict.get('text'): if data_dict.get('svg') or data_dict.get('icon'): btn.text_beside_icon() else: btn.text_only() else: btn.icon_only() if self._button_group.buttons(): separator = label.BaseLabel(self._separator).secondary() self._separators.append(separator) self.main_layout.insertWidget(self.main_layout.count() - 1, separator) self.main_layout.insertWidget(self.main_layout.count() - 1, btn) if index is None: self._button_group.addButton(btn) else: self._button_group.addButton(btn, index) def set_from_path(self, file_path): """ Creates a proper Breadcrumb list for given path and sets the text """ self._widgets = list() file_path = os.path.dirname(file_path) folders = path.get_folders_from_path(file_path) data_list = list() for folder in folders: data_list.append({'text': folder}) self.set_items(data_list) def get_breadcumbs(self): """ Returns current list of breadcumb texts :return: list(str) """ return [btn.text() for btn in self._button_group.buttons()]
class RenamerView(base.BaseWidget, object): def __init__(self, model, controller, parent=None): self._model = model self._controller = controller super(RenamerView, self).__init__(parent=parent) self.refresh() @property def model(self): return self._model @property def controller(self): return self._controller def ui(self): super(RenamerView, self).ui() top_layout = layouts.HorizontalLayout(spacing=2, margins=(2, 2, 2, 2)) top_layout.setAlignment(Qt.AlignLeft) self._buttons_grp = QButtonGroup(self) self._buttons_grp.setExclusive(True) self.main_layout.addLayout(top_layout) self.main_layout.addLayout(dividers.DividerLayout()) self._categories_layout = layouts.HorizontalLayout(spacing=2, margins=(2, 2, 2, 2)) self._categories_layout.setAlignment(Qt.AlignLeft) selection_layout = layouts.HorizontalLayout(spacing=2, margins=(4, 0, 4, 0)) top_layout.addLayout(selection_layout) self._all_radio = buttons.BaseRadioButton('All', parent=self) self._all_radio.setFixedHeight(19) self._all_radio.setAutoExclusive(True) self._selected_radio = buttons.BaseRadioButton('Selected', parent=self) self._selected_radio.setFixedHeight(19) self._selected_radio.setChecked(True) self._selected_radio.setAutoExclusive(True) self._hierarchy_cbx = checkbox.BaseCheckBox('Hierarchy', parent=self) self._hierarchy_cbx.setFixedHeight(19) self._node_types_combo = combobox.BaseComboBox(parent=self) self._auto_rename_shapes_cbx = None self._auto_rename_shapes_cbx = checkbox.BaseCheckBox( 'Auto Rename Shapes', parent=self) self._auto_rename_shapes_cbx.setChecked(True) if not dcc.client().is_maya(): self._auto_rename_shapes_cbx.setVisible(False) selection_layout.addWidget(self._selected_radio) selection_layout.addWidget(self._all_radio) selection_layout.addItem( QSpacerItem(10, 0, QSizePolicy.Fixed, QSizePolicy.Fixed)) selection_layout.addWidget(self._hierarchy_cbx) selection_layout.addItem( QSpacerItem(10, 0, QSizePolicy.Fixed, QSizePolicy.Fixed)) selection_layout.addWidget(self._node_types_combo) if self._auto_rename_shapes_cbx: selection_layout.addItem( QSpacerItem(10, 0, QSizePolicy.Fixed, QSizePolicy.Fixed)) selection_layout.addWidget(self._auto_rename_shapes_cbx) self._splitter = splitter.CollapsibleSplitter(parent=self) self._splitter.setOrientation(Qt.Horizontal) self._splitter.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Expanding) self._splitter.setMinimumHeight(750) self.main_layout.addWidget(self._splitter) self._rename_tab = tabs.BaseTabWidget(parent=self) self._splitter.addWidget(self._rename_tab) self._manual_rename_widget = manualrenamewidget.ManualRenameWidget( model=self._model, controller=self._controller, parent=self) self._auto_rename_widget = autorenamewidget.AutoRenameWidget( model=self._model, controller=self._controller, parent=self) self._rename_tab.addTab(self._manual_rename_widget, 'Manual') self._rename_tab.addTab(self._auto_rename_widget, 'Auto') self._stack = stack.SlidingStackedWidget() # splitter_right_widget = QWidget() # splitter_right_layout = layouts.VerticalLayout(spacing=0, margins=(0, 0, 0, 0)) # splitter_right_layout.addLayout(self._categories_layout) # splitter_right_layout.addWidget(self._stack) # splitter_right_widget.setLayout(splitter_right_layout) # self._splitter.addWidget(splitter_right_widget) # # no_items_widget = QFrame() # no_items_widget.setFrameShape(QFrame.StyledPanel) # no_items_widget.setFrameShadow(QFrame.Sunken) # no_items_widget.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Expanding) # no_items_layout = layouts.VerticalLayout(spacing=0, margins=(0, 0, 0, 0)) # no_items_widget.setLayout(no_items_layout) # no_items_lbl = label.BaseLabel() # no_items_pixmap = tp.ResourcesMgr().pixmap('no_items') # no_items_lbl.setPixmap(no_items_pixmap) # no_items_lbl.setAlignment(Qt.AlignCenter) # no_items_layout.addItem(QSpacerItem(0, 10, QSizePolicy.Preferred, QSizePolicy.Expanding)) # no_items_layout.addWidget(no_items_lbl) # no_items_layout.addItem(QSpacerItem(0, 10, QSizePolicy.Preferred, QSizePolicy.Expanding)) # # self._stack.addWidget(no_items_widget) self._splitter.handle(0).collapse() self._splitter.setSizes([1, 0]) def setup_signals(self): self._stack.animFinished.connect(self._on_stack_anim_finished) self._rename_tab.currentChanged.connect(self._on_tab_changed) self._selected_radio.clicked.connect(self._controller.set_selected) self._all_radio.clicked.connect(self._controller.set_all_selection) self._hierarchy_cbx.toggled.connect( self._controller.hierarchy_check_toggle) self._node_types_combo.currentIndexChanged.connect( self._controller.set_filter_type) self._auto_rename_shapes_cbx.toggled.connect( self._controller.auto_rename_shapes_check_toggle) self._model.hierarchyCheckChanged.connect( self._on_toggle_hierarchy_cbx) self._model.renameShapeChanged.connect( self._on_toggle_auto_rename_shape_cbx) self._model.filterTypeChanged.connect(self._on_filter_type_changed) def refresh(self): """ Syncs view to the current state of its model """ self._hierarchy_cbx.setChecked(self._model.hierarchy_check) self._node_types_combo.clear() for btn in self._buttons_grp.buttons(): self._buttons_grp.removeButton(btn) node_types = self._model.node_types if not node_types: self._node_types_combo.setVisible(False) else: self._node_types_combo.setVisible(True) for node_type in node_types: self._node_types_combo.addItem(str(node_type).split('.')[-1]) categories = self._model.categories or dict() nodes_to_discard = self._model.nodes_to_discard types_to_discard = self._model.types_to_discard for node_type in types_to_discard: nodes_to_discard.extend(dcc.list_nodes(node_type=node_type)) for i, category in enumerate(categories): for category_name, category_data in category.items(): title = category_data.get('title', category) icon = category_data.get('icon', None) types = category_data.get('types', dict()) category_btn = buttons.BaseButton(title) category_btn.setCheckable(True) if icon: category_btn.setIcon(resources.icon(icon)) if i == 0: category_btn.setChecked(True) self._buttons_grp.addButton(category_btn) self._categories_layout.addWidget(category_btn) category_widget = categorywidget.CategoryWidget( types=types, nodes_to_discard=nodes_to_discard) self._stack.addWidget(category_widget) # category_widget.doRefresh.connect(self._on_refresh_category) # category_widget.doPreview.connect(self._set_preview_names) # category_widget.togglePreview.connect(self.update_current_items) # category_widget.doRename.connect(self._on_rename) # category_btn.clicked.connect(partial(self._on_category_selected, i + 1)) self._auto_rename_widget.refresh() self._controller.update_rules() def _on_stack_anim_finished(self, index): """ Internal callback function that is called when stack animation is completed :param index: :return: """ for btn in self._buttons_grp.buttons(): btn.setEnabled(True) category_widget = self._stack.current_widget if not category_widget: return # self._on_refresh_category(category_widget) def _on_tab_changed(self, tab_index): """ Internal callback function that is called when tab widget is changed by the user :param tab_index: int """ if tab_index != 1: return self._controller.update_rules() def _on_toggle_hierarchy_cbx(self, flag): """ Internal callback function called when the user toggles hierarchy check box :param flag: bool """ self._hierarchy_cbx.setChecked(flag) def _on_toggle_auto_rename_shape_cbx(self, flag): """ Internal callback function called when the user toggles auto rename shape check box :param flag: bool """ self._auto_rename_shapes_cbx.setChecked(flag) def _on_filter_type_changed(self, value): """ Internal callback function that is called when user changes the filter type combo box :param value: str """ self._node_types_combo.setCurrentText(value)