Merge pull request #137 from mauricioabreu/missing-map-on-dumps

Missing EXT-X-MAP attributes on dumps

m3u8/model.py

@@ -263,6 +263,13 @@ def dumps(self):
             output.append(str(self.start))
         if self.is_i_frames_only:
             output.append('#EXT-X-I-FRAMES-ONLY')
+        if self.segment_map:
+            map_output = []
+            if self.segment_map.get('uri'):
+                map_output.append('URI=' + quoted(self.segment_map['uri']))
+            if self.segment_map.get('byterange'):
+                map_output.append('BYTERANGE=' + self.segment_map['byterange'])
+            output.append('#EXT-X-MAP:' + ','.join(map_output))
         if self.is_variant:
             if self.media:
                 output.append(str(self.media))

tests/test_model.py

@@ -483,6 +483,16 @@ def test_dump_should_include_program_date_time():
     assert "EXT-X-PROGRAM-DATE-TIME:2014-08-13T13:36:33+00:00" in obj.dumps().strip()
 
 
+def test_dump_should_include_segment_level_program_date_time():
+    obj = m3u8.M3U8(playlists.DISCONTINUITY_PLAYLIST_WITH_PROGRAM_DATE_TIME)
+    # Tag being expected is in the segment level, not the global one
+    assert "#EXT-X-PROGRAM-DATE-TIME:2014-08-13T13:36:55+00:00" in obj.dumps().strip()
+
+def test_dump_should_include_map_attributes():
+    obj = m3u8.M3U8(playlists.MAP_URI_PLAYLIST_WITH_BYTERANGE)
+
+    assert 'EXT-X-MAP:URI="main.mp4",BYTERANGE="812@0"' in obj.dumps().strip()
+
 def test_dump_should_work_for_playlists_using_byteranges():
     obj = m3u8.M3U8(playlists.PLAYLIST_USING_BYTERANGES)