management/commands/refresh_highlighted_buttons.py

@@ -18,10 +18,15 @@ class Command(BaseCommand):
         return parser
 
     def handle(self, *args, **kwargs):
-        queryset = RecurrentTransaction.objects.filter(template__display=True, created_at__gte=timezone.now() - timedelta(days=30)).values("template").annotate(transaction_count=Count("template")).order_by("-transaction_count")[:10]
+        queryset = RecurrentTransaction.objects.filter(
+            template__display=True,
+            valid=True,
+            created_at__gte=timezone.now() - timedelta(days=30),
+        ).values("template").annotate(transaction_count=Count("template")).order_by("-transaction_count")[:10]
         for d in queryset.all():
             button_id = d["template"]
             button = TransactionTemplate.objects.get(pk=button_id)
-            self.stdout.write(self.style.WARNING("Highlight button {name} ({count:d} transactions)...".format(name=button.name, count=d["transaction_count"])))
+            self.stdout.write(self.style.WARNING("Highlight button {name} ({count:d} transactions)..."
+                                                 .format(name=button.name, count=d["transaction_count"])))
             button.highlighted = True
             button.save()