Main

pywidevine/clients/blim/manifest_parse.py

@@ -0,0 +1,104 @@
+import isodate
+
+def get_mpd_list(mpd):
+    def get_height(width, height):
+        if width == '1920':
+            return '1080'
+        elif width in ('1280', '1248'):
+            return '720'
+        else:
+            return height
+
+    length = isodate.parse_duration(mpd['MPD']['@mediaPresentationDuration']).total_seconds()
+    period = mpd['MPD']['Period']
+    base_url = period['BaseURL']
+    tracks = period['AdaptationSet']
+                    
+    video_list = []
+    for video_tracks in tracks:
+        if video_tracks['@mimeType'] == 'video/mp4':
+            for x in video_tracks['Representation']:
+                try:
+                    codecs = x['@codecs']
+                except KeyError:
+                    codecs = video_tracks['@codecs']
+
+                videoDict = {
+                        'Height':get_height(x['@width'], x['@height']), 
+                        'Width':x['@width'], 
+                        'Bandwidth':x['@bandwidth'], 
+                        'ID':x['@id'], 
+                        'Codec':codecs}
+                video_list.append(videoDict)
+
+    def list_representation(x):
+        if isinstance(x['Representation'], list):
+            X = x['Representation']
+        else:
+            X = [x['Representation']]
+        return X
+
+    def replace_code_lang(x):
+        X = x.replace('es', 'es-la').replace('en', 'es-la')
+        return X
+
+    audio_list = []
+    for audio_tracks in tracks:
+        if audio_tracks['@mimeType'] == 'audio/mp4':
+            for x in list_representation(audio_tracks):
+                try:
+                    codecs = x['@codecs']
+                except KeyError:
+                    codecs = audio_tracks['@codecs']
+                audio_dict = {
+                        'Bandwidth':x['@bandwidth'],  
+                        'ID':x['@id'], 
+                        'Language':audio_tracks["@lang"], 
+                        'Codec':codecs}
+                audio_list.append(audio_dict)
+
+    subs_list = []
+    for subs_tracks in tracks:
+        if subs_tracks['@mimeType'] == 'text/vtt':
+            for x in list_representation(subs_tracks):
+                subs_dict = {
+                        'ID':x['@id'], 
+                        'Language':replace_code_lang(subs_tracks["@lang"]), 
+                        'Codec':subs_tracks['@mimeType'],
+                        'File_URL':base_url + x['BaseURL'].replace('../', '')}
+                subs_list.append(subs_dict)
+
+    return length, video_list, audio_list, subs_list
+
+def get_ism_list(ism):
+    length = float(ism['SmoothStreamingMedia']['@Duration'][:-7])
+    tracks = ism['SmoothStreamingMedia']["StreamIndex"]
+
+    video_list = []
+    for video_tracks in tracks:
+        if video_tracks['@Type'] == 'video':
+            for x in video_tracks['QualityLevel']:
+                videoDict = {
+                        'Height':x['@MaxHeight'], 
+                        'Width':x['@MaxWidth'],
+                        'ID':'0',
+                        'Bandwidth':x['@Bitrate'], 
+                        'Codec':x["@FourCC"]}
+                video_list.append(videoDict)
+
+    def replace_code_lang(x):
+        X = x.replace('255', 'es-la')
+        return X
+
+    audio_list = []
+    for audio_tracks in tracks:
+        if audio_tracks['@Type'] == 'audio':
+            for x in audio_tracks["QualityLevel"]:
+                audio_dict = {
+                        'Bandwidth':x['@Bitrate'],
+                        'ID':'0', 
+                        'Language':replace_code_lang(x["@AudioTag"]), 
+                        'Codec':x["@FourCC"]}
+                audio_list.append(audio_dict)
+
+    return length, video_list, audio_list, []