From 357f02daba5ea8f0689abe9716f7963cca4f22de Mon Sep 17 00:00:00 2001 From: Ian Adam Naval Date: Mon, 1 Apr 2019 20:52:55 -0400 Subject: [PATCH] Add support for solo flights --- autopilot/reservation.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/autopilot/reservation.py b/autopilot/reservation.py index 9b22fcd..080049d 100644 --- a/autopilot/reservation.py +++ b/autopilot/reservation.py @@ -24,9 +24,17 @@ class Reservation(object): title = "Ground School" return title + @property + def lesson_suffix(self): + if self.instructor: + suffix = f"with {self.instructor}" + else: + suffix = "Solo" + return suffix + @property def summary(self): - return f"{self.lesson_title} with {self.instructor}" + return f"{self.lesson_title} {self.lesson_suffix}" @property def description(self):