Fix list_transactions for the case when workbaskets are empty.

uktrade · Aug 26, 2022 · 3ad256c · 3ad256c
1 parent f03bc89
commit 3ad256c
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/pii-ner-exclude.txt b/pii-ner-exclude.txt
@@ -1168,3 +1168,4 @@ measure_qs)\n
 MeasureComponent
 Components
 TP2000-452
+WorkBasketOutputFormat Enum
diff --git a/workbaskets/management/util.py b/workbaskets/management/util.py
@@ -22,8 +22,14 @@ def _output_workbasket_readable(
         self.stdout.write(f"{spaces}reason: {first_line_of(workbasket.reason)}")
         self.stdout.write(f"{spaces}status: {workbasket.status}")
         if show_transaction_info:
+            transactions = workbasket.transactions
+            first_pk = (
+                workbasket.transactions.first().pk if transactions.count() else "-"
+            )
+            last_pk = workbasket.transactions.last().pk if transactions.count() else "-"
+
             self.stdout.write(
-                f"{spaces}transactions: {workbasket.transactions.first().pk} - {workbasket.transactions.last().pk}",
+                f"{spaces}transactions: {first_pk} - {last_pk} [{transactions.count()}]",
             )
 
     def _output_workbasket_compact(self, workbasket, show_transaction_info, **kwargs):
@@ -32,14 +38,19 @@ def _output_workbasket_compact(self, workbasket, show_transaction_info, **kwargs
             ending="" if show_transaction_info else "\n",
         )
         if show_transaction_info:
+            transactions = workbasket.transactions
+            first_pk = (
+                workbasket.transactions.first().pk if transactions.count() else "-"
+            )
+            last_pk = workbasket.transactions.last().pk if transactions.count() else "-"
             self.stdout.write(
-                f", {workbasket.transactions.first().pk} - {workbasket.transactions.last().pk}",
+                f", {first_pk} - {last_pk} [{transactions.count()}]",
             )
 
     def output_workbasket(
         self,
         workbasket,
-        show_transaction_info=False,
+        show_transaction_info,
         output_format=WorkBasketOutputFormat.READABLE,
         **kwargs,
     ):