yt_dlp/extractor/eporner.py

@@ -1,14 +1,15 @@
+import re
 from .common import InfoExtractor
 from ..utils import (
-    ExtractorError,
     encode_base_n,
-    get_elements_by_class,
+    ExtractorError,
     int_or_none,
-    join_nonempty,
     merge_dicts,
     parse_duration,
     str_to_int,
     url_or_none,
+    get_elements_by_class,
+    join_nonempty
 )
 
 
@@ -83,7 +84,7 @@ class EpornerIE(InfoExtractor):
         sources = video['sources']
 
         formats = []
-        has_av1 = bool(get_elements_by_class('download-av1', webpage))
+        av1 = True if len(get_elements_by_class("download-av1", webpage)) > 0 else False
         for kind, formats_dict in sources.items():
             if not isinstance(formats_dict, dict):
                 continue
@@ -109,10 +110,10 @@ class EpornerIE(InfoExtractor):
                         'height': height,
                         'fps': fps,
                     })
-                    if has_av1:
+                    if av1:
                         formats.append({
-                            'url': src.replace('.mp4', '-av1.mp4'),
+                            'url': re.sub('(.*)(.mp4)', '\\1-av1\\2', src),
-                            'format_id': join_nonempty('av1', format_id),
+                            'format_id': join_nonempty('AV1', format_id),
                             'height': height,
                             'fps': fps,
                             'vcodec': 'av1',