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']
-
-            # Get format index
-            for format_index, fmt in enumerate(info_dict['formats']):
-                if fmt['format_id'] == info_dict['format_id']:
-                    break
+            format_index = next((i for i, fmt in enumerate(info_dict['formats']) if fmt['format_id'] == info_dict['format_id']))

            # Get video info
            total_duration = 0