Implement __add__ to Tracks class

devine-dl · Feb 20, 2024 · 0c20160 · 0c20160
1 parent eef397f
commit 0c20160
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/devine/core/tracks/tracks.py b/devine/core/tracks/tracks.py
@@ -51,6 +51,12 @@ def __iter__(self) -> Iterator[AnyTrack]:
     def __len__(self) -> int:
         return len(self.videos) + len(self.audio) + len(self.subtitles)
 
+    def __add__(self, other: Tracks) -> Tracks:
+        if not isinstance(other, Tracks):
+            raise TypeError(f"Cannot only add {Tracks} objects with one another, not {type(other)}")
+
+        return Tracks(list(self) + list(other))
+
     def __repr__(self) -> str:
         return "{name}({items})".format(
             name=self.__class__.__name__,