Update models.py

taiga/models/models.py

@@ -984,6 +984,21 @@ def import_(self, project, subject, priority, status, issue_type, severity, **at
         return self.instance.parse(self.requester, response.json())
 
 
+class EpicAttribute(CustomAttribute):
+    """
+    EpicAttribute model
+    """
+
+    endpoint = "epic-custom-attributes"
+
+
+class EpicAttributes(CustomAttributes):
+    """
+    EpicAttributes factory
+    """
+
+    instance = IssueAttribute
+
 class IssueAttribute(CustomAttribute):
     """
     IssueAttribute model
@@ -1582,6 +1597,21 @@ def list_wikilinks(self):
         """
         return WikiLinks(self.requester).list(project=self.id)
 
+    def add_epic_attribute(self, name, **attrs):
+        """
+        Add a new Epic attribute and return a :class:`IssueAttribute` object.
+
+        :param name: name of the :class:`EpicAttribute`
+        :param attrs: optional attributes for :class:`EpicAttribute`
+        """
+        return EpicAttributes(self.requester).create(self.id, name, **attrs)
+
+    def list_epic_attributes(self):
+        """
+        Get the list of :class:`EpicAttribute` resources for the project.
+        """
+        return EpicAttributes(self.requester).list(project=self.id)
+
     def add_issue_attribute(self, name, **attrs):
         """
         Add a new Issue attribute and return a :class:`IssueAttribute` object.