Add support for multiple dashboards.
Add support for removing dashboard widgets. Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
@@ -45,6 +45,43 @@ class Collection(object):
|
||||
return self._model.objects.all()
|
||||
|
||||
|
||||
class Dashboard(object):
|
||||
_registry = {}
|
||||
|
||||
@classmethod
|
||||
def get(cls, name):
|
||||
return cls._registry[name]
|
||||
|
||||
def __init__(self, name, label):
|
||||
self.name = name
|
||||
self.label = label
|
||||
self.widgets = {}
|
||||
self.removed_widgets = []
|
||||
self.__class__._registry[name] = self
|
||||
|
||||
def add_widget(self, widget, order=0):
|
||||
self.widgets[widget] = {'widget': widget, 'order': order}
|
||||
|
||||
def get_widgets(self):
|
||||
"""
|
||||
Returns a list of widgets sorted by their 'order'.
|
||||
If two or more widgets have the same 'order', sort by label.
|
||||
"""
|
||||
return map(
|
||||
lambda x: x['widget'],
|
||||
filter(
|
||||
lambda x: x['widget'] not in self.removed_widgets,
|
||||
sorted(
|
||||
self.widgets.values(),
|
||||
key=lambda x: (x['order'], x['widget'].label)
|
||||
)
|
||||
)
|
||||
)
|
||||
|
||||
def remove_widget(self, widget):
|
||||
self.removed_widgets.append(widget)
|
||||
|
||||
|
||||
class DashboardWidget(object):
|
||||
_registry = []
|
||||
|
||||
|
||||
Reference in New Issue
Block a user