Skip to content

Commit

Permalink
Merge pull request #411 from MPIB/feature_historical_modelbar
Browse files Browse the repository at this point in the history
Sort models by year for Machine Models Bar dashboard widget
  • Loading branch information
grahamgilbert authored Jan 7, 2021
2 parents 783bbba + 1e915e1 commit 48bee7d
Showing 1 changed file with 10 additions and 3 deletions.
13 changes: 10 additions & 3 deletions server/plugins/machinemodelsbar/machinemodelsbar.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
from collections import Counter
from operator import itemgetter
import re

from django.db.models import Q

Expand Down Expand Up @@ -30,9 +31,15 @@ def get_context(self, queryset, **kwargs):
machine.machine_model, machine.machine_model)
machine_models[machine_model_friendly] += 1

data = [{"label": model, "value": machine_models[model]}
for model in machine_models]
sorted_data = sorted(data, key=itemgetter("value"))
def getyear(model):
try:
return re.findall(r'(\d{4})', model)[0]
except IndexError:
return 'unknown'

data = [{"label": model, "value": machine_models[model],
"year": getyear(model)} for model in machine_models]
sorted_data = sorted(data, key=itemgetter("year"))

context["data"] = sorted_data

Expand Down

0 comments on commit 48bee7d

Please sign in to comment.