List typing

m3u8/model.py

@@ -3,6 +3,7 @@
 # license that can be found in the LICENSE file.
 import decimal
 import os
+from typing import TypeVar
 
 from m3u8.mixins import BasePathMixin, GroupedBasePathMixin
 from m3u8.parser import format_date_time, parse
@@ -449,6 +450,15 @@ def _create_sub_directories(self, filename):
             os.makedirs(basename, exist_ok=True)
 
 
+T = TypeVar("T")
+
+
+class TagList(list[T]):
+    def __str__(self):
+        output = [str(tag) for tag in self]
+        return "\n".join(output)
+
+
 class Segment(BasePathMixin):
     """
     A video segment from a M3U8 playlist
@@ -706,7 +716,7 @@ def base_uri(self, newbase_uri):
             self.init_section.base_uri = newbase_uri
 
 
-class SegmentList(list, GroupedBasePathMixin):
+class SegmentList(list[Segment], GroupedBasePathMixin):
     def dumps(self, timespec="milliseconds", infspec="auto"):
         output = []
         last_segment = None
@@ -821,7 +831,7 @@ def __str__(self):
         return self.dumps(None)
 
 
-class PartialSegmentList(list, GroupedBasePathMixin):
+class PartialSegmentList(list[PartialSegment], GroupedBasePathMixin):
     def __str__(self):
         output = [str(part) for part in self]
         return "\n".join(output)
@@ -1009,6 +1019,10 @@ def __str__(self):
         return "#EXT-X-STREAM-INF:" + ",".join(stream_inf) + "\n" + self.uri
 
 
+class PlaylistList(TagList[Playlist], GroupedBasePathMixin):
+    pass
+
+
 class IFramePlaylist(BasePathMixin):
     """
     IFramePlaylist object representing a link to a
@@ -1260,26 +1274,12 @@ def __str__(self):
         return self.dumps()
 
 
-class TagList(list):
-    def __str__(self):
-        output = [str(tag) for tag in self]
-        return "\n".join(output)
-
-
-class MediaList(TagList, GroupedBasePathMixin):
+class MediaList(TagList[Media], GroupedBasePathMixin):
     @property
     def uri(self):
         return [media.uri for media in self]
 
 
-class PlaylistList(TagList, GroupedBasePathMixin):
-    pass
-
-
-class SessionDataList(TagList):
-    pass
-
-
 class Start:
     def __init__(self, time_offset, precise=None):
         self.time_offset = float(time_offset)
@@ -1314,7 +1314,7 @@ def __str__(self):
         return self.dumps()
 
 
-class RenditionReportList(list, GroupedBasePathMixin):
+class RenditionReportList(list[RenditionReport], GroupedBasePathMixin):
     def __str__(self):
         output = [str(report) for report in self]
         return "\n".join(output)
@@ -1443,7 +1443,7 @@ def __str__(self):
         return self.dumps()
 
 
-class DateRangeList(TagList):
+class SessionDataList(TagList[SessionData]):
     pass
 
 
@@ -1502,6 +1502,10 @@ def __str__(self):
         return self.dumps()
 
 
+class DateRangeList(TagList[DateRange]):
+    pass
+
+
 class ContentSteering(BasePathMixin):
     def __init__(self, base_uri, server_uri, pathway_id=None):
         self.base_uri = base_uri