class DateEditor(CustomEditor):
"""Widget for editing date values"""
calendar_action_trigger = QtCore.pyqtSignal()
special_date_icon = Icon('tango/16x16/apps/office-calendar.png')
def __init__(self, parent = None,
editable = True,
nullable = True,
field_name = 'date',
**kwargs):
CustomEditor.__init__(self, parent)
self.setObjectName( field_name )
self.date_format = local_date_format()
self.line_edit = DecoratedLineEdit()
self.line_edit.set_minimum_width( len( self.date_format ) )
self.line_edit.set_background_text( 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 = QtGui.QMenu(self)
calendar_widget_action = QtGui.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 = QtGui.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(QtGui.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 = QtGui.QHBoxLayout()
self.hlayout.addWidget(self.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(self.line_edit)
self.line_edit.editingFinished.connect( self.line_edit_finished )
self.line_edit.textEdited.connect(self.text_edited)
special_date_menu.triggered.connect(self.set_special_date)
def calendar_widget_activated(self, date):
self.calendar_action_trigger.emit()
self.set_value(date)
self.editingFinished.emit()
self.line_edit.setFocus()
def line_edit_finished(self):
self.setProperty( 'value', QtCore.QVariant( self.get_value() ) )
self.valueChanged.emit()
self.editingFinished.emit()
def focusOutEvent(self, event):
# explicitely set value on focus out to format the date in case
# it was entered unformatted
value = self.get_value()
self.set_value( value )
self.editingFinished.emit()
def set_value(self, value):
value = CustomEditor.set_value(self, value)
self.setProperty( 'value', QtCore.QVariant( value ) )
if value:
qdate = QtCore.QDate(value)
formatted_date = qdate.toString(self.date_format)
self.line_edit.set_user_input(formatted_date)
self.calendar_widget.setSelectedDate(qdate)
else:
self.line_edit.set_user_input('')
self.valueChanged.emit()
def text_edited(self, text ):
try:
date_from_string( self.line_edit.user_input() )
self.line_edit.set_valid(True)
self.valueChanged.emit()
except ParsingError:
self.line_edit.set_valid(False)
def get_value(self):
try:
value = date_from_string( self.line_edit.user_input() )
except ParsingError:
value = None
return CustomEditor.get_value(self) or value
def set_field_attributes(self, editable = True,
background_color = None,
tooltip = None, **kwargs):
self.set_enabled(editable)
self.set_background_color(background_color)
self.line_edit.setToolTip(unicode(tooltip or ''))
def set_background_color(self, background_color):
set_background_color_palette( self.line_edit, background_color )
def set_enabled(self, editable=True):
self.line_edit.setEnabled(editable)
if editable:
self.special_date.show()
else:
self.special_date.hide()
def set_special_date(self, action):
if action.text().compare(_('Today')) == 0:
self.set_value(datetime.date.today())
elif action.text().compare(_('Far future')) == 0:
self.set_value(datetime.date( year = 2400, month = 12, day = 31 ))
elif action.text().compare(_('Clear')) == 0:
self.set_value(None)
self.line_edit.setFocus()
self.editingFinished.emit()
class SimpleSearchControl(QtGui.QWidget):
"""A control that displays a single text field in which search keywords can
be typed
emits a search and a cancel signal if the user starts or cancels the search
"""
expand_search_options_signal = QtCore.pyqtSignal()
cancel_signal = QtCore.pyqtSignal()
search_signal = QtCore.pyqtSignal(str)
def __init__(self, parent):
QtGui.QWidget.__init__(self, parent)
layout = QtGui.QHBoxLayout()
layout.setSpacing(0)
layout.setMargin(3)
# Search button
self.search_button = QtGui.QToolButton()
icon = Icon('tango/16x16/actions/system-search.png').getQIcon()
self.search_button.setIcon(icon)
self.search_button.setIconSize(QtCore.QSize(14, 14))
self.search_button.setAutoRaise(True)
self.search_button.setToolTip(_('Expand or collapse search options'))
self.search_button.clicked.connect(self.emit_expand_search_options)
# Search input
from camelot.view.controls.decorated_line_edit import DecoratedLineEdit
self.search_input = DecoratedLineEdit(self)
self.search_input.set_background_text(_('Search...'))
self.search_input.setToolTip(_('type words to search for'))
#self.search_input.setStyleSheet('QLineEdit{ border-radius: 0.25em;}')
self.search_input.returnPressed.connect(self.emit_search)
self.search_input.textEdited.connect(self.emit_search)
self.setFocusProxy(self.search_input)
# Cancel button
self.cancel_button = QtGui.QToolButton()
icon = Icon('tango/16x16/actions/edit-clear.png').getQIcon()
self.cancel_button.setIcon(icon)
self.cancel_button.setIconSize(QtCore.QSize(14, 14))
self.cancel_button.setAutoRaise(True)
self.cancel_button.clicked.connect(self.emit_cancel)
# Setup layout
layout.addWidget(self.search_button)
layout.addWidget(self.search_input)
layout.addWidget(self.cancel_button)
self.setLayout(layout)
def search(self, search_text):
"""Start searching for search_text"""
self.search_input.setText(search_text)
self.emit_search()
@QtCore.pyqtSlot()
def emit_expand_search_options(self):
self.expand_search_options_signal.emit()
@QtCore.pyqtSlot()
@QtCore.pyqtSlot(str)
def emit_search(self, str=''):
text = unicode(self.search_input.user_input())
self.search_signal.emit(text)
@QtCore.pyqtSlot()
def emit_cancel(self):
self.search_input.setText('')
self.cancel_signal.emit()
class DateEditor(CustomEditor):
"""Widget for editing date values"""
calendar_action_trigger = QtCore.pyqtSignal()
special_date_icon = Icon('tango/16x16/apps/office-calendar.png')
def __init__(self,
parent=None,
editable=True,
nullable=True,
**kwargs):
CustomEditor.__init__(self, parent)
self.date_format = local_date_format()
self.line_edit = DecoratedLineEdit()
self.line_edit.set_background_text( 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 = QtGui.QMenu(self)
calendar_widget_action = QtGui.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 = QtGui.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(QtGui.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 = QtGui.QHBoxLayout()
self.hlayout.addWidget(self.line_edit)
self.hlayout.addWidget(self.special_date)
self.hlayout.setContentsMargins(0, 0, 0, 0)
self.hlayout.setMargin(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(self.line_edit)
self.line_edit.editingFinished.connect( self.line_edit_finished )
self.line_edit.textEdited.connect(self.text_edited)
special_date_menu.triggered.connect(self.set_special_date)
def calendar_widget_activated(self, date):
self.calendar_action_trigger.emit()
self.set_value(date)
self.editingFinished.emit()
def line_edit_finished(self):
self.setProperty( 'value', QtCore.QVariant( self.get_value() ) )
self.valueChanged.emit()
self.editingFinished.emit()
def focusOutEvent(self, event):
self.setProperty( 'value', QtCore.QVariant( self.get_value() ) )
self.valueChanged.emit()
self.editingFinished.emit()
def set_value(self, value):
value = CustomEditor.set_value(self, value)
self.setProperty( 'value', QtCore.QVariant( value ) )
if value:
qdate = QtCore.QDate(value)
formatted_date = qdate.toString(self.date_format)
self.line_edit.set_user_input(formatted_date)
self.calendar_widget.setSelectedDate(qdate)
else:
self.line_edit.set_user_input('')
self.valueChanged.emit()
def text_edited(self, text ):
try:
date_from_string( self.line_edit.user_input() )
self.line_edit.set_valid(True)
self.valueChanged.emit()
except ParsingError:
self.line_edit.set_valid(False)
def get_value(self):
try:
value = date_from_string( self.line_edit.user_input() )
except ParsingError:
value = None
return CustomEditor.get_value(self) or value
def set_enabled(self, editable=True):
self.line_edit.setEnabled(editable)
if editable:
self.special_date.show()
else:
self.special_date.hide()
def set_special_date(self, action):
if action.text().compare(_('Today')) == 0:
self.set_value(datetime.date.today())
elif action.text().compare(_('Far future')) == 0:
self.set_value(datetime.date( year = 2400, month = 12, day = 31 ))
elif action.text().compare(_('Clear')) == 0:
self.set_value(None)
self.editingFinished.emit()
class SimpleSearchControl(QtGui.QWidget):
"""A control that displays a single text field in which search keywords can
be typed
emits a search and a cancel signal if the user starts or cancels the search
"""
expand_search_options_signal = QtCore.pyqtSignal()
cancel_signal = QtCore.pyqtSignal()
search_signal = QtCore.pyqtSignal(str)
def __init__(self, parent):
QtGui.QWidget.__init__(self, parent)
layout = QtGui.QHBoxLayout()
layout.setSpacing(0)
layout.setMargin(3)
# Search button
self.search_button = QtGui.QToolButton()
icon = Icon('tango/16x16/actions/system-search.png').getQIcon()
self.search_button.setIcon(icon)
self.search_button.setIconSize(QtCore.QSize(14, 14))
self.search_button.setAutoRaise(True)
self.search_button.setToolTip(_('Expand or collapse search options'))
self.search_button.clicked.connect( self.emit_expand_search_options )
# Search input
from camelot.view.controls.decorated_line_edit import DecoratedLineEdit
self.search_input = DecoratedLineEdit(self)
self.search_input.set_background_text(_('Search...'))
self.search_input.setToolTip(_('type words to search for'))
#self.search_input.setStyleSheet('QLineEdit{ border-radius: 0.25em;}')
self.search_input.returnPressed.connect( self.emit_search )
self.search_input.textEdited.connect( self.emit_search )
self.setFocusProxy( self.search_input )
# Cancel button
self.cancel_button = QtGui.QToolButton()
icon = Icon('tango/16x16/actions/edit-clear.png').getQIcon()
self.cancel_button.setIcon(icon)
self.cancel_button.setIconSize(QtCore.QSize(14, 14))
self.cancel_button.setAutoRaise(True)
self.cancel_button.clicked.connect( self.emit_cancel )
# Setup layout
layout.addWidget(self.search_button)
layout.addWidget(self.search_input)
layout.addWidget(self.cancel_button)
self.setLayout(layout)
def search(self, search_text):
"""Start searching for search_text"""
self.search_input.setText(search_text)
self.emit_search()
@QtCore.pyqtSlot()
def emit_expand_search_options(self):
self.expand_search_options_signal.emit()
@QtCore.pyqtSlot()
@QtCore.pyqtSlot(str)
def emit_search(self, str=''):
text = unicode(self.search_input.user_input())
self.search_signal.emit( text )
@QtCore.pyqtSlot()
def emit_cancel(self):
self.search_input.setText('')
self.cancel_signal.emit()
class SimpleSearchControl(AbstractSearchWidget):
"""A control that displays a single text field in which search keywords can
be typed
emits a search and a cancel signal if the user starts or cancels the search
"""
def __init__(self, parent):
QtGui.QWidget.__init__(self, parent)
layout = QtGui.QHBoxLayout()
layout.setSpacing(0)
layout.setContentsMargins(3, 3, 3, 3)
#
# The search timer reduced the number of search signals that are
# emitted, by waiting for the next keystroke before emitting the
# search signal
#
timer = QtCore.QTimer( self )
timer.setInterval( 300 )
timer.setSingleShot( True )
timer.setObjectName( 'timer' )
timer.timeout.connect( self.emit_search )
# Search button
self.search_button = QtGui.QToolButton()
icon = Icon('tango/16x16/actions/system-search.png').getQIcon()
self.search_button.setIcon(icon)
self.search_button.setIconSize(QtCore.QSize(14, 14))
self.search_button.setAutoRaise(True)
self.search_button.setToolTip(_('Expand or collapse search options'))
self.search_button.clicked.connect( self.emit_expand_search_options )
# Search input
from camelot.view.controls.decorated_line_edit import DecoratedLineEdit
self.search_input = DecoratedLineEdit(self)
self.search_input.set_background_text(_('Search...'))
#self.search_input.setStyleSheet('QLineEdit{ border-radius: 0.25em;}')
self.search_input.returnPressed.connect( self.emit_search )
self.search_input.textEdited.connect( self._start_search_timer )
self.search_input.arrow_down_key_pressed.connect(self.on_arrow_down_key_pressed)
self.setFocusProxy( self.search_input )
# Cancel button
self.cancel_button = QtGui.QToolButton()
icon = Icon('tango/16x16/actions/edit-clear.png').getQIcon()
self.cancel_button.setIcon(icon)
self.cancel_button.setIconSize(QtCore.QSize(14, 14))
self.cancel_button.setAutoRaise(True)
self.cancel_button.clicked.connect( self.emit_cancel )
# Setup layout
layout.addWidget(self.search_button)
layout.addWidget(self.search_input)
layout.addWidget(self.cancel_button)
self.setLayout(layout)
def search(self, search_text):
"""Start searching for search_text"""
self.search_input.setText(search_text)
self.emit_search()
@QtCore.pyqtSlot()
@QtCore.pyqtSlot(str)
def _start_search_timer(self, str=''):
timer = self.findChild( QtCore.QTimer, 'timer' )
if timer:
timer.stop()
timer.start()
@QtCore.pyqtSlot()
def emit_expand_search_options(self):
self.expand_search_options_signal.emit()
@QtCore.pyqtSlot()
@QtCore.pyqtSlot(str)
def emit_search(self, str=''):
timer = self.findChild( QtCore.QTimer, 'timer' )
if timer:
timer.stop()
text = unicode(self.search_input.user_input())
self.search_signal.emit( text )
@QtCore.pyqtSlot()
def emit_cancel(self):
timer = self.findChild( QtCore.QTimer, 'timer' )
if timer:
timer.stop()
self.search_input.setText('')
self.cancel_signal.emit()
@QtCore.pyqtSlot()
def on_arrow_down_key_pressed(self):
self.on_arrow_down_signal.emit()
class SimpleSearchControl(AbstractSearchWidget):
"""A control that displays a single text field in which search keywords can
be typed
emits a search and a cancel signal if the user starts or cancels the search
"""
def __init__(self, parent):
QtGui.QWidget.__init__(self, parent)
layout = QtGui.QHBoxLayout()
layout.setSpacing(0)
layout.setContentsMargins(3, 3, 3, 3)
#
# The search timer reduced the number of search signals that are
# emitted, by waiting for the next keystroke before emitting the
# search signal
#
timer = QtCore.QTimer(self)
timer.setInterval(300)
timer.setSingleShot(True)
timer.setObjectName('timer')
timer.timeout.connect(self.emit_search)
# Search button
self.search_button = QtGui.QToolButton()
icon = Icon('tango/16x16/actions/system-search.png').getQIcon()
self.search_button.setIcon(icon)
self.search_button.setIconSize(QtCore.QSize(14, 14))
self.search_button.setAutoRaise(True)
self.search_button.setToolTip(_('Expand or collapse search options'))
self.search_button.clicked.connect(self.emit_expand_search_options)
# Search input
from camelot.view.controls.decorated_line_edit import DecoratedLineEdit
self.search_input = DecoratedLineEdit(self)
self.search_input.set_background_text(_('Search...'))
#self.search_input.setStyleSheet('QLineEdit{ border-radius: 0.25em;}')
self.search_input.returnPressed.connect(self.emit_search)
self.search_input.textEdited.connect(self._start_search_timer)
self.search_input.arrow_down_key_pressed.connect(
self.on_arrow_down_key_pressed)
self.setFocusProxy(self.search_input)
# Cancel button
self.cancel_button = QtGui.QToolButton()
icon = Icon('tango/16x16/actions/edit-clear.png').getQIcon()
self.cancel_button.setIcon(icon)
self.cancel_button.setIconSize(QtCore.QSize(14, 14))
self.cancel_button.setAutoRaise(True)
self.cancel_button.clicked.connect(self.emit_cancel)
# Setup layout
layout.addWidget(self.search_button)
layout.addWidget(self.search_input)
layout.addWidget(self.cancel_button)
self.setLayout(layout)
def search(self, search_text):
"""Start searching for search_text"""
self.search_input.setText(search_text)
self.emit_search()
@QtCore.pyqtSlot()
@QtCore.pyqtSlot(str)
def _start_search_timer(self, str=''):
timer = self.findChild(QtCore.QTimer, 'timer')
if timer:
timer.stop()
timer.start()
@QtCore.pyqtSlot()
def emit_expand_search_options(self):
self.expand_search_options_signal.emit()
@QtCore.pyqtSlot()
@QtCore.pyqtSlot(str)
def emit_search(self, str=''):
timer = self.findChild(QtCore.QTimer, 'timer')
if timer:
timer.stop()
text = unicode(self.search_input.user_input())
self.search_signal.emit(text)
@QtCore.pyqtSlot()
def emit_cancel(self):
timer = self.findChild(QtCore.QTimer, 'timer')
if timer:
timer.stop()
self.search_input.setText('')
self.cancel_signal.emit()
@QtCore.pyqtSlot()
def on_arrow_down_key_pressed(self):
self.on_arrow_down_signal.emit()
class Many2OneEditor(CustomEditor, AbstractManyToOneEditor):
"""Widget for editing many 2 one relations"""
new_icon = Icon('tango/16x16/actions/document-new.png')
search_icon = Icon('tango/16x16/actions/system-search.png')
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):
if role == Qt.DisplayRole:
return QtCore.QVariant(self._completions[index.row()][0])
elif role == Qt.EditRole:
return QtCore.QVariant(self._completions[index.row()][1])
return QtCore.QVariant()
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, **kwargs):
""":param entity_admin : The Admin interface for the object on the one
side of the relation
"""
CustomEditor.__init__(self, parent)
self.admin = admin
self.entity_set = False
self._editable = editable
self._entity_representation = ''
self.entity_instance_getter = None
self._last_highlighted_entity_getter = None
self.layout = QtGui.QHBoxLayout()
self.layout.setSpacing(0)
self.layout.setMargin(0)
# Search button
self.search_button = QtGui.QToolButton()
self.search_button.setAutoRaise(True)
self.search_button.setFocusPolicy(Qt.ClickFocus)
self.search_button.setFixedHeight(self.get_height())
self.search_button.clicked.connect(self.searchButtonClicked)
self.search_button.setIcon(
Icon('tango/16x16/actions/edit-clear.png').getQIcon()
)
self.search_button.setToolTip(unicode(_('clear')))
# Open button
self.open_button = QtGui.QToolButton()
self.open_button.setAutoRaise(True)
self.open_button.setFocusPolicy(Qt.ClickFocus)
self.open_button.setFixedHeight(self.get_height())
self.open_button.clicked.connect(self.openButtonClicked)
self.open_button.setIcon( self.new_icon.getQIcon() )
self.open_button.setToolTip(unicode(_('new')))
# Search input
self.search_input = DecoratedLineEdit(self)
self.search_input.set_background_text(_('Search...'))
self.search_input.textEdited.connect(self.textEdited)
# 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.connect(self.completionActivated)
#self.completer.highlighted.connect(self.completion_highlighted)
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.layout.addWidget(self.search_button)
self.layout.addWidget(self.open_button)
self.setLayout(self.layout)
def set_field_attributes(self, editable=True, background_color=None, **kw):
self.set_editable(editable)
self.set_background_color(background_color)
def set_editable(self, editable):
self._editable = editable
self.search_input.setEnabled(editable)
self.search_button.setEnabled(editable)
def textEdited(self, text):
self._last_highlighted_entity_getter = None
text = self.search_input.user_input()
def create_search_completion(text):
return lambda: self.search_completions(text)
post(
create_search_completion(unicode(text)),
self.display_search_completions
)
self.completer.complete()
@model_function
def search_completions(self, text):
"""Search for object that match text, to fill the list of completions
:return: a list of tuples of (object_representation, object_getter)
"""
search_decorator = create_entity_search_query_decorator(
self.admin, text
)
if search_decorator:
sresult = [
(unicode(e), create_constant_function(e))
for e in search_decorator(self.admin.entity.query).limit(20)
]
return text, sresult
return text, []
@gui_function
def display_search_completions(self, prefix_and_completions):
prefix, completions = prefix_and_completions
self.completions_model.setCompletions(completions)
self.completer.setCompletionPrefix(prefix)
self.completer.complete()
def completionActivated(self, index):
object_getter = index.data(Qt.EditRole)
self.setEntity(variant_to_pyobject(object_getter))
def completion_highlighted(self, index ):
object_getter = index.data(Qt.EditRole)
pyob = variant_to_pyobject(object_getter)
self._last_highlighted_entity_getter = pyob
def openButtonClicked(self):
if self.entity_set:
return self.createFormView()
else:
return self.createNew()
def returnPressed(self):
if not self.entity_set:
self.createSelectView()
def searchButtonClicked(self):
if self.entity_set:
self.setEntity(lambda:None)
else:
self.createSelectView()
def trashButtonClicked(self):
self.setEntity(lambda:None)
@gui_function
def createNew(self):
@model_function
def get_has_subclasses():
return len(self.admin.get_subclass_tree())
post(get_has_subclasses, self.show_new_view)
@gui_function
def show_new_view(self, has_subclasses):
from camelot.view.workspace import show_top_level
selected = QtGui.QDialog.Accepted
admin = self.admin
if has_subclasses:
from camelot.view.controls.inheritance import SubclassDialog
select_subclass = SubclassDialog(self, self.admin)
select_subclass.setWindowTitle(_('select'))
selected = select_subclass.exec_()
admin = select_subclass.selected_subclass
if selected:
form = admin.create_new_view()
form.entity_created_signal.connect(self.selectEntity)
# @todo: dirty trick to keep reference
#self.__new_form = form
show_top_level( form, self )
def createFormView(self):
if self.entity_instance_getter:
def get_admin_and_title():
obj = self.entity_instance_getter()
admin = self.admin.get_related_entity_admin(obj.__class__)
return admin, ''
post(get_admin_and_title, self.show_form_view)
def show_form_view(self, admin_and_title):
from camelot.view.workspace import show_top_level
admin, title = admin_and_title
def create_collection_getter(instance_getter):
return lambda:[instance_getter()]
from camelot.view.proxy.collection_proxy import CollectionProxy
model = CollectionProxy(
admin,
create_collection_getter(self.entity_instance_getter),
admin.get_fields
)
model.dataChanged.connect(self.dataChanged)
form = admin.create_form_view(title, model, 0)
# @todo : dirty trick to keep reference
#self.__form = form
show_top_level( form, self )
def dataChanged(self, index1, index2):
self.setEntity(self.entity_instance_getter, False)
def search_input_editing_finished(self):
if not self.entity_set:
# 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.setEntity(self._last_highlighted_entity_getter)
elif not self.entity_set and self.completions_model.rowCount()==1:
# There is only one possible option
index = self.completions_model.index(0,0)
entity_getter = variant_to_pyobject(index.data(Qt.EditRole))
self.setEntity(entity_getter)
self.search_input.set_user_input(self._entity_representation)
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
value = CustomEditor.set_value(self, value)
if value:
self.setEntity(value, propagate = False)
def get_value(self):
""":return: a function that returns the selected entity or ValueLoading
or None"""
value = CustomEditor.get_value(self)
if not value:
value = self.entity_instance_getter
return value
@QtCore.pyqtSlot(tuple)
def set_instance_represenation(self, representation_and_propagate):
"""Update the gui"""
((desc, pk), propagate) = representation_and_propagate
self._entity_representation = desc
self.search_input.set_user_input(desc)
if pk != False:
self.open_button.setIcon(
Icon('tango/16x16/places/folder.png').getQIcon()
)
self.open_button.setToolTip(unicode(_('open')))
self.open_button.setEnabled(True)
self.search_button.setIcon(
Icon('tango/16x16/actions/edit-clear.png').getQIcon()
)
self.search_button.setToolTip(unicode(_('clear')))
self.entity_set = True
else:
self.open_button.setIcon( self.new_icon.getQIcon() )
self.open_button.setToolTip(unicode(_('new')))
self.open_button.setEnabled(self._editable)
self.search_button.setIcon( self.search_icon.getQIcon() )
self.search_button.setToolTip(_('Search'))
self.entity_set = False
if propagate:
self.editingFinished.emit()
def setEntity(self, entity_instance_getter, propagate=True):
def create_instance_getter(entity_instance):
return lambda:entity_instance
def get_instance_represenation():
"""Get a representation of the instance
:return: (unicode, pk) its unicode representation and its primary
key or ('', False) if the instance was None"""
entity = entity_instance_getter()
self.entity_instance_getter = create_instance_getter(entity)
if entity and hasattr(entity, 'id'):
return ((unicode(entity), entity.id), propagate)
elif entity:
return ((unicode(entity), False), propagate)
return ((None, False), propagate)
post(get_instance_represenation, self.set_instance_represenation)
def selectEntity(self, entity_instance_getter):
self.setEntity(entity_instance_getter)
class Many2OneEditor(CustomEditor, AbstractManyToOneEditor):
"""Widget for editing many 2 one relations"""
new_icon = Icon('tango/16x16/actions/document-new.png')
search_icon = Icon('tango/16x16/actions/system-search.png')
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):
if role == Qt.DisplayRole:
return QtCore.QVariant(self._completions[index.row()][0])
elif role == Qt.EditRole:
return QtCore.QVariant(self._completions[index.row()][1])
return QtCore.QVariant()
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, **kwargs):
""":param entity_admin : The Admin interface for the object on the one
side of the relation
"""
CustomEditor.__init__(self, parent)
self.admin = admin
self.entity_set = False
self._editable = editable
self._entity_representation = ''
self.entity_instance_getter = None
self._last_highlighted_entity_getter = None
self.layout = QtGui.QHBoxLayout()
self.layout.setSpacing(0)
self.layout.setMargin(0)
# Search button
self.search_button = QtGui.QToolButton()
self.search_button.setAutoRaise(True)
self.search_button.setFocusPolicy(Qt.ClickFocus)
self.search_button.setFixedHeight(self.get_height())
self.search_button.clicked.connect(self.searchButtonClicked)
self.search_button.setIcon(
Icon('tango/16x16/actions/edit-clear.png').getQIcon())
self.search_button.setToolTip(unicode(_('clear')))
# Open button
self.open_button = QtGui.QToolButton()
self.open_button.setAutoRaise(True)
self.open_button.setFocusPolicy(Qt.ClickFocus)
self.open_button.setFixedHeight(self.get_height())
self.open_button.clicked.connect(self.openButtonClicked)
self.open_button.setIcon(self.new_icon.getQIcon())
self.open_button.setToolTip(unicode(_('new')))
# Search input
self.search_input = DecoratedLineEdit(self)
self.search_input.set_background_text(_('Search...'))
self.search_input.textEdited.connect(self.textEdited)
# 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.connect(self.completionActivated)
#self.completer.highlighted.connect(self.completion_highlighted)
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.layout.addWidget(self.search_button)
self.layout.addWidget(self.open_button)
self.setLayout(self.layout)
def set_field_attributes(self, editable=True, background_color=None, **kw):
self.set_editable(editable)
self.set_background_color(background_color)
def set_editable(self, editable):
self._editable = editable
self.search_input.setEnabled(editable)
self.search_button.setEnabled(editable)
def textEdited(self, text):
self._last_highlighted_entity_getter = None
text = self.search_input.user_input()
def create_search_completion(text):
return lambda: self.search_completions(text)
post(create_search_completion(unicode(text)),
self.display_search_completions)
self.completer.complete()
@model_function
def search_completions(self, text):
"""Search for object that match text, to fill the list of completions
:return: a list of tuples of (object_representation, object_getter)
"""
search_decorator = create_entity_search_query_decorator(
self.admin, text)
if search_decorator:
sresult = [
(unicode(e), create_constant_function(e))
for e in search_decorator(self.admin.entity.query).limit(20)
]
return text, sresult
return text, []
@gui_function
def display_search_completions(self, prefix_and_completions):
prefix, completions = prefix_and_completions
self.completions_model.setCompletions(completions)
self.completer.setCompletionPrefix(prefix)
self.completer.complete()
def completionActivated(self, index):
object_getter = index.data(Qt.EditRole)
self.setEntity(variant_to_pyobject(object_getter))
def completion_highlighted(self, index):
object_getter = index.data(Qt.EditRole)
pyob = variant_to_pyobject(object_getter)
self._last_highlighted_entity_getter = pyob
def openButtonClicked(self):
if self.entity_set:
return self.createFormView()
else:
return self.createNew()
def returnPressed(self):
if not self.entity_set:
self.createSelectView()
def searchButtonClicked(self):
if self.entity_set:
self.setEntity(lambda: None)
else:
self.createSelectView()
def trashButtonClicked(self):
self.setEntity(lambda: None)
@gui_function
def createNew(self):
@model_function
def get_has_subclasses():
return len(self.admin.get_subclass_tree())
post(get_has_subclasses, self.show_new_view)
@gui_function
def show_new_view(self, has_subclasses):
from camelot.view.workspace import show_top_level
selected = QtGui.QDialog.Accepted
admin = self.admin
if has_subclasses:
from camelot.view.controls.inheritance import SubclassDialog
select_subclass = SubclassDialog(self, self.admin)
select_subclass.setWindowTitle(_('select'))
selected = select_subclass.exec_()
admin = select_subclass.selected_subclass
if selected:
form = admin.create_new_view()
form.entity_created_signal.connect(self.selectEntity)
# @todo: dirty trick to keep reference
#self.__new_form = form
show_top_level(form, self)
def createFormView(self):
if self.entity_instance_getter:
def get_admin_and_title():
obj = self.entity_instance_getter()
admin = self.admin.get_related_entity_admin(obj.__class__)
return admin, ''
post(get_admin_and_title, self.show_form_view)
def show_form_view(self, admin_and_title):
from camelot.view.workspace import show_top_level
admin, title = admin_and_title
def create_collection_getter(instance_getter):
return lambda: [instance_getter()]
from camelot.view.proxy.collection_proxy import CollectionProxy
model = CollectionProxy(
admin, create_collection_getter(self.entity_instance_getter),
admin.get_fields)
model.dataChanged.connect(self.dataChanged)
form = admin.create_form_view(title, model, 0)
# @todo : dirty trick to keep reference
#self.__form = form
show_top_level(form, self)
def dataChanged(self, index1, index2):
self.setEntity(self.entity_instance_getter, False)
def search_input_editing_finished(self):
if not self.entity_set:
# 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.setEntity(self._last_highlighted_entity_getter)
elif not self.entity_set and self.completions_model.rowCount(
) == 1:
# There is only one possible option
index = self.completions_model.index(0, 0)
entity_getter = variant_to_pyobject(index.data(Qt.EditRole))
self.setEntity(entity_getter)
self.search_input.set_user_input(self._entity_representation)
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
value = CustomEditor.set_value(self, value)
if value:
self.setEntity(value, propagate=False)
def get_value(self):
""":return: a function that returns the selected entity or ValueLoading
or None"""
value = CustomEditor.get_value(self)
if not value:
value = self.entity_instance_getter
return value
@QtCore.pyqtSlot(tuple)
def set_instance_represenation(self, representation_and_propagate):
"""Update the gui"""
((desc, pk), propagate) = representation_and_propagate
self._entity_representation = desc
self.search_input.set_user_input(desc)
if pk != False:
self.open_button.setIcon(
Icon('tango/16x16/places/folder.png').getQIcon())
self.open_button.setToolTip(unicode(_('open')))
self.open_button.setEnabled(True)
self.search_button.setIcon(
Icon('tango/16x16/actions/edit-clear.png').getQIcon())
self.search_button.setToolTip(unicode(_('clear')))
self.entity_set = True
else:
self.open_button.setIcon(self.new_icon.getQIcon())
self.open_button.setToolTip(unicode(_('new')))
self.open_button.setEnabled(self._editable)
self.search_button.setIcon(self.search_icon.getQIcon())
self.search_button.setToolTip(_('Search'))
self.entity_set = False
if propagate:
self.editingFinished.emit()
def setEntity(self, entity_instance_getter, propagate=True):
def create_instance_getter(entity_instance):
return lambda: entity_instance
def get_instance_represenation():
"""Get a representation of the instance
:return: (unicode, pk) its unicode representation and its primary
key or ('', False) if the instance was None"""
entity = entity_instance_getter()
self.entity_instance_getter = create_instance_getter(entity)
if entity and hasattr(entity, 'id'):
return ((unicode(entity), entity.id), propagate)
elif entity:
return ((unicode(entity), False), propagate)
return ((None, False), propagate)
post(get_instance_represenation, self.set_instance_represenation)
def selectEntity(self, entity_instance_getter):
self.setEntity(entity_instance_getter)
