Nice engagement duration text for summary box on detail page

1 files changed, 14 insertions, 0 deletions

diff --git a/engagements/utils.py b/engagements/utils.py
index 3b414e9..b80e74d 100644
--- a/engagements/utils.py
+++ b/engagements/utils.py
@@ -18,6 +18,20 @@ from instruments.models import Instrument, SubInstrument
 from myuser.models import Team, TeamUser
 
 
+def duration_formatter(d1: datetime.date, d2: datetime.date) -> str:
+    trip_delta = d2 - d1
+    if trip_delta.days == 0:
+        # this is a single day event
+        return f"{d1.strftime("%d %B %Y")} (1 day)"
+    else:
+        # if months are different...
+        if d1.month != d2.month:
+            # we need to format this differently
+            return f"{d1.strftime("%d %B")} - {d2.strftime("%d %B %Y")} ({trip_delta.days +1} days)"
+        else:
+            return f"{d1.strftime("%d")}-{d2.strftime("%d %B %Y")} ({trip_delta.days + 1} days)"
+
+
 def populate_database():
     out = defaultdict(list)
     fake = Faker(locale="en_GB")