Warning

This document is for an in-development version of Galaxy. You can alternatively view this page in the latest release if it exists or view the top of the latest release's documentation.

Source code for galaxy.web.framework.helpers.grids

import logging
import math
from collections import OrderedDict
from json import dumps, loads

from markupsafe import escape
from sqlalchemy.sql.expression import and_, false, func, null, or_, true

from galaxy.model.item_attrs import get_foreign_key, UsesAnnotations, UsesItemRatings
from galaxy.util import restore_text, sanitize_text, unicodify
from galaxy.web.framework import decorators, url_for


log = logging.getLogger(__name__)


[docs]class Grid: """ Specifies the content and format of a grid (data table). """ title = "" model_class = None show_item_checkboxes = False use_hide_message = True global_actions = [] columns = [] operations = [] standard_filters = [] # Any columns that are filterable (either standard or advanced) should have a default value set in the default filter. default_filter = {} default_sort_key = None use_paging = False num_rows_per_page = 25 num_page_links = 10 # Set preference names. cur_filter_pref_name = ".filter" cur_sort_key_pref_name = ".sort_key" legend = None info_text = None
[docs] def __init__(self): # Determine if any multiple row operations are defined self.has_multiple_item_operations = False for operation in self.operations: if operation.allow_multiple: self.has_multiple_item_operations = True break # If a column does not have a model class, set the column's model class # to be the grid's model class. for column in self.columns: if not column.model_class: column.model_class = self.model_class
def __call__(self, trans, **kwargs): # Get basics. # FIXME: pretty sure this is only here to pass along, can likely be eliminated status = kwargs.get('status', None) message = kwargs.get('message', None) # Build a base filter and sort key that is the combination of the saved state and defaults. # Saved state takes preference over defaults. base_filter = {} if self.default_filter: # default_filter is a dictionary that provides a default set of filters based on the grid's columns. base_filter = self.default_filter.copy() base_sort_key = self.default_sort_key # Build initial query query = self.build_initial_query(trans, **kwargs) query = self.apply_query_filter(trans, query, **kwargs) # Maintain sort state in generated urls extra_url_args = {} # Determine whether use_default_filter flag is set. use_default_filter_str = kwargs.get('use_default_filter') use_default_filter = False if use_default_filter_str: use_default_filter = (use_default_filter_str.lower() == 'true') # Process filtering arguments to (a) build a query that represents the filter and (b) build a # dictionary that denotes the current filter. cur_filter_dict = {} for column in self.columns: if column.key: # Get the filter criterion for the column. Precedence is (a) if using default filter, only look there; otherwise, # (b) look in kwargs; and (c) look in base filter. column_filter = None if use_default_filter: if self.default_filter: column_filter = self.default_filter.get(column.key) elif "f-" + column.model_class.__name__ + ".%s" % column.key in kwargs: # Queries that include table joins cannot guarantee unique column names. This problem is # handled by setting the column_filter value to <TableName>.<ColumnName>. column_filter = kwargs.get("f-" + column.model_class.__name__ + ".%s" % column.key) elif "f-" + column.key in kwargs: column_filter = kwargs.get("f-" + column.key) elif column.key in base_filter: column_filter = base_filter.get(column.key) # Method (1) combines a mix of strings and lists of strings into a single string and (2) attempts to de-jsonify all strings. def loads_recurse(item): decoded_list = [] if isinstance(item, str): try: # Not clear what we're decoding, so recurse to ensure that we catch everything. decoded_item = loads(item) if isinstance(decoded_item, list): decoded_list = loads_recurse(decoded_item) else: decoded_list = [str(decoded_item)] except ValueError: decoded_list = [str(item)] elif isinstance(item, list): for element in item: a_list = loads_recurse(element) decoded_list = decoded_list + a_list return decoded_list # If column filter found, apply it. if column_filter is not None: # TextColumns may have a mix of json and strings. if isinstance(column, TextColumn): column_filter = loads_recurse(column_filter) if len(column_filter) == 1: column_filter = column_filter[0] # Interpret ',' as a separator for multiple terms. if isinstance(column_filter, str) and column_filter.find(',') != -1: column_filter = column_filter.split(',') # Check if filter is empty if isinstance(column_filter, list): # Remove empty strings from filter list column_filter = [x for x in column_filter if x != ''] if len(column_filter) == 0: continue elif isinstance(column_filter, str): # If filter criterion is empty, do nothing. if column_filter == '': continue # Update query. query = column.filter(trans, trans.user, query, column_filter) # Upate current filter dict. # Column filters are rendered in various places, sanitize them all here. cur_filter_dict[column.key] = sanitize_text(column_filter) # Carry filter along to newly generated urls; make sure filter is a string so # that we can encode to UTF-8 and thus handle user input to filters. if isinstance(column_filter, list): # Filter is a list; process each item. extra_url_args["f-" + column.key] = dumps(column_filter) else: # Process singleton filter. extra_url_args["f-" + column.key] = column_filter # Process sort arguments. sort_key = None if 'sort' in kwargs: sort_key = kwargs['sort'] elif base_sort_key: sort_key = base_sort_key if sort_key: ascending = not(sort_key.startswith("-")) # Queries that include table joins cannot guarantee unique column names. This problem is # handled by setting the column_filter value to <TableName>.<ColumnName>. table_name = None if sort_key.find('.') > -1: a_list = sort_key.split('.') if ascending: table_name = a_list[0] else: table_name = a_list[0][1:] column_name = a_list[1] elif ascending: column_name = sort_key else: column_name = sort_key[1:] # Sort key is a column key. for column in self.columns: if column.key and column.key.find('.') > -1: column_key = column.key.split('.')[1] else: column_key = column.key if (table_name is None or table_name == column.model_class.__name__) and column_key == column_name: query = column.sort(trans, query, ascending, column_name=column_name) break extra_url_args['sort'] = sort_key # There might be a current row current_item = self.get_current_item(trans, **kwargs) # Process page number. num_pages = None total_row_count_query = query # query without limit applied to get total number of rows. if self.use_paging: if 'page' in kwargs: if kwargs['page'] == 'all': page_num = 0 else: page_num = int(kwargs['page']) else: page_num = 1 if page_num == 0: num_pages = 1 page_num = 1 else: query = query.limit(self.num_rows_per_page).offset((page_num - 1) * self.num_rows_per_page) else: # Defaults. page_num = 1 # There are some places in grid templates where it's useful for a grid # to have its current filter. self.cur_filter_dict = cur_filter_dict # Log grid view. context = str(self.__class__.__name__) params = cur_filter_dict.copy() params['sort'] = sort_key # Render grid. def url(*args, **kwargs): route_name = kwargs.pop('__route_name__', None) # Only include sort/filter arguments if not linking to another # page. This is a bit of a hack. if 'action' in kwargs: new_kwargs = dict() else: new_kwargs = dict(extra_url_args) # Extend new_kwargs with first argument if found if len(args) > 0: new_kwargs.update(args[0]) new_kwargs.update(kwargs) # We need to encode item ids if 'id' in new_kwargs: id = new_kwargs['id'] if isinstance(id, list): new_kwargs['id'] = [trans.security.encode_id(i) for i in id] else: new_kwargs['id'] = trans.security.encode_id(id) # The url_for invocation *must* include a controller and action. if 'controller' not in new_kwargs: new_kwargs['controller'] = trans.controller if 'action' not in new_kwargs: new_kwargs['action'] = trans.action if route_name: return url_for(route_name, **new_kwargs) return url_for(**new_kwargs) self.use_panels = (kwargs.get('use_panels', False) in [True, 'True', 'true']) self.advanced_search = (kwargs.get('advanced_search', False) in [True, 'True', 'true']) # Currently, filling the template returns a str object; this requires decoding the string into a # unicode object within mako templates. What probably should be done is to return the template as # utf-8 unicode; however, this would require encoding the object as utf-8 before returning the grid # results via a controller method, which is require substantial changes. Hence, for now, return grid # as str. grid_config = { 'title' : self.title, 'title_id' : getattr(self, "title_id", None), 'url_base' : trans.request.path_url, 'async_ops' : [], 'categorical_filters' : {}, 'filters' : cur_filter_dict, 'sort_key' : sort_key, 'show_item_checkboxes' : self.show_item_checkboxes or kwargs.get('show_item_checkboxes', '') in ['True', 'true'], 'cur_page_num' : page_num, 'num_page_links' : self.num_page_links, 'status' : status, 'message' : restore_text(message), 'global_actions' : [], 'operations' : [], 'items' : [], 'columns' : [], 'model_class' : str(self.model_class), 'use_paging' : self.use_paging, 'legend' : self.legend, 'current_item_id' : False, 'use_hide_message' : self.use_hide_message, 'default_filter_dict' : self.default_filter, 'advanced_search' : self.advanced_search, 'info_text' : self.info_text, 'url' : url(dict()), 'refresh_frames' : kwargs.get('refresh_frames', []) } if current_item: grid_config['current_item_id'] = current_item.id for column in self.columns: extra = '' if column.sortable: if sort_key.endswith(column.key): if not sort_key.startswith("-"): extra = "&darr;" else: extra = "&uarr;" grid_config['columns'].append({ 'key' : column.key, 'visible' : column.visible, 'nowrap' : column.nowrap, 'attach_popup' : column.attach_popup, 'label_id_prefix' : column.label_id_prefix, 'sortable' : column.sortable, 'label' : column.label, 'filterable' : column.filterable, 'delayed' : column.delayed, 'is_text' : isinstance(column, TextColumn), 'extra' : extra }) for operation in self.operations: grid_config['operations'].append({ 'allow_multiple' : operation.allow_multiple, 'allow_popup' : operation.allow_popup, 'target' : operation.target, 'label' : operation.label, 'confirm' : operation.confirm, 'href' : url(**operation.url_args) if isinstance(operation.url_args, dict) else None, 'global_operation' : False }) if operation.allow_multiple: grid_config['show_item_checkboxes'] = True if operation.global_operation: grid_config['global_operation'] = url(** (operation.global_operation())) for action in self.global_actions: grid_config['global_actions'].append({ 'url_args' : url(**action.url_args), 'label' : action.label, 'target' : action.target }) for operation in [op for op in self.operations if op.async_compatible]: grid_config['async_ops'].append(operation.label.lower()) for column in self.columns: if column.filterable is not None and not isinstance(column, TextColumn): grid_config['categorical_filters'][column.key] = {filter.label: filter.args for filter in column.get_accepted_filters()} for i, item in enumerate(query): item_dict = { 'id' : item.id, 'encode_id' : trans.security.encode_id(item.id), 'link' : [], 'operation_config' : {}, 'column_config' : {} } for column in self.columns: if column.visible: link = column.get_link(trans, self, item) if link: link = url(**link) else: link = None target = column.target value = unicodify(column.get_value(trans, self, item)) if value: value = value.replace('/', '//') item_dict['column_config'][column.label] = { 'link' : link, 'value' : value, 'target' : target } for operation in self.operations: item_dict['operation_config'][operation.label] = { 'allowed' : operation.allowed(item), 'url_args' : url(**operation.get_url_args(item)), 'target' : operation.target } grid_config['items'].append(item_dict) if self.use_paging and num_pages is None: # TODO: it would be better to just return this as None, render, and fire # off a second request for this count I think. total_num_rows = total_row_count_query.count() num_pages = int(math.ceil(float(total_num_rows) / self.num_rows_per_page)) grid_config["num_pages"] = num_pages trans.log_action(trans.get_user(), "grid.view", context, params) return grid_config
[docs] def get_ids(self, **kwargs): id = [] if 'id' in kwargs: id = kwargs['id'] # Coerce ids to list if not isinstance(id, list): id = id.split(",") # Ensure ids are integers try: id = list(map(int, id)) except Exception: decorators.error("Invalid id") return id
# ---- Override these ----------------------------------------------------
[docs] def handle_operation(self, trans, operation, ids, **kwargs): pass
[docs] def get_current_item(self, trans, **kwargs): return None
[docs] def build_initial_query(self, trans, **kwargs): return trans.sa_session.query(self.model_class)
[docs] def apply_query_filter(self, trans, query, **kwargs): # Applies a database filter that holds for all items in the grid. # (gvk) Is this method necessary? Why not simply build the entire query, # including applying filters in the build_initial_query() method? return query
[docs]class GridColumn:
[docs] def __init__(self, label, key=None, model_class=None, method=None, format=None, link=None, attach_popup=False, visible=True, nowrap=False, # Valid values for filterable are ['standard', 'advanced', None] filterable=None, sortable=True, label_id_prefix=None, target=None, delayed=False): """Create a grid column.""" self.label = label self.key = key self.model_class = model_class self.method = method self.format = format self.link = link self.target = target self.nowrap = nowrap self.attach_popup = attach_popup self.visible = visible self.filterable = filterable self.delayed = delayed # Column must have a key to be sortable. self.sortable = (self.key is not None and sortable) self.label_id_prefix = label_id_prefix or ''
[docs] def get_value(self, trans, grid, item): if self.method: value = getattr(grid, self.method)(trans, item) elif self.key and hasattr(item, self.key): value = getattr(item, self.key) else: value = None if self.format: value = self.format(value) return escape(unicodify(value))
[docs] def filter(self, trans, user, query, column_filter): """ Modify query to reflect the column filter. """ if column_filter == "All": pass if column_filter == "True": query = query.filter_by(**{self.key: True}) elif column_filter == "False": query = query.filter_by(**{self.key: False}) return query
[docs] def get_accepted_filters(self): """ Returns a list of accepted filters for this column. """ accepted_filters_vals = ["False", "True", "All"] accepted_filters = [] for val in accepted_filters_vals: args = {self.key: val} accepted_filters.append(GridColumnFilter(val, args)) return accepted_filters
[docs] def sort(self, trans, query, ascending, column_name=None): """Sort query using this column.""" if column_name is None: column_name = self.key if ascending: query = query.order_by(self.model_class.table.c.get(column_name).asc()) else: query = query.order_by(self.model_class.table.c.get(column_name).desc()) return query
[docs]class ReverseSortColumn(GridColumn): """ Column that reverses sorting; this is useful when the natural sort is descending. """
[docs] def sort(self, trans, query, ascending, column_name=None): return GridColumn.sort(self, trans, query, (not ascending), column_name=column_name)
[docs]class TextColumn(GridColumn): """ Generic column that employs freetext and, hence, supports freetext, case-independent filtering. """
[docs] def filter(self, trans, user, query, column_filter): """ Modify query to filter using free text, case independence. """ if column_filter == "All": pass elif column_filter: query = query.filter(self.get_filter(trans, user, column_filter)) return query
[docs] def get_filter(self, trans, user, column_filter): """ Returns a SQLAlchemy criterion derived from column_filter. """ if isinstance(column_filter, str): return self.get_single_filter(user, column_filter) elif isinstance(column_filter, list): clause_list = [] for filter in column_filter: clause_list.append(self.get_single_filter(user, filter)) return and_(*clause_list)
[docs] def get_single_filter(self, user, a_filter): """ Returns a SQLAlchemy criterion derived for a single filter. Single filter is the most basic filter--usually a string--and cannot be a list. """ # Queries that include table joins cannot guarantee that table column names will be # unique, so check to see if a_filter is of type <TableName>.<ColumnName>. if self.key.find('.') > -1: a_key = self.key.split('.')[1] else: a_key = self.key model_class_key_field = getattr(self.model_class, a_key) return func.lower(model_class_key_field).like("%" + a_filter.lower() + "%")
[docs] def sort(self, trans, query, ascending, column_name=None): """Sort column using case-insensitive alphabetical sorting.""" if column_name is None: column_name = self.key if ascending: query = query.order_by(func.lower(self.model_class.table.c.get(column_name)).asc()) else: query = query.order_by(func.lower(self.model_class.table.c.get(column_name)).desc()) return query
[docs]class DateTimeColumn(TextColumn):
[docs] def sort(self, trans, query, ascending, column_name=None): """Sort query using this column.""" return GridColumn.sort(self, trans, query, ascending, column_name=column_name)
[docs]class BooleanColumn(TextColumn):
[docs] def sort(self, trans, query, ascending, column_name=None): """Sort query using this column.""" return GridColumn.sort(self, trans, query, ascending, column_name=column_name)
[docs] def get_single_filter(self, user, a_filter): if self.key.find('.') > -1: a_key = self.key.split('.')[1] else: a_key = self.key model_class_key_field = getattr(self.model_class, a_key) return model_class_key_field == a_filter
[docs]class IntegerColumn(TextColumn): """ Integer column that employs freetext, but checks that the text is an integer, so support filtering on integer values. IMPORTANT NOTE: grids that use this column type should not include the column in the cols_to_filter list of MulticolFilterColumn ( i.e., searching on this column type should not be performed in the grid's standard search - it won't throw exceptions, but it also will not find what you're looking for ). Grids that search on this column should use 'filterable="advanced"' so that searching is only performed in the advanced search component, restricting the search to the specific column. This is useful for searching on object ids or other integer columns. See the JobIdColumn column in the SpecifiedDateListGrid class in the jobs controller of the reports webapp for an example. """
[docs] def get_single_filter(self, user, a_filter): model_class_key_field = getattr(self.model_class, self.key) assert int(a_filter), "The search entry must be an integer" return model_class_key_field == int(a_filter)
[docs] def sort(self, trans, query, ascending, column_name=None): """Sort query using this column.""" return GridColumn.sort(self, trans, query, ascending, column_name=column_name)
[docs]class CommunityRatingColumn(GridColumn, UsesItemRatings): """ Column that displays community ratings for an item. """
[docs] def get_value(self, trans, grid, item): if not hasattr(item, "average_rating"): # No prefetched column property, generate it on the fly. ave_item_rating, num_ratings = self.get_ave_item_rating_data(trans.sa_session, item, webapp_model=trans.model) else: ave_item_rating = item.average_rating num_ratings = 2 # just used for pluralization if not ave_item_rating: ave_item_rating = 0 return trans.fill_template("tool_shed_rating.mako", ave_item_rating=ave_item_rating, num_ratings=num_ratings, item_id=trans.security.encode_id(item.id))
[docs] def sort(self, trans, query, ascending, column_name=None): # Get the columns that connect item's table and item's rating association table. item_rating_assoc_class = getattr(trans.model, '%sRatingAssociation' % self.model_class.__name__) foreign_key = get_foreign_key(item_rating_assoc_class, self.model_class) fk_col = foreign_key.parent referent_col = foreign_key.get_referent(self.model_class.table) # Do sorting using a subquery. # Subquery to get average rating for each item. ave_rating_subquery = trans.sa_session.query(fk_col, func.avg(item_rating_assoc_class.table.c.rating).label('avg_rating')) \ .group_by(fk_col).subquery() # Integrate subquery into main query. query = query.outerjoin((ave_rating_subquery, referent_col == ave_rating_subquery.columns[fk_col.name])) # Sort using subquery results; use coalesce to avoid null values. if not ascending: # TODO: for now, reverse sorting b/c first sort is ascending, and that should be the natural sort. query = query.order_by(func.coalesce(ave_rating_subquery.c.avg_rating, 0).asc()) else: query = query.order_by(func.coalesce(ave_rating_subquery.c.avg_rating, 0).desc()) return query
[docs]class OwnerAnnotationColumn(TextColumn, UsesAnnotations): """ Column that displays and filters item owner's annotations. """
[docs] def __init__(self, col_name, key, model_class=None, model_annotation_association_class=None, filterable=None): GridColumn.__init__(self, col_name, key=key, model_class=model_class, filterable=filterable) self.sortable = False self.model_annotation_association_class = model_annotation_association_class
[docs] def get_value(self, trans, grid, item): """ Returns first 150 characters of annotation. """ annotation = self.get_item_annotation_str(trans.sa_session, item.user, item) if annotation: ann_snippet = annotation[:155] if len(annotation) > 155: ann_snippet = ann_snippet[:ann_snippet.rfind(' ')] ann_snippet += "..." else: ann_snippet = "" return escape(ann_snippet)
[docs] def get_single_filter(self, user, a_filter): """ Filter by annotation and annotation owner. """ return self.model_class.annotations.any( and_(func.lower(self.model_annotation_association_class.annotation).like("%" + a_filter.lower() + "%"), # TODO: not sure why, to filter by owner's annotations, we have to do this rather than # 'self.model_class.user==self.model_annotation_association_class.user' self.model_annotation_association_class.table.c.user_id == self.model_class.table.c.user_id))
[docs]class CommunityTagsColumn(TextColumn): """ Column that supports community tags. """
[docs] def __init__(self, col_name, key, model_class=None, model_tag_association_class=None, filterable=None, grid_name=None): GridColumn.__init__(self, col_name, key=key, model_class=model_class, nowrap=True, filterable=filterable, sortable=False) self.model_tag_association_class = model_tag_association_class # Column-specific attributes. self.grid_name = grid_name
[docs] def get_value(self, trans, grid, item): return trans.fill_template("/tagging_common.mako", tag_type="community", trans=trans, user=trans.get_user(), tagged_item=item, elt_context=self.grid_name, tag_click_fn="add_tag_to_grid_filter", use_toggle_link=True)
[docs] def filter(self, trans, user, query, column_filter): """ Modify query to filter model_class by tag. Multiple filters are ANDed. """ if column_filter == "All": pass elif column_filter: query = query.filter(self.get_filter(trans, user, column_filter)) return query
[docs] def get_filter(self, trans, user, column_filter): # Parse filter to extract multiple tags. if isinstance(column_filter, list): # Collapse list of tags into a single string; this is redundant but effective. TODO: fix this by iterating over tags. column_filter = ",".join(column_filter) raw_tags = trans.app.tag_handler.parse_tags(column_filter) clause_list = [] for name, value in raw_tags: if name: # Filter by all tags. clause_list.append(self.model_class.tags.any(func.lower(self.model_tag_association_class.user_tname).like("%" + name.lower() + "%"))) if value: # Filter by all values. clause_list.append(self.model_class.tags.any(func.lower(self.model_tag_association_class.user_value).like("%" + value.lower() + "%"))) return and_(*clause_list)
[docs]class IndividualTagsColumn(CommunityTagsColumn): """ Column that supports individual tags. """
[docs] def get_value(self, trans, grid, item): return trans.fill_template("/tagging_common.mako", tag_type="individual", user=trans.user, tagged_item=item, elt_context=self.grid_name, tag_click_fn="add_tag_to_grid_filter", use_toggle_link=True)
[docs] def get_filter(self, trans, user, column_filter): # Parse filter to extract multiple tags. if isinstance(column_filter, list): # Collapse list of tags into a single string; this is redundant but effective. TODO: fix this by iterating over tags. column_filter = ",".join(column_filter) raw_tags = trans.app.tag_handler.parse_tags(column_filter) clause_list = [] for name, value in raw_tags: if name: # Filter by individual's tag names. clause_list.append(self.model_class.tags.any(and_(func.lower(self.model_tag_association_class.user_tname).like("%" + name.lower() + "%"), self.model_tag_association_class.user == user))) if value: # Filter by individual's tag values. clause_list.append(self.model_class.tags.any(and_(func.lower(self.model_tag_association_class.user_value).like("%" + value.lower() + "%"), self.model_tag_association_class.user == user))) return and_(*clause_list)
[docs]class MulticolFilterColumn(TextColumn): """ Column that performs multicolumn filtering. """
[docs] def __init__(self, col_name, cols_to_filter, key, visible, filterable="default"): GridColumn.__init__(self, col_name, key=key, visible=visible, filterable=filterable) self.cols_to_filter = cols_to_filter
[docs] def filter(self, trans, user, query, column_filter): """ Modify query to filter model_class by tag. Multiple filters are ANDed. """ if column_filter == "All": return query if isinstance(column_filter, list): clause_list = [] for filter in column_filter: part_clause_list = [] for column in self.cols_to_filter: part_clause_list.append(column.get_filter(trans, user, filter)) clause_list.append(or_(*part_clause_list)) complete_filter = and_(*clause_list) else: clause_list = [] for column in self.cols_to_filter: clause_list.append(column.get_filter(trans, user, column_filter)) complete_filter = or_(*clause_list) return query.filter(complete_filter)
[docs]class OwnerColumn(TextColumn): """ Column that lists item's owner. """
[docs] def get_value(self, trans, grid, item): return item.user.username
[docs] def sort(self, trans, query, ascending, column_name=None): """ Sort column using case-insensitive alphabetical sorting on item's username. """ if ascending: query = query.order_by(func.lower(self.model_class.username).asc()) else: query = query.order_by(func.lower(self.model_class.username).desc()) return query
[docs]class PublicURLColumn(TextColumn): """ Column displays item's public URL based on username and slug. """
[docs]class DeletedColumn(GridColumn): """ Column that tracks and filters for items with deleted attribute. """
[docs] def get_accepted_filters(self): """ Returns a list of accepted filters for this column. """ accepted_filter_labels_and_vals = {"active" : "False", "deleted" : "True", "all": "All"} accepted_filters = [] for label, val in accepted_filter_labels_and_vals.items(): args = {self.key: val} accepted_filters.append(GridColumnFilter(label, args)) return accepted_filters
[docs] def filter(self, trans, user, query, column_filter): """Modify query to filter self.model_class by state.""" if column_filter == "All": pass elif column_filter in ["True", "False"]: query = query.filter(self.model_class.deleted == (column_filter == "True")) return query
[docs]class PurgedColumn(GridColumn): """ Column that tracks and filters for items with purged attribute. """
[docs] def get_accepted_filters(self): """ Returns a list of accepted filters for this column. """ accepted_filter_labels_and_vals = {"nonpurged" : "False", "purged" : "True", "all": "All"} accepted_filters = [] for label, val in accepted_filter_labels_and_vals.items(): args = {self.key: val} accepted_filters.append(GridColumnFilter(label, args)) return accepted_filters
[docs] def filter(self, trans, user, query, column_filter): """Modify query to filter self.model_class by state.""" if column_filter == "All": pass elif column_filter in ["True", "False"]: query = query.filter(self.model_class.purged == (column_filter == "True")) return query
[docs]class StateColumn(GridColumn): """ Column that tracks and filters for items with state attribute. IMPORTANT NOTE: self.model_class must have a states Bunch or dict if this column type is used in the grid. """
[docs] def get_value(self, trans, grid, item): return item.state
[docs] def filter(self, trans, user, query, column_filter): """Modify query to filter self.model_class by state.""" if column_filter == "All": pass elif column_filter in [v for k, v in self.model_class.states.items()]: query = query.filter(self.model_class.state == column_filter) return query
[docs] def get_accepted_filters(self): """Returns a list of accepted filters for this column.""" all = GridColumnFilter('all', {self.key : 'All'}) accepted_filters = [all] for k, v in self.model_class.states.items(): args = {self.key: v} accepted_filters.append(GridColumnFilter(v, args)) return accepted_filters
[docs]class SharingStatusColumn(GridColumn): """ Grid column to indicate sharing status. """
[docs] def __init__(self, *args, **kwargs): self.use_shared_with_count = kwargs.pop("use_shared_with_count", False) super().__init__(*args, **kwargs)
[docs] def get_value(self, trans, grid, item): # Delete items cannot be shared. if item.deleted: return "" # Build a list of sharing for this item. sharing_statuses = [] if self._is_shared(item): sharing_statuses.append("Shared") if item.importable: sharing_statuses.append("Accessible") if item.published: sharing_statuses.append("Published") return ", ".join(sharing_statuses)
def _is_shared(self, item): if self.use_shared_with_count: # optimization to skip join for users_shared_with and loading in that data. return item.users_shared_with_count > 0 return item.users_shared_with
[docs] def filter(self, trans, user, query, column_filter): """ Modify query to filter histories by sharing status. """ if column_filter == "All": pass elif column_filter: if column_filter == "private": query = query.filter(self.model_class.users_shared_with == null()) query = query.filter(self.model_class.importable == false()) elif column_filter == "shared": query = query.filter(self.model_class.users_shared_with != null()) elif column_filter == "accessible": query = query.filter(self.model_class.importable == true()) elif column_filter == "published": query = query.filter(self.model_class.published == true()) return query
[docs] def get_accepted_filters(self): """ Returns a list of accepted filters for this column. """ accepted_filter_labels_and_vals = OrderedDict() accepted_filter_labels_and_vals["private"] = "private" accepted_filter_labels_and_vals["shared"] = "shared" accepted_filter_labels_and_vals["accessible"] = "accessible" accepted_filter_labels_and_vals["published"] = "published" accepted_filter_labels_and_vals["all"] = "All" accepted_filters = [] for label, val in accepted_filter_labels_and_vals.items(): args = {self.key: val} accepted_filters.append(GridColumnFilter(label, args)) return accepted_filters
[docs]class GridOperation:
[docs] def __init__(self, label, key=None, condition=None, allow_multiple=True, allow_popup=True, target=None, url_args=None, async_compatible=False, confirm=None, global_operation=None): self.label = label self.key = key self.allow_multiple = allow_multiple self.allow_popup = allow_popup self.condition = condition self.target = target self.url_args = url_args self.async_compatible = async_compatible # if 'confirm' is set, then ask before completing the operation self.confirm = confirm # specify a general operation that acts on the full grid # this should be a function returning a dictionary with parameters # to pass to the URL, similar to GridColumn links: # global_operation=(lambda: dict(operation="download") self.global_operation = global_operation
[docs] def get_url_args(self, item): if self.url_args: if hasattr(self.url_args, '__call__'): url_args = self.url_args(item) else: url_args = dict(self.url_args) url_args['id'] = item.id return url_args else: return dict(operation=self.label, id=item.id)
[docs] def allowed(self, item): if self.condition: return bool(self.condition(item)) else: return True
[docs]class DisplayByUsernameAndSlugGridOperation(GridOperation): """ Operation to display an item by username and slug. """
[docs] def get_url_args(self, item): return {'action' : 'display_by_username_and_slug', 'username' : item.user.username, 'slug' : item.slug}
[docs]class GridAction:
[docs] def __init__(self, label=None, url_args=None, target=None): self.label = label self.url_args = url_args self.target = target
[docs]class GridColumnFilter:
[docs] def __init__(self, label, args=None): self.label = label self.args = args
[docs] def get_url_args(self): rval = {} for k, v in self.args.items(): rval["f-" + k] = v return rval