class StringEditor(BaseEditor, object):
attr_type = 'str'
def __init__(self, parent=None, **kwargs):
super(StringEditor, self).__init__(parent=parent, **kwargs)
self._default_value = ""
self._clean_value = kwargs.get('clean', True)
self.value_line = QLineEdit(self)
reg_exp = QRegExp('^([a-zA-Z0-9_]+)')
self.main_layout.addWidget(self.value_line)
self.value_line.textEdited.connect(self._validate_text)
self.value_line.editingFinished.connect(self.OnValueUpdated)
self.value_line.returnPressed.connect(self.OnValueUpdated)
def get_value(self):
return str(self.value_line.text())
value = property(get_value)
def initialize_editor(self):
editor_value = self.default_value
node_values = self.values
if node_values:
if len(node_values) > 1:
pass
elif len(node_values) == 1:
if node_values[0]:
editor_value = node_values[0]
self.value_line.blockSignals(True)
self.value_line.setText(str(editor_value))
self.value_line.blockSignals(False)
def set_connected(self, conn):
if conn != self._connection:
self._connection = conn
self.value_line.setText(conn)
self.value_line.setEnabled(False)
self.value_line.setProperty('class', 'Connected')
def _validate_text(self, text):
"""
Validates the given value and update the current text
:param text: str, text to validate
"""
current_text = self.value
if self._clean_value:
cursor_pos = self.value_line.cursorPosition()
cleaned = string_utils.clean_string(text=text)
self.value_line.blockSignals(True)
self.value_line.setText(cleaned)
self.value_line.blockSignals(False)
self.value_line.setCursorPosition(cursor_pos)
class StringInputWidget(InputWidgetSingle):
"""
String data input widget
"""
def __init__(self, parent=None, **kwds):
super(StringInputWidget, self).__init__(parent=parent, **kwds)
self.le = QLineEdit(self)
self.le.setContextMenuPolicy(QtCore.Qt.NoContextMenu)
self.setWidget(self.le)
self.le.textChanged.connect(lambda val: self.dataSetCallback(val))
def blockWidgetSignals(self, bLocked):
self.le.blockSignals(bLocked)
def setWidgetValue(self, val):
self.le.setText(str(val))
class RowTableWidget(QFrame):
double_clicked = Signal(object)
def __init__(self,
auto_resize=False,
single_row_select=True,
context_menu_callback=None,
last_column_stretch=True,
has_counters=False,
parent=None
):
QFrame.__init__(self, parent)
self._has_counters = has_counters
self.model = None
self.table_view = RowTableView(auto_resize, single_row_select, context_menu_callback, last_column_stretch)
self.table_view.doubleClicked.connect(self._double_clicked)
if has_counters:
self.counters = Counters()
self.counters.button_clicked.connect(self._counter_clicked)
self.search_bar = QLineEdit()
self.search_bar.setFixedHeight(SEARCHBAR_HEIGHT)
self.search_bar.textChanged.connect(self.set_search_text)
self.search_bar.setToolTip("Search bar")
self.auto_size_button = QPushButton('<>')
self.auto_size_button.setFixedSize(SEARCHBAR_HEIGHT, SEARCHBAR_HEIGHT)
self.auto_size_button.clicked.connect(self._auto_size_clicked)
self.auto_size_button.setToolTip("Auto size")
self.status_label = QLabel(STATUS_LABEL_MESSAGE.format(0, 0))
self.status_label.setFixedWidth(STATUS_LABEL_WIDTH)
self.progress_bar = QProgressBar()
self.progress_bar.setFormat('')
layout = QGridLayout()
layout.addWidget(self.search_bar, 0, 0, 1, 3)
layout.addWidget(self.auto_size_button, 0, 3)
if has_counters:
layout.addWidget(self.counters, 1, 0, 1, 2)
layout.addWidget(self.table_view, 1, 2, 1, 2)
else:
layout.addWidget(self.table_view, 1, 0, 1, 4)
layout.addWidget(self.status_label, 2, 0)
layout.addWidget(self.progress_bar, 2, 1, 1, 3)
layout.setColumnStretch(2, 100)
self.setLayout(layout)
def set_model(self, model):
self.model = model
self.table_view.setModel(model)
model.modelReset.connect(self._set_progress_maximum)
model.modelReset.connect(self._update_status)
if self._has_counters:
model.modelReset.connect(self._update_counters)
self._update_counters()
model.progress_updated.connect(self._update_progress)
self._set_progress_maximum()
self._update_status()
self.progress_bar.setVisible(model.has_background_callback)
@property
def selected_rows(self):
return [self.model.data(index, Qt.UserRole) for index
in self.table_view.selectionModel().selectedIndexes() if index.column() == 0]
@property
def search_text(self):
return self.search_bar.text()
def set_search_text(self, text):
self.search_bar.blockSignals(True)
self.search_bar.setText(text)
self.search_bar.blockSignals(False)
self.model.set_search_text(text)
def _set_progress_maximum(self):
self.progress_bar.setMaximum(self.model.progress_max) # do better ?
def _update_progress(self, value):
self.progress_bar.setValue(value)
def _update_status(self):
self.status_label.setText(STATUS_LABEL_MESSAGE.format(
self.model.rowCount(), self.model.total_row_count
))
def _counter_clicked(self, entry, checked_buttons):
entries = [entry for entry, active in checked_buttons.items() if active]
self.model.set_search_counters(entries)
def _update_counters(self):
self.counters.set(self.model.counters)
def _auto_size_clicked(self):
with Hourglass():
self.table_view.resizeColumnsToContents()
def _double_clicked(self, index):
row = self.model.data(index, Qt.UserRole)
self.double_clicked.emit(row)
def state(self):
header_sizes = list()
header = self.table_view.horizontalHeader()
for section_index in range(header.count()):
header_sizes.append(header.sectionSize(section_index))
return {
'header_sizes': header_sizes,
'search_text': self.search_bar.text()
}
def load_state(self, state):
header = self.table_view.horizontalHeader()
for section_index, size in enumerate(state['header_sizes']):
header.resizeSection(section_index, size)
self.search_bar.setText(state['search_text'])