Added back display names as they are used in Home Assistant

costastf · Nov 30, 2019 · a6844a8 · a6844a8
1 parent a268fa9
commit a6844a8
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/toonapilib/toonapilib.py b/toonapilib/toonapilib.py
@@ -154,6 +154,16 @@ def _get_agreements(self, session):
             raise AgreementsRetrievalError(response.text)
         return agreements
 
+    @property
+    def display_names(self):
+        """The ids of all the agreements.
+
+        Returns:
+            list: A list of the agreement ids.
+
+        """
+        return [agreement.display_common_name.lower() for agreement in self.agreements]
+
     def _reset(self):
         self.agreements = None
         self.agreement = None