class MuscleSplineView(base.BaseWidget, object):
def __init__(self, model, controller, parent=None):
self._model = model
self._controller = controller
super(MuscleSplineView, self).__init__(parent=parent)
self.refresh()
def get_main_layout(self):
return layouts.VerticalLayout(spacing=0, margins=(0, 0, 0, 0))
def ui(self):
super(MuscleSplineView, self).ui()
expander = expandables.ExpanderWidget()
muscle_setup_widget = QWidget()
muscle_setup_layout = layouts.VerticalLayout(spacing=2,
margins=(2, 2, 2, 2))
muscle_setup_widget.setLayout(muscle_setup_layout)
muscle_setup_base_layout = layouts.GridLayout(spacing=5,
margins=(2, 2, 2, 2))
name_label = label.BaseLabel('Name:', parent=self)
self._name_line = lineedit.BaseLineEdit(parent=self)
size_label = label.BaseLabel('Size:')
self._size_spn = spinbox.BaseDoubleSpinBox(parent=self)
self._size_spn.setMinimum(1.0)
insertion_ctrls_lbl = label.BaseLabel('N. Controls:', parent=self)
insertion_ctrls_lbl.setToolTip('Number Insertion Controls')
self._insertion_ctrls_spn = spinbox.BaseSpinBox(parent=self)
self._insertion_ctrls_spn.setMinimum(2)
self._insertion_ctrls_spn.setMaximum(24)
insertion_type_lbl = label.BaseLabel('Type:', parent=self)
self._insertion_type_combo = combobox.BaseComboBox(parent=self)
num_driven_lbl = label.BaseLabel('N. Driven:', parent=self)
num_driven_lbl.setToolTip('Number Driven Joints')
self._num_driven_spn = spinbox.BaseSpinBox(parent=self)
self._num_driven_spn.setMinimum(1)
self._num_driven_spn.setMaximum(64)
num_driven_type_lbl = label.BaseLabel('Type:')
self._num_driven_type_combo = combobox.BaseComboBox(parent=self)
extras_layout = layouts.HorizontalLayout(spacing=2,
margins=(2, 2, 2, 2))
self._cns_mid_ctrls_cbx = checkbox.BaseCheckBox(
'Constraint Mid Controls', parent=self)
self._lock_ctrls_scale_cbx = checkbox.BaseCheckBox(
'Lock Controls Scale', parent=self)
self._lock_jiggle_attrs_cbx = checkbox.BaseCheckBox(
'Lock Jiggle Attributes', parent=self)
extras_layout.addStretch()
extras_layout.addWidget(self._cns_mid_ctrls_cbx)
extras_layout.addWidget(self._lock_ctrls_scale_cbx)
extras_layout.addWidget(self._lock_jiggle_attrs_cbx)
extras_layout.addStretch()
muscle_setup_layout.addLayout(muscle_setup_base_layout)
muscle_setup_layout.addWidget(dividers.Divider())
muscle_setup_layout.addLayout(extras_layout)
muscle_setup_base_layout.addWidget(name_label, 0, 0, Qt.AlignRight)
muscle_setup_base_layout.addWidget(self._name_line, 0, 1)
muscle_setup_base_layout.addWidget(size_label, 1, 0, Qt.AlignRight)
muscle_setup_base_layout.addWidget(self._size_spn, 1, 1)
muscle_setup_base_layout.addWidget(dividers.Divider(), 2, 0, 1, 2)
muscle_setup_base_layout.addWidget(insertion_ctrls_lbl, 3, 0,
Qt.AlignRight)
muscle_setup_base_layout.addWidget(self._insertion_ctrls_spn, 3, 1)
muscle_setup_base_layout.addWidget(insertion_type_lbl, 4, 0,
Qt.AlignRight)
muscle_setup_base_layout.addWidget(self._insertion_type_combo, 4, 1)
muscle_setup_base_layout.addWidget(dividers.Divider(), 5, 0, 1, 2)
muscle_setup_base_layout.addWidget(num_driven_lbl, 6, 0, Qt.AlignRight)
muscle_setup_base_layout.addWidget(self._num_driven_spn, 6, 1)
muscle_setup_base_layout.addWidget(num_driven_type_lbl, 7, 0,
Qt.AlignRight)
muscle_setup_base_layout.addWidget(self._num_driven_type_combo, 7, 1)
advanced_setup_widget = QWidget()
advanced_setup_layout = layouts.VerticalLayout(spacing=2,
margins=(2, 2, 2, 2))
advanced_setup_widget.setLayout(advanced_setup_layout)
self._advanced_widgets_widget = QWidget()
advanced_widgets_layout = layouts.VerticalLayout(spacing=0,
margins=(0, 0, 0, 0))
self._advanced_widgets_widget.setLayout(advanced_widgets_layout)
advanced_setup_base_layout = layouts.GridLayout(spacing=5,
margins=(2, 2, 2, 2))
self._advanced_enable_cbx = checkbox.BaseCheckBox('Enable',
parent=self)
ctrl_suffix_lbl = label.BaseLabel('Control Suffix:', parent=self)
self._ctrl_suffix_line = lineedit.BaseLineEdit(parent=self)
joint_suffix_lbl = label.BaseLabel('Joint Suffix:', parent=self)
self._joint_suffix_line = lineedit.BaseLineEdit(parent=self)
grp_suffix_lbl = label.BaseLabel('Group Suffix:', parent=self)
self._grp_suffix_line = lineedit.BaseLineEdit(parent=self)
drv_suffix_lbl = label.BaseLabel('Driven Suffix:', parent=self)
self._drv_suffix_line = lineedit.BaseLineEdit(parent=self)
advanced_setup_set_layout = layouts.GridLayout(spacing=5,
margins=(2, 2, 2, 2))
self._create_sets_cbx = checkbox.BaseCheckBox('Create Sets',
parent=self)
muscle_set_lbl = label.BaseLabel('Main Muscle Set Name:', parent=self)
self._main_muscle_set_name_line = lineedit.BaseLineEdit(parent=self)
set_suffix_lbl = label.BaseLabel('Muscle Set Suffix:', parent=self)
self._muscle_set_suffix_line = lineedit.BaseLineEdit(parent=self)
muscle_spline_name_lbl = label.BaseLabel('Muscle Spline Name:',
parent=self)
self._muscle_spline_name_line = lineedit.BaseLineEdit(parent=self)
controls_group_suffix_lbl = label.BaseLabel('Controls Group Suffix:',
parent=self)
self._controls_group_suffix_line = lineedit.BaseLineEdit(parent=self)
joints_group_suffix_lbl = label.BaseLabel('Joints Group Suffix:')
self._joints_group_suffix_line = lineedit.BaseLineEdit(parent=self)
root_group_suffix_lbl = label.BaseLabel('Root Group Suffix:',
parent=self)
self._root_group_suffix_line = lineedit.BaseLineEdit(parent=self)
auto_group_suffix_lbl = label.BaseLabel('Auto Group Suffix:',
parent=self)
self._auto_group_suffix_line = lineedit.BaseLineEdit(parent=self)
create_layout = layouts.HorizontalLayout(spacing=0,
margins=(0, 0, 0, 0))
self._create_btn = buttons.BaseButton('Create', parent=self)
create_layout.addStretch()
create_layout.addWidget(self._create_btn)
create_layout.addStretch()
advanced_setup_layout.addWidget(self._advanced_enable_cbx)
advanced_setup_layout.addWidget(dividers.Divider())
advanced_setup_layout.addWidget(self._advanced_widgets_widget)
advanced_widgets_layout.addLayout(advanced_setup_base_layout)
advanced_widgets_layout.addLayout(advanced_setup_set_layout)
advanced_setup_base_layout.addWidget(ctrl_suffix_lbl, 0, 0,
Qt.AlignRight)
advanced_setup_base_layout.addWidget(self._ctrl_suffix_line, 0, 1)
advanced_setup_base_layout.addWidget(joint_suffix_lbl, 1, 0,
Qt.AlignRight)
advanced_setup_base_layout.addWidget(self._joint_suffix_line, 1, 1)
advanced_setup_base_layout.addWidget(grp_suffix_lbl, 2, 0,
Qt.AlignRight)
advanced_setup_base_layout.addWidget(self._grp_suffix_line, 2, 1)
advanced_setup_base_layout.addWidget(drv_suffix_lbl, 3, 0,
Qt.AlignRight)
advanced_setup_base_layout.addWidget(self._drv_suffix_line, 3, 1)
advanced_setup_base_layout.addWidget(dividers.Divider(), 4, 0, 1, 2)
advanced_setup_set_layout.addWidget(self._create_sets_cbx, 0, 0)
advanced_setup_set_layout.addWidget(muscle_set_lbl, 1, 0,
Qt.AlignRight)
advanced_setup_set_layout.addWidget(self._main_muscle_set_name_line, 1,
1)
advanced_setup_set_layout.addWidget(set_suffix_lbl, 2, 0,
Qt.AlignRight)
advanced_setup_set_layout.addWidget(self._muscle_set_suffix_line, 2, 1)
advanced_setup_set_layout.addWidget(dividers.Divider(), 3, 0, 1, 2)
advanced_setup_set_layout.addWidget(muscle_spline_name_lbl, 4, 0,
Qt.AlignRight)
advanced_setup_set_layout.addWidget(self._muscle_spline_name_line, 4,
1)
advanced_setup_set_layout.addWidget(dividers.Divider(), 5, 0, 1, 2)
advanced_setup_set_layout.addWidget(controls_group_suffix_lbl, 6, 0,
Qt.AlignRight)
advanced_setup_set_layout.addWidget(self._controls_group_suffix_line,
6, 1)
advanced_setup_set_layout.addWidget(joints_group_suffix_lbl, 7, 0,
Qt.AlignRight)
advanced_setup_set_layout.addWidget(self._joints_group_suffix_line, 7,
1)
advanced_setup_set_layout.addWidget(root_group_suffix_lbl, 8, 0,
Qt.AlignRight)
advanced_setup_set_layout.addWidget(self._root_group_suffix_line, 8, 1)
advanced_setup_set_layout.addWidget(auto_group_suffix_lbl, 9, 0,
Qt.AlignRight)
advanced_setup_set_layout.addWidget(self._auto_group_suffix_line, 9, 1)
expander.addItem('Advanced Setup', advanced_setup_widget)
expander.addItem('Muscle Setup', muscle_setup_widget)
self.main_layout.addWidget(expander)
self.main_layout.addWidget(dividers.Divider())
self.main_layout.addLayout(create_layout)
def setup_signals(self):
self._name_line.textChanged.connect(self._controller.change_name)
self._size_spn.valueChanged.connect(self._controller.change_size)
self._insertion_ctrls_spn.valueChanged.connect(
self._controller.change_insertion_controls)
self._insertion_type_combo.currentTextChanged.connect(
self._controller.change_control_type)
self._num_driven_spn.valueChanged.connect(
self._controller.change_number_driven_joints)
self._num_driven_type_combo.currentTextChanged.connect(
self._controller.change_driven_type)
self._cns_mid_ctrls_cbx.toggled.connect(
self._controller.change_constraint_mid_controls)
self._lock_ctrls_scale_cbx.toggled.connect(
self._controller.change_lock_controls_scale)
self._lock_jiggle_attrs_cbx.toggled.connect(
self._controller.change_lock_jiggle_attributes)
self._advanced_enable_cbx.toggled.connect(
self._controller.change_enable_advanced)
self._ctrl_suffix_line.textChanged.connect(
self._controller.change_control_suffix)
self._joint_suffix_line.textChanged.connect(
self._controller.change_joint_suffix)
self._grp_suffix_line.textChanged.connect(
self._controller.change_group_suffix)
self._drv_suffix_line.textChanged.connect(
self._controller.change_driven_suffix)
self._create_sets_cbx.toggled.connect(
self._controller.change_create_sets)
self._main_muscle_set_name_line.textChanged.connect(
self._controller.change_main_muscle_set_name)
self._muscle_set_suffix_line.textChanged.connect(
self._controller.change_muscle_set_suffix)
self._muscle_spline_name_line.textChanged.connect(
self._controller.change_muscle_spline_name)
self._controls_group_suffix_line.textChanged.connect(
self._controller.change_controls_group_suffix)
self._joints_group_suffix_line.textChanged.connect(
self._controller.change_joints_group_suffix)
self._root_group_suffix_line.textChanged.connect(
self._controller.change_root_group_suffix)
self._auto_group_suffix_line.textChanged.connect(
self._controller.change_auto_group_suffix)
self._model.nameChanged.connect(self._on_name_changed)
self._model.sizeChanged.connect(self._size_spn.setValue)
self._model.insertionControlsChanged.connect(
self._insertion_ctrls_spn.setValue)
self._model.controlTypeChanged.connect(
self._insertion_type_combo.setCurrentText)
self._model.drivenJointsChanged.connect(self._num_driven_spn.setValue)
self._model.drivenTypeChanged.connect(
self._num_driven_type_combo.setCurrentText)
self._model.constraintMidControlsChanged.connect(
self._cns_mid_ctrls_cbx.setChecked)
self._model.lockControlsScaleChanged.connect(
self._lock_ctrls_scale_cbx.setChecked)
self._model.lockJiggleAttributesChanged.connect(
self._lock_jiggle_attrs_cbx.setChecked)
self._model.enableAdvancedChanged.connect(
self._on_enable_advanced_changed)
self._model.controlSuffixChanged.connect(
self._ctrl_suffix_line.setText)
self._model.jointSuffixChanged.connect(self._joint_suffix_line.setText)
self._model.groupSuffixChanged.connect(self._grp_suffix_line.setText)
self._model.drivenSuffixChanged.connect(self._drv_suffix_line.setText)
self._model.createSetsChanged.connect(self._create_sets_cbx.setChecked)
self._model.mainMuscleSetNameChanged.connect(
self._main_muscle_set_name_line.setText)
self._model.muscleSetSuffixChanged.connect(
self._muscle_set_suffix_line.setText)
self._model.muscleSplineNameChanged.connect(
self._muscle_spline_name_line.setText)
self._model.controlsGroupSuffixChanged.connect(
self._controls_group_suffix_line.setText)
self._model.jointsGroupSuffixChanged.connect(
self._joints_group_suffix_line.setText)
self._model.rootGroupSuffixChanged.connect(
self._root_group_suffix_line.setText)
self._model.autoGroupSuffixChanged.connect(
self._auto_group_suffix_line.setText)
self._create_btn.clicked.connect(self._controller.create_muscle_spline)
def refresh(self):
self._insertion_type_combo.clear()
self._num_driven_type_combo.clear()
self._name_line.setText(self._model.name)
self._size_spn.setValue(self._model.size)
self._insertion_type_combo.addItems(self._model.insertion_types)
self._num_driven_type_combo.addItems(self._model.driven_types)
self._insertion_ctrls_spn.setValue(self._model.insertion_controls)
self._insertion_type_combo.setCurrentText(self._model.control_type)
self._num_driven_spn.setValue(self._model.driven_joints)
self._num_driven_type_combo.setCurrentText(self._model.driven_type)
self._cns_mid_ctrls_cbx.setChecked(self._model.constraint_mid_controls)
self._lock_ctrls_scale_cbx.setChecked(self._model.lock_controls_scale)
self._lock_jiggle_attrs_cbx.setChecked(
self._model.lock_jiggle_attributes)
self._advanced_enable_cbx.setChecked(self._model.enable_advanced)
self._ctrl_suffix_line.setText(self._model.control_suffix)
self._joint_suffix_line.setText(self._model.joint_suffix)
self._grp_suffix_line.setText(self._model.group_suffix)
self._drv_suffix_line.setText(self._model.driven_suffix)
self._create_sets_cbx.setChecked(self._model.create_sets)
self._main_muscle_set_name_line.setText(
self._model.main_muscle_set_name)
self._muscle_set_suffix_line.setText(self._model.muscle_set_suffix)
self._muscle_spline_name_line.setText(self._model.muscle_spline_name)
self._controls_group_suffix_line.setText(
self._model.controls_group_suffix)
self._joints_group_suffix_line.setText(self._model.joints_group_suffix)
self._root_group_suffix_line.setText(self._model.root_group_suffix)
self._auto_group_suffix_line.setText(self._model.auto_group_suffix)
self._check_ui()
def _check_ui(self):
self._create_btn.setEnabled(not self._model.name == '')
self._advanced_widgets_widget.setEnabled(self._model.enable_advanced)
def _on_name_changed(self, value):
self._name_line.setText(value)
self._check_ui()
def _on_enable_advanced_changed(self, flag):
self._advanced_enable_cbx.setChecked(flag)
self._check_ui()
class NameIt(base.BaseWidget, object):
ACTIVE_RULE = None
def __init__(self, naming_lib, data_file=None, parent=None):
self._naming_lib = naming_lib
self._data_file = None
super(NameIt, self).__init__(parent=parent)
self.set_data_file(data_file)
def get_active_rule(self):
"""
Returns the current naming active rule
"""
return self._naming_lib.active_rule()
def set_active_rule(self, name):
"""
Sets the current active rule
:param name: str
"""
# First, we clean the status of the naming library
self._naming_lib.remove_all_tokens()
self._naming_lib.remove_all_rules()
self._naming_lib.load_session()
# Load rules from the naming manager
rules = self._naming_lib.rules
for rule in rules:
expressions = rule.get_expression_tokens()
self._naming_lib.add_rule(rule.name, rule.iterator_format,
*expressions)
# Load tokens from the naming manager
tokens = self._naming_lib.tokens
for token in tokens:
tokens_keywords = token.get_values_as_keyword()
self._naming_lib.add_token(token.name, **tokens_keywords)
self._naming_lib.set_active_rule(name)
def set_active_rule_iterator(self, iterator_format):
active_rule = self.get_active_rule()
if not active_rule:
return
def set_active_rule_auto_fix(self, auto_fix):
active_rule = self.get_active_rule()
if not active_rule:
return
self._naming_lib.set_rule_auto_fix(active_rule.name(), auto_fix)
def solve(self, *args, **kwargs):
if len(args) > 0 and len(kwargs) > 0:
return self._naming_lib.solve(*args, **kwargs)
else:
if len(args) > 0:
return self._naming_lib.solve(*args)
else:
return self._naming_lib.solve(**kwargs)
def ui(self):
super(NameIt, self).ui()
tools_toolbar = toolbar.ToolBar('Main ToolBar', parent=self)
tools_toolbar.setMovable(False)
# tools_toolbar.setAllowedAreas(Qt.TopToolBarArea | Qt.BottomToolBarArea)
self.main_layout.addWidget(tools_toolbar)
refresh_btn = buttons.BaseToolButton(parent=tools_toolbar)
refresh_btn.setIcon(resources.icon('refresh'))
refresh_btn.clicked.connect(self._on_refresh)
self._refresh_btn = tools_toolbar.addWidget(refresh_btn)
self._refresh_btn.setEnabled(False)
save_btn = buttons.BaseToolButton(parent=tools_toolbar)
save_btn.setIcon(resources.icon('save'))
save_btn.clicked.connect(self._on_save)
self._save_btn = tools_toolbar.addWidget(save_btn)
self._save_btn.setEnabled(False)
if self._is_renamer_tool_available():
renamer_btn = buttons.BaseToolButton(parent=tools_toolbar)
renamer_btn.setIcon(resources.icon('rename'))
renamer_btn.clicked.connect(self._on_open_renamer_tool)
tools_toolbar.addWidget(renamer_btn)
self._name_file_line = directory.SelectFile(label_text='Naming File',
parent=tools_toolbar)
tools_toolbar.addWidget(self._name_file_line)
base_layout = layouts.HorizontalLayout(spacing=0, margins=(0, 0, 0, 0))
self.main_layout.addLayout(base_layout)
panels_splitter = QSplitter(parent=self)
panels_splitter.setSizePolicy(QSizePolicy.Expanding,
QSizePolicy.Expanding)
base_layout.addWidget(panels_splitter)
left_panel_widget = QWidget()
left_panel_layout = layouts.VerticalLayout(margins=(5, 0, 5, 0))
left_panel_widget.setLayout(left_panel_layout)
panels_splitter.addWidget(left_panel_widget)
# Tab Widget
rules_tab = QWidget()
tokens_tab = QWidget()
templates_tab = QWidget()
templates_tokens_tab = QWidget()
self.tabs = tabs.BaseTabWidget()
self.tabs.addTab(rules_tab, 'Rules')
self.tabs.addTab(tokens_tab, 'Tokens')
self.tabs.addTab(templates_tab, 'Templates')
self.tabs.addTab(templates_tokens_tab, 'Templates Tokens')
left_panel_layout.addWidget(self.tabs)
# Rules Tab
rules_main_layout = layouts.VerticalLayout(spacing=0,
margins=(5, 5, 5, 5))
self.rules_list = QListWidget()
rules_main_layout.addWidget(self.rules_list)
left_panel_buttons_layout_rules = layouts.HorizontalLayout(margins=(5,
5,
5,
0))
rules_main_layout.addLayout(left_panel_buttons_layout_rules)
self.add_rule_btn = buttons.BaseButton(parent=self)
self.remove_rule_btn = buttons.BaseButton(parent=self)
self.add_rule_btn.setIcon(resources.icon('plus'))
self.remove_rule_btn.setIcon(resources.icon('minus'))
left_panel_buttons_layout_rules.addStretch()
left_panel_buttons_layout_rules.addWidget(self.add_rule_btn)
left_panel_buttons_layout_rules.addWidget(self.remove_rule_btn)
rules_tab.setLayout(rules_main_layout)
# Tokens Tab
tokens_main_layout = layouts.VerticalLayout(margins=(5, 5, 5, 5))
tokens_main_layout.setSpacing(0)
self.tokens_list = QListWidget()
tokens_main_layout.addWidget(self.tokens_list)
left_panel_buttons_layout_tokens = layouts.HorizontalLayout(
margins=(5, 5, 5, 0))
tokens_main_layout.addLayout(left_panel_buttons_layout_tokens)
self.add_token_btn = buttons.BaseButton(parent=self)
self.remove_token_btn = buttons.BaseButton(parent=self)
self.add_token_btn.setIcon(resources.icon('plus'))
self.remove_token_btn.setIcon(resources.icon('minus'))
left_panel_buttons_layout_tokens.addStretch()
left_panel_buttons_layout_tokens.addWidget(self.add_token_btn)
left_panel_buttons_layout_tokens.addWidget(self.remove_token_btn)
tokens_tab.setLayout(tokens_main_layout)
# Templates Tab
templates_main_layout = layouts.VerticalLayout(spacing=0,
margins=(5, 5, 5, 5))
self.templates_list = QListWidget()
templates_main_layout.addWidget(self.templates_list)
left_panel_buttons_layout_templates = layouts.HorizontalLayout(
margins=(5, 5, 5, 0))
templates_main_layout.addLayout(left_panel_buttons_layout_templates)
self.add_template_btn = buttons.BaseButton(parent=self)
self.remove_template_btn = buttons.BaseButton(parent=self)
self.add_template_btn.setIcon(resources.icon('plus'))
self.remove_template_btn.setIcon(resources.icon('minus'))
left_panel_buttons_layout_templates.addStretch()
left_panel_buttons_layout_templates.addWidget(self.add_template_btn)
left_panel_buttons_layout_templates.addWidget(self.remove_template_btn)
templates_tab.setLayout(templates_main_layout)
# Template Tokens Tab
templates_tokens_main_layout = layouts.VerticalLayout(spacing=0,
margins=(5, 5, 5,
5))
self.template_tokens_list = QListWidget()
templates_tokens_main_layout.addWidget(self.template_tokens_list)
left_panel_buttons_layout_templates_tokens = layouts.HorizontalLayout(
margins=(5, 5, 5, 0))
left_panel_buttons_layout_templates_tokens.setContentsMargins(
5, 5, 5, 0)
templates_tokens_main_layout.addLayout(
left_panel_buttons_layout_templates_tokens)
self.add_template_token_btn = buttons.BaseButton(parent=self)
self.remove_template_token_btn = buttons.BaseButton(parent=self)
self.add_template_token_btn.setIcon(resources.icon('plus'))
self.remove_template_token_btn.setIcon(resources.icon('minus'))
left_panel_buttons_layout_templates_tokens.addStretch()
left_panel_buttons_layout_templates_tokens.addWidget(
self.add_template_token_btn)
left_panel_buttons_layout_templates_tokens.addWidget(
self.remove_template_token_btn)
templates_tokens_tab.setLayout(templates_tokens_main_layout)
# === PROPERTIES === #
main_group = QGroupBox('Properties')
panels_splitter.addWidget(main_group)
self.group_layout = layouts.VerticalLayout(spacing=0,
margins=(5, 5, 5, 5))
main_group.setLayout(self.group_layout)
# Rules Panel
self.rules_widget = QWidget(self)
rules_layout = layouts.GridLayout()
self.rules_widget.setLayout(rules_layout)
expression_lbl = label.BaseLabel('Expression', parent=self)
self.expression_line = lineedit.BaseLineEdit(parent=self)
self.expression_btn = buttons.BaseButton(' <', parent=self)
self.expression_btn.setEnabled(False)
self.expression_btn.setStyleSheet(
"QPushButton::menu-indicator{image:url(none.jpg);}")
self.expression_menu = QMenu(self)
self.expression_btn.setMenu(self.expression_menu)
iterator_lbl = label.BaseLabel('Iterator', parent=self)
self.iterator_cbx = combobox.BaseComboBox(parent=self)
for it_format in [
'@', '@^', '#', '##', '###', '####', '#####', 'None'
]:
self.iterator_cbx.addItem(it_format)
description_rule_lbl = label.BaseLabel('Description', parent=self)
self.description_rule_text = QTextEdit(parent=self)
self.group_layout.addWidget(self.rules_widget)
rules_layout.addWidget(expression_lbl, 0, 0, Qt.AlignRight)
rules_layout.addWidget(self.expression_line, 0, 1)
rules_layout.addWidget(self.expression_btn, 0, 2)
rules_layout.addWidget(iterator_lbl, 1, 0, Qt.AlignRight | Qt.AlignTop)
rules_layout.addWidget(self.iterator_cbx, 1, 1, 1, 2)
rules_layout.addWidget(description_rule_lbl, 2, 0,
Qt.AlignRight | Qt.AlignTop)
rules_layout.addWidget(self.description_rule_text, 2, 1, 1, 2)
# Tokens Panel
self.tokens_widget = QWidget(parent=self)
tokens_layout = layouts.GridLayout()
self.tokens_widget.setLayout(tokens_layout)
values_lbl = label.BaseLabel('Values')
data = {'key': list(), 'value': list()}
self.values_table = TokensTable(data, 0, 2)
self.values_table.setSizePolicy(QSizePolicy.Expanding,
QSizePolicy.Expanding)
values_buttons_layout = layouts.HorizontalLayout(margins=(5, 5, 5, 0))
self.add_key_value_btn = buttons.BaseButton(parent=self)
self.remove_key_value_btn = buttons.BaseButton(parent=self)
self.add_key_value_btn.setIcon(resources.icon('plus'))
self.remove_key_value_btn.setIcon(resources.icon('minus'))
values_buttons_layout.addWidget(self.add_key_value_btn)
values_buttons_layout.addWidget(self.remove_key_value_btn)
values_buttons_layout.addStretch()
default_lbl = label.BaseLabel('Default', parent=self)
self.default_cbx = combobox.BaseComboBox(parent=self)
description_tokens_lbl = label.BaseLabel('Description', parent=self)
self.description_token_text = QTextEdit(parent=self)
self.group_layout.addWidget(self.tokens_widget)
self.tokens_widget.hide()
tokens_layout.addWidget(values_lbl, 0, 0, Qt.AlignRight | Qt.AlignTop)
tokens_layout.addWidget(self.values_table, 0, 1, 2, 1)
tokens_layout.addLayout(values_buttons_layout, 0, 2)
tokens_layout.addWidget(default_lbl, 4, 0, Qt.AlignRight)
tokens_layout.addWidget(self.default_cbx, 4, 1, 1, 2)
tokens_layout.addWidget(description_tokens_lbl, 5, 0,
Qt.AlignRight | Qt.AlignTop)
tokens_layout.addWidget(self.description_token_text, 5, 1, 1, 2)
# Templates Panel
self.templates_widget = QWidget(parent=self)
templates_layout = layouts.VerticalLayout()
self.templates_widget.setLayout(templates_layout)
pattern_layout = layouts.HorizontalLayout(spacing=5,
margins=(5, 5, 5, 5))
pattern_lbl = label.BaseLabel('Pattern: ', parent=self)
self.pattern_line = lineedit.BaseLineEdit(parent=self)
pattern_layout.addWidget(pattern_lbl)
pattern_layout.addWidget(self.pattern_line)
templates_layout.addLayout(pattern_layout)
templates_layout.addLayout(dividers.DividerLayout())
self.template_tokens_layout = layouts.GridLayout()
self.template_tokens_layout.setAlignment(Qt.AlignTop)
template_tokens_frame = QFrame(parent=self)
template_tokens_frame.setFrameShape(QFrame.StyledPanel)
template_tokens_frame.setFrameShadow(QFrame.Sunken)
template_tokens_frame.setLayout(self.template_tokens_layout)
templates_layout.addWidget(template_tokens_frame)
self.group_layout.addWidget(self.templates_widget)
self.templates_widget.hide()
# Templates Tokens Panel
self.templates_tokens_widget = QWidget(parent=self)
templates_tokens_layout = layouts.VerticalLayout()
self.templates_tokens_widget.setLayout(templates_tokens_layout)
description_templates_token_layout = layouts.HorizontalLayout(
spacing=5, margins=(5, 5, 5, 5))
description_tokens_layout = layouts.VerticalLayout()
description_templates_token_lbl = label.BaseLabel('Description: ',
parent=self)
description_tokens_layout.addWidget(description_templates_token_lbl)
description_tokens_layout.addStretch()
self.description_templates_token_text = QTextEdit(parent=self)
description_templates_token_layout.addLayout(description_tokens_layout)
description_templates_token_layout.addWidget(
self.description_templates_token_text)
templates_tokens_layout.addLayout(description_templates_token_layout)
self.group_layout.addWidget(self.templates_tokens_widget)
self.templates_tokens_widget.hide()
# Initialize database
self._init_db()
# First update of the UI
self.update_expression_menu()
self.update_tokens_properties_state()
self.update_rules_properties_state()
self.update_templates_properties_state()
def setup_signals(self):
super(NameIt, self).setup_signals()
self.tabs.currentChanged.connect(self._on_change_tab)
self._name_file_line.directoryChanged.connect(
self._on_naming_file_changed)
self.add_rule_btn.clicked.connect(self._on_add_rule)
self.remove_rule_btn.clicked.connect(self._on_remove_rule)
self.rules_list.currentItemChanged.connect(self._on_change_rule)
self.rules_list.itemChanged.connect(self._on_edit_rule_name)
self.expression_line.textChanged.connect(self._on_edit_rule_expression)
self.description_rule_text.textChanged.connect(
self._on_edit_rule_description)
self.iterator_cbx.currentIndexChanged.connect(
self._on_edit_rule_iterator)
self.add_token_btn.clicked.connect(self._on_add_token)
self.remove_token_btn.clicked.connect(self._on_remove_token)
self.tokens_list.currentItemChanged.connect(self._on_change_token)
self.tokens_list.itemChanged.connect(self._on_edit_token_name)
self.values_table.itemChanged.connect(self._on_change_token_value)
self.add_key_value_btn.clicked.connect(self._on_add_token_value)
self.remove_key_value_btn.clicked.connect(self._on_remove_token_value)
self.description_token_text.textChanged.connect(
self._on_edit_token_description)
self.default_cbx.currentIndexChanged.connect(
self._on_edit_token_default)
self.add_template_btn.clicked.connect(self._on_add_template)
self.remove_template_btn.clicked.connect(self._on_remove_template)
self.templates_list.currentItemChanged.connect(
self._on_change_template)
self.templates_list.itemChanged.connect(self._on_edit_template_name)
self.pattern_line.textChanged.connect(self._on_edit_template_pattern)
self.add_template_token_btn.clicked.connect(
self._on_add_template_token)
self.remove_template_token_btn.clicked.connect(
self._on_remove_template_token)
self.template_tokens_list.currentItemChanged.connect(
self._on_change_template_token)
self.template_tokens_list.itemChanged.connect(
self._on_edit_template_token_name)
self.description_templates_token_text.textChanged.connect(
self._on_edit_template_token_description)
def set_data_file(self, data_file):
"""
Sets the data file used by the naming library
:param data_file: str
"""
if data_file and os.path.isfile(data_file):
self._naming_lib.naming_file = data_file
self._data_file = data_file
else:
naming_file = self._naming_lib.naming_file
if naming_file and os.path.isfile(naming_file):
self._data_file = self._naming_lib.naming_file
else:
self._data_file = self._get_default_data_file()
self._naming_lib.naming_file = self._data_file
self._init_db()
def add_expression(self, name):
"""
Add an expression to the list of expressions
:param str name: Expression name
:return: None
"""
if self.expression_line.text() == '':
self.expression_line.setText('{' + name + '}')
else:
self.expression_line.setText(self.expression_line.text() + '_{' +
name + '}')
def update_expression_menu(self):
"""
Updates the expression menu
:return:
"""
# First, we clear the expression menu
self.expression_menu.clear()
tokens = self._naming_lib.tokens
if tokens and len(tokens) > 0:
self.expression_btn.setEnabled(True)
for token in tokens:
self.expression_menu.addAction(
token.name, partial(self.add_expression, token.name))
else:
self.expression_btn.setEnabled(False)
def update_expression_state(self):
# TODO: This method is used to check the name of the expression (its different parts) and
# TODO: check if some token of the expression does not exist and in that case, update the expression
# TODO: so it becomes a valid expression
pass
def update_rules_properties_state(self):
if self.rules_list.count() <= 0 or self.rules_list.currentItem(
) is None:
self.expression_line.setText('')
self.description_rule_text.setText('')
self.iterator_cbx.setCurrentIndex(0)
self.rules_widget.setEnabled(False)
else:
rule = self._naming_lib.get_rule(
self.rules_list.currentItem().text())
if rule is not None:
self.expression_line.setText(rule.expression)
self.description_rule_text.setText(rule.description)
self.iterator_cbx.setCurrentText(rule.iterator_format)
self.rules_widget.setEnabled(True)
def update_tokens_properties_state(self):
if self.tokens_list.currentItem() is None:
self.tokens_widget.setEnabled(False)
else:
self.tokens_widget.setEnabled(True)
def update_default_token_list(self):
self.default_cbx.blockSignals(True)
for i in range(self.default_cbx.count()):
self.default_cbx.removeItem(self.default_cbx.count() - 1)
item_text = self.tokens_list.currentItem().text()
self.default_cbx.addItem('')
tokens = self._naming_lib.tokens
for token in tokens:
if token.name != item_text:
continue
for value in token.values['key']:
self.default_cbx.addItem(value)
token = self._naming_lib.get_token(item_text)
if token:
self.default_cbx.setCurrentIndex(token.default)
self.default_cbx.blockSignals(False)
def update_tokens_key_table(self):
item_text = self.tokens_list.currentItem().text()
self.clean_tokens_key_table()
if self.tokens_list.count() > 0:
keys = []
values = []
tokens = self._naming_lib.tokens
for token in tokens:
if token.name != item_text:
continue
for i in range(len(token.values['key'])):
self.values_table.insertRow(self.values_table.rowCount())
keys.append(QTableWidgetItem())
values.append(QTableWidgetItem())
for index, value in enumerate(token.values['key']):
keys[index].setText(value)
self.values_table.setItem(index, 0, keys[index])
for index, value in enumerate(token.values['value']):
values[index].setText(value)
self.values_table.setItem(index, 1, values[index])
def clean_tokens_key_table(self):
for i in range(self.values_table.rowCount()):
self.values_table.removeRow(self.values_table.rowCount() - 1)
def update_templates_properties_state(self):
if self.templates_list.count() <= 0 or self.templates_list.currentItem(
) is None:
self.pattern_line.setText('')
self.templates_widget.setEnabled(False)
else:
template = self._naming_lib.get_template(
self.templates_list.currentItem().text())
if template is not None:
self.pattern_line.setText(template.pattern)
self.templates_widget.setEnabled(True)
self._update_template_tokens(template)
def _init_db(self):
"""
Initializes the naming data base
"""
self._naming_lib.init_naming_data()
self._init_data()
self._name_file_line.set_directory(str(self._naming_lib.naming_file))
def _init_data(self):
if self._load_rules():
self._load_tokens()
self._load_templates()
self._load_template_tokens()
def _get_default_data_file(self):
"""
Internal function that returns default path to nameing data file
:return: str
"""
return os.path.join(
os.path.dirname(os.path.dirname(os.path.realpath(__file__))),
'data', 'naming_data.json')
def _load_rules(self):
"""
Internal function that load rules from data file
"""
self.rules_list.clear()
rules = self._naming_lib.rules
if not rules:
return False
for rule in rules:
if rule == '_active':
continue
self._on_add_rule(rule)
return True
def _load_tokens(self):
"""
Load tokens from data file
"""
self.tokens_list.clear()
tokens = self._naming_lib.tokens
if not tokens:
return False
for token in tokens:
self._on_add_token(token)
return True
def _load_templates(self):
"""
Internal function that loads templates from DB
"""
try:
templates = self._naming_lib.templates
self.templates_list.clear()
if templates is not None:
for template in templates:
self._on_add_template(template)
return True
except Exception as e:
logger.error(
'Error while loading templates from: {} | {} | {}'.format(
self.DATA_FILE, e, traceback.format_exc()))
return False
def _load_template_tokens(self):
"""
Internal function that loads template tokens from DB
"""
try:
template_tokens = self._naming_lib.template_tokens
self.template_tokens_list.clear()
if template_tokens is not None:
for template_token in template_tokens:
self._on_add_template_token(template_token)
return True
except Exception as e:
logger.error(
'Error while loading template tokens from: {} | {} | {}'.
format(self.DATA_FILE, e, traceback.format_exc()))
return False
def _clear_template_tokens(self):
"""
Intenral function that clears all template tokens from layout
"""
for i in range(self.template_tokens_layout.count(), -1, -1):
item = self.template_tokens_layout.itemAt(i)
if item is None:
continue
item.widget().setParent(None)
self.template_tokens_layout.removeItem(item)
def _add_template_token(self,
template_token_name,
template_token_description=None):
"""
Adds template token to layout
:param template_token_name: str
:param template_token_data: dict
"""
row = 0
while self.template_tokens_layout.itemAtPosition(row, 0) is not None:
row += 1
self.template_tokens_layout.addWidget(
label.BaseLabel(template_token_name), row, 0)
self.template_tokens_layout.addWidget(
label.BaseLabel(template_token_description
if template_token_description else '< NOT FOUND >',
parent=self), row, 1)
def _update_template_tokens(self, template):
"""
Internal function that updates template tokens currently loaded
:param template: Template
"""
if not template:
return
temp_tokens = list()
try:
temp = self._naming_lib.get_template(template.name)
# temp = lucidity.Template(template.name, template.pattern)
temp_template = temp.template
temp_template.duplicate_placeholder_mode = lucidity.Template.STRICT
temp_tokens = temp_template.keys()
except (ValueError, lucidity.error.ResolveError) as exc:
self._clear_template_tokens()
return
template_tokens = self._naming_lib.template_tokens
self._clear_template_tokens()
for token in temp_tokens:
token_found = False
for template_token in template_tokens:
if token == template_token.name:
self._add_template_token(token, template_token.description)
token_found = True
if not token_found:
self._add_template_token(token)
def _on_change_tab(self, tab_index):
"""
This methods changes the properties tab widgets
:param tab_index: Index of the current tab (0:rules tab, 1:tokens tab)
:return: None
"""
if tab_index == 0:
self.rules_widget.show()
self.tokens_widget.hide()
self.templates_widget.hide()
self.templates_tokens_widget.hide()
self.update_expression_menu()
self.update_expression_state()
elif tab_index == 1:
self.rules_widget.hide()
self.tokens_widget.show()
self.templates_widget.hide()
self.templates_tokens_widget.hide()
self.update_tokens_properties_state()
elif tab_index == 2:
self.rules_widget.hide()
self.tokens_widget.hide()
self.templates_widget.show()
self.templates_tokens_widget.hide()
self.update_templates_properties_state()
else:
self.rules_widget.hide()
self.tokens_widget.hide()
self.templates_widget.hide()
self.templates_tokens_widget.show()
def _on_naming_file_changed(self, file_path):
"""
Internal callback function that is called when a new file path is set
:param file_path: str
"""
self._naming_lib.naming_file = file_path
if not file_path or not os.path.isfile(file_path):
self._refresh_btn.setEnabled(False)
self._save_btn.setEnabled(False)
else:
self._refresh_btn.setEnabled(True)
self._save_btn.setEnabled(True)
self._on_refresh()
def _on_add_rule(self, *args):
"""
Creates a new standard rule and add it to the Naming Manager
:return:
"""
load_rule = True
if len(args) == 0:
load_rule = False
self.description_rule_text.blockSignals(True)
rule = None
if not load_rule:
rule = self._naming_lib.get_rule_unique_name(name='New_Rule')
elif load_rule and len(args) == 1:
rule = args[0].name
if rule is not None:
# Create a new item based on the rule name and add it
item = QListWidgetItem(rule)
item.setFlags(item.flags() | Qt.ItemIsEditable)
self.rules_list.addItem(item)
# Add the data of the rule to our JSON data file
if len(args) == 0:
self._naming_lib.add_rule(rule)
# Update necessary UI Widgets
if not load_rule:
self.rules_list.setCurrentItem(item)
self.update_expression_menu()
self.update_rules_properties_state()
self.description_rule_text.blockSignals(False)
def _on_remove_rule(self):
"""
Remove the selected rule from the list of rules
:return: bool, True if the element deletion is successful or False otherwise
"""
self.description_rule_text.blockSignals(True)
curr_rule = self.rules_list.currentItem()
if curr_rule is not None:
rule_name = self.rules_list.currentItem().text()
rule = self._naming_lib.get_rule(rule_name)
if rule is not None:
self._naming_lib.remove_rule(rule_name)
self.rules_list.takeItem(
self.rules_list.row(self.rules_list.currentItem()))
self.update_rules_properties_state()
self.description_rule_text.blockSignals(False)
def _on_change_rule(self, rule_item):
"""
Change the selected rule
:param rule_item: new QListWidgetItem selected
:return: None
"""
if rule_item is not None:
if rule_item.listWidget().count() > 0:
rule = self._naming_lib.get_rule(rule_item.text())
if rule is not None:
self.description_rule_text.setText(rule.description)
self.expression_line.setText(rule.expression)
self.iterator_cbx.setCurrentText(rule.iterator_format)
self.update_expression_menu()
self.update_rules_properties_state()
def _on_edit_rule_name(self, rule_item):
"""
Changes name of the rule
:param rule_item: Renamed QListWidgetItem
:return: None
"""
rule_index = rule_item.listWidget().currentRow()
rule = self._naming_lib.get_rule_by_index(rule_index)
if rule:
rule.name = rule_item.text()
def _on_edit_rule_expression(self):
"""
Changes expression of the selected rule
:return: None
"""
selected_rule = self.rules_list.currentItem()
if not selected_rule:
return
rule_name = selected_rule.text()
rule = self._naming_lib.get_rule(rule_name)
if rule:
rule.expression = self.expression_line.text()
def _on_edit_rule_description(self):
"""
Changes description of the selected rule
:return: None
"""
rule_name = self.rules_list.currentItem().text()
rule = self._naming_lib.get_rule(rule_name)
if rule:
rule.description = self.description_rule_text.toPlainText()
def _on_edit_rule_iterator(self, iterator_index):
"""
Changes iterator of the selected rule
:param iterator_index: int
:return: None
"""
rule_item = self.rules_list.currentItem()
if not rule_item:
return
rule_name = rule_item.text()
rule = self._naming_lib.get_rule(rule_name)
if rule:
rule.iterator_format = self.iterator_cbx.itemText(iterator_index)
def _on_add_token(self, *args):
"""
Creates a new token and add it to the Naming Manager
:return: None
"""
load_token = True
if len(args) == 0:
load_token = False
token = None
if not load_token:
token = self._naming_lib.get_token_unique_name(name='New_Token')
elif load_token and len(args) == 1:
token = args[0].name
if token:
# Create a new item based on the token name and add it
item = QListWidgetItem(token)
item.setFlags(item.flags() | Qt.ItemIsEditable)
self.tokens_list.addItem(item)
# Add the data of the token to our JSON data file
if len(args) == 0:
self._naming_lib.add_token(token)
# Update necessary UI wigdets
if not load_token:
self.tokens_list.setCurrentItem(item)
def _on_remove_token(self):
"""
Remove the selected token from the list of tokens
:return: True if the element deletion is successfull or False otherwise
"""
curr_token = self.tokens_list.currentItem()
if curr_token is not None:
token_index = self.tokens_list.currentRow()
name = self.tokens_list.currentItem().text()
if token_index > -1 and name is not None:
token = self._naming_lib.get_token(name)
if token is not None:
if token.name == name:
self._naming_lib.remove_token(name)
self.tokens_list.takeItem(
self.tokens_list.row(
self.tokens_list.currentItem()))
def _on_change_token(self, token_item):
"""
Change the selected token
:param token_item: new QListWidgetItem selected
:return: None
"""
if token_item is not None:
if token_item.listWidget().count() > 0:
token = self._naming_lib.get_token(token_item.text())
if token:
try:
self.description_rule_text.blockSignals(True)
self.default_cbx.blockSignals(True)
self.description_token_text.setText(token.description)
self.default_cbx.setCurrentIndex(int(token.default))
finally:
self.description_rule_text.blockSignals(False)
self.default_cbx.blockSignals(False)
self.update_tokens_properties_state()
self.update_tokens_key_table()
self.update_default_token_list()
def _on_edit_token_name(self, token_item):
"""
Changes name of the token
:param token_item: Renamed QListWidgetItem
:return:
"""
token_index = self.tokens_list.currentRow()
token = self._naming_lib.get_token_by_index(token_index)
if token:
token.name = token_item.text()
def _on_change_token_value(self, item):
"""
Called when we change a token value name
:return: None
"""
token_text = self.tokens_list.currentItem().text()
token = self._naming_lib.get_token(token_text)
if not token:
return
if self.tokens_list.currentRow() > -1 and item.row() > -1:
if item.column() == 0:
token.set_token_key(item.row(), item.text())
else:
token.set_token_value(item.row(), item.text())
self.update_default_token_list()
def _on_add_token_value(self, *args):
self.description_rule_text.blockSignals(True)
token_text = self.tokens_list.currentItem().text()
token = self._naming_lib.get_token(token_text)
if token:
key_data = token.add_token_value()
if key_data:
self.clean_tokens_key_table()
keys = list()
values = list()
for i in range(len(key_data['key'])):
self.values_table.insertRow(self.values_table.rowCount())
keys.append(QTableWidgetItem())
values.append(QTableWidgetItem())
for index, value in enumerate(key_data['key']):
keys[index].setText(value)
self.values_table.setItem(index, 0, keys[index])
for index, value in enumerate(key_data['value']):
values[index].setText(value)
self.values_table.setItem(index, 1, values[index])
self.update_default_token_list()
# new_index = self.default_cbx.currentIndex() + 2
# self.default_cbx.setCurrentIndex(new_index)
self.description_rule_text.blockSignals(False)
def _on_remove_token_value(self):
"""
Removes a token value from the list of tokens values
"""
self.description_rule_text.blockSignals(True)
token_text = self.tokens_list.currentItem().text()
token = self._naming_lib.get_token(token_text)
if token:
key_data = token.remove_token_value(self.values_table.currentRow())
if key_data:
self.clean_tokens_key_table()
keys = list()
values = list()
for i in range(len(key_data['key'])):
self.values_table.insertRow(self.values_table.rowCount())
keys.append(QTableWidgetItem())
values.append(QTableWidgetItem())
for index, value in enumerate(key_data['key']):
keys[index].setText(value)
self.values_table.setItem(index, 0, keys[index])
for index, value in enumerate(key_data['value']):
values[index].setText(value)
self.values_table.setItem(index, 1, values[index])
self.update_default_token_list()
new_index = self.default_cbx.currentIndex()
token.default = new_index
self.default_cbx.setCurrentIndex(new_index)
self.description_rule_text.blockSignals(False)
def _on_edit_token_default(self, index):
"""
Edits the default token
:param index: int, index of the token to edit
"""
token_text = self.tokens_list.currentItem().text()
token = self._naming_lib.get_token(token_text)
if token:
token.default = index
def _on_edit_token_description(self):
"""
Edits the token description
"""
token_text = self.tokens_list.currentItem().text()
token = self._naming_lib.get_token(token_text)
if token:
token.description = self.description_token_text.toPlainText(
).strip()
def _on_add_template(self, *args):
"""
Creates a new template and add it to the Naming Manager
:return:
"""
load_template = True
if len(args) == 0:
load_template = False
template = None
if not load_template:
template = self._naming_lib.get_template_unique_name(
'New_Template')
elif load_template and len(args) == 1:
template = args[0].name
if template is not None:
item = QListWidgetItem(template)
item.setFlags(item.flags() | Qt.ItemIsEditable)
self.templates_list.addItem(item)
if len(args) == 0:
self._naming_lib.add_template(template)
if not load_template:
self.templates_list.setCurrentItem(item)
def _on_remove_template(self):
"""
Removes the selected template from the list of templates
:return: bool, True if the element deletion is successful or False otherwise
"""
current_template = self.templates_list.currentItem()
if current_template is not None:
template_index = self.templates_list.currentRow()
name = self.templates_list.currentItem().text()
if template_index > -1 and name is not None:
template = self._naming_lib.get_template(name)
if template is not None:
if template.name == name:
valid_remove = self._naming_lib.remove_template(name)
if valid_remove:
self.templates_list.takeItem(
self.templates_list.row(
self.templates_list.currentItem()))
def _on_change_template(self, template_item):
"""
Changes the selected template
:param template_item: new QlistWidgetItem selected
"""
if not template_item or not template_item.listWidget().count() > 0:
return
template_name = template_item.text()
template = self._naming_lib.get_template(template_name)
if not template:
return
self.pattern_line.setText(template.pattern)
self.update_templates_properties_state()
def _on_edit_template_name(self, template_item):
"""
Changes name of the template
:param template_item: Renamed QListWidgetItem
"""
template_index = self.templates_list.currentRow()
template = self._naming_lib.get_template_by_index(template_index)
if template:
template.name = template_item.text()
def _on_edit_template_pattern(self, pattern_item):
"""
Changes template pattern
:param pattern_item:
"""
template_index = self.templates_list.currentRow()
template = self._naming_lib.get_template_by_index(template_index)
if template:
template.pattern = self.pattern_line.text()
self._update_template_tokens(template)
def _on_add_template_token(self, *args):
"""
Creates a new template token
:param args:
:return:
"""
load_template_token = True
if len(args) == 0:
load_template_token = False
template_token = None
if not load_template_token:
template_token = self._naming_lib.get_template_token_unique_name(
'New_Template_Token')
elif load_template_token and len(args) == 1:
template_token = args[0].name
if template_token is not None:
item = QListWidgetItem(template_token)
item.setFlags(item.flags() | Qt.ItemIsEditable)
self.template_tokens_list.addItem(item)
if len(args) == 0:
self._naming_lib.add_template_token(template_token)
if not load_template_token:
self.template_tokens_list.setCurrentItem(item)
def _on_remove_template_token(self):
"""
Remove the selected template token from the list of template tokens
:return:
"""
curr_template_token = self.template_tokens_list.currentItem()
if curr_template_token is not None:
template_token_index = self.template_tokens_list.currentRow()
name = self.template_tokens_list.currentItem().text()
if template_token_index > -1 and name is not None:
template = self._naming_lib.get_template_token(name)
if template.name == name:
valid_remove = self._naming_lib.remove_template_token(name)
if valid_remove:
self.template_tokens_list.takeItem(
self.template_tokens_list.row(
self.template_tokens_list.currentItem()))
def _on_change_template_token(self, template_token_item):
"""
Changes the selected template token
:param template_item: new QlistWidgetItem selected
"""
if not template_token_item or not template_token_item.listWidget(
).count() > 0:
return
template_tokien_name = template_token_item.text()
template_token = self._naming_lib.get_template_token(
template_tokien_name)
if not template_token:
return
self.description_templates_token_text.setText(
template_token.description)
def _on_edit_template_token_name(self, token_template_item):
"""
Changes name of the token
:param token_template_item: Renamed QListWidgetItem
:return:
"""
token_index = self.template_tokens_list.currentRow()
template_token = self._naming_lib.get_template_token_by_index(
token_index)
if template_token:
template_token.name = token_template_item.text()
def _on_edit_template_token_description(self):
"""
Edits the template token description
"""
template_token_text = self.template_tokens_list.currentItem().text()
template_token = self._naming_lib.get_template_token(
template_token_text)
if template_token:
template_token.description = self.description_templates_token_text.toPlainText(
).rstrip()
def _on_open_renamer_tool(self):
"""
Internal function that is used by toolbar to open Renamer Tool
"""
try:
tools.ToolsManager().launch_tool_by_id(
'tpDcc-tools-renamer',
naming_file=self._naming_lib.naming_file)
except Exception:
logger.warning('tpDcc-tools-renamer is not available!')
return None
def _is_renamer_tool_available(self):
"""
Returns whether or not tpRenamer tool is available or not
:return: bool
"""
try:
import tpDcc.tools.renamer
except Exception:
return False
return True
def _on_refresh(self):
"""
Internal function that is called when save button is pressed
"""
self._naming_lib.load_session()
self._init_data()
def _on_save(self):
"""
Internal function that is called when save button is pressed
"""
self._naming_lib.save_session()