adds Collection

Author: meisnate12

VERSION

@@ -1 +1 @@
-1.2.9
+1.3.0

arrapi/objs/base.py

@@ -113,6 +113,8 @@ def _parse(self, data: Any = None, attrs: Optional[Union[str, list]] = None, val
                 return default
         elif value_type == "date":
             return datetime.strptime(value[:-1].split(".")[0], "%Y-%m-%dT%H:%M:%S")
+        elif value_type == "collection":
+            return arrapi.objs.simple.Collection(self._arr, value)
         elif value_type == "image":
             return arrapi.objs.simple.Image(self._arr, value)
         elif value_type == "season":

arrapi/objs/reload.py

@@ -286,6 +286,7 @@ def _load(self, data):
         self.added = self._parse(attrs="added", value_type="date")
         self.rating_votes = self._parse(attrs=["rating", "votes"], value_type="int")
         self.rating_value = self._parse(attrs=["rating", "value"], value_type="float")
+        self.collection = self._parse(attrs="collection", value_type="collection")
         self.id = self._parse(attrs="id", value_type="int", default_is_none=True)
         if self._raw.v3:
             self.originalTitle = self._parse(attrs="originalTitle")

arrapi/objs/simple.py

@@ -8,6 +8,25 @@ def _load(self, data):
         super()._load(data)
 
 
+class Collection(SimpleObj):
+    """ Represents a single Collection.
+
+        Attributes:
+            name (str): Name of the Collection.
+            tmdbId (int): TMDb Collection ID of the Collection.
+    """
+
+    def _load(self, data):
+        super()._load(data)
+        self.name = self._parse(attrs="name")
+        self.tmdbId = self._parse(attrs="tmdbId", value_type="int")
+        self._finish(self.name)
+
+    def get_url(self):
+        """ URL of the Collection on TMDb"""
+        return f"https://www.themoviedb.org/collection/{self.tmdbId}"
+
+
 class Image(SimpleObj):
     """ Represents a single Image.
 

docs/objs.rst

@@ -75,6 +75,13 @@ Command
     :members:
 
 
+Collection
+--------------------
+
+.. autoclass:: arrapi.objs.simple.Collection
+    :members:
+
+
 Remote Path Mapping
 --------------------