def _gen_stacked_chart(chart_id, chart_type):
chart = ChartConfig(chart_id, chart_type)
chart.add_source('anpa_category.qcode', {
'a': {1: 1, 3: 1},
'b': {1: 1, 3: 2},
'c': {1: 2, 3: 1, 5: 1},
})
chart.add_source('urgency', {1: 4, 3: 4, 5: 1})
return chart
def test_translate_source(self):
chart = ChartConfig('source', 'bar')
chart.add_source('source', {'aap': 3, 'ftp': 1, 'ap': 5})
self.assertEqual(chart.translations, {})
chart.load_translations()
self.assertEqual(chart.translations, {
'source': {
'title': 'Source',
'names': {}
}
})
def generate_highcharts_config(self, docs, args):
params = args.get('params') or {}
aggs = args.get('aggs') or {}
group = aggs.get('group') or {}
subgroup = aggs.get('subgroup') or {}
translations = args.get('translations') or {}
chart = params.get('chart') or {}
chart_type = chart.get('type') or 'bar'
report = self.generate_report(docs, args)
chart_config = ChartConfig('content_publishing', chart_type)
chart_config.add_source(group.get('field'), report.get('groups'))
if report.get('subgroups'):
chart_config.add_source(subgroup.get('field'), report['subgroups'])
def gen_title():
if chart.get('title'):
return chart['title']
group_type = group.get('field')
group_title = chart_config.get_source_name(group_type)
if subgroup.get('field'):
subgroup_type = subgroup.get('field')
subgroup_title = chart_config.get_source_name(subgroup_type)
return 'Published Stories per {} with {} breakdown'.format(
group_title,
subgroup_title
)
return 'Published Stories per {}'.format(group_title)
def gen_subtitle():
return ChartConfig.gen_subtitle_for_dates(params)
chart_config.get_title = gen_title
chart_config.get_subtitle = gen_subtitle
chart_config.sort_order = chart.get('sort_order') or 'desc'
chart_config.translations = translations
report['highcharts'] = [chart_config.gen_config()]
return report
def test_translate_authors(self):
chart = ChartConfig('author', 'bar')
chart.add_source('authors.parent', {'user1': 3, 'user2': 4, 'user3': 5})
self.assertEqual(chart.translations, {})
chart.load_translations()
self.assertEqual(chart.translations, {
'authors_parent': {
'title': 'Author',
'names': {
'user1': 'first user',
'user2': 'second user',
'user3': 'last user'
}
}
})
def test_translate_user(self):
chart = ChartConfig('user', 'bar')
chart.add_source('task.user', {'user1': 3, 'user2': 4, 'user3': 5})
self.assertEqual(chart.translations, {})
chart.load_translations()
self.assertEqual(chart.translations, {
'task_user': {
'title': 'User',
'names': {
'user1': 'first user',
'user2': 'second user',
'user3': 'last user'
}
}
})
def test_translate_desk(self):
chart = ChartConfig('desk', 'bar')
chart.add_source('task.desk', {'desk1': 4, 'desk2': 5, 'desk3': 1})
self.assertEqual(chart.translations, {})
chart.load_translations()
self.assertEqual(chart.translations, {
'task_desk': {
'title': 'Desk',
'names': {
'desk1': 'Politic Desk',
'desk2': 'Sports Desk',
'desk3': 'System Desk'
}
}
})
def test_translate_genre(self):
chart = ChartConfig('genre', 'bar')
chart.add_source('genre.qcode', {'Article': 4, 'Sidebar': 5, 'Factbox': 1})
self.assertEqual(chart.translations, {})
chart.load_translations()
self.assertEqual(chart.translations, {
'genre_qcode': {
'title': 'Genre',
'names': {
'Article': 'Article (news)',
'Sidebar': 'Sidebar',
'Factbox': 'Factbox'
}
}
})
def test_translate_anpa_category(self):
chart = ChartConfig('catrgory', 'bar')
chart.add_source('anpa_category.qcode', {'a': 3, 'b': 4, 'c': 1})
self.assertEqual(chart.translations, {})
chart.load_translations()
self.assertEqual(chart.translations, {
'anpa_category_qcode': {
'title': 'Category',
'names': {
'a': 'Advisories',
'b': 'Basketball',
'c': 'Cricket'
}
}
})
def test_translate_state(self):
chart = ChartConfig('state', 'bar')
chart.add_source('state', {'published': 3, 'killed': 1, 'updated': 5})
self.assertEqual(chart.translations, {})
chart.load_translations()
self.assertEqual(chart.translations, {
'state': {
'title': 'State',
'names': {
'published': 'Published',
'killed': 'Killed',
'corrected': 'Corrected',
'updated': 'Updated',
}
}
})
def test_translate_urgency(self):
chart = ChartConfig('urgency', 'bar')
chart.add_source('urgency', {1: 4, 3: 4, 5: 1})
self.assertEqual(chart.translations, {})
chart.load_translations()
self.assertEqual(chart.translations, {
'urgency': {
'title': 'Urgency',
'names': {
1: 1,
2: 2,
3: 3,
4: 4,
5: 5
}
}
})
def generate_highcharts_config(self, docs, args):
params = args.get('params') or {}
aggs = args.get('aggs') or {}
group = aggs.get('group') or {}
subgroup = aggs.get('subgroup') or {}
translations = args.get('translations') or {}
chart = params.get('chart') or {}
chart_type = chart.get('type') or 'bar'
report = self.generate_report(docs, args)
chart_config = ChartConfig('content_publishing', chart_type)
group_keys = list(report['groups'].keys())
if len(group_keys) == 1 and report.get('subgroups'):
chart_config.add_source(
subgroup.get('field'),
report['subgroups']
)
chart_config.load_translations(group.get('field'))
else:
chart_config.add_source(group.get('field'), report.get('groups'))
if report.get('subgroups'):
chart_config.add_source(subgroup.get('field'), report['subgroups'])
def gen_title():
if chart.get('title'):
return chart['title']
group_type = group.get('field')
group_title = chart_config.get_source_name(group_type)
if len(group_keys) == 1 and report.get('subgroups'):
data_name = chart_config.get_source_title(
group_type,
group_keys[0]
)
return 'Published Stories for {}: {}'.format(
group_title,
data_name
)
return 'Published Stories per {}'.format(group_title)
def gen_subtitle():
return ChartConfig.gen_subtitle_for_dates(params)
chart_config.get_title = gen_title
chart_config.get_subtitle = gen_subtitle
chart_config.sort_order = chart.get('sort_order') or 'desc'
chart_config.translations = translations
report['highcharts'] = [chart_config.gen_config()]
return report
def generate_highcharts_config(self, docs, args):
params = args.get('params') or {}
chart = params.get('chart') or {}
chart_type = chart.get('type') or 'bar'
report = self.generate_report(docs, args)
chart_config = ChartConfig('planning_usage', chart_type)
chart_config.add_source('task.user', report.get('group'))
chart_config.add_source('planning_usage', report.get('subgroup'))
def gen_title():
if chart.get('title'):
return chart['title']
return 'Planning Module Usage'
def gen_subtitle():
return ChartConfig.gen_subtitle_for_dates(params)
def get_y_axis_title():
return 'Items Created'
chart_config.get_title = gen_title
chart_config.get_subtitle = gen_subtitle
chart_config.get_y_axis_title = get_y_axis_title
chart_config.sort_order = chart.get('sort_order') or 'desc'
translations = args.get('translations') or {}
chart_config.translations = translations
report['highcharts'] = [chart_config.gen_config()]
return report
def _gen_single_chart(chart_id, chart_type):
chart = ChartConfig(chart_id, chart_type)
chart.add_source('anpa_category.qcode', {'a': 3, 'b': 4, 'c': 1})
return chart