class BasicDatabaseReport(base.Report):
database_entity = 'test.entities.AllTheData'
filters = [
('user_is_active',
database.QueryFilter(
lambda entity, user_input: entity.user_is_active == user_input
if user_input is not None else None,
widget=widgets.Select(label='User Is Active',
choices=ACTIVE_USER_CHOICES))),
]
keys = ('user_id', key_range.SourceKeyRange)
columns = [
('user_id',
database.GroupBy('user_id',
format=formats.Integer(label='User ID',
grouping=False))),
('user_is_active',
database.First('user_is_active',
format=formats.Boolean(label='Active?'))),
('num_widgets',
database.Count('widget_id',
distinct=True,
format=formats.Integer(label='Widgets'))),
('_sum_widget_price', database.Sum('widget_price')),
('average_widget_price',
derived.Value(
lambda row: row['_sum_widget_price'] / row['num_widgets'],
format=formats.Bling)),
]
default_sort = ('average_widget_price', 'desc')
class BasicMergeReport(base.Report):
merged_reports = {
'one': BasicDatabaseReport,
'two': BasicDatabaseReport,
}
filters = [
('num_widgets',
merge.PostFilter(lambda row: row['double_num_widgets'] > 0)),
('user_is_active',
merge.DelegatedFilter(database.QueryFilter(
lambda entity, user_input: entity.user_is_active == user_input
if user_input is not None else None),
widget=widgets.Select(
label='User Is Active',
choices=ACTIVE_USER_CHOICES))),
('include',
merge.ReportFilter('one',
widget=widgets.Checkbox(label='Include',
default=True))),
]
keys = ('user_id', key_range.SourceKeyRange)
columns = [
('user_id',
merge.First(format=formats.Integer(label='ID', grouping=False),
footer=False)),
('user_is_active', merge.First(format=formats.Boolean)),
('double_num_widgets', merge.Sum('num_widgets',
format=formats.Integer)),
('single_num_widgets',
derived.Value(lambda row: row['double_num_widgets'] / 2,
format=formats.Integer)),
]
default_sort = ('single_num_widgets', 'asc')
def test_integer_format(self):
format = formats.Integer(label='Many')
self.assertEqual(
format.header_info, {
'label': 'Many',
'sortable': True,
'data_type': 'integer',
'className': 'num',
})
self.assertEqual(format.format_html(123456), '123,456')
self.assertEqual(format.format_csv(123456), '123456')
self.assertEqual(format.format_csv(None), '0')
format = formats.Integer(label='Many', grouping=False)
self.assertEqual(format.format_html(123456), '123456')
self.assertEqual(format.format_csv(123456), '123456')