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_derived_column(self):
col = derived.Value(lambda row: row['x'] / row['y'])
self.assertEqual(col.get_derived_value({'x': Decimal('5.0'), 'y': Decimal('10.0')}), Decimal('0.5'))
self.assertEqual(col.get_derived_value({'x': None, 'y': Decimal('10.0')}), None)
self.assertEqual(col.get_derived_value({'x': Decimal('5.0'), 'y': Decimal('0.0')}), Decimal('0.00'))
self.assertEqual(col.get_derived_value({'x': 2, 'y': 0}), Decimal('0.00'))
self.assertEqual(col.increment_footer(None, Decimal('1.2')), None)
self.assertEqual(col.finalize_footer(None, {'x': Decimal('20.5'), 'y': Decimal('0.5'), 'othervalue': 'string'}), Decimal('41.0'))
self.assertEqual(col.finalize_footer(None, {'x': Decimal('20.5'), 'y': Decimal('0.0'), 'othervalue': 'string'}), Decimal('0.00'))
self.assertEqual(col.finalize_footer(None, {'x': Decimal('20.5'), 'y': None, 'othervalue': 'string'}), None)