pythonic way to get the format_index

11 months ago · feaefd8ec6
parent dcefdfe508
commit feaefd8ec6
1 changed files with 1 additions and 5 deletions
--- a/yt_dlp/downloader/niconico.py
+++ b/yt_dlp/downloader/niconico.py
@ -181,11 +181,7 @@ class NiconicoLiveFD(FragmentFD):
            ie = NiconicoIE(self.ydl)
            video_id = info_dict['id']
-
+            format_index = next((i for i, fmt in enumerate(info_dict['formats']) if fmt['format_id'] == info_dict['format_id']))
            # Get format index
            for format_index, fmt in enumerate(info_dict['formats']):
                if fmt['format_id'] == info_dict['format_id']:
                    break
            # Get video info
            total_duration = 0