# ============================================================================
#
# Copyright (C) 2007-2011 Conceptive Engineering bvba. All rights reserved.
# www.conceptive.be / project-camelot@conceptive.be
#
# This file is part of the Camelot Library.
#
# This file may be used under the terms of the GNU General Public
# License version 2.0 as published by the Free Software Foundation
# and appearing in the file license.txt included in the packaging of
# this file. Please review this information to ensure GNU
# General Public Licensing requirements will be met.
#
# If you are unsure which license is appropriate for your use, please
# visit www.python-camelot.com or contact project-camelot@conceptive.be
#
# This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
# WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
#
# For use of this library in commercial applications, please contact
# project-camelot@conceptive.be
#
# ============================================================================
"""Tableview"""
import functools
import logging
logger = logging.getLogger( 'camelot.view.controls.tableview' )
from PyQt4 import QtGui
from PyQt4 import QtCore
from PyQt4.QtCore import Qt
from PyQt4.QtGui import QSizePolicy
from camelot.admin.action.list_action import ListActionGuiContext
from camelot.core.utils import ugettext as _
from camelot.view.proxy.queryproxy import QueryTableProxy
from camelot.view.controls.view import AbstractView
from camelot.view.controls.user_translatable_label import UserTranslatableLabel
from camelot.view.controls.progress_dialog import ProgressDialog
from camelot.view.model_thread import post
from camelot.view.model_thread import gui_function
from camelot.view.model_thread import model_function
from camelot.view import register
from search import SimpleSearchControl
[docs]class FrozenTableWidget( QtGui.QTableView ):
"""A table widget to be used as the frozen table widget inside a table
widget."""
def __init__(self, parent, columns_frozen):
super(FrozenTableWidget, self).__init__(parent)
self.setSelectionBehavior( QtGui.QAbstractItemView.SelectRows )
self.setEditTriggers( QtGui.QAbstractItemView.SelectedClicked |
QtGui.QAbstractItemView.DoubleClicked |
QtGui.QAbstractItemView.CurrentChanged )
self._columns_frozen = columns_frozen
@QtCore.pyqtSlot(QtCore.QModelIndex, QtCore.QModelIndex)
[docs] def currentChanged(self, current, previous):
"""When the current index has changed, prevent it to jump to
a column that is not frozen"""
if current.column() >= self._columns_frozen:
current = self.model().index( current.row(), -1 )
if previous.column() >= self._columns_frozen:
previous = self.model().index( previous.row(), -1 )
super(FrozenTableWidget, self).currentChanged(current, previous)
[docs]class TableWidget( QtGui.QTableView ):
"""A widget displaying a table, to be used within a TableView. This is a
pumped up version of the QTableView widget providing extra functions such as
frozen columns. But it does not rely on the model being Camelot specific, or
a Collection Proxy.
.. attribute:: margin
margin, specified as a number of pixels, used to calculate the height of a row
in the table, the minimum row height will allow for this number of pixels below
and above the text.
"""
margin = 5
keyboard_selection_signal = QtCore.pyqtSignal()
def __init__( self, parent = None, columns_frozen = 0, lines_per_row = 1 ):
"""
:param columns_frozen: the number of columns on the left that don't scroll
:param lines_per_row: the number of lines of text that should be viewable in a single row.
"""
QtGui.QTableView.__init__( self, parent )
logger.debug( 'create TableWidget' )
self._columns_frozen = columns_frozen
self.setSelectionBehavior( QtGui.QAbstractItemView.SelectRows )
self.setEditTriggers( QtGui.QAbstractItemView.SelectedClicked |
QtGui.QAbstractItemView.DoubleClicked |
QtGui.QAbstractItemView.CurrentChanged )
self.setSizePolicy( QSizePolicy.Expanding, QSizePolicy.Expanding )
self.horizontalHeader().setClickable( True )
self._header_font_required = QtGui.QApplication.font()
self._header_font_required.setBold( True )
line_height = QtGui.QFontMetrics(QtGui.QApplication.font()).lineSpacing()
self._minimal_row_height = line_height * lines_per_row + 2*self.margin
self.verticalHeader().setDefaultSectionSize( self._minimal_row_height )
self.setHorizontalScrollMode(QtGui.QAbstractItemView.ScrollPerPixel)
self.setVerticalScrollMode(QtGui.QAbstractItemView.ScrollPerPixel)
self.horizontalHeader().sectionClicked.connect(
self.horizontal_section_clicked )
if columns_frozen:
frozen_table_view = FrozenTableWidget(self, columns_frozen)
frozen_table_view.setObjectName( 'frozen_table_view' )
frozen_table_view.verticalHeader().setDefaultSectionSize( self._minimal_row_height )
frozen_table_view.verticalHeader().hide()
frozen_table_view.horizontalHeader().setResizeMode(QtGui.QHeaderView.Fixed)
frozen_table_view.horizontalHeader().sectionClicked.connect(
self.horizontal_section_clicked )
self.horizontalHeader().sectionResized.connect( self._update_section_width )
self.verticalHeader().sectionResized.connect( self._update_section_height )
frozen_table_view.verticalScrollBar().valueChanged.connect( self.verticalScrollBar().setValue )
self.verticalScrollBar().valueChanged.connect( frozen_table_view.verticalScrollBar().setValue )
self.viewport().stackUnder(frozen_table_view)
frozen_table_view.setStyleSheet("QTableView { border: none;}")
frozen_table_view.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
frozen_table_view.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
frozen_table_view.show()
frozen_table_view.setVerticalScrollMode(QtGui.QAbstractItemView.ScrollPerPixel)
@QtCore.pyqtSlot(int, int, int)
def _update_section_width(self, logical_index, _int, new_size):
frozen_table_view = self.findChild(QtGui.QWidget, 'frozen_table_view' )
if logical_index<self._columns_frozen and frozen_table_view:
frozen_table_view.setColumnWidth( logical_index, new_size)
self._update_frozen_table()
@QtCore.pyqtSlot(int, int, int)
def _update_section_height(self, logical_index, _int, new_size):
frozen_table_view = self.findChild(QtGui.QWidget, 'frozen_table_view' )
if frozen_table_view:
frozen_table_view.setRowHeight(logical_index, new_size)
[docs] def setItemDelegate(self, item_delegate):
super(TableWidget, self).setItemDelegate(item_delegate)
frozen_table_view = self.findChild(QtGui.QWidget, 'frozen_table_view' )
if frozen_table_view:
frozen_table_view.setItemDelegate(item_delegate)
[docs] def resizeEvent(self, event):
super(TableWidget, self).resizeEvent(event)
self._update_frozen_table()
[docs] def moveCursor(self, cursorAction, modifiers):
current = super(TableWidget, self).moveCursor(cursorAction, modifiers)
frozen_table_view = self.findChild(QtGui.QWidget, 'frozen_table_view' )
if frozen_table_view:
frozen_width = 0
last_frozen = min(self._columns_frozen, self.model().columnCount())
for column in range(0, last_frozen):
frozen_width += self.columnWidth(column)
if cursorAction == QtGui.QAbstractItemView.MoveLeft and current.column() >= last_frozen and \
self.visualRect(current).topLeft().x() < frozen_width:
new_value = self.horizontalScrollBar().value() + self.visualRect(current).topLeft().x() - frozen_width
self.horizontalScrollBar().setValue(new_value)
return current
[docs] def scrollTo(self, index, hint):
if(index.column()>=self._columns_frozen):
super(TableWidget, self).scrollTo(index, hint)
[docs] def edit(self, index, trigger=None, event=None):
#
# columns in the frozen part should never be edited, because this might result
# in an editor opening below the frozen column that contains the old value
# which will be committed again when closed
#
if index.column() >= self._columns_frozen:
if trigger==None and event==None:
return super( TableWidget, self ).edit( index )
return super( TableWidget, self ).edit( index, trigger, event )
return False
@QtCore.pyqtSlot()
def _update_frozen_table(self):
frozen_table_view = self.findChild(QtGui.QWidget, 'frozen_table_view' )
if frozen_table_view:
selection_model = self.selectionModel()
if selection_model != None:
# explicitly check if there is a selection model, because
# setting the selection model to None will cause an assertion
# failure in Qt
frozen_table_view.setSelectionModel( selection_model )
last_frozen = min(self._columns_frozen, self.model().columnCount())
frozen_width = 0
for column in range(0, last_frozen):
frozen_width += self.columnWidth( column )
frozen_table_view.setColumnWidth( column,
self.columnWidth(column) )
for column in range(last_frozen, self.model().columnCount()):
frozen_table_view.setColumnHidden(column, True)
frozen_table_view.setGeometry( self.verticalHeader().width() + self.frameWidth(),
self.frameWidth(),
frozen_width,
self.viewport().height() + self.horizontalHeader().height() )
@QtCore.pyqtSlot( int )
[docs] def horizontal_section_clicked( self, logical_index ):
"""Update the sorting of the model and the header"""
header = self.horizontalHeader()
order = Qt.AscendingOrder
if not header.isSortIndicatorShown():
header.setSortIndicatorShown( True )
elif header.sortIndicatorSection()==logical_index:
# apparently, the sort order on the header is already switched
# when the section was clicked, so there is no need to reverse it
order = header.sortIndicatorOrder()
header.setSortIndicator( logical_index, order )
self.model().sort( logical_index, order )
[docs] def close_editor(self):
"""Close the active editor, this method is used to prevent assertion
failures in QT when an editor is still open in the view for a cell
that no longer exists in the model
those assertion failures only exist in QT debug builds.
"""
current_index = self.currentIndex()
if not current_index:
return
if(current_index.column()>=self._columns_frozen):
table_widget = self
else:
table_widget = self.findChild(QtGui.QWidget, 'frozen_table_view' )
if table_widget:
table_widget.closePersistentEditor( current_index )
[docs] def setModel( self, model ):
#
# An editor might be open that is no longer available for the new
# model. Not closing this editor, results in assertion failures
# in qt, resulting in segfaults in the debug build.
#
self.close_editor()
#
# Editor, closed. it should be safe to change the model
#
QtGui.QTableView.setModel( self, model )
frozen_table_view = self.findChild(QtGui.QWidget, 'frozen_table_view' )
if frozen_table_view:
model.layoutChanged.connect( self._update_frozen_table )
frozen_table_view.setModel( model )
self._update_frozen_table()
register.register( model, self )
self.selectionModel().currentChanged.connect( self.activated )
@QtCore.pyqtSlot(QtCore.QModelIndex, QtCore.QModelIndex)
[docs] def activated( self, selectedIndex, previousSelectedIndex ):
option = QtGui.QStyleOptionViewItem()
new_size = self.itemDelegate( selectedIndex ).sizeHint( option,
selectedIndex )
row = selectedIndex.row()
if previousSelectedIndex.row() >= 0:
previous_row = previousSelectedIndex.row()
self.setRowHeight( previous_row, self._minimal_row_height )
self.setRowHeight( row, max( new_size.height(),
self._minimal_row_height ) )
[docs] def keyPressEvent(self, e):
if self.hasFocus() and e.key() in (QtCore.Qt.Key_Enter, QtCore.Qt.Key_Return):
self.keyboard_selection_signal.emit()
else:
super(TableWidget, self).keyPressEvent(e)
[docs]class AdminTableWidget(TableWidget):
"""A table widget that inspects the admin class and changes the behavior
of the table as specified in the admin class"""
def __init__(self, admin, parent=None):
self._admin = admin
super(AdminTableWidget, self).__init__( columns_frozen = admin.list_columns_frozen,
lines_per_row = admin.lines_per_row,
parent=parent )
@QtCore.pyqtSlot()
[docs] def delete_selected_rows(self):
logger.debug( 'delete selected rows called' )
confirmed = True
rows = set( index.row() for index in self.selectedIndexes() )
if not rows:
return
if self._admin.get_delete_mode()=='on_confirm':
if QtGui.QMessageBox.question(self,
_('Please confirm'),
unicode(self._admin.get_delete_message(None)),
QtGui.QMessageBox.Yes,
QtGui.QMessageBox.No) == QtGui.QMessageBox.No:
confirmed = False
if confirmed:
#
# if there is an open editor on a row that will be deleted, there
# might be an assertion failure in QT, or the data of the editor
# might be pushed to the row that replaces the deleted one
#
progress_dialog = ProgressDialog(_('Removing'))
self.model().rows_removed_signal.connect( progress_dialog.finished )
self.model().exception_signal.connect( progress_dialog.exception )
self.close_editor()
self.model().remove_rows( set( rows ) )
progress_dialog.exec_()
@QtCore.pyqtSlot()
[docs] def copy_selected_rows(self):
for row in set( map( lambda x: x.row(), self.selectedIndexes() ) ):
self.model().copy_row( row )
[docs]class RowsWidget( QtGui.QLabel ):
"""Widget that is part of the header widget, displaying the number of rows
in the table view"""
_number_of_rows_font = QtGui.QApplication.font()
def __init__( self, parent ):
QtGui.QLabel.__init__( self, parent )
self.setFont( self._number_of_rows_font )
[docs]class HeaderWidget( QtGui.QWidget ):
"""HeaderWidget for a tableview, containing the title, the search widget,
and the number of rows in the table"""
search_widget = SimpleSearchControl
rows_widget = RowsWidget
filters_changed_signal = QtCore.pyqtSignal()
_title_font = QtGui.QApplication.font()
_title_font.setBold( True )
def __init__( self, parent, admin ):
QtGui.QWidget.__init__( self, parent )
self._admin = admin
layout = QtGui.QVBoxLayout()
widget_layout = QtGui.QHBoxLayout()
search = self.search_widget( self )
search.expand_search_options_signal.connect(
self.expand_search_options )
title = UserTranslatableLabel( admin.get_verbose_name_plural(),
self )
title.setFont( self._title_font )
widget_layout.addWidget( title )
widget_layout.addWidget( search )
if self.rows_widget:
self.number_of_rows = self.rows_widget( self )
widget_layout.addWidget( self.number_of_rows )
else:
self.number_of_rows = None
layout.addLayout( widget_layout )
self._expanded_filters_created = False
self._expanded_search = QtGui.QWidget()
self._expanded_search.hide()
layout.addWidget(self._expanded_search)
self.setLayout( layout )
self.setSizePolicy( QSizePolicy.Minimum, QSizePolicy.Fixed )
self.setNumberOfRows( 0 )
self.search = search
def _fill_expanded_search_options(self, columns):
"""Given the columns in the table view, present the user
with more options to filter rows in the table
:param columns: a list of tuples with field names and attributes
"""
from camelot.view.controls.filter_operator import FilterOperator
layout = QtGui.QHBoxLayout()
layout.setSpacing( 2 )
layout.setMargin( 2 )
for i, (field, attributes) in enumerate(columns):
if 'operators' in attributes and attributes['operators']:
box = QtGui.QGroupBox()
box_layout = QtGui.QVBoxLayout()
box_layout.setMargin( 2 )
widget = FilterOperator( self._admin.entity,
field, attributes,
box )
box_layout.addWidget( widget )
box.setLayout( box_layout )
widget.filter_changed_signal.connect( self._filter_changed )
layout.addWidget( box )
layout.addStretch()
self._expanded_search.setLayout( layout )
self._expanded_filters_created = True
def _filter_changed(self):
self.filters_changed_signal.emit()
[docs] def decorate_query(self, query):
"""Apply expanded filters on the query"""
if self._expanded_filters_created:
for i in range(self._expanded_search.layout().count()):
box = self._expanded_search.layout().itemAt(i).widget()
if box:
widget = box.layout().itemAt(0).widget()
if widget:
query = widget.decorate_query(query)
return query
@QtCore.pyqtSlot()
[docs] def expand_search_options(self):
if self._expanded_search.isHidden():
if not self._expanded_filters_created:
post( self._admin.get_expanded_search_fields,
self._fill_expanded_search_options )
self._expanded_search.show()
else:
self._expanded_search.hide()
@gui_function
[docs] def setNumberOfRows( self, rows ):
if self.number_of_rows:
self.number_of_rows.setNumberOfRows( rows )
[docs]class SplitterHandle( QtGui.QSplitterHandle ):
"""Custom implementation of QSplitterHandle to provide more functions,
such as hiding a widget by clicking the handle"""
def __init__ (self, orientation, splitter, widget_to_hide = None):
super(SplitterHandle, self).__init__ (orientation, splitter)
self.setToolTip('Click to close')
self._widget_to_hide = widget_to_hide
[docs] def mousePressEvent(self, event):
splitter = self.splitter()
print splitter.count()
splitter.widget( splitter.count() - 1 ).hide()
[docs]class Splitter(QtGui.QSplitter):
"""Custom implementation of QSplitter to use the custom SplitterHandle"""
[docs] def createHandle(self):
return SplitterHandle( self.orientation(), self, self._widget_to_hide )
[docs]class TableView( AbstractView ):
"""A generic tableview widget that puts together some other widgets. The behaviour of this class and
the resulting interface can be tuned by specifying specific class attributes which define the underlying
widgets used ::
class MovieRentalTableView(TableView):
title_format = 'Grand overview of recent movie rentals'
The attributes that can be specified are :
.. attribute:: header_widget
The widget class to be used as a header in the table view::
header_widget = HeaderWidget
.. attribute:: table_widget
The widget class used to display a table within the table view ::
table_widget = TableWidget
.. attribute:: title_format
A string used to format the title of the view ::
title_format = '%(verbose_name_plural)s'
.. attribute:: table_model
A class implementing QAbstractTableModel that will be used as a model for the table view ::
table_model = QueryTableProxy
- emits the row_selected signal when a row has been selected
"""
header_widget = HeaderWidget
AdminTableWidget = AdminTableWidget
#
# The proxy class to use
#
table_model = QueryTableProxy
#
# Format to use as the window title
#
title_format = '%(verbose_name_plural)s'
row_selected_signal = QtCore.pyqtSignal(int)
def __init__( self, admin, search_text = None, parent = None ):
super(TableView, self).__init__( parent )
self.admin = admin
post( self.get_title, self.change_title )
widget_layout = QtGui.QVBoxLayout()
if self.header_widget:
self.header = self.header_widget( self, admin )
widget_layout.addWidget( self.header )
self.header.search.search_signal.connect( self.startSearch )
self.header.search.cancel_signal.connect( self.cancelSearch )
self.header.search.on_arrow_down_signal.connect(self.focusTable)
if search_text:
self.header.search.search( search_text )
else:
self.header = None
widget_layout.setSpacing( 0 )
widget_layout.setContentsMargins(0, 0, 0, 0)
splitter = QtGui.QSplitter( self )
splitter.setObjectName('splitter')
widget_layout.addWidget( splitter )
table_widget = QtGui.QWidget( self )
filters_widget = QtGui.QWidget( self )
self.table_layout = QtGui.QVBoxLayout()
self.table_layout.setSpacing( 0 )
self.table_layout.setContentsMargins(0, 0, 0, 0)
self.table = None
self.filters_layout = QtGui.QVBoxLayout()
self.filters_layout.setSpacing( 0 )
self.filters_layout.setContentsMargins(0, 0, 0, 0)
self.actions = None
table_widget.setLayout( self.table_layout )
filters_widget.setLayout( self.filters_layout )
#filters_widget.hide()
self.set_admin( admin )
splitter.addWidget( table_widget )
splitter.addWidget( filters_widget )
self.setLayout( widget_layout )
self.search_filter = lambda q: q
shortcut = QtGui.QShortcut(QtGui.QKeySequence(QtGui.QKeySequence.Find), self)
shortcut.activated.connect( self.activate_search )
if self.header_widget:
self.header.filters_changed_signal.connect( self.rebuild_query )
# give the table widget focus to prevent the header and its search control to
# receive default focus, as this would prevent the displaying of 'Search...' in the
# search control, but this conflicts with the MDI, resulting in the window not
# being active and the menus not to work properly
#table_widget.setFocus( QtCore.Qt.OtherFocusReason )
#self.setFocusProxy(table_widget)
#self.setFocus( QtCore.Qt.OtherFocusReason )
post( self.admin.get_subclass_tree, self.setSubclassTree )
@QtCore.pyqtSlot()
@model_function
[docs] def get_title( self ):
return self.title_format % {'verbose_name_plural':self.admin.get_verbose_name_plural()}
@QtCore.pyqtSlot(object)
@gui_function
[docs] def setSubclassTree( self, subclasses ):
if len( subclasses ) > 0:
from inheritance import SubclassTree
splitter = self.findChild(QtGui.QWidget, 'splitter' )
class_tree = SubclassTree( self.admin, splitter )
splitter.insertWidget( 0, class_tree )
class_tree.subclass_clicked_signal.connect( self.set_admin )
@QtCore.pyqtSlot(int)
[docs] def sectionClicked( self, section ):
"""emits a row_selected signal"""
self.admin.list_action.gui_run( self.gui_context )
[docs] def copy_selected_rows( self ):
"""Copy the selected rows in this tableview"""
logger.debug( 'delete selected rows called' )
if self.table and self.table.model():
for row in set( map( lambda x: x.row(), self.table.selectedIndexes() ) ):
self.table.model().copy_row( row )
[docs] def create_table_model( self, admin ):
"""Create a table model for the given admin interface"""
return self.table_model( admin,
None,
admin.get_columns )
@QtCore.pyqtSlot( object )
@gui_function
[docs] def set_admin( self, admin ):
"""Switch to a different subclass, where admin is the admin object of the
subclass"""
logger.debug('set_admin called')
self.admin = admin
if self.table:
self.table.model().layoutChanged.disconnect( self.tableLayoutChanged )
self.table_layout.removeWidget(self.table)
self.table.deleteLater()
self.table.model().deleteLater()
splitter = self.findChild( QtGui.QWidget, 'splitter' )
self.table = self.AdminTableWidget( self.admin, splitter )
self.table.setObjectName('AdminTableWidget')
new_model = self.create_table_model( admin )
self.table.setModel( new_model )
self.table.verticalHeader().sectionClicked.connect( self.sectionClicked )
self.table.keyboard_selection_signal.connect(self.on_keyboard_selection_signal)
self.table.model().layoutChanged.connect( self.tableLayoutChanged )
self.tableLayoutChanged()
self.table_layout.insertWidget( 1, self.table )
self.gui_context = ListActionGuiContext()
self.gui_context.admin = self.admin
self.gui_context.item_view = self.table
def get_filters_and_actions():
return ( admin.get_filters(), admin.get_list_actions() )
post( get_filters_and_actions, self.set_filters_and_actions )
@QtCore.pyqtSlot()
[docs] def on_keyboard_selection_signal(self):
self.sectionClicked( self.table.currentIndex().row() )
@QtCore.pyqtSlot()
@gui_function
[docs] def tableLayoutChanged( self ):
logger.debug('tableLayoutChanged')
model = self.table.model()
if self.header:
self.header.setNumberOfRows( model.rowCount() )
item_delegate = model.getItemDelegate()
if item_delegate:
self.table.setItemDelegate( item_delegate )
for i in range( model.columnCount() ):
self.table.setColumnWidth( i, model.headerData( i, Qt.Horizontal, Qt.SizeHintRole ).toSize().width() )
[docs] def deleteSelectedRows( self ):
"""delete the selected rows in this tableview"""
self.table.delete_selected_rows()
@gui_function
[docs] def newRow( self ):
"""Create a new row in the tableview"""
from camelot.view.workspace import show_top_level
form = self.admin.create_new_view( related_collection_proxy=self.get_model(),
parent = None )
show_top_level( form, self )
[docs] def closeEvent( self, event ):
"""reimplements close event"""
logger.debug( 'tableview closed' )
event.accept()
[docs] def makeImport(self):
pass
# for row in data:
# o = self.admin.entity()
# #For example, setattr(x, 'foobar', 123) is equivalent to x.foobar = 123
# # if you want to import all attributes, you must link them to other objects
# #for example: a movie has a director, this isn't a primitive like a string
# # but a object fetched from the db
# setattr(o, object_attributes[0], row[0])
# name = row[2].split( ' ' ) #director
# o.short_description = "korte beschrijving"
# o.genre = ""
# from sqlalchemy.orm.session import Session
# Session.object_session(o).flush([o])
#
# post( makeImport )
@gui_function
[docs] def selectedTableIndexes( self ):
"""returns a list of selected rows indexes"""
return self.table.selectedIndexes()
[docs] def getColumns( self ):
"""return the columns to be displayed in the table view"""
return self.admin.get_columns()
[docs] def getData( self ):
"""generator for data queried by table model"""
for d in self.table.model().getData():
yield d
[docs] def getTitle( self ):
"""return the name of the entity managed by the admin attribute"""
return self.admin.get_verbose_name()
[docs] def viewLast( self ):
"""selects last row"""
self.selectTableRow( self.table.model().rowCount() - 1 )
[docs] def viewNext( self ):
"""selects next row"""
first = self.selectedTableIndexes()[0]
next = ( first.row() + 1 ) % self.table.model().rowCount()
self.selectTableRow( next )
[docs] def viewPrevious( self ):
"""selects previous row"""
first = self.selectedTableIndexes()[0]
prev = ( first.row() - 1 ) % self.table.model().rowCount()
self.selectTableRow( prev )
@QtCore.pyqtSlot(object)
def _set_query(self, query_getter):
if isinstance(self.table.model(), QueryTableProxy):
self.table.model().setQuery(query_getter)
self.table.clearSelection()
@QtCore.pyqtSlot()
@QtCore.pyqtSlot()
[docs] def rebuild_query( self ):
"""resets the table model query"""
from filterlist import FilterList
def rebuild_query():
query = self.admin.get_query()
# a table view is not required to have a header
if self.header:
query = self.header.decorate_query(query)
filters = self.findChild(FilterList, 'filters')
if filters:
query = filters.decorate_query( query )
if self.search_filter:
query = self.search_filter( query )
query_getter = lambda:query
return query_getter
post( rebuild_query, self._set_query )
@QtCore.pyqtSlot(str)
[docs] def startSearch( self, text ):
"""rebuilds query based on filtering text"""
from camelot.view.search import create_entity_search_query_decorator
logger.debug( 'search %s' % text )
self.search_filter = create_entity_search_query_decorator( self.admin, unicode(text) )
self.rebuild_query()
@QtCore.pyqtSlot()
[docs] def cancelSearch( self ):
"""resets search filtering to default"""
logger.debug( 'cancel search' )
self.search_filter = lambda q: q
self.rebuild_query()
@gui_function
[docs] def get_selection_getter(self):
""":return: a function that when called return an iterable with all the
objects corresponding to the selected rows in the table."""
selected_rows = set( map( lambda x: x.row(), self.table.selectedIndexes() ) )
def selection_getter(table, selected_rows):
selection = []
model = table.model()
for row in selected_rows:
selection.append( model._get_object(row) )
return selection
return functools.partial( selection_getter, self.table, selected_rows )
@gui_function
[docs] def get_collection_getter(self):
""":return: a list with all the objects corresponding to the rows in the table
"""
def get_collection(table):
return table.model().get_collection()
return functools.partial( get_collection, self.table )
@QtCore.pyqtSlot(object)
@gui_function
[docs] def set_filters_and_actions( self, filters_and_actions ):
"""sets filters for the tableview"""
filters, actions = filters_and_actions
from camelot.view.controls.filterlist import FilterList
from camelot.view.controls.actionsbox import ActionsBox
logger.debug( 'setting filters for tableview' )
filters_widget = self.findChild(FilterList, 'filters')
actions_widget = self.findChild(ActionsBox, 'actions')
if filters_widget:
filters_widget.filters_changed_signal.disconnect( self.rebuild_query )
self.filters_layout.removeWidget(filters_widget)
filters_widget.deleteLater()
if actions_widget:
self.filters_layout.removeWidget(actions_widget)
actions_widget.deleteLater()
if filters:
splitter = self.findChild( QtGui.QWidget, 'splitter' )
filters_widget = FilterList( filters, parent=splitter )
filters_widget.setObjectName('filters')
self.filters_layout.addWidget( filters_widget )
filters_widget.filters_changed_signal.connect( self.rebuild_query )
#
# filters might have default values, so we can only build the queries now
#
self.rebuild_query()
if actions:
actions_widget = ActionsBox( parent = self,
gui_context = self.gui_context )
actions_widget.setObjectName( 'actions' )
actions_widget.set_actions( actions )
self.filters_layout.addWidget( actions_widget )
[docs] def to_html( self ):
"""generates html of the table"""
if self.table and self.table.model():
query_getter = self.table.model().get_query_getter()
table = [[getattr( row, col[0] ) for col in self.admin.get_columns()]
for row in query_getter().all()]
context = {
'title': self.admin.get_verbose_name_plural(),
'table': table,
'columns': [field_attributes['name'] for _field, field_attributes in self.admin.get_columns()],
}
from camelot.view.templates import loader
from jinja2 import Environment
env = Environment( loader = loader )
tp = env.get_template( 'table_view.html' )
return tp.render( context )
[docs] def importFromFile( self ):
""""import data : the data will be imported in the activeMdiChild """
logger.info( 'call import method' )
from camelot.view.wizard.importwizard import ImportWizard
wizard = ImportWizard(self, self.admin)
wizard.exec_()
@QtCore.pyqtSlot()
[docs] def focusTable(self):
if self.table and self.table.model().rowCount() > 0:
self.table.setFocus()
self.table.selectRow(0)