ninjajiraiya
/
devine
forked from DRMTalks/devine
1
0
Fork 0
Code Pull Requests Activity

Simplify `Tracks.__add__` method, support Chapter(s) & Track objects

This commit is contained in:
rlaphoenix 2024-02-29 23:19:05 +00:00
parent 97efb59e5f
commit be0ed0b0fb
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
devine/core/tracks/tracks.py
View File

@ -51,11 +51,12 @@ class Tracks:
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 __add__(
self,
other: Union[Tracks, Sequence[Union[AnyTrack, Chapter, Chapters]], Track, Chapter, Chapters]
) -> Tracks:
self.add(other)
return self
def __repr__(self) -> str:
return "{name}({items})".format(