SciPost Code Repository

Skip to content
Snippets Groups Projects
Commit 0c2eebed authored by George Katsikas's avatar George Katsikas :goat:
Browse files

refactor(graphs): :lipstick: use group field display when possible

parent 68d9279d
No related branches found
No related tags found
No related merge requests found
...@@ -466,7 +466,16 @@ class BarPlot(PlotKind): ...@@ -466,7 +466,16 @@ class BarPlot(PlotKind):
if qs.exists(): if qs.exists():
groups, vals = zip(*qs.values_list(group_key, "agg")) groups, vals = zip(*qs.values_list(group_key, "agg"))
return [str(group) for group in groups], vals
# Attempt to convert the group values to display labels if possible
try:
field_choices = self.plotter.model._meta.get_field(group_key).choices
group_display_labels = dict(field_choices)
groups = [group_display_labels.get(group, group) for group in groups]
except Exception:
groups = [str(group) for group in groups]
return groups, vals
else: else:
return [], [] return [], []
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment