Parse Bitrate value as integer not string

davemevans · Aug 7, 2024 · f9cbdfb · f9cbdfb
1 parent f9360bd
commit f9cbdfb
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/m3u8/model.py b/m3u8/model.py
@@ -671,7 +671,7 @@ def dumps(self, last_segment, timespec="milliseconds", infspec="auto"):
                 output.append("#EXT-X-BYTERANGE:%s\n" % self.byterange)
 
             if self.bitrate:
-                output.append("#EXT-X-BITRATE:%s\n" % self.bitrate)
+                output.append("#EXT-X-BITRATE:%d\n" % self.bitrate)
 
             if self.gap_tag:
                 output.append("#EXT-X-GAP\n")

diff --git a/m3u8/parser.py b/m3u8/parser.py
@@ -439,10 +439,12 @@ def _parse_variant_playlist(line, data, state, **kwargs):
     state["expect_playlist"] = False
 
 
-def _parse_bitrate(line, state, **kwargs):
+def _parse_bitrate(state, **kwargs):
     if "segment" not in state:
         state["segment"] = {}
-    state["segment"]["bitrate"] = line.replace(protocol.ext_x_bitrate + ":", "")
+    state["segment"]["bitrate"] = _parse_simple_parameter(
+        cast_to=int, **kwargs
+    )
 
 
 def _parse_byterange(line, state, **kwargs):

diff --git a/tests/test_parser.py b/tests/test_parser.py
@@ -856,8 +856,8 @@ def test_delta_playlist_daterange_skipping():
 
 def test_bitrate():
     data = m3u8.parse(playlists.BITRATE_PLAYLIST)
-    assert data["segments"][0]["bitrate"] == "1674"
-    assert data["segments"][1]["bitrate"] == "1625"
+    assert data["segments"][0]["bitrate"] == 1674
+    assert data["segments"][1]["bitrate"] == 1625
 
 
 def test_content_steering():