class Many2OneEditor(CustomEditor): """Widget for editing many 2 one relations""" arrow_down_key_pressed = QtCore.qt_signal() class CompletionsModel(QtCore.QAbstractListModel): def __init__(self, parent=None): QtCore.QAbstractListModel.__init__(self, parent) self._completions = [] def setCompletions(self, completions): self._completions = completions self.layoutChanged.emit() def data(self, index, role): return py_to_variant(self._completions[index.row()].get(role)) def rowCount(self, index=None): return len(self._completions) def columnCount(self, index=None): return 1 def __init__(self, admin=None, parent=None, editable=True, field_name='manytoone', actions=[ field_action.ClearObject(), field_action.SelectObject(), field_action.NewObject(), field_action.OpenObject() ], **kwargs): """ :param entity_admin : The Admin interface for the object on the one side of the relation """ CustomEditor.__init__(self, parent) self.setSizePolicy(QtGui.QSizePolicy.Preferred, QtGui.QSizePolicy.Fixed) self.setObjectName(field_name) self.admin = admin self.new_value = None self._entity_representation = '' self.obj = None self._last_highlighted_entity_getter = None self.layout = QtWidgets.QHBoxLayout() self.layout.setSpacing(0) self.layout.setContentsMargins(0, 0, 0, 0) # Search input self.search_input = DecoratedLineEdit(self) self.search_input.setPlaceholderText(_('Search...')) self.search_input.textEdited.connect(self.textEdited) self.search_input.set_minimum_width(20) self.search_input.arrow_down_key_pressed.connect( self.on_arrow_down_key_pressed) # suppose garbage was entered, we need to refresh the content self.search_input.editingFinished.connect( self.search_input_editing_finished) self.setFocusProxy(self.search_input) # Search Completer self.completer = QtGui.QCompleter() self.completions_model = self.CompletionsModel(self.completer) self.completer.setModel(self.completions_model) self.completer.setCaseSensitivity(Qt.CaseInsensitive) self.completer.setCompletionMode( QtGui.QCompleter.UnfilteredPopupCompletion) self.completer.activated[QtCore.QModelIndex].connect( self.completionActivated) self.completer.highlighted[QtCore.QModelIndex].connect( self.completion_highlighted) self.search_input.setCompleter(self.completer) # Setup layout self.layout.addWidget(self.search_input) self.setLayout(self.layout) self.add_actions(actions, self.layout) get_signal_handler().connect_signals(self) def set_field_attributes(self, **kwargs): super(Many2OneEditor, self).set_field_attributes(**kwargs) set_background_color_palette(self.search_input, kwargs.get('background_color')) self.search_input.setToolTip(kwargs.get('tooltip') or '') self.search_input.setEnabled(kwargs.get('editable', False)) self.update_actions() def on_arrow_down_key_pressed(self): self.arrow_down_key_pressed.emit() def textEdited(self, text): self._last_highlighted_entity_getter = None text = six.text_type(self.search_input.text()) def create_search_completion(text): return lambda: self.search_completions(text) post(create_search_completion(six.text_type(text)), self.display_search_completions) self.completer.complete() def search_completions(self, text): """Search for object that match text, to fill the list of completions :return: a list of tuples of (dict_of_object_representation, object) """ search_decorator = create_entity_search_query_decorator( self.admin, text) if search_decorator: sresult = [ self.admin.get_search_identifiers(e) for e in search_decorator(self.admin.get_query()).limit(20) ] return text, sresult return text, [] def display_search_completions(self, prefix_and_completions): assert object_thread(self) prefix, completions = prefix_and_completions self.completions_model.setCompletions(completions) self.completer.setCompletionPrefix(prefix) self.completer.complete() def completionActivated(self, index): obj = index.data(Qt.EditRole) self.set_object(variant_to_py(obj)) def completion_highlighted(self, index): obj = index.data(Qt.EditRole) self._last_highlighted_entity_getter = variant_to_py(obj) @QtCore.qt_slot(object, object) def handle_entity_update(self, sender, entity): if entity is self.get_value(): self.set_object(entity, False) @QtCore.qt_slot(object, object) def handle_entity_delete(self, sender, entity): if entity is self.get_value(): self.set_object(None, False) @QtCore.qt_slot(object, object) def handle_entity_create(self, sender, entity): if entity is self.new_value: self.new_value = None self.set_object(entity) def search_input_editing_finished(self): if self.obj is None: # Only try to 'guess' what the user meant when no entity is set # to avoid inappropriate removal of data, (eg when the user presses # Esc, editingfinished will be called as well, and we should not # overwrite the current entity set) if self._last_highlighted_entity_getter: self.set_object(self._last_highlighted_entity_getter) elif self.completions_model.rowCount() == 1: # There is only one possible option index = self.completions_model.index(0, 0) entity_getter = variant_to_py(index.data(Qt.EditRole)) self.set_object(entity_getter) self.search_input.setText(self._entity_representation or u'') def set_value(self, value): """:param value: either ValueLoading, or a function that returns None or the entity to be shown in the editor""" self._last_highlighted_entity_getter = None self.new_value = None value = CustomEditor.set_value(self, value) self.set_object(value, propagate=False) self.update_actions() def get_value(self): """:return: a function that returns the selected entity or ValueLoading or None""" value = CustomEditor.get_value(self) if value is not None: return value return self.obj @QtCore.qt_slot(tuple) def set_instance_representation(self, representation_and_propagate): """Update the gui""" (desc, propagate) = representation_and_propagate self._entity_representation = desc self.search_input.setText(desc or u'') if propagate: self.editingFinished.emit() def set_object(self, obj, propagate=True): self.obj = obj def get_instance_representation(obj, propagate): """Get a representation of the instance""" if obj is not None: return (self.admin.get_verbose_object_name(obj), propagate) return (None, propagate) post( update_wrapper( partial(get_instance_representation, obj, propagate), get_instance_representation), self.set_instance_representation) selected_object = property(fset=set_object)
def __init__(self, parent=None, editable=True, nullable=True, field_name='date', validator=DateValidator(), **kwargs): CustomEditor.__init__(self, parent) self.setSizePolicy(QtGui.QSizePolicy.Preferred, QtGui.QSizePolicy.Fixed) self.setObjectName(field_name) self.date_format = local_date_format() line_edit = DecoratedLineEdit() line_edit.setValidator(validator) line_edit.setObjectName('date_line_edit') line_edit.set_minimum_width( six.text_type( QtCore.QDate(2000, 12, 22).toString(self.date_format))) line_edit.setPlaceholderText( QtCore.QDate(2000, 1, 1).toString(self.date_format)) # The order of creation of this widgets and their parenting # seems very sensitive under windows and creates system crashes # so don't change this without extensive testing on windows special_date_menu = QtWidgets.QMenu(self) calendar_widget_action = QtWidgets.QWidgetAction(special_date_menu) self.calendar_widget = QtGui.QCalendarWidget(special_date_menu) self.calendar_widget.activated.connect(self.calendar_widget_activated) self.calendar_widget.clicked.connect(self.calendar_widget_activated) calendar_widget_action.setDefaultWidget(self.calendar_widget) self.calendar_action_trigger.connect(special_date_menu.hide) special_date_menu.addAction(calendar_widget_action) special_date_menu.addAction(_('Today')) special_date_menu.addAction(_('Far future')) self.special_date = QtWidgets.QToolButton(self) self.special_date.setIcon(self.special_date_icon.getQIcon()) self.special_date.setAutoRaise(True) self.special_date.setToolTip(_('Calendar and special dates')) self.special_date.setMenu(special_date_menu) self.special_date.setPopupMode(QtWidgets.QToolButton.InstantPopup) self.special_date.setFixedHeight(self.get_height()) self.special_date.setFocusPolicy(Qt.ClickFocus) # end of sensitive part if nullable: special_date_menu.addAction(_('Clear')) self.hlayout = QtWidgets.QHBoxLayout() self.hlayout.addWidget(line_edit) self.hlayout.addWidget(self.special_date) self.hlayout.setContentsMargins(0, 0, 0, 0) self.hlayout.setSpacing(0) self.hlayout.setAlignment(Qt.AlignRight | Qt.AlignVCenter) self.setContentsMargins(0, 0, 0, 0) self.setLayout(self.hlayout) self.minimum = datetime.date.min self.maximum = datetime.date.max self.setFocusProxy(line_edit) line_edit.editingFinished.connect(self.line_edit_finished) special_date_menu.triggered.connect(self.set_special_date)