]> jfr.im git - yt-dlp.git/blob - yt_dlp/utils.py
bf2eb6b3807cf1efcaead0014e07e0c9f90ec5dd
[yt-dlp.git] / yt_dlp / utils.py
1 import asyncio
2 import atexit
3 import base64
4 import binascii
5 import calendar
6 import codecs
7 import collections
8 import collections.abc
9 import contextlib
10 import datetime
11 import email.header
12 import email.utils
13 import errno
14 import gzip
15 import hashlib
16 import hmac
17 import html.entities
18 import html.parser
19 import http.client
20 import http.cookiejar
21 import inspect
22 import io
23 import itertools
24 import json
25 import locale
26 import math
27 import mimetypes
28 import operator
29 import os
30 import platform
31 import random
32 import re
33 import shlex
34 import socket
35 import ssl
36 import struct
37 import subprocess
38 import sys
39 import tempfile
40 import time
41 import traceback
42 import types
43 import unicodedata
44 import urllib.error
45 import urllib.parse
46 import urllib.request
47 import xml.etree.ElementTree
48 import zlib
49
50 from .compat import functools # isort: split
51 from .compat import (
52 compat_etree_fromstring,
53 compat_expanduser,
54 compat_HTMLParseError,
55 compat_os_name,
56 compat_shlex_quote,
57 )
58 from .dependencies import brotli, certifi, websockets, xattr
59 from .socks import ProxyType, sockssocket
60
61
62 def register_socks_protocols():
63 # "Register" SOCKS protocols
64 # In Python < 2.6.5, urlsplit() suffers from bug https://bugs.python.org/issue7904
65 # URLs with protocols not in urlparse.uses_netloc are not handled correctly
66 for scheme in ('socks', 'socks4', 'socks4a', 'socks5'):
67 if scheme not in urllib.parse.uses_netloc:
68 urllib.parse.uses_netloc.append(scheme)
69
70
71 # This is not clearly defined otherwise
72 compiled_regex_type = type(re.compile(''))
73
74
75 def random_user_agent():
76 _USER_AGENT_TPL = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/%s Safari/537.36'
77 _CHROME_VERSIONS = (
78 '90.0.4430.212',
79 '90.0.4430.24',
80 '90.0.4430.70',
81 '90.0.4430.72',
82 '90.0.4430.85',
83 '90.0.4430.93',
84 '91.0.4472.101',
85 '91.0.4472.106',
86 '91.0.4472.114',
87 '91.0.4472.124',
88 '91.0.4472.164',
89 '91.0.4472.19',
90 '91.0.4472.77',
91 '92.0.4515.107',
92 '92.0.4515.115',
93 '92.0.4515.131',
94 '92.0.4515.159',
95 '92.0.4515.43',
96 '93.0.4556.0',
97 '93.0.4577.15',
98 '93.0.4577.63',
99 '93.0.4577.82',
100 '94.0.4606.41',
101 '94.0.4606.54',
102 '94.0.4606.61',
103 '94.0.4606.71',
104 '94.0.4606.81',
105 '94.0.4606.85',
106 '95.0.4638.17',
107 '95.0.4638.50',
108 '95.0.4638.54',
109 '95.0.4638.69',
110 '95.0.4638.74',
111 '96.0.4664.18',
112 '96.0.4664.45',
113 '96.0.4664.55',
114 '96.0.4664.93',
115 '97.0.4692.20',
116 )
117 return _USER_AGENT_TPL % random.choice(_CHROME_VERSIONS)
118
119
120 SUPPORTED_ENCODINGS = [
121 'gzip', 'deflate'
122 ]
123 if brotli:
124 SUPPORTED_ENCODINGS.append('br')
125
126 std_headers = {
127 'User-Agent': random_user_agent(),
128 'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
129 'Accept-Language': 'en-us,en;q=0.5',
130 'Sec-Fetch-Mode': 'navigate',
131 }
132
133
134 USER_AGENTS = {
135 'Safari': 'Mozilla/5.0 (X11; Linux x86_64; rv:10.0) AppleWebKit/533.20.25 (KHTML, like Gecko) Version/5.0.4 Safari/533.20.27',
136 }
137
138
139 NO_DEFAULT = object()
140 IDENTITY = lambda x: x
141
142 ENGLISH_MONTH_NAMES = [
143 'January', 'February', 'March', 'April', 'May', 'June',
144 'July', 'August', 'September', 'October', 'November', 'December']
145
146 MONTH_NAMES = {
147 'en': ENGLISH_MONTH_NAMES,
148 'fr': [
149 'janvier', 'février', 'mars', 'avril', 'mai', 'juin',
150 'juillet', 'août', 'septembre', 'octobre', 'novembre', 'décembre'],
151 # these follow the genitive grammatical case (dopełniacz)
152 # some websites might be using nominative, which will require another month list
153 # https://en.wikibooks.org/wiki/Polish/Noun_cases
154 'pl': ['stycznia', 'lutego', 'marca', 'kwietnia', 'maja', 'czerwca',
155 'lipca', 'sierpnia', 'września', 'października', 'listopada', 'grudnia'],
156 }
157
158 # From https://github.com/python/cpython/blob/3.11/Lib/email/_parseaddr.py#L36-L42
159 TIMEZONE_NAMES = {
160 'UT': 0, 'UTC': 0, 'GMT': 0, 'Z': 0,
161 'AST': -4, 'ADT': -3, # Atlantic (used in Canada)
162 'EST': -5, 'EDT': -4, # Eastern
163 'CST': -6, 'CDT': -5, # Central
164 'MST': -7, 'MDT': -6, # Mountain
165 'PST': -8, 'PDT': -7 # Pacific
166 }
167
168 # needed for sanitizing filenames in restricted mode
169 ACCENT_CHARS = dict(zip('ÂÃÄÀÁÅÆÇÈÉÊËÌÍÎÏÐÑÒÓÔÕÖŐØŒÙÚÛÜŰÝÞßàáâãäåæçèéêëìíîïðñòóôõöőøœùúûüűýþÿ',
170 itertools.chain('AAAAAA', ['AE'], 'CEEEEIIIIDNOOOOOOO', ['OE'], 'UUUUUY', ['TH', 'ss'],
171 'aaaaaa', ['ae'], 'ceeeeiiiionooooooo', ['oe'], 'uuuuuy', ['th'], 'y')))
172
173 DATE_FORMATS = (
174 '%d %B %Y',
175 '%d %b %Y',
176 '%B %d %Y',
177 '%B %dst %Y',
178 '%B %dnd %Y',
179 '%B %drd %Y',
180 '%B %dth %Y',
181 '%b %d %Y',
182 '%b %dst %Y',
183 '%b %dnd %Y',
184 '%b %drd %Y',
185 '%b %dth %Y',
186 '%b %dst %Y %I:%M',
187 '%b %dnd %Y %I:%M',
188 '%b %drd %Y %I:%M',
189 '%b %dth %Y %I:%M',
190 '%Y %m %d',
191 '%Y-%m-%d',
192 '%Y.%m.%d.',
193 '%Y/%m/%d',
194 '%Y/%m/%d %H:%M',
195 '%Y/%m/%d %H:%M:%S',
196 '%Y%m%d%H%M',
197 '%Y%m%d%H%M%S',
198 '%Y%m%d',
199 '%Y-%m-%d %H:%M',
200 '%Y-%m-%d %H:%M:%S',
201 '%Y-%m-%d %H:%M:%S.%f',
202 '%Y-%m-%d %H:%M:%S:%f',
203 '%d.%m.%Y %H:%M',
204 '%d.%m.%Y %H.%M',
205 '%Y-%m-%dT%H:%M:%SZ',
206 '%Y-%m-%dT%H:%M:%S.%fZ',
207 '%Y-%m-%dT%H:%M:%S.%f0Z',
208 '%Y-%m-%dT%H:%M:%S',
209 '%Y-%m-%dT%H:%M:%S.%f',
210 '%Y-%m-%dT%H:%M',
211 '%b %d %Y at %H:%M',
212 '%b %d %Y at %H:%M:%S',
213 '%B %d %Y at %H:%M',
214 '%B %d %Y at %H:%M:%S',
215 '%H:%M %d-%b-%Y',
216 )
217
218 DATE_FORMATS_DAY_FIRST = list(DATE_FORMATS)
219 DATE_FORMATS_DAY_FIRST.extend([
220 '%d-%m-%Y',
221 '%d.%m.%Y',
222 '%d.%m.%y',
223 '%d/%m/%Y',
224 '%d/%m/%y',
225 '%d/%m/%Y %H:%M:%S',
226 '%d-%m-%Y %H:%M',
227 ])
228
229 DATE_FORMATS_MONTH_FIRST = list(DATE_FORMATS)
230 DATE_FORMATS_MONTH_FIRST.extend([
231 '%m-%d-%Y',
232 '%m.%d.%Y',
233 '%m/%d/%Y',
234 '%m/%d/%y',
235 '%m/%d/%Y %H:%M:%S',
236 ])
237
238 PACKED_CODES_RE = r"}\('(.+)',(\d+),(\d+),'([^']+)'\.split\('\|'\)"
239 JSON_LD_RE = r'(?is)<script[^>]+type=(["\']?)application/ld\+json\1[^>]*>\s*(?P<json_ld>{.+?}|\[.+?\])\s*</script>'
240
241 NUMBER_RE = r'\d+(?:\.\d+)?'
242
243
244 @functools.cache
245 def preferredencoding():
246 """Get preferred encoding.
247
248 Returns the best encoding scheme for the system, based on
249 locale.getpreferredencoding() and some further tweaks.
250 """
251 try:
252 pref = locale.getpreferredencoding()
253 'TEST'.encode(pref)
254 except Exception:
255 pref = 'UTF-8'
256
257 return pref
258
259
260 def write_json_file(obj, fn):
261 """ Encode obj as JSON and write it to fn, atomically if possible """
262
263 tf = tempfile.NamedTemporaryFile(
264 prefix=f'{os.path.basename(fn)}.', dir=os.path.dirname(fn),
265 suffix='.tmp', delete=False, mode='w', encoding='utf-8')
266
267 try:
268 with tf:
269 json.dump(obj, tf, ensure_ascii=False)
270 if sys.platform == 'win32':
271 # Need to remove existing file on Windows, else os.rename raises
272 # WindowsError or FileExistsError.
273 with contextlib.suppress(OSError):
274 os.unlink(fn)
275 with contextlib.suppress(OSError):
276 mask = os.umask(0)
277 os.umask(mask)
278 os.chmod(tf.name, 0o666 & ~mask)
279 os.rename(tf.name, fn)
280 except Exception:
281 with contextlib.suppress(OSError):
282 os.remove(tf.name)
283 raise
284
285
286 def find_xpath_attr(node, xpath, key, val=None):
287 """ Find the xpath xpath[@key=val] """
288 assert re.match(r'^[a-zA-Z_-]+$', key)
289 expr = xpath + ('[@%s]' % key if val is None else f"[@{key}='{val}']")
290 return node.find(expr)
291
292 # On python2.6 the xml.etree.ElementTree.Element methods don't support
293 # the namespace parameter
294
295
296 def xpath_with_ns(path, ns_map):
297 components = [c.split(':') for c in path.split('/')]
298 replaced = []
299 for c in components:
300 if len(c) == 1:
301 replaced.append(c[0])
302 else:
303 ns, tag = c
304 replaced.append('{%s}%s' % (ns_map[ns], tag))
305 return '/'.join(replaced)
306
307
308 def xpath_element(node, xpath, name=None, fatal=False, default=NO_DEFAULT):
309 def _find_xpath(xpath):
310 return node.find(xpath)
311
312 if isinstance(xpath, str):
313 n = _find_xpath(xpath)
314 else:
315 for xp in xpath:
316 n = _find_xpath(xp)
317 if n is not None:
318 break
319
320 if n is None:
321 if default is not NO_DEFAULT:
322 return default
323 elif fatal:
324 name = xpath if name is None else name
325 raise ExtractorError('Could not find XML element %s' % name)
326 else:
327 return None
328 return n
329
330
331 def xpath_text(node, xpath, name=None, fatal=False, default=NO_DEFAULT):
332 n = xpath_element(node, xpath, name, fatal=fatal, default=default)
333 if n is None or n == default:
334 return n
335 if n.text is None:
336 if default is not NO_DEFAULT:
337 return default
338 elif fatal:
339 name = xpath if name is None else name
340 raise ExtractorError('Could not find XML element\'s text %s' % name)
341 else:
342 return None
343 return n.text
344
345
346 def xpath_attr(node, xpath, key, name=None, fatal=False, default=NO_DEFAULT):
347 n = find_xpath_attr(node, xpath, key)
348 if n is None:
349 if default is not NO_DEFAULT:
350 return default
351 elif fatal:
352 name = f'{xpath}[@{key}]' if name is None else name
353 raise ExtractorError('Could not find XML attribute %s' % name)
354 else:
355 return None
356 return n.attrib[key]
357
358
359 def get_element_by_id(id, html, **kwargs):
360 """Return the content of the tag with the specified ID in the passed HTML document"""
361 return get_element_by_attribute('id', id, html, **kwargs)
362
363
364 def get_element_html_by_id(id, html, **kwargs):
365 """Return the html of the tag with the specified ID in the passed HTML document"""
366 return get_element_html_by_attribute('id', id, html, **kwargs)
367
368
369 def get_element_by_class(class_name, html):
370 """Return the content of the first tag with the specified class in the passed HTML document"""
371 retval = get_elements_by_class(class_name, html)
372 return retval[0] if retval else None
373
374
375 def get_element_html_by_class(class_name, html):
376 """Return the html of the first tag with the specified class in the passed HTML document"""
377 retval = get_elements_html_by_class(class_name, html)
378 return retval[0] if retval else None
379
380
381 def get_element_by_attribute(attribute, value, html, **kwargs):
382 retval = get_elements_by_attribute(attribute, value, html, **kwargs)
383 return retval[0] if retval else None
384
385
386 def get_element_html_by_attribute(attribute, value, html, **kargs):
387 retval = get_elements_html_by_attribute(attribute, value, html, **kargs)
388 return retval[0] if retval else None
389
390
391 def get_elements_by_class(class_name, html, **kargs):
392 """Return the content of all tags with the specified class in the passed HTML document as a list"""
393 return get_elements_by_attribute(
394 'class', r'[^\'"]*(?<=[\'"\s])%s(?=[\'"\s])[^\'"]*' % re.escape(class_name),
395 html, escape_value=False)
396
397
398 def get_elements_html_by_class(class_name, html):
399 """Return the html of all tags with the specified class in the passed HTML document as a list"""
400 return get_elements_html_by_attribute(
401 'class', r'[^\'"]*(?<=[\'"\s])%s(?=[\'"\s])[^\'"]*' % re.escape(class_name),
402 html, escape_value=False)
403
404
405 def get_elements_by_attribute(*args, **kwargs):
406 """Return the content of the tag with the specified attribute in the passed HTML document"""
407 return [content for content, _ in get_elements_text_and_html_by_attribute(*args, **kwargs)]
408
409
410 def get_elements_html_by_attribute(*args, **kwargs):
411 """Return the html of the tag with the specified attribute in the passed HTML document"""
412 return [whole for _, whole in get_elements_text_and_html_by_attribute(*args, **kwargs)]
413
414
415 def get_elements_text_and_html_by_attribute(attribute, value, html, *, tag=r'[\w:.-]+', escape_value=True):
416 """
417 Return the text (content) and the html (whole) of the tag with the specified
418 attribute in the passed HTML document
419 """
420 if not value:
421 return
422
423 quote = '' if re.match(r'''[\s"'`=<>]''', value) else '?'
424
425 value = re.escape(value) if escape_value else value
426
427 partial_element_re = rf'''(?x)
428 <(?P<tag>{tag})
429 (?:\s(?:[^>"']|"[^"]*"|'[^']*')*)?
430 \s{re.escape(attribute)}\s*=\s*(?P<_q>['"]{quote})(?-x:{value})(?P=_q)
431 '''
432
433 for m in re.finditer(partial_element_re, html):
434 content, whole = get_element_text_and_html_by_tag(m.group('tag'), html[m.start():])
435
436 yield (
437 unescapeHTML(re.sub(r'^(?P<q>["\'])(?P<content>.*)(?P=q)$', r'\g<content>', content, flags=re.DOTALL)),
438 whole
439 )
440
441
442 class HTMLBreakOnClosingTagParser(html.parser.HTMLParser):
443 """
444 HTML parser which raises HTMLBreakOnClosingTagException upon reaching the
445 closing tag for the first opening tag it has encountered, and can be used
446 as a context manager
447 """
448
449 class HTMLBreakOnClosingTagException(Exception):
450 pass
451
452 def __init__(self):
453 self.tagstack = collections.deque()
454 html.parser.HTMLParser.__init__(self)
455
456 def __enter__(self):
457 return self
458
459 def __exit__(self, *_):
460 self.close()
461
462 def close(self):
463 # handle_endtag does not return upon raising HTMLBreakOnClosingTagException,
464 # so data remains buffered; we no longer have any interest in it, thus
465 # override this method to discard it
466 pass
467
468 def handle_starttag(self, tag, _):
469 self.tagstack.append(tag)
470
471 def handle_endtag(self, tag):
472 if not self.tagstack:
473 raise compat_HTMLParseError('no tags in the stack')
474 while self.tagstack:
475 inner_tag = self.tagstack.pop()
476 if inner_tag == tag:
477 break
478 else:
479 raise compat_HTMLParseError(f'matching opening tag for closing {tag} tag not found')
480 if not self.tagstack:
481 raise self.HTMLBreakOnClosingTagException()
482
483
484 # XXX: This should be far less strict
485 def get_element_text_and_html_by_tag(tag, html):
486 """
487 For the first element with the specified tag in the passed HTML document
488 return its' content (text) and the whole element (html)
489 """
490 def find_or_raise(haystack, needle, exc):
491 try:
492 return haystack.index(needle)
493 except ValueError:
494 raise exc
495 closing_tag = f'</{tag}>'
496 whole_start = find_or_raise(
497 html, f'<{tag}', compat_HTMLParseError(f'opening {tag} tag not found'))
498 content_start = find_or_raise(
499 html[whole_start:], '>', compat_HTMLParseError(f'malformed opening {tag} tag'))
500 content_start += whole_start + 1
501 with HTMLBreakOnClosingTagParser() as parser:
502 parser.feed(html[whole_start:content_start])
503 if not parser.tagstack or parser.tagstack[0] != tag:
504 raise compat_HTMLParseError(f'parser did not match opening {tag} tag')
505 offset = content_start
506 while offset < len(html):
507 next_closing_tag_start = find_or_raise(
508 html[offset:], closing_tag,
509 compat_HTMLParseError(f'closing {tag} tag not found'))
510 next_closing_tag_end = next_closing_tag_start + len(closing_tag)
511 try:
512 parser.feed(html[offset:offset + next_closing_tag_end])
513 offset += next_closing_tag_end
514 except HTMLBreakOnClosingTagParser.HTMLBreakOnClosingTagException:
515 return html[content_start:offset + next_closing_tag_start], \
516 html[whole_start:offset + next_closing_tag_end]
517 raise compat_HTMLParseError('unexpected end of html')
518
519
520 class HTMLAttributeParser(html.parser.HTMLParser):
521 """Trivial HTML parser to gather the attributes for a single element"""
522
523 def __init__(self):
524 self.attrs = {}
525 html.parser.HTMLParser.__init__(self)
526
527 def handle_starttag(self, tag, attrs):
528 self.attrs = dict(attrs)
529 raise compat_HTMLParseError('done')
530
531
532 class HTMLListAttrsParser(html.parser.HTMLParser):
533 """HTML parser to gather the attributes for the elements of a list"""
534
535 def __init__(self):
536 html.parser.HTMLParser.__init__(self)
537 self.items = []
538 self._level = 0
539
540 def handle_starttag(self, tag, attrs):
541 if tag == 'li' and self._level == 0:
542 self.items.append(dict(attrs))
543 self._level += 1
544
545 def handle_endtag(self, tag):
546 self._level -= 1
547
548
549 def extract_attributes(html_element):
550 """Given a string for an HTML element such as
551 <el
552 a="foo" B="bar" c="&98;az" d=boz
553 empty= noval entity="&amp;"
554 sq='"' dq="'"
555 >
556 Decode and return a dictionary of attributes.
557 {
558 'a': 'foo', 'b': 'bar', c: 'baz', d: 'boz',
559 'empty': '', 'noval': None, 'entity': '&',
560 'sq': '"', 'dq': '\''
561 }.
562 """
563 parser = HTMLAttributeParser()
564 with contextlib.suppress(compat_HTMLParseError):
565 parser.feed(html_element)
566 parser.close()
567 return parser.attrs
568
569
570 def parse_list(webpage):
571 """Given a string for an series of HTML <li> elements,
572 return a dictionary of their attributes"""
573 parser = HTMLListAttrsParser()
574 parser.feed(webpage)
575 parser.close()
576 return parser.items
577
578
579 def clean_html(html):
580 """Clean an HTML snippet into a readable string"""
581
582 if html is None: # Convenience for sanitizing descriptions etc.
583 return html
584
585 html = re.sub(r'\s+', ' ', html)
586 html = re.sub(r'(?u)\s?<\s?br\s?/?\s?>\s?', '\n', html)
587 html = re.sub(r'(?u)<\s?/\s?p\s?>\s?<\s?p[^>]*>', '\n', html)
588 # Strip html tags
589 html = re.sub('<.*?>', '', html)
590 # Replace html entities
591 html = unescapeHTML(html)
592 return html.strip()
593
594
595 class LenientJSONDecoder(json.JSONDecoder):
596 # TODO: Write tests
597 def __init__(self, *args, transform_source=None, ignore_extra=False, close_objects=0, **kwargs):
598 self.transform_source, self.ignore_extra = transform_source, ignore_extra
599 self._close_attempts = 2 * close_objects
600 super().__init__(*args, **kwargs)
601
602 @staticmethod
603 def _close_object(err):
604 doc = err.doc[:err.pos]
605 # We need to add comma first to get the correct error message
606 if err.msg.startswith('Expecting \',\''):
607 return doc + ','
608 elif not doc.endswith(','):
609 return
610
611 if err.msg.startswith('Expecting property name'):
612 return doc[:-1] + '}'
613 elif err.msg.startswith('Expecting value'):
614 return doc[:-1] + ']'
615
616 def decode(self, s):
617 if self.transform_source:
618 s = self.transform_source(s)
619 for attempt in range(self._close_attempts + 1):
620 try:
621 if self.ignore_extra:
622 return self.raw_decode(s.lstrip())[0]
623 return super().decode(s)
624 except json.JSONDecodeError as e:
625 if e.pos is None:
626 raise
627 elif attempt < self._close_attempts:
628 s = self._close_object(e)
629 if s is not None:
630 continue
631 raise type(e)(f'{e.msg} in {s[e.pos-10:e.pos+10]!r}', s, e.pos)
632 assert False, 'Too many attempts to decode JSON'
633
634
635 def sanitize_open(filename, open_mode):
636 """Try to open the given filename, and slightly tweak it if this fails.
637
638 Attempts to open the given filename. If this fails, it tries to change
639 the filename slightly, step by step, until it's either able to open it
640 or it fails and raises a final exception, like the standard open()
641 function.
642
643 It returns the tuple (stream, definitive_file_name).
644 """
645 if filename == '-':
646 if sys.platform == 'win32':
647 import msvcrt
648
649 # stdout may be any IO stream, e.g. when using contextlib.redirect_stdout
650 with contextlib.suppress(io.UnsupportedOperation):
651 msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
652 return (sys.stdout.buffer if hasattr(sys.stdout, 'buffer') else sys.stdout, filename)
653
654 for attempt in range(2):
655 try:
656 try:
657 if sys.platform == 'win32':
658 # FIXME: An exclusive lock also locks the file from being read.
659 # Since windows locks are mandatory, don't lock the file on windows (for now).
660 # Ref: https://github.com/yt-dlp/yt-dlp/issues/3124
661 raise LockingUnsupportedError()
662 stream = locked_file(filename, open_mode, block=False).__enter__()
663 except OSError:
664 stream = open(filename, open_mode)
665 return stream, filename
666 except OSError as err:
667 if attempt or err.errno in (errno.EACCES,):
668 raise
669 old_filename, filename = filename, sanitize_path(filename)
670 if old_filename == filename:
671 raise
672
673
674 def timeconvert(timestr):
675 """Convert RFC 2822 defined time string into system timestamp"""
676 timestamp = None
677 timetuple = email.utils.parsedate_tz(timestr)
678 if timetuple is not None:
679 timestamp = email.utils.mktime_tz(timetuple)
680 return timestamp
681
682
683 def sanitize_filename(s, restricted=False, is_id=NO_DEFAULT):
684 """Sanitizes a string so it could be used as part of a filename.
685 @param restricted Use a stricter subset of allowed characters
686 @param is_id Whether this is an ID that should be kept unchanged if possible.
687 If unset, yt-dlp's new sanitization rules are in effect
688 """
689 if s == '':
690 return ''
691
692 def replace_insane(char):
693 if restricted and char in ACCENT_CHARS:
694 return ACCENT_CHARS[char]
695 elif not restricted and char == '\n':
696 return '\0 '
697 elif is_id is NO_DEFAULT and not restricted and char in '"*:<>?|/\\':
698 # Replace with their full-width unicode counterparts
699 return {'/': '\u29F8', '\\': '\u29f9'}.get(char, chr(ord(char) + 0xfee0))
700 elif char == '?' or ord(char) < 32 or ord(char) == 127:
701 return ''
702 elif char == '"':
703 return '' if restricted else '\''
704 elif char == ':':
705 return '\0_\0-' if restricted else '\0 \0-'
706 elif char in '\\/|*<>':
707 return '\0_'
708 if restricted and (char in '!&\'()[]{}$;`^,#' or char.isspace() or ord(char) > 127):
709 return '\0_'
710 return char
711
712 # Replace look-alike Unicode glyphs
713 if restricted and (is_id is NO_DEFAULT or not is_id):
714 s = unicodedata.normalize('NFKC', s)
715 s = re.sub(r'[0-9]+(?::[0-9]+)+', lambda m: m.group(0).replace(':', '_'), s) # Handle timestamps
716 result = ''.join(map(replace_insane, s))
717 if is_id is NO_DEFAULT:
718 result = re.sub(r'(\0.)(?:(?=\1)..)+', r'\1', result) # Remove repeated substitute chars
719 STRIP_RE = r'(?:\0.|[ _-])*'
720 result = re.sub(f'^\0.{STRIP_RE}|{STRIP_RE}\0.$', '', result) # Remove substitute chars from start/end
721 result = result.replace('\0', '') or '_'
722
723 if not is_id:
724 while '__' in result:
725 result = result.replace('__', '_')
726 result = result.strip('_')
727 # Common case of "Foreign band name - English song title"
728 if restricted and result.startswith('-_'):
729 result = result[2:]
730 if result.startswith('-'):
731 result = '_' + result[len('-'):]
732 result = result.lstrip('.')
733 if not result:
734 result = '_'
735 return result
736
737
738 def sanitize_path(s, force=False):
739 """Sanitizes and normalizes path on Windows"""
740 if sys.platform == 'win32':
741 force = False
742 drive_or_unc, _ = os.path.splitdrive(s)
743 elif force:
744 drive_or_unc = ''
745 else:
746 return s
747
748 norm_path = os.path.normpath(remove_start(s, drive_or_unc)).split(os.path.sep)
749 if drive_or_unc:
750 norm_path.pop(0)
751 sanitized_path = [
752 path_part if path_part in ['.', '..'] else re.sub(r'(?:[/<>:"\|\\?\*]|[\s.]$)', '#', path_part)
753 for path_part in norm_path]
754 if drive_or_unc:
755 sanitized_path.insert(0, drive_or_unc + os.path.sep)
756 elif force and s and s[0] == os.path.sep:
757 sanitized_path.insert(0, os.path.sep)
758 return os.path.join(*sanitized_path)
759
760
761 def sanitize_url(url, *, scheme='http'):
762 # Prepend protocol-less URLs with `http:` scheme in order to mitigate
763 # the number of unwanted failures due to missing protocol
764 if url is None:
765 return
766 elif url.startswith('//'):
767 return f'{scheme}:{url}'
768 # Fix some common typos seen so far
769 COMMON_TYPOS = (
770 # https://github.com/ytdl-org/youtube-dl/issues/15649
771 (r'^httpss://', r'https://'),
772 # https://bx1.be/lives/direct-tv/
773 (r'^rmtp([es]?)://', r'rtmp\1://'),
774 )
775 for mistake, fixup in COMMON_TYPOS:
776 if re.match(mistake, url):
777 return re.sub(mistake, fixup, url)
778 return url
779
780
781 def extract_basic_auth(url):
782 parts = urllib.parse.urlsplit(url)
783 if parts.username is None:
784 return url, None
785 url = urllib.parse.urlunsplit(parts._replace(netloc=(
786 parts.hostname if parts.port is None
787 else '%s:%d' % (parts.hostname, parts.port))))
788 auth_payload = base64.b64encode(
789 ('%s:%s' % (parts.username, parts.password or '')).encode())
790 return url, f'Basic {auth_payload.decode()}'
791
792
793 def sanitized_Request(url, *args, **kwargs):
794 url, auth_header = extract_basic_auth(escape_url(sanitize_url(url)))
795 if auth_header is not None:
796 headers = args[1] if len(args) >= 2 else kwargs.setdefault('headers', {})
797 headers['Authorization'] = auth_header
798 return urllib.request.Request(url, *args, **kwargs)
799
800
801 def expand_path(s):
802 """Expand shell variables and ~"""
803 return os.path.expandvars(compat_expanduser(s))
804
805
806 def orderedSet(iterable, *, lazy=False):
807 """Remove all duplicates from the input iterable"""
808 def _iter():
809 seen = [] # Do not use set since the items can be unhashable
810 for x in iterable:
811 if x not in seen:
812 seen.append(x)
813 yield x
814
815 return _iter() if lazy else list(_iter())
816
817
818 def _htmlentity_transform(entity_with_semicolon):
819 """Transforms an HTML entity to a character."""
820 entity = entity_with_semicolon[:-1]
821
822 # Known non-numeric HTML entity
823 if entity in html.entities.name2codepoint:
824 return chr(html.entities.name2codepoint[entity])
825
826 # TODO: HTML5 allows entities without a semicolon.
827 # E.g. '&Eacuteric' should be decoded as 'Éric'.
828 if entity_with_semicolon in html.entities.html5:
829 return html.entities.html5[entity_with_semicolon]
830
831 mobj = re.match(r'#(x[0-9a-fA-F]+|[0-9]+)', entity)
832 if mobj is not None:
833 numstr = mobj.group(1)
834 if numstr.startswith('x'):
835 base = 16
836 numstr = '0%s' % numstr
837 else:
838 base = 10
839 # See https://github.com/ytdl-org/youtube-dl/issues/7518
840 with contextlib.suppress(ValueError):
841 return chr(int(numstr, base))
842
843 # Unknown entity in name, return its literal representation
844 return '&%s;' % entity
845
846
847 def unescapeHTML(s):
848 if s is None:
849 return None
850 assert isinstance(s, str)
851
852 return re.sub(
853 r'&([^&;]+;)', lambda m: _htmlentity_transform(m.group(1)), s)
854
855
856 def escapeHTML(text):
857 return (
858 text
859 .replace('&', '&amp;')
860 .replace('<', '&lt;')
861 .replace('>', '&gt;')
862 .replace('"', '&quot;')
863 .replace("'", '&#39;')
864 )
865
866
867 def process_communicate_or_kill(p, *args, **kwargs):
868 deprecation_warning(f'"{__name__}.process_communicate_or_kill" is deprecated and may be removed '
869 f'in a future version. Use "{__name__}.Popen.communicate_or_kill" instead')
870 return Popen.communicate_or_kill(p, *args, **kwargs)
871
872
873 class Popen(subprocess.Popen):
874 if sys.platform == 'win32':
875 _startupinfo = subprocess.STARTUPINFO()
876 _startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
877 else:
878 _startupinfo = None
879
880 @staticmethod
881 def _fix_pyinstaller_ld_path(env):
882 """Restore LD_LIBRARY_PATH when using PyInstaller
883 Ref: https://github.com/pyinstaller/pyinstaller/blob/develop/doc/runtime-information.rst#ld_library_path--libpath-considerations
884 https://github.com/yt-dlp/yt-dlp/issues/4573
885 """
886 if not hasattr(sys, '_MEIPASS'):
887 return
888
889 def _fix(key):
890 orig = env.get(f'{key}_ORIG')
891 if orig is None:
892 env.pop(key, None)
893 else:
894 env[key] = orig
895
896 _fix('LD_LIBRARY_PATH') # Linux
897 _fix('DYLD_LIBRARY_PATH') # macOS
898
899 def __init__(self, *args, env=None, text=False, **kwargs):
900 if env is None:
901 env = os.environ.copy()
902 self._fix_pyinstaller_ld_path(env)
903
904 self.__text_mode = kwargs.get('encoding') or kwargs.get('errors') or text or kwargs.get('universal_newlines')
905 if text is True:
906 kwargs['universal_newlines'] = True # For 3.6 compatibility
907 kwargs.setdefault('encoding', 'utf-8')
908 kwargs.setdefault('errors', 'replace')
909 super().__init__(*args, env=env, **kwargs, startupinfo=self._startupinfo)
910
911 def communicate_or_kill(self, *args, **kwargs):
912 try:
913 return self.communicate(*args, **kwargs)
914 except BaseException: # Including KeyboardInterrupt
915 self.kill(timeout=None)
916 raise
917
918 def kill(self, *, timeout=0):
919 super().kill()
920 if timeout != 0:
921 self.wait(timeout=timeout)
922
923 @classmethod
924 def run(cls, *args, timeout=None, **kwargs):
925 with cls(*args, **kwargs) as proc:
926 default = '' if proc.__text_mode else b''
927 stdout, stderr = proc.communicate_or_kill(timeout=timeout)
928 return stdout or default, stderr or default, proc.returncode
929
930
931 def get_subprocess_encoding():
932 if sys.platform == 'win32' and sys.getwindowsversion()[0] >= 5:
933 # For subprocess calls, encode with locale encoding
934 # Refer to http://stackoverflow.com/a/9951851/35070
935 encoding = preferredencoding()
936 else:
937 encoding = sys.getfilesystemencoding()
938 if encoding is None:
939 encoding = 'utf-8'
940 return encoding
941
942
943 def encodeFilename(s, for_subprocess=False):
944 assert isinstance(s, str)
945 return s
946
947
948 def decodeFilename(b, for_subprocess=False):
949 return b
950
951
952 def encodeArgument(s):
953 # Legacy code that uses byte strings
954 # Uncomment the following line after fixing all post processors
955 # assert isinstance(s, str), 'Internal error: %r should be of type %r, is %r' % (s, str, type(s))
956 return s if isinstance(s, str) else s.decode('ascii')
957
958
959 def decodeArgument(b):
960 return b
961
962
963 def decodeOption(optval):
964 if optval is None:
965 return optval
966 if isinstance(optval, bytes):
967 optval = optval.decode(preferredencoding())
968
969 assert isinstance(optval, str)
970 return optval
971
972
973 _timetuple = collections.namedtuple('Time', ('hours', 'minutes', 'seconds', 'milliseconds'))
974
975
976 def timetuple_from_msec(msec):
977 secs, msec = divmod(msec, 1000)
978 mins, secs = divmod(secs, 60)
979 hrs, mins = divmod(mins, 60)
980 return _timetuple(hrs, mins, secs, msec)
981
982
983 def formatSeconds(secs, delim=':', msec=False):
984 time = timetuple_from_msec(secs * 1000)
985 if time.hours:
986 ret = '%d%s%02d%s%02d' % (time.hours, delim, time.minutes, delim, time.seconds)
987 elif time.minutes:
988 ret = '%d%s%02d' % (time.minutes, delim, time.seconds)
989 else:
990 ret = '%d' % time.seconds
991 return '%s.%03d' % (ret, time.milliseconds) if msec else ret
992
993
994 def _ssl_load_windows_store_certs(ssl_context, storename):
995 # Code adapted from _load_windows_store_certs in https://github.com/python/cpython/blob/main/Lib/ssl.py
996 try:
997 certs = [cert for cert, encoding, trust in ssl.enum_certificates(storename)
998 if encoding == 'x509_asn' and (
999 trust is True or ssl.Purpose.SERVER_AUTH.oid in trust)]
1000 except PermissionError:
1001 return
1002 for cert in certs:
1003 with contextlib.suppress(ssl.SSLError):
1004 ssl_context.load_verify_locations(cadata=cert)
1005
1006
1007 def make_HTTPS_handler(params, **kwargs):
1008 opts_check_certificate = not params.get('nocheckcertificate')
1009 context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
1010 context.check_hostname = opts_check_certificate
1011 if params.get('legacyserverconnect'):
1012 context.options |= 4 # SSL_OP_LEGACY_SERVER_CONNECT
1013 # Allow use of weaker ciphers in Python 3.10+. See https://bugs.python.org/issue43998
1014 context.set_ciphers('DEFAULT')
1015 elif (
1016 sys.version_info < (3, 10)
1017 and ssl.OPENSSL_VERSION_INFO >= (1, 1, 1)
1018 and not ssl.OPENSSL_VERSION.startswith('LibreSSL')
1019 ):
1020 # Backport the default SSL ciphers and minimum TLS version settings from Python 3.10 [1].
1021 # This is to ensure consistent behavior across Python versions, and help avoid fingerprinting
1022 # in some situations [2][3].
1023 # Python 3.10 only supports OpenSSL 1.1.1+ [4]. Because this change is likely
1024 # untested on older versions, we only apply this to OpenSSL 1.1.1+ to be safe.
1025 # LibreSSL is excluded until further investigation due to cipher support issues [5][6].
1026 # 1. https://github.com/python/cpython/commit/e983252b516edb15d4338b0a47631b59ef1e2536
1027 # 2. https://github.com/yt-dlp/yt-dlp/issues/4627
1028 # 3. https://github.com/yt-dlp/yt-dlp/pull/5294
1029 # 4. https://peps.python.org/pep-0644/
1030 # 5. https://peps.python.org/pep-0644/#libressl-support
1031 # 6. https://github.com/yt-dlp/yt-dlp/commit/5b9f253fa0aee996cf1ed30185d4b502e00609c4#commitcomment-89054368
1032 context.set_ciphers('@SECLEVEL=2:ECDH+AESGCM:ECDH+CHACHA20:ECDH+AES:DHE+AES:!aNULL:!eNULL:!aDSS:!SHA1:!AESCCM')
1033 context.minimum_version = ssl.TLSVersion.TLSv1_2
1034
1035 context.verify_mode = ssl.CERT_REQUIRED if opts_check_certificate else ssl.CERT_NONE
1036 if opts_check_certificate:
1037 if has_certifi and 'no-certifi' not in params.get('compat_opts', []):
1038 context.load_verify_locations(cafile=certifi.where())
1039 else:
1040 try:
1041 context.load_default_certs()
1042 # Work around the issue in load_default_certs when there are bad certificates. See:
1043 # https://github.com/yt-dlp/yt-dlp/issues/1060,
1044 # https://bugs.python.org/issue35665, https://bugs.python.org/issue45312
1045 except ssl.SSLError:
1046 # enum_certificates is not present in mingw python. See https://github.com/yt-dlp/yt-dlp/issues/1151
1047 if sys.platform == 'win32' and hasattr(ssl, 'enum_certificates'):
1048 for storename in ('CA', 'ROOT'):
1049 _ssl_load_windows_store_certs(context, storename)
1050 context.set_default_verify_paths()
1051
1052 client_certfile = params.get('client_certificate')
1053 if client_certfile:
1054 try:
1055 context.load_cert_chain(
1056 client_certfile, keyfile=params.get('client_certificate_key'),
1057 password=params.get('client_certificate_password'))
1058 except ssl.SSLError:
1059 raise YoutubeDLError('Unable to load client certificate')
1060
1061 # Some servers may reject requests if ALPN extension is not sent. See:
1062 # https://github.com/python/cpython/issues/85140
1063 # https://github.com/yt-dlp/yt-dlp/issues/3878
1064 with contextlib.suppress(NotImplementedError):
1065 context.set_alpn_protocols(['http/1.1'])
1066
1067 return YoutubeDLHTTPSHandler(params, context=context, **kwargs)
1068
1069
1070 def bug_reports_message(before=';'):
1071 from .update import REPOSITORY
1072
1073 msg = (f'please report this issue on https://github.com/{REPOSITORY}/issues?q= , '
1074 'filling out the appropriate issue template. Confirm you are on the latest version using yt-dlp -U')
1075
1076 before = before.rstrip()
1077 if not before or before.endswith(('.', '!', '?')):
1078 msg = msg[0].title() + msg[1:]
1079
1080 return (before + ' ' if before else '') + msg
1081
1082
1083 class YoutubeDLError(Exception):
1084 """Base exception for YoutubeDL errors."""
1085 msg = None
1086
1087 def __init__(self, msg=None):
1088 if msg is not None:
1089 self.msg = msg
1090 elif self.msg is None:
1091 self.msg = type(self).__name__
1092 super().__init__(self.msg)
1093
1094
1095 network_exceptions = [urllib.error.URLError, http.client.HTTPException, socket.error]
1096 if hasattr(ssl, 'CertificateError'):
1097 network_exceptions.append(ssl.CertificateError)
1098 network_exceptions = tuple(network_exceptions)
1099
1100
1101 class ExtractorError(YoutubeDLError):
1102 """Error during info extraction."""
1103
1104 def __init__(self, msg, tb=None, expected=False, cause=None, video_id=None, ie=None):
1105 """ tb, if given, is the original traceback (so that it can be printed out).
1106 If expected is set, this is a normal error message and most likely not a bug in yt-dlp.
1107 """
1108 if sys.exc_info()[0] in network_exceptions:
1109 expected = True
1110
1111 self.orig_msg = str(msg)
1112 self.traceback = tb
1113 self.expected = expected
1114 self.cause = cause
1115 self.video_id = video_id
1116 self.ie = ie
1117 self.exc_info = sys.exc_info() # preserve original exception
1118 if isinstance(self.exc_info[1], ExtractorError):
1119 self.exc_info = self.exc_info[1].exc_info
1120 super().__init__(self.__msg)
1121
1122 @property
1123 def __msg(self):
1124 return ''.join((
1125 format_field(self.ie, None, '[%s] '),
1126 format_field(self.video_id, None, '%s: '),
1127 self.orig_msg,
1128 format_field(self.cause, None, ' (caused by %r)'),
1129 '' if self.expected else bug_reports_message()))
1130
1131 def format_traceback(self):
1132 return join_nonempty(
1133 self.traceback and ''.join(traceback.format_tb(self.traceback)),
1134 self.cause and ''.join(traceback.format_exception(None, self.cause, self.cause.__traceback__)[1:]),
1135 delim='\n') or None
1136
1137 def __setattr__(self, name, value):
1138 super().__setattr__(name, value)
1139 if getattr(self, 'msg', None) and name not in ('msg', 'args'):
1140 self.msg = self.__msg or type(self).__name__
1141 self.args = (self.msg, ) # Cannot be property
1142
1143
1144 class UnsupportedError(ExtractorError):
1145 def __init__(self, url):
1146 super().__init__(
1147 'Unsupported URL: %s' % url, expected=True)
1148 self.url = url
1149
1150
1151 class RegexNotFoundError(ExtractorError):
1152 """Error when a regex didn't match"""
1153 pass
1154
1155
1156 class GeoRestrictedError(ExtractorError):
1157 """Geographic restriction Error exception.
1158
1159 This exception may be thrown when a video is not available from your
1160 geographic location due to geographic restrictions imposed by a website.
1161 """
1162
1163 def __init__(self, msg, countries=None, **kwargs):
1164 kwargs['expected'] = True
1165 super().__init__(msg, **kwargs)
1166 self.countries = countries
1167
1168
1169 class UserNotLive(ExtractorError):
1170 """Error when a channel/user is not live"""
1171
1172 def __init__(self, msg=None, **kwargs):
1173 kwargs['expected'] = True
1174 super().__init__(msg or 'The channel is not currently live', **kwargs)
1175
1176
1177 class DownloadError(YoutubeDLError):
1178 """Download Error exception.
1179
1180 This exception may be thrown by FileDownloader objects if they are not
1181 configured to continue on errors. They will contain the appropriate
1182 error message.
1183 """
1184
1185 def __init__(self, msg, exc_info=None):
1186 """ exc_info, if given, is the original exception that caused the trouble (as returned by sys.exc_info()). """
1187 super().__init__(msg)
1188 self.exc_info = exc_info
1189
1190
1191 class EntryNotInPlaylist(YoutubeDLError):
1192 """Entry not in playlist exception.
1193
1194 This exception will be thrown by YoutubeDL when a requested entry
1195 is not found in the playlist info_dict
1196 """
1197 msg = 'Entry not found in info'
1198
1199
1200 class SameFileError(YoutubeDLError):
1201 """Same File exception.
1202
1203 This exception will be thrown by FileDownloader objects if they detect
1204 multiple files would have to be downloaded to the same file on disk.
1205 """
1206 msg = 'Fixed output name but more than one file to download'
1207
1208 def __init__(self, filename=None):
1209 if filename is not None:
1210 self.msg += f': {filename}'
1211 super().__init__(self.msg)
1212
1213
1214 class PostProcessingError(YoutubeDLError):
1215 """Post Processing exception.
1216
1217 This exception may be raised by PostProcessor's .run() method to
1218 indicate an error in the postprocessing task.
1219 """
1220
1221
1222 class DownloadCancelled(YoutubeDLError):
1223 """ Exception raised when the download queue should be interrupted """
1224 msg = 'The download was cancelled'
1225
1226
1227 class ExistingVideoReached(DownloadCancelled):
1228 """ --break-on-existing triggered """
1229 msg = 'Encountered a video that is already in the archive, stopping due to --break-on-existing'
1230
1231
1232 class RejectedVideoReached(DownloadCancelled):
1233 """ --break-on-reject triggered """
1234 msg = 'Encountered a video that did not match filter, stopping due to --break-on-reject'
1235
1236
1237 class MaxDownloadsReached(DownloadCancelled):
1238 """ --max-downloads limit has been reached. """
1239 msg = 'Maximum number of downloads reached, stopping due to --max-downloads'
1240
1241
1242 class ReExtractInfo(YoutubeDLError):
1243 """ Video info needs to be re-extracted. """
1244
1245 def __init__(self, msg, expected=False):
1246 super().__init__(msg)
1247 self.expected = expected
1248
1249
1250 class ThrottledDownload(ReExtractInfo):
1251 """ Download speed below --throttled-rate. """
1252 msg = 'The download speed is below throttle limit'
1253
1254 def __init__(self):
1255 super().__init__(self.msg, expected=False)
1256
1257
1258 class UnavailableVideoError(YoutubeDLError):
1259 """Unavailable Format exception.
1260
1261 This exception will be thrown when a video is requested
1262 in a format that is not available for that video.
1263 """
1264 msg = 'Unable to download video'
1265
1266 def __init__(self, err=None):
1267 if err is not None:
1268 self.msg += f': {err}'
1269 super().__init__(self.msg)
1270
1271
1272 class ContentTooShortError(YoutubeDLError):
1273 """Content Too Short exception.
1274
1275 This exception may be raised by FileDownloader objects when a file they
1276 download is too small for what the server announced first, indicating
1277 the connection was probably interrupted.
1278 """
1279
1280 def __init__(self, downloaded, expected):
1281 super().__init__(f'Downloaded {downloaded} bytes, expected {expected} bytes')
1282 # Both in bytes
1283 self.downloaded = downloaded
1284 self.expected = expected
1285
1286
1287 class XAttrMetadataError(YoutubeDLError):
1288 def __init__(self, code=None, msg='Unknown error'):
1289 super().__init__(msg)
1290 self.code = code
1291 self.msg = msg
1292
1293 # Parsing code and msg
1294 if (self.code in (errno.ENOSPC, errno.EDQUOT)
1295 or 'No space left' in self.msg or 'Disk quota exceeded' in self.msg):
1296 self.reason = 'NO_SPACE'
1297 elif self.code == errno.E2BIG or 'Argument list too long' in self.msg:
1298 self.reason = 'VALUE_TOO_LONG'
1299 else:
1300 self.reason = 'NOT_SUPPORTED'
1301
1302
1303 class XAttrUnavailableError(YoutubeDLError):
1304 pass
1305
1306
1307 def _create_http_connection(ydl_handler, http_class, is_https, *args, **kwargs):
1308 hc = http_class(*args, **kwargs)
1309 source_address = ydl_handler._params.get('source_address')
1310
1311 if source_address is not None:
1312 # This is to workaround _create_connection() from socket where it will try all
1313 # address data from getaddrinfo() including IPv6. This filters the result from
1314 # getaddrinfo() based on the source_address value.
1315 # This is based on the cpython socket.create_connection() function.
1316 # https://github.com/python/cpython/blob/master/Lib/socket.py#L691
1317 def _create_connection(address, timeout=socket._GLOBAL_DEFAULT_TIMEOUT, source_address=None):
1318 host, port = address
1319 err = None
1320 addrs = socket.getaddrinfo(host, port, 0, socket.SOCK_STREAM)
1321 af = socket.AF_INET if '.' in source_address[0] else socket.AF_INET6
1322 ip_addrs = [addr for addr in addrs if addr[0] == af]
1323 if addrs and not ip_addrs:
1324 ip_version = 'v4' if af == socket.AF_INET else 'v6'
1325 raise OSError(
1326 "No remote IP%s addresses available for connect, can't use '%s' as source address"
1327 % (ip_version, source_address[0]))
1328 for res in ip_addrs:
1329 af, socktype, proto, canonname, sa = res
1330 sock = None
1331 try:
1332 sock = socket.socket(af, socktype, proto)
1333 if timeout is not socket._GLOBAL_DEFAULT_TIMEOUT:
1334 sock.settimeout(timeout)
1335 sock.bind(source_address)
1336 sock.connect(sa)
1337 err = None # Explicitly break reference cycle
1338 return sock
1339 except OSError as _:
1340 err = _
1341 if sock is not None:
1342 sock.close()
1343 if err is not None:
1344 raise err
1345 else:
1346 raise OSError('getaddrinfo returns an empty list')
1347 if hasattr(hc, '_create_connection'):
1348 hc._create_connection = _create_connection
1349 hc.source_address = (source_address, 0)
1350
1351 return hc
1352
1353
1354 def handle_youtubedl_headers(headers):
1355 filtered_headers = headers
1356
1357 if 'Youtubedl-no-compression' in filtered_headers:
1358 filtered_headers = {k: v for k, v in filtered_headers.items() if k.lower() != 'accept-encoding'}
1359 del filtered_headers['Youtubedl-no-compression']
1360
1361 return filtered_headers
1362
1363
1364 class YoutubeDLHandler(urllib.request.HTTPHandler):
1365 """Handler for HTTP requests and responses.
1366
1367 This class, when installed with an OpenerDirector, automatically adds
1368 the standard headers to every HTTP request and handles gzipped and
1369 deflated responses from web servers. If compression is to be avoided in
1370 a particular request, the original request in the program code only has
1371 to include the HTTP header "Youtubedl-no-compression", which will be
1372 removed before making the real request.
1373
1374 Part of this code was copied from:
1375
1376 http://techknack.net/python-urllib2-handlers/
1377
1378 Andrew Rowls, the author of that code, agreed to release it to the
1379 public domain.
1380 """
1381
1382 def __init__(self, params, *args, **kwargs):
1383 urllib.request.HTTPHandler.__init__(self, *args, **kwargs)
1384 self._params = params
1385
1386 def http_open(self, req):
1387 conn_class = http.client.HTTPConnection
1388
1389 socks_proxy = req.headers.get('Ytdl-socks-proxy')
1390 if socks_proxy:
1391 conn_class = make_socks_conn_class(conn_class, socks_proxy)
1392 del req.headers['Ytdl-socks-proxy']
1393
1394 return self.do_open(functools.partial(
1395 _create_http_connection, self, conn_class, False),
1396 req)
1397
1398 @staticmethod
1399 def deflate(data):
1400 if not data:
1401 return data
1402 try:
1403 return zlib.decompress(data, -zlib.MAX_WBITS)
1404 except zlib.error:
1405 return zlib.decompress(data)
1406
1407 @staticmethod
1408 def brotli(data):
1409 if not data:
1410 return data
1411 return brotli.decompress(data)
1412
1413 def http_request(self, req):
1414 # According to RFC 3986, URLs can not contain non-ASCII characters, however this is not
1415 # always respected by websites, some tend to give out URLs with non percent-encoded
1416 # non-ASCII characters (see telemb.py, ard.py [#3412])
1417 # urllib chokes on URLs with non-ASCII characters (see http://bugs.python.org/issue3991)
1418 # To work around aforementioned issue we will replace request's original URL with
1419 # percent-encoded one
1420 # Since redirects are also affected (e.g. http://www.southpark.de/alle-episoden/s18e09)
1421 # the code of this workaround has been moved here from YoutubeDL.urlopen()
1422 url = req.get_full_url()
1423 url_escaped = escape_url(url)
1424
1425 # Substitute URL if any change after escaping
1426 if url != url_escaped:
1427 req = update_Request(req, url=url_escaped)
1428
1429 for h, v in self._params.get('http_headers', std_headers).items():
1430 # Capitalize is needed because of Python bug 2275: http://bugs.python.org/issue2275
1431 # The dict keys are capitalized because of this bug by urllib
1432 if h.capitalize() not in req.headers:
1433 req.add_header(h, v)
1434
1435 if 'Accept-encoding' not in req.headers:
1436 req.add_header('Accept-encoding', ', '.join(SUPPORTED_ENCODINGS))
1437
1438 req.headers = handle_youtubedl_headers(req.headers)
1439
1440 return super().do_request_(req)
1441
1442 def http_response(self, req, resp):
1443 old_resp = resp
1444 # gzip
1445 if resp.headers.get('Content-encoding', '') == 'gzip':
1446 content = resp.read()
1447 gz = gzip.GzipFile(fileobj=io.BytesIO(content), mode='rb')
1448 try:
1449 uncompressed = io.BytesIO(gz.read())
1450 except OSError as original_ioerror:
1451 # There may be junk add the end of the file
1452 # See http://stackoverflow.com/q/4928560/35070 for details
1453 for i in range(1, 1024):
1454 try:
1455 gz = gzip.GzipFile(fileobj=io.BytesIO(content[:-i]), mode='rb')
1456 uncompressed = io.BytesIO(gz.read())
1457 except OSError:
1458 continue
1459 break
1460 else:
1461 raise original_ioerror
1462 resp = urllib.request.addinfourl(uncompressed, old_resp.headers, old_resp.url, old_resp.code)
1463 resp.msg = old_resp.msg
1464 # deflate
1465 if resp.headers.get('Content-encoding', '') == 'deflate':
1466 gz = io.BytesIO(self.deflate(resp.read()))
1467 resp = urllib.request.addinfourl(gz, old_resp.headers, old_resp.url, old_resp.code)
1468 resp.msg = old_resp.msg
1469 # brotli
1470 if resp.headers.get('Content-encoding', '') == 'br':
1471 resp = urllib.request.addinfourl(
1472 io.BytesIO(self.brotli(resp.read())), old_resp.headers, old_resp.url, old_resp.code)
1473 resp.msg = old_resp.msg
1474 # Percent-encode redirect URL of Location HTTP header to satisfy RFC 3986 (see
1475 # https://github.com/ytdl-org/youtube-dl/issues/6457).
1476 if 300 <= resp.code < 400:
1477 location = resp.headers.get('Location')
1478 if location:
1479 # As of RFC 2616 default charset is iso-8859-1 that is respected by python 3
1480 location = location.encode('iso-8859-1').decode()
1481 location_escaped = escape_url(location)
1482 if location != location_escaped:
1483 del resp.headers['Location']
1484 resp.headers['Location'] = location_escaped
1485 return resp
1486
1487 https_request = http_request
1488 https_response = http_response
1489
1490
1491 def make_socks_conn_class(base_class, socks_proxy):
1492 assert issubclass(base_class, (
1493 http.client.HTTPConnection, http.client.HTTPSConnection))
1494
1495 url_components = urllib.parse.urlparse(socks_proxy)
1496 if url_components.scheme.lower() == 'socks5':
1497 socks_type = ProxyType.SOCKS5
1498 elif url_components.scheme.lower() in ('socks', 'socks4'):
1499 socks_type = ProxyType.SOCKS4
1500 elif url_components.scheme.lower() == 'socks4a':
1501 socks_type = ProxyType.SOCKS4A
1502
1503 def unquote_if_non_empty(s):
1504 if not s:
1505 return s
1506 return urllib.parse.unquote_plus(s)
1507
1508 proxy_args = (
1509 socks_type,
1510 url_components.hostname, url_components.port or 1080,
1511 True, # Remote DNS
1512 unquote_if_non_empty(url_components.username),
1513 unquote_if_non_empty(url_components.password),
1514 )
1515
1516 class SocksConnection(base_class):
1517 def connect(self):
1518 self.sock = sockssocket()
1519 self.sock.setproxy(*proxy_args)
1520 if isinstance(self.timeout, (int, float)):
1521 self.sock.settimeout(self.timeout)
1522 self.sock.connect((self.host, self.port))
1523
1524 if isinstance(self, http.client.HTTPSConnection):
1525 if hasattr(self, '_context'): # Python > 2.6
1526 self.sock = self._context.wrap_socket(
1527 self.sock, server_hostname=self.host)
1528 else:
1529 self.sock = ssl.wrap_socket(self.sock)
1530
1531 return SocksConnection
1532
1533
1534 class YoutubeDLHTTPSHandler(urllib.request.HTTPSHandler):
1535 def __init__(self, params, https_conn_class=None, *args, **kwargs):
1536 urllib.request.HTTPSHandler.__init__(self, *args, **kwargs)
1537 self._https_conn_class = https_conn_class or http.client.HTTPSConnection
1538 self._params = params
1539
1540 def https_open(self, req):
1541 kwargs = {}
1542 conn_class = self._https_conn_class
1543
1544 if hasattr(self, '_context'): # python > 2.6
1545 kwargs['context'] = self._context
1546 if hasattr(self, '_check_hostname'): # python 3.x
1547 kwargs['check_hostname'] = self._check_hostname
1548
1549 socks_proxy = req.headers.get('Ytdl-socks-proxy')
1550 if socks_proxy:
1551 conn_class = make_socks_conn_class(conn_class, socks_proxy)
1552 del req.headers['Ytdl-socks-proxy']
1553
1554 try:
1555 return self.do_open(
1556 functools.partial(_create_http_connection, self, conn_class, True), req, **kwargs)
1557 except urllib.error.URLError as e:
1558 if (isinstance(e.reason, ssl.SSLError)
1559 and getattr(e.reason, 'reason', None) == 'SSLV3_ALERT_HANDSHAKE_FAILURE'):
1560 raise YoutubeDLError('SSLV3_ALERT_HANDSHAKE_FAILURE: Try using --legacy-server-connect')
1561 raise
1562
1563
1564 def is_path_like(f):
1565 return isinstance(f, (str, bytes, os.PathLike))
1566
1567
1568 class YoutubeDLCookieJar(http.cookiejar.MozillaCookieJar):
1569 """
1570 See [1] for cookie file format.
1571
1572 1. https://curl.haxx.se/docs/http-cookies.html
1573 """
1574 _HTTPONLY_PREFIX = '#HttpOnly_'
1575 _ENTRY_LEN = 7
1576 _HEADER = '''# Netscape HTTP Cookie File
1577 # This file is generated by yt-dlp. Do not edit.
1578
1579 '''
1580 _CookieFileEntry = collections.namedtuple(
1581 'CookieFileEntry',
1582 ('domain_name', 'include_subdomains', 'path', 'https_only', 'expires_at', 'name', 'value'))
1583
1584 def __init__(self, filename=None, *args, **kwargs):
1585 super().__init__(None, *args, **kwargs)
1586 if is_path_like(filename):
1587 filename = os.fspath(filename)
1588 self.filename = filename
1589
1590 @staticmethod
1591 def _true_or_false(cndn):
1592 return 'TRUE' if cndn else 'FALSE'
1593
1594 @contextlib.contextmanager
1595 def open(self, file, *, write=False):
1596 if is_path_like(file):
1597 with open(file, 'w' if write else 'r', encoding='utf-8') as f:
1598 yield f
1599 else:
1600 if write:
1601 file.truncate(0)
1602 yield file
1603
1604 def _really_save(self, f, ignore_discard=False, ignore_expires=False):
1605 now = time.time()
1606 for cookie in self:
1607 if (not ignore_discard and cookie.discard
1608 or not ignore_expires and cookie.is_expired(now)):
1609 continue
1610 name, value = cookie.name, cookie.value
1611 if value is None:
1612 # cookies.txt regards 'Set-Cookie: foo' as a cookie
1613 # with no name, whereas http.cookiejar regards it as a
1614 # cookie with no value.
1615 name, value = '', name
1616 f.write('%s\n' % '\t'.join((
1617 cookie.domain,
1618 self._true_or_false(cookie.domain.startswith('.')),
1619 cookie.path,
1620 self._true_or_false(cookie.secure),
1621 str_or_none(cookie.expires, default=''),
1622 name, value
1623 )))
1624
1625 def save(self, filename=None, *args, **kwargs):
1626 """
1627 Save cookies to a file.
1628 Code is taken from CPython 3.6
1629 https://github.com/python/cpython/blob/8d999cbf4adea053be6dbb612b9844635c4dfb8e/Lib/http/cookiejar.py#L2091-L2117 """
1630
1631 if filename is None:
1632 if self.filename is not None:
1633 filename = self.filename
1634 else:
1635 raise ValueError(http.cookiejar.MISSING_FILENAME_TEXT)
1636
1637 # Store session cookies with `expires` set to 0 instead of an empty string
1638 for cookie in self:
1639 if cookie.expires is None:
1640 cookie.expires = 0
1641
1642 with self.open(filename, write=True) as f:
1643 f.write(self._HEADER)
1644 self._really_save(f, *args, **kwargs)
1645
1646 def load(self, filename=None, ignore_discard=False, ignore_expires=False):
1647 """Load cookies from a file."""
1648 if filename is None:
1649 if self.filename is not None:
1650 filename = self.filename
1651 else:
1652 raise ValueError(http.cookiejar.MISSING_FILENAME_TEXT)
1653
1654 def prepare_line(line):
1655 if line.startswith(self._HTTPONLY_PREFIX):
1656 line = line[len(self._HTTPONLY_PREFIX):]
1657 # comments and empty lines are fine
1658 if line.startswith('#') or not line.strip():
1659 return line
1660 cookie_list = line.split('\t')
1661 if len(cookie_list) != self._ENTRY_LEN:
1662 raise http.cookiejar.LoadError('invalid length %d' % len(cookie_list))
1663 cookie = self._CookieFileEntry(*cookie_list)
1664 if cookie.expires_at and not cookie.expires_at.isdigit():
1665 raise http.cookiejar.LoadError('invalid expires at %s' % cookie.expires_at)
1666 return line
1667
1668 cf = io.StringIO()
1669 with self.open(filename) as f:
1670 for line in f:
1671 try:
1672 cf.write(prepare_line(line))
1673 except http.cookiejar.LoadError as e:
1674 if f'{line.strip()} '[0] in '[{"':
1675 raise http.cookiejar.LoadError(
1676 'Cookies file must be Netscape formatted, not JSON. See '
1677 'https://github.com/yt-dlp/yt-dlp/wiki/FAQ#how-do-i-pass-cookies-to-yt-dlp')
1678 write_string(f'WARNING: skipping cookie file entry due to {e}: {line!r}\n')
1679 continue
1680 cf.seek(0)
1681 self._really_load(cf, filename, ignore_discard, ignore_expires)
1682 # Session cookies are denoted by either `expires` field set to
1683 # an empty string or 0. MozillaCookieJar only recognizes the former
1684 # (see [1]). So we need force the latter to be recognized as session
1685 # cookies on our own.
1686 # Session cookies may be important for cookies-based authentication,
1687 # e.g. usually, when user does not check 'Remember me' check box while
1688 # logging in on a site, some important cookies are stored as session
1689 # cookies so that not recognizing them will result in failed login.
1690 # 1. https://bugs.python.org/issue17164
1691 for cookie in self:
1692 # Treat `expires=0` cookies as session cookies
1693 if cookie.expires == 0:
1694 cookie.expires = None
1695 cookie.discard = True
1696
1697
1698 class YoutubeDLCookieProcessor(urllib.request.HTTPCookieProcessor):
1699 def __init__(self, cookiejar=None):
1700 urllib.request.HTTPCookieProcessor.__init__(self, cookiejar)
1701
1702 def http_response(self, request, response):
1703 return urllib.request.HTTPCookieProcessor.http_response(self, request, response)
1704
1705 https_request = urllib.request.HTTPCookieProcessor.http_request
1706 https_response = http_response
1707
1708
1709 class YoutubeDLRedirectHandler(urllib.request.HTTPRedirectHandler):
1710 """YoutubeDL redirect handler
1711
1712 The code is based on HTTPRedirectHandler implementation from CPython [1].
1713
1714 This redirect handler solves two issues:
1715 - ensures redirect URL is always unicode under python 2
1716 - introduces support for experimental HTTP response status code
1717 308 Permanent Redirect [2] used by some sites [3]
1718
1719 1. https://github.com/python/cpython/blob/master/Lib/urllib/request.py
1720 2. https://developer.mozilla.org/en-US/docs/Web/HTTP/Status/308
1721 3. https://github.com/ytdl-org/youtube-dl/issues/28768
1722 """
1723
1724 http_error_301 = http_error_303 = http_error_307 = http_error_308 = urllib.request.HTTPRedirectHandler.http_error_302
1725
1726 def redirect_request(self, req, fp, code, msg, headers, newurl):
1727 """Return a Request or None in response to a redirect.
1728
1729 This is called by the http_error_30x methods when a
1730 redirection response is received. If a redirection should
1731 take place, return a new Request to allow http_error_30x to
1732 perform the redirect. Otherwise, raise HTTPError if no-one
1733 else should try to handle this url. Return None if you can't
1734 but another Handler might.
1735 """
1736 m = req.get_method()
1737 if (not (code in (301, 302, 303, 307, 308) and m in ("GET", "HEAD")
1738 or code in (301, 302, 303) and m == "POST")):
1739 raise urllib.error.HTTPError(req.full_url, code, msg, headers, fp)
1740 # Strictly (according to RFC 2616), 301 or 302 in response to
1741 # a POST MUST NOT cause a redirection without confirmation
1742 # from the user (of urllib.request, in this case). In practice,
1743 # essentially all clients do redirect in this case, so we do
1744 # the same.
1745
1746 # Be conciliant with URIs containing a space. This is mainly
1747 # redundant with the more complete encoding done in http_error_302(),
1748 # but it is kept for compatibility with other callers.
1749 newurl = newurl.replace(' ', '%20')
1750
1751 CONTENT_HEADERS = ("content-length", "content-type")
1752 # NB: don't use dict comprehension for python 2.6 compatibility
1753 newheaders = {k: v for k, v in req.headers.items() if k.lower() not in CONTENT_HEADERS}
1754
1755 # A 303 must either use GET or HEAD for subsequent request
1756 # https://datatracker.ietf.org/doc/html/rfc7231#section-6.4.4
1757 if code == 303 and m != 'HEAD':
1758 m = 'GET'
1759 # 301 and 302 redirects are commonly turned into a GET from a POST
1760 # for subsequent requests by browsers, so we'll do the same.
1761 # https://datatracker.ietf.org/doc/html/rfc7231#section-6.4.2
1762 # https://datatracker.ietf.org/doc/html/rfc7231#section-6.4.3
1763 if code in (301, 302) and m == 'POST':
1764 m = 'GET'
1765
1766 return urllib.request.Request(
1767 newurl, headers=newheaders, origin_req_host=req.origin_req_host,
1768 unverifiable=True, method=m)
1769
1770
1771 def extract_timezone(date_str):
1772 m = re.search(
1773 r'''(?x)
1774 ^.{8,}? # >=8 char non-TZ prefix, if present
1775 (?P<tz>Z| # just the UTC Z, or
1776 (?:(?<=.\b\d{4}|\b\d{2}:\d\d)| # preceded by 4 digits or hh:mm or
1777 (?<!.\b[a-zA-Z]{3}|[a-zA-Z]{4}|..\b\d\d)) # not preceded by 3 alpha word or >= 4 alpha or 2 digits
1778 [ ]? # optional space
1779 (?P<sign>\+|-) # +/-
1780 (?P<hours>[0-9]{2}):?(?P<minutes>[0-9]{2}) # hh[:]mm
1781 $)
1782 ''', date_str)
1783 if not m:
1784 m = re.search(r'\d{1,2}:\d{1,2}(?:\.\d+)?(?P<tz>\s*[A-Z]+)$', date_str)
1785 timezone = TIMEZONE_NAMES.get(m and m.group('tz').strip())
1786 if timezone is not None:
1787 date_str = date_str[:-len(m.group('tz'))]
1788 timezone = datetime.timedelta(hours=timezone or 0)
1789 else:
1790 date_str = date_str[:-len(m.group('tz'))]
1791 if not m.group('sign'):
1792 timezone = datetime.timedelta()
1793 else:
1794 sign = 1 if m.group('sign') == '+' else -1
1795 timezone = datetime.timedelta(
1796 hours=sign * int(m.group('hours')),
1797 minutes=sign * int(m.group('minutes')))
1798 return timezone, date_str
1799
1800
1801 def parse_iso8601(date_str, delimiter='T', timezone=None):
1802 """ Return a UNIX timestamp from the given date """
1803
1804 if date_str is None:
1805 return None
1806
1807 date_str = re.sub(r'\.[0-9]+', '', date_str)
1808
1809 if timezone is None:
1810 timezone, date_str = extract_timezone(date_str)
1811
1812 with contextlib.suppress(ValueError):
1813 date_format = f'%Y-%m-%d{delimiter}%H:%M:%S'
1814 dt = datetime.datetime.strptime(date_str, date_format) - timezone
1815 return calendar.timegm(dt.timetuple())
1816
1817
1818 def date_formats(day_first=True):
1819 return DATE_FORMATS_DAY_FIRST if day_first else DATE_FORMATS_MONTH_FIRST
1820
1821
1822 def unified_strdate(date_str, day_first=True):
1823 """Return a string with the date in the format YYYYMMDD"""
1824
1825 if date_str is None:
1826 return None
1827 upload_date = None
1828 # Replace commas
1829 date_str = date_str.replace(',', ' ')
1830 # Remove AM/PM + timezone
1831 date_str = re.sub(r'(?i)\s*(?:AM|PM)(?:\s+[A-Z]+)?', '', date_str)
1832 _, date_str = extract_timezone(date_str)
1833
1834 for expression in date_formats(day_first):
1835 with contextlib.suppress(ValueError):
1836 upload_date = datetime.datetime.strptime(date_str, expression).strftime('%Y%m%d')
1837 if upload_date is None:
1838 timetuple = email.utils.parsedate_tz(date_str)
1839 if timetuple:
1840 with contextlib.suppress(ValueError):
1841 upload_date = datetime.datetime(*timetuple[:6]).strftime('%Y%m%d')
1842 if upload_date is not None:
1843 return str(upload_date)
1844
1845
1846 def unified_timestamp(date_str, day_first=True):
1847 if date_str is None:
1848 return None
1849
1850 date_str = re.sub(r'\s+', ' ', re.sub(
1851 r'(?i)[,|]|(mon|tues?|wed(nes)?|thu(rs)?|fri|sat(ur)?)(day)?', '', date_str))
1852
1853 pm_delta = 12 if re.search(r'(?i)PM', date_str) else 0
1854 timezone, date_str = extract_timezone(date_str)
1855
1856 # Remove AM/PM + timezone
1857 date_str = re.sub(r'(?i)\s*(?:AM|PM)(?:\s+[A-Z]+)?', '', date_str)
1858
1859 # Remove unrecognized timezones from ISO 8601 alike timestamps
1860 m = re.search(r'\d{1,2}:\d{1,2}(?:\.\d+)?(?P<tz>\s*[A-Z]+)$', date_str)
1861 if m:
1862 date_str = date_str[:-len(m.group('tz'))]
1863
1864 # Python only supports microseconds, so remove nanoseconds
1865 m = re.search(r'^([0-9]{4,}-[0-9]{1,2}-[0-9]{1,2}T[0-9]{1,2}:[0-9]{1,2}:[0-9]{1,2}\.[0-9]{6})[0-9]+$', date_str)
1866 if m:
1867 date_str = m.group(1)
1868
1869 for expression in date_formats(day_first):
1870 with contextlib.suppress(ValueError):
1871 dt = datetime.datetime.strptime(date_str, expression) - timezone + datetime.timedelta(hours=pm_delta)
1872 return calendar.timegm(dt.timetuple())
1873
1874 timetuple = email.utils.parsedate_tz(date_str)
1875 if timetuple:
1876 return calendar.timegm(timetuple) + pm_delta * 3600 - timezone.total_seconds()
1877
1878
1879 def determine_ext(url, default_ext='unknown_video'):
1880 if url is None or '.' not in url:
1881 return default_ext
1882 guess = url.partition('?')[0].rpartition('.')[2]
1883 if re.match(r'^[A-Za-z0-9]+$', guess):
1884 return guess
1885 # Try extract ext from URLs like http://example.com/foo/bar.mp4/?download
1886 elif guess.rstrip('/') in KNOWN_EXTENSIONS:
1887 return guess.rstrip('/')
1888 else:
1889 return default_ext
1890
1891
1892 def subtitles_filename(filename, sub_lang, sub_format, expected_real_ext=None):
1893 return replace_extension(filename, sub_lang + '.' + sub_format, expected_real_ext)
1894
1895
1896 def datetime_from_str(date_str, precision='auto', format='%Y%m%d'):
1897 R"""
1898 Return a datetime object from a string.
1899 Supported format:
1900 (now|today|yesterday|DATE)([+-]\d+(microsecond|second|minute|hour|day|week|month|year)s?)?
1901
1902 @param format strftime format of DATE
1903 @param precision Round the datetime object: auto|microsecond|second|minute|hour|day
1904 auto: round to the unit provided in date_str (if applicable).
1905 """
1906 auto_precision = False
1907 if precision == 'auto':
1908 auto_precision = True
1909 precision = 'microsecond'
1910 today = datetime_round(datetime.datetime.utcnow(), precision)
1911 if date_str in ('now', 'today'):
1912 return today
1913 if date_str == 'yesterday':
1914 return today - datetime.timedelta(days=1)
1915 match = re.match(
1916 r'(?P<start>.+)(?P<sign>[+-])(?P<time>\d+)(?P<unit>microsecond|second|minute|hour|day|week|month|year)s?',
1917 date_str)
1918 if match is not None:
1919 start_time = datetime_from_str(match.group('start'), precision, format)
1920 time = int(match.group('time')) * (-1 if match.group('sign') == '-' else 1)
1921 unit = match.group('unit')
1922 if unit == 'month' or unit == 'year':
1923 new_date = datetime_add_months(start_time, time * 12 if unit == 'year' else time)
1924 unit = 'day'
1925 else:
1926 if unit == 'week':
1927 unit = 'day'
1928 time *= 7
1929 delta = datetime.timedelta(**{unit + 's': time})
1930 new_date = start_time + delta
1931 if auto_precision:
1932 return datetime_round(new_date, unit)
1933 return new_date
1934
1935 return datetime_round(datetime.datetime.strptime(date_str, format), precision)
1936
1937
1938 def date_from_str(date_str, format='%Y%m%d', strict=False):
1939 R"""
1940 Return a date object from a string using datetime_from_str
1941
1942 @param strict Restrict allowed patterns to "YYYYMMDD" and
1943 (now|today|yesterday)(-\d+(day|week|month|year)s?)?
1944 """
1945 if strict and not re.fullmatch(r'\d{8}|(now|today|yesterday)(-\d+(day|week|month|year)s?)?', date_str):
1946 raise ValueError(f'Invalid date format "{date_str}"')
1947 return datetime_from_str(date_str, precision='microsecond', format=format).date()
1948
1949
1950 def datetime_add_months(dt, months):
1951 """Increment/Decrement a datetime object by months."""
1952 month = dt.month + months - 1
1953 year = dt.year + month // 12
1954 month = month % 12 + 1
1955 day = min(dt.day, calendar.monthrange(year, month)[1])
1956 return dt.replace(year, month, day)
1957
1958
1959 def datetime_round(dt, precision='day'):
1960 """
1961 Round a datetime object's time to a specific precision
1962 """
1963 if precision == 'microsecond':
1964 return dt
1965
1966 unit_seconds = {
1967 'day': 86400,
1968 'hour': 3600,
1969 'minute': 60,
1970 'second': 1,
1971 }
1972 roundto = lambda x, n: ((x + n / 2) // n) * n
1973 timestamp = calendar.timegm(dt.timetuple())
1974 return datetime.datetime.utcfromtimestamp(roundto(timestamp, unit_seconds[precision]))
1975
1976
1977 def hyphenate_date(date_str):
1978 """
1979 Convert a date in 'YYYYMMDD' format to 'YYYY-MM-DD' format"""
1980 match = re.match(r'^(\d\d\d\d)(\d\d)(\d\d)$', date_str)
1981 if match is not None:
1982 return '-'.join(match.groups())
1983 else:
1984 return date_str
1985
1986
1987 class DateRange:
1988 """Represents a time interval between two dates"""
1989
1990 def __init__(self, start=None, end=None):
1991 """start and end must be strings in the format accepted by date"""
1992 if start is not None:
1993 self.start = date_from_str(start, strict=True)
1994 else:
1995 self.start = datetime.datetime.min.date()
1996 if end is not None:
1997 self.end = date_from_str(end, strict=True)
1998 else:
1999 self.end = datetime.datetime.max.date()
2000 if self.start > self.end:
2001 raise ValueError('Date range: "%s" , the start date must be before the end date' % self)
2002
2003 @classmethod
2004 def day(cls, day):
2005 """Returns a range that only contains the given day"""
2006 return cls(day, day)
2007
2008 def __contains__(self, date):
2009 """Check if the date is in the range"""
2010 if not isinstance(date, datetime.date):
2011 date = date_from_str(date)
2012 return self.start <= date <= self.end
2013
2014 def __str__(self):
2015 return f'{self.start.isoformat()} - {self.end.isoformat()}'
2016
2017 def __eq__(self, other):
2018 return (isinstance(other, DateRange)
2019 and self.start == other.start and self.end == other.end)
2020
2021
2022 def platform_name():
2023 """ Returns the platform name as a str """
2024 deprecation_warning(f'"{__name__}.platform_name" is deprecated, use "platform.platform" instead')
2025 return platform.platform()
2026
2027
2028 @functools.cache
2029 def system_identifier():
2030 python_implementation = platform.python_implementation()
2031 if python_implementation == 'PyPy' and hasattr(sys, 'pypy_version_info'):
2032 python_implementation += ' version %d.%d.%d' % sys.pypy_version_info[:3]
2033 libc_ver = []
2034 with contextlib.suppress(OSError): # We may not have access to the executable
2035 libc_ver = platform.libc_ver()
2036
2037 return 'Python %s (%s %s %s) - %s (%s%s)' % (
2038 platform.python_version(),
2039 python_implementation,
2040 platform.machine(),
2041 platform.architecture()[0],
2042 platform.platform(),
2043 ssl.OPENSSL_VERSION,
2044 format_field(join_nonempty(*libc_ver, delim=' '), None, ', %s'),
2045 )
2046
2047
2048 @functools.cache
2049 def get_windows_version():
2050 ''' Get Windows version. returns () if it's not running on Windows '''
2051 if compat_os_name == 'nt':
2052 return version_tuple(platform.win32_ver()[1])
2053 else:
2054 return ()
2055
2056
2057 def write_string(s, out=None, encoding=None):
2058 assert isinstance(s, str)
2059 out = out or sys.stderr
2060
2061 if compat_os_name == 'nt' and supports_terminal_sequences(out):
2062 s = re.sub(r'([\r\n]+)', r' \1', s)
2063
2064 enc, buffer = None, out
2065 if 'b' in getattr(out, 'mode', ''):
2066 enc = encoding or preferredencoding()
2067 elif hasattr(out, 'buffer'):
2068 buffer = out.buffer
2069 enc = encoding or getattr(out, 'encoding', None) or preferredencoding()
2070
2071 buffer.write(s.encode(enc, 'ignore') if enc else s)
2072 out.flush()
2073
2074
2075 def deprecation_warning(msg, *, printer=None, stacklevel=0, **kwargs):
2076 from . import _IN_CLI
2077 if _IN_CLI:
2078 if msg in deprecation_warning._cache:
2079 return
2080 deprecation_warning._cache.add(msg)
2081 if printer:
2082 return printer(f'{msg}{bug_reports_message()}', **kwargs)
2083 return write_string(f'ERROR: {msg}{bug_reports_message()}\n', **kwargs)
2084 else:
2085 import warnings
2086 warnings.warn(DeprecationWarning(msg), stacklevel=stacklevel + 3)
2087
2088
2089 deprecation_warning._cache = set()
2090
2091
2092 def bytes_to_intlist(bs):
2093 if not bs:
2094 return []
2095 if isinstance(bs[0], int): # Python 3
2096 return list(bs)
2097 else:
2098 return [ord(c) for c in bs]
2099
2100
2101 def intlist_to_bytes(xs):
2102 if not xs:
2103 return b''
2104 return struct.pack('%dB' % len(xs), *xs)
2105
2106
2107 class LockingUnsupportedError(OSError):
2108 msg = 'File locking is not supported'
2109
2110 def __init__(self):
2111 super().__init__(self.msg)
2112
2113
2114 # Cross-platform file locking
2115 if sys.platform == 'win32':
2116 import ctypes
2117 import ctypes.wintypes
2118 import msvcrt
2119
2120 class OVERLAPPED(ctypes.Structure):
2121 _fields_ = [
2122 ('Internal', ctypes.wintypes.LPVOID),
2123 ('InternalHigh', ctypes.wintypes.LPVOID),
2124 ('Offset', ctypes.wintypes.DWORD),
2125 ('OffsetHigh', ctypes.wintypes.DWORD),
2126 ('hEvent', ctypes.wintypes.HANDLE),
2127 ]
2128
2129 kernel32 = ctypes.WinDLL('kernel32')
2130 LockFileEx = kernel32.LockFileEx
2131 LockFileEx.argtypes = [
2132 ctypes.wintypes.HANDLE, # hFile
2133 ctypes.wintypes.DWORD, # dwFlags
2134 ctypes.wintypes.DWORD, # dwReserved
2135 ctypes.wintypes.DWORD, # nNumberOfBytesToLockLow
2136 ctypes.wintypes.DWORD, # nNumberOfBytesToLockHigh
2137 ctypes.POINTER(OVERLAPPED) # Overlapped
2138 ]
2139 LockFileEx.restype = ctypes.wintypes.BOOL
2140 UnlockFileEx = kernel32.UnlockFileEx
2141 UnlockFileEx.argtypes = [
2142 ctypes.wintypes.HANDLE, # hFile
2143 ctypes.wintypes.DWORD, # dwReserved
2144 ctypes.wintypes.DWORD, # nNumberOfBytesToLockLow
2145 ctypes.wintypes.DWORD, # nNumberOfBytesToLockHigh
2146 ctypes.POINTER(OVERLAPPED) # Overlapped
2147 ]
2148 UnlockFileEx.restype = ctypes.wintypes.BOOL
2149 whole_low = 0xffffffff
2150 whole_high = 0x7fffffff
2151
2152 def _lock_file(f, exclusive, block):
2153 overlapped = OVERLAPPED()
2154 overlapped.Offset = 0
2155 overlapped.OffsetHigh = 0
2156 overlapped.hEvent = 0
2157 f._lock_file_overlapped_p = ctypes.pointer(overlapped)
2158
2159 if not LockFileEx(msvcrt.get_osfhandle(f.fileno()),
2160 (0x2 if exclusive else 0x0) | (0x0 if block else 0x1),
2161 0, whole_low, whole_high, f._lock_file_overlapped_p):
2162 # NB: No argument form of "ctypes.FormatError" does not work on PyPy
2163 raise BlockingIOError(f'Locking file failed: {ctypes.FormatError(ctypes.GetLastError())!r}')
2164
2165 def _unlock_file(f):
2166 assert f._lock_file_overlapped_p
2167 handle = msvcrt.get_osfhandle(f.fileno())
2168 if not UnlockFileEx(handle, 0, whole_low, whole_high, f._lock_file_overlapped_p):
2169 raise OSError('Unlocking file failed: %r' % ctypes.FormatError())
2170
2171 else:
2172 try:
2173 import fcntl
2174
2175 def _lock_file(f, exclusive, block):
2176 flags = fcntl.LOCK_EX if exclusive else fcntl.LOCK_SH
2177 if not block:
2178 flags |= fcntl.LOCK_NB
2179 try:
2180 fcntl.flock(f, flags)
2181 except BlockingIOError:
2182 raise
2183 except OSError: # AOSP does not have flock()
2184 fcntl.lockf(f, flags)
2185
2186 def _unlock_file(f):
2187 try:
2188 fcntl.flock(f, fcntl.LOCK_UN)
2189 except OSError:
2190 fcntl.lockf(f, fcntl.LOCK_UN)
2191
2192 except ImportError:
2193
2194 def _lock_file(f, exclusive, block):
2195 raise LockingUnsupportedError()
2196
2197 def _unlock_file(f):
2198 raise LockingUnsupportedError()
2199
2200
2201 class locked_file:
2202 locked = False
2203
2204 def __init__(self, filename, mode, block=True, encoding=None):
2205 if mode not in {'r', 'rb', 'a', 'ab', 'w', 'wb'}:
2206 raise NotImplementedError(mode)
2207 self.mode, self.block = mode, block
2208
2209 writable = any(f in mode for f in 'wax+')
2210 readable = any(f in mode for f in 'r+')
2211 flags = functools.reduce(operator.ior, (
2212 getattr(os, 'O_CLOEXEC', 0), # UNIX only
2213 getattr(os, 'O_BINARY', 0), # Windows only
2214 getattr(os, 'O_NOINHERIT', 0), # Windows only
2215 os.O_CREAT if writable else 0, # O_TRUNC only after locking
2216 os.O_APPEND if 'a' in mode else 0,
2217 os.O_EXCL if 'x' in mode else 0,
2218 os.O_RDONLY if not writable else os.O_RDWR if readable else os.O_WRONLY,
2219 ))
2220
2221 self.f = os.fdopen(os.open(filename, flags, 0o666), mode, encoding=encoding)
2222
2223 def __enter__(self):
2224 exclusive = 'r' not in self.mode
2225 try:
2226 _lock_file(self.f, exclusive, self.block)
2227 self.locked = True
2228 except OSError:
2229 self.f.close()
2230 raise
2231 if 'w' in self.mode:
2232 try:
2233 self.f.truncate()
2234 except OSError as e:
2235 if e.errno not in (
2236 errno.ESPIPE, # Illegal seek - expected for FIFO
2237 errno.EINVAL, # Invalid argument - expected for /dev/null
2238 ):
2239 raise
2240 return self
2241
2242 def unlock(self):
2243 if not self.locked:
2244 return
2245 try:
2246 _unlock_file(self.f)
2247 finally:
2248 self.locked = False
2249
2250 def __exit__(self, *_):
2251 try:
2252 self.unlock()
2253 finally:
2254 self.f.close()
2255
2256 open = __enter__
2257 close = __exit__
2258
2259 def __getattr__(self, attr):
2260 return getattr(self.f, attr)
2261
2262 def __iter__(self):
2263 return iter(self.f)
2264
2265
2266 @functools.cache
2267 def get_filesystem_encoding():
2268 encoding = sys.getfilesystemencoding()
2269 return encoding if encoding is not None else 'utf-8'
2270
2271
2272 def shell_quote(args):
2273 quoted_args = []
2274 encoding = get_filesystem_encoding()
2275 for a in args:
2276 if isinstance(a, bytes):
2277 # We may get a filename encoded with 'encodeFilename'
2278 a = a.decode(encoding)
2279 quoted_args.append(compat_shlex_quote(a))
2280 return ' '.join(quoted_args)
2281
2282
2283 def smuggle_url(url, data):
2284 """ Pass additional data in a URL for internal use. """
2285
2286 url, idata = unsmuggle_url(url, {})
2287 data.update(idata)
2288 sdata = urllib.parse.urlencode(
2289 {'__youtubedl_smuggle': json.dumps(data)})
2290 return url + '#' + sdata
2291
2292
2293 def unsmuggle_url(smug_url, default=None):
2294 if '#__youtubedl_smuggle' not in smug_url:
2295 return smug_url, default
2296 url, _, sdata = smug_url.rpartition('#')
2297 jsond = urllib.parse.parse_qs(sdata)['__youtubedl_smuggle'][0]
2298 data = json.loads(jsond)
2299 return url, data
2300
2301
2302 def format_decimal_suffix(num, fmt='%d%s', *, factor=1000):
2303 """ Formats numbers with decimal sufixes like K, M, etc """
2304 num, factor = float_or_none(num), float(factor)
2305 if num is None or num < 0:
2306 return None
2307 POSSIBLE_SUFFIXES = 'kMGTPEZY'
2308 exponent = 0 if num == 0 else min(int(math.log(num, factor)), len(POSSIBLE_SUFFIXES))
2309 suffix = ['', *POSSIBLE_SUFFIXES][exponent]
2310 if factor == 1024:
2311 suffix = {'k': 'Ki', '': ''}.get(suffix, f'{suffix}i')
2312 converted = num / (factor ** exponent)
2313 return fmt % (converted, suffix)
2314
2315
2316 def format_bytes(bytes):
2317 return format_decimal_suffix(bytes, '%.2f%sB', factor=1024) or 'N/A'
2318
2319
2320 def lookup_unit_table(unit_table, s, strict=False):
2321 num_re = NUMBER_RE if strict else NUMBER_RE.replace(R'\.', '[,.]')
2322 units_re = '|'.join(re.escape(u) for u in unit_table)
2323 m = (re.fullmatch if strict else re.match)(
2324 rf'(?P<num>{num_re})\s*(?P<unit>{units_re})\b', s)
2325 if not m:
2326 return None
2327
2328 num = float(m.group('num').replace(',', '.'))
2329 mult = unit_table[m.group('unit')]
2330 return round(num * mult)
2331
2332
2333 def parse_bytes(s):
2334 """Parse a string indicating a byte quantity into an integer"""
2335 return lookup_unit_table(
2336 {u: 1024**i for i, u in enumerate(['', *'KMGTPEZY'])},
2337 s.upper(), strict=True)
2338
2339
2340 def parse_filesize(s):
2341 if s is None:
2342 return None
2343
2344 # The lower-case forms are of course incorrect and unofficial,
2345 # but we support those too
2346 _UNIT_TABLE = {
2347 'B': 1,
2348 'b': 1,
2349 'bytes': 1,
2350 'KiB': 1024,
2351 'KB': 1000,
2352 'kB': 1024,
2353 'Kb': 1000,
2354 'kb': 1000,
2355 'kilobytes': 1000,
2356 'kibibytes': 1024,
2357 'MiB': 1024 ** 2,
2358 'MB': 1000 ** 2,
2359 'mB': 1024 ** 2,
2360 'Mb': 1000 ** 2,
2361 'mb': 1000 ** 2,
2362 'megabytes': 1000 ** 2,
2363 'mebibytes': 1024 ** 2,
2364 'GiB': 1024 ** 3,
2365 'GB': 1000 ** 3,
2366 'gB': 1024 ** 3,
2367 'Gb': 1000 ** 3,
2368 'gb': 1000 ** 3,
2369 'gigabytes': 1000 ** 3,
2370 'gibibytes': 1024 ** 3,
2371 'TiB': 1024 ** 4,
2372 'TB': 1000 ** 4,
2373 'tB': 1024 ** 4,
2374 'Tb': 1000 ** 4,
2375 'tb': 1000 ** 4,
2376 'terabytes': 1000 ** 4,
2377 'tebibytes': 1024 ** 4,
2378 'PiB': 1024 ** 5,
2379 'PB': 1000 ** 5,
2380 'pB': 1024 ** 5,
2381 'Pb': 1000 ** 5,
2382 'pb': 1000 ** 5,
2383 'petabytes': 1000 ** 5,
2384 'pebibytes': 1024 ** 5,
2385 'EiB': 1024 ** 6,
2386 'EB': 1000 ** 6,
2387 'eB': 1024 ** 6,
2388 'Eb': 1000 ** 6,
2389 'eb': 1000 ** 6,
2390 'exabytes': 1000 ** 6,
2391 'exbibytes': 1024 ** 6,
2392 'ZiB': 1024 ** 7,
2393 'ZB': 1000 ** 7,
2394 'zB': 1024 ** 7,
2395 'Zb': 1000 ** 7,
2396 'zb': 1000 ** 7,
2397 'zettabytes': 1000 ** 7,
2398 'zebibytes': 1024 ** 7,
2399 'YiB': 1024 ** 8,
2400 'YB': 1000 ** 8,
2401 'yB': 1024 ** 8,
2402 'Yb': 1000 ** 8,
2403 'yb': 1000 ** 8,
2404 'yottabytes': 1000 ** 8,
2405 'yobibytes': 1024 ** 8,
2406 }
2407
2408 return lookup_unit_table(_UNIT_TABLE, s)
2409
2410
2411 def parse_count(s):
2412 if s is None:
2413 return None
2414
2415 s = re.sub(r'^[^\d]+\s', '', s).strip()
2416
2417 if re.match(r'^[\d,.]+$', s):
2418 return str_to_int(s)
2419
2420 _UNIT_TABLE = {
2421 'k': 1000,
2422 'K': 1000,
2423 'm': 1000 ** 2,
2424 'M': 1000 ** 2,
2425 'kk': 1000 ** 2,
2426 'KK': 1000 ** 2,
2427 'b': 1000 ** 3,
2428 'B': 1000 ** 3,
2429 }
2430
2431 ret = lookup_unit_table(_UNIT_TABLE, s)
2432 if ret is not None:
2433 return ret
2434
2435 mobj = re.match(r'([\d,.]+)(?:$|\s)', s)
2436 if mobj:
2437 return str_to_int(mobj.group(1))
2438
2439
2440 def parse_resolution(s, *, lenient=False):
2441 if s is None:
2442 return {}
2443
2444 if lenient:
2445 mobj = re.search(r'(?P<w>\d+)\s*[xX×,]\s*(?P<h>\d+)', s)
2446 else:
2447 mobj = re.search(r'(?<![a-zA-Z0-9])(?P<w>\d+)\s*[xX×,]\s*(?P<h>\d+)(?![a-zA-Z0-9])', s)
2448 if mobj:
2449 return {
2450 'width': int(mobj.group('w')),
2451 'height': int(mobj.group('h')),
2452 }
2453
2454 mobj = re.search(r'(?<![a-zA-Z0-9])(\d+)[pPiI](?![a-zA-Z0-9])', s)
2455 if mobj:
2456 return {'height': int(mobj.group(1))}
2457
2458 mobj = re.search(r'\b([48])[kK]\b', s)
2459 if mobj:
2460 return {'height': int(mobj.group(1)) * 540}
2461
2462 return {}
2463
2464
2465 def parse_bitrate(s):
2466 if not isinstance(s, str):
2467 return
2468 mobj = re.search(r'\b(\d+)\s*kbps', s)
2469 if mobj:
2470 return int(mobj.group(1))
2471
2472
2473 def month_by_name(name, lang='en'):
2474 """ Return the number of a month by (locale-independently) English name """
2475
2476 month_names = MONTH_NAMES.get(lang, MONTH_NAMES['en'])
2477
2478 try:
2479 return month_names.index(name) + 1
2480 except ValueError:
2481 return None
2482
2483
2484 def month_by_abbreviation(abbrev):
2485 """ Return the number of a month by (locale-independently) English
2486 abbreviations """
2487
2488 try:
2489 return [s[:3] for s in ENGLISH_MONTH_NAMES].index(abbrev) + 1
2490 except ValueError:
2491 return None
2492
2493
2494 def fix_xml_ampersands(xml_str):
2495 """Replace all the '&' by '&amp;' in XML"""
2496 return re.sub(
2497 r'&(?!amp;|lt;|gt;|apos;|quot;|#x[0-9a-fA-F]{,4};|#[0-9]{,4};)',
2498 '&amp;',
2499 xml_str)
2500
2501
2502 def setproctitle(title):
2503 assert isinstance(title, str)
2504
2505 # Workaround for https://github.com/yt-dlp/yt-dlp/issues/4541
2506 try:
2507 import ctypes
2508 except ImportError:
2509 return
2510
2511 try:
2512 libc = ctypes.cdll.LoadLibrary('libc.so.6')
2513 except OSError:
2514 return
2515 except TypeError:
2516 # LoadLibrary in Windows Python 2.7.13 only expects
2517 # a bytestring, but since unicode_literals turns
2518 # every string into a unicode string, it fails.
2519 return
2520 title_bytes = title.encode()
2521 buf = ctypes.create_string_buffer(len(title_bytes))
2522 buf.value = title_bytes
2523 try:
2524 libc.prctl(15, buf, 0, 0, 0)
2525 except AttributeError:
2526 return # Strange libc, just skip this
2527
2528
2529 def remove_start(s, start):
2530 return s[len(start):] if s is not None and s.startswith(start) else s
2531
2532
2533 def remove_end(s, end):
2534 return s[:-len(end)] if s is not None and s.endswith(end) else s
2535
2536
2537 def remove_quotes(s):
2538 if s is None or len(s) < 2:
2539 return s
2540 for quote in ('"', "'", ):
2541 if s[0] == quote and s[-1] == quote:
2542 return s[1:-1]
2543 return s
2544
2545
2546 def get_domain(url):
2547 """
2548 This implementation is inconsistent, but is kept for compatibility.
2549 Use this only for "webpage_url_domain"
2550 """
2551 return remove_start(urllib.parse.urlparse(url).netloc, 'www.') or None
2552
2553
2554 def url_basename(url):
2555 path = urllib.parse.urlparse(url).path
2556 return path.strip('/').split('/')[-1]
2557
2558
2559 def base_url(url):
2560 return re.match(r'https?://[^?#]+/', url).group()
2561
2562
2563 def urljoin(base, path):
2564 if isinstance(path, bytes):
2565 path = path.decode()
2566 if not isinstance(path, str) or not path:
2567 return None
2568 if re.match(r'^(?:[a-zA-Z][a-zA-Z0-9+-.]*:)?//', path):
2569 return path
2570 if isinstance(base, bytes):
2571 base = base.decode()
2572 if not isinstance(base, str) or not re.match(
2573 r'^(?:https?:)?//', base):
2574 return None
2575 return urllib.parse.urljoin(base, path)
2576
2577
2578 class HEADRequest(urllib.request.Request):
2579 def get_method(self):
2580 return 'HEAD'
2581
2582
2583 class PUTRequest(urllib.request.Request):
2584 def get_method(self):
2585 return 'PUT'
2586
2587
2588 def int_or_none(v, scale=1, default=None, get_attr=None, invscale=1):
2589 if get_attr and v is not None:
2590 v = getattr(v, get_attr, None)
2591 try:
2592 return int(v) * invscale // scale
2593 except (ValueError, TypeError, OverflowError):
2594 return default
2595
2596
2597 def str_or_none(v, default=None):
2598 return default if v is None else str(v)
2599
2600
2601 def str_to_int(int_str):
2602 """ A more relaxed version of int_or_none """
2603 if isinstance(int_str, int):
2604 return int_str
2605 elif isinstance(int_str, str):
2606 int_str = re.sub(r'[,\.\+]', '', int_str)
2607 return int_or_none(int_str)
2608
2609
2610 def float_or_none(v, scale=1, invscale=1, default=None):
2611 if v is None:
2612 return default
2613 try:
2614 return float(v) * invscale / scale
2615 except (ValueError, TypeError):
2616 return default
2617
2618
2619 def bool_or_none(v, default=None):
2620 return v if isinstance(v, bool) else default
2621
2622
2623 def strip_or_none(v, default=None):
2624 return v.strip() if isinstance(v, str) else default
2625
2626
2627 def url_or_none(url):
2628 if not url or not isinstance(url, str):
2629 return None
2630 url = url.strip()
2631 return url if re.match(r'^(?:(?:https?|rt(?:m(?:pt?[es]?|fp)|sp[su]?)|mms|ftps?):)?//', url) else None
2632
2633
2634 def request_to_url(req):
2635 if isinstance(req, urllib.request.Request):
2636 return req.get_full_url()
2637 else:
2638 return req
2639
2640
2641 def strftime_or_none(timestamp, date_format, default=None):
2642 datetime_object = None
2643 try:
2644 if isinstance(timestamp, (int, float)): # unix timestamp
2645 # Using naive datetime here can break timestamp() in Windows
2646 # Ref: https://github.com/yt-dlp/yt-dlp/issues/5185, https://github.com/python/cpython/issues/94414
2647 datetime_object = datetime.datetime.fromtimestamp(timestamp, datetime.timezone.utc)
2648 elif isinstance(timestamp, str): # assume YYYYMMDD
2649 datetime_object = datetime.datetime.strptime(timestamp, '%Y%m%d')
2650 date_format = re.sub( # Support %s on windows
2651 r'(?<!%)(%%)*%s', rf'\g<1>{int(datetime_object.timestamp())}', date_format)
2652 return datetime_object.strftime(date_format)
2653 except (ValueError, TypeError, AttributeError):
2654 return default
2655
2656
2657 def parse_duration(s):
2658 if not isinstance(s, str):
2659 return None
2660 s = s.strip()
2661 if not s:
2662 return None
2663
2664 days, hours, mins, secs, ms = [None] * 5
2665 m = re.match(r'''(?x)
2666 (?P<before_secs>
2667 (?:(?:(?P<days>[0-9]+):)?(?P<hours>[0-9]+):)?(?P<mins>[0-9]+):)?
2668 (?P<secs>(?(before_secs)[0-9]{1,2}|[0-9]+))
2669 (?P<ms>[.:][0-9]+)?Z?$
2670 ''', s)
2671 if m:
2672 days, hours, mins, secs, ms = m.group('days', 'hours', 'mins', 'secs', 'ms')
2673 else:
2674 m = re.match(
2675 r'''(?ix)(?:P?
2676 (?:
2677 [0-9]+\s*y(?:ears?)?,?\s*
2678 )?
2679 (?:
2680 [0-9]+\s*m(?:onths?)?,?\s*
2681 )?
2682 (?:
2683 [0-9]+\s*w(?:eeks?)?,?\s*
2684 )?
2685 (?:
2686 (?P<days>[0-9]+)\s*d(?:ays?)?,?\s*
2687 )?
2688 T)?
2689 (?:
2690 (?P<hours>[0-9]+)\s*h(?:ours?)?,?\s*
2691 )?
2692 (?:
2693 (?P<mins>[0-9]+)\s*m(?:in(?:ute)?s?)?,?\s*
2694 )?
2695 (?:
2696 (?P<secs>[0-9]+)(?P<ms>\.[0-9]+)?\s*s(?:ec(?:ond)?s?)?\s*
2697 )?Z?$''', s)
2698 if m:
2699 days, hours, mins, secs, ms = m.groups()
2700 else:
2701 m = re.match(r'(?i)(?:(?P<hours>[0-9.]+)\s*(?:hours?)|(?P<mins>[0-9.]+)\s*(?:mins?\.?|minutes?)\s*)Z?$', s)
2702 if m:
2703 hours, mins = m.groups()
2704 else:
2705 return None
2706
2707 if ms:
2708 ms = ms.replace(':', '.')
2709 return sum(float(part or 0) * mult for part, mult in (
2710 (days, 86400), (hours, 3600), (mins, 60), (secs, 1), (ms, 1)))
2711
2712
2713 def prepend_extension(filename, ext, expected_real_ext=None):
2714 name, real_ext = os.path.splitext(filename)
2715 return (
2716 f'{name}.{ext}{real_ext}'
2717 if not expected_real_ext or real_ext[1:] == expected_real_ext
2718 else f'{filename}.{ext}')
2719
2720
2721 def replace_extension(filename, ext, expected_real_ext=None):
2722 name, real_ext = os.path.splitext(filename)
2723 return '{}.{}'.format(
2724 name if not expected_real_ext or real_ext[1:] == expected_real_ext else filename,
2725 ext)
2726
2727
2728 def check_executable(exe, args=[]):
2729 """ Checks if the given binary is installed somewhere in PATH, and returns its name.
2730 args can be a list of arguments for a short output (like -version) """
2731 try:
2732 Popen.run([exe] + args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
2733 except OSError:
2734 return False
2735 return exe
2736
2737
2738 def _get_exe_version_output(exe, args):
2739 try:
2740 # STDIN should be redirected too. On UNIX-like systems, ffmpeg triggers
2741 # SIGTTOU if yt-dlp is run in the background.
2742 # See https://github.com/ytdl-org/youtube-dl/issues/955#issuecomment-209789656
2743 stdout, _, ret = Popen.run([encodeArgument(exe)] + args, text=True,
2744 stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
2745 if ret:
2746 return None
2747 except OSError:
2748 return False
2749 return stdout
2750
2751
2752 def detect_exe_version(output, version_re=None, unrecognized='present'):
2753 assert isinstance(output, str)
2754 if version_re is None:
2755 version_re = r'version\s+([-0-9._a-zA-Z]+)'
2756 m = re.search(version_re, output)
2757 if m:
2758 return m.group(1)
2759 else:
2760 return unrecognized
2761
2762
2763 def get_exe_version(exe, args=['--version'],
2764 version_re=None, unrecognized=('present', 'broken')):
2765 """ Returns the version of the specified executable,
2766 or False if the executable is not present """
2767 unrecognized = variadic(unrecognized)
2768 assert len(unrecognized) in (1, 2)
2769 out = _get_exe_version_output(exe, args)
2770 if out is None:
2771 return unrecognized[-1]
2772 return out and detect_exe_version(out, version_re, unrecognized[0])
2773
2774
2775 def frange(start=0, stop=None, step=1):
2776 """Float range"""
2777 if stop is None:
2778 start, stop = 0, start
2779 sign = [-1, 1][step > 0] if step else 0
2780 while sign * start < sign * stop:
2781 yield start
2782 start += step
2783
2784
2785 class LazyList(collections.abc.Sequence):
2786 """Lazy immutable list from an iterable
2787 Note that slices of a LazyList are lists and not LazyList"""
2788
2789 class IndexError(IndexError):
2790 pass
2791
2792 def __init__(self, iterable, *, reverse=False, _cache=None):
2793 self._iterable = iter(iterable)
2794 self._cache = [] if _cache is None else _cache
2795 self._reversed = reverse
2796
2797 def __iter__(self):
2798 if self._reversed:
2799 # We need to consume the entire iterable to iterate in reverse
2800 yield from self.exhaust()
2801 return
2802 yield from self._cache
2803 for item in self._iterable:
2804 self._cache.append(item)
2805 yield item
2806
2807 def _exhaust(self):
2808 self._cache.extend(self._iterable)
2809 self._iterable = [] # Discard the emptied iterable to make it pickle-able
2810 return self._cache
2811
2812 def exhaust(self):
2813 """Evaluate the entire iterable"""
2814 return self._exhaust()[::-1 if self._reversed else 1]
2815
2816 @staticmethod
2817 def _reverse_index(x):
2818 return None if x is None else ~x
2819
2820 def __getitem__(self, idx):
2821 if isinstance(idx, slice):
2822 if self._reversed:
2823 idx = slice(self._reverse_index(idx.start), self._reverse_index(idx.stop), -(idx.step or 1))
2824 start, stop, step = idx.start, idx.stop, idx.step or 1
2825 elif isinstance(idx, int):
2826 if self._reversed:
2827 idx = self._reverse_index(idx)
2828 start, stop, step = idx, idx, 0
2829 else:
2830 raise TypeError('indices must be integers or slices')
2831 if ((start or 0) < 0 or (stop or 0) < 0
2832 or (start is None and step < 0)
2833 or (stop is None and step > 0)):
2834 # We need to consume the entire iterable to be able to slice from the end
2835 # Obviously, never use this with infinite iterables
2836 self._exhaust()
2837 try:
2838 return self._cache[idx]
2839 except IndexError as e:
2840 raise self.IndexError(e) from e
2841 n = max(start or 0, stop or 0) - len(self._cache) + 1
2842 if n > 0:
2843 self._cache.extend(itertools.islice(self._iterable, n))
2844 try:
2845 return self._cache[idx]
2846 except IndexError as e:
2847 raise self.IndexError(e) from e
2848
2849 def __bool__(self):
2850 try:
2851 self[-1] if self._reversed else self[0]
2852 except self.IndexError:
2853 return False
2854 return True
2855
2856 def __len__(self):
2857 self._exhaust()
2858 return len(self._cache)
2859
2860 def __reversed__(self):
2861 return type(self)(self._iterable, reverse=not self._reversed, _cache=self._cache)
2862
2863 def __copy__(self):
2864 return type(self)(self._iterable, reverse=self._reversed, _cache=self._cache)
2865
2866 def __repr__(self):
2867 # repr and str should mimic a list. So we exhaust the iterable
2868 return repr(self.exhaust())
2869
2870 def __str__(self):
2871 return repr(self.exhaust())
2872
2873
2874 class PagedList:
2875
2876 class IndexError(IndexError):
2877 pass
2878
2879 def __len__(self):
2880 # This is only useful for tests
2881 return len(self.getslice())
2882
2883 def __init__(self, pagefunc, pagesize, use_cache=True):
2884 self._pagefunc = pagefunc
2885 self._pagesize = pagesize
2886 self._pagecount = float('inf')
2887 self._use_cache = use_cache
2888 self._cache = {}
2889
2890 def getpage(self, pagenum):
2891 page_results = self._cache.get(pagenum)
2892 if page_results is None:
2893 page_results = [] if pagenum > self._pagecount else list(self._pagefunc(pagenum))
2894 if self._use_cache:
2895 self._cache[pagenum] = page_results
2896 return page_results
2897
2898 def getslice(self, start=0, end=None):
2899 return list(self._getslice(start, end))
2900
2901 def _getslice(self, start, end):
2902 raise NotImplementedError('This method must be implemented by subclasses')
2903
2904 def __getitem__(self, idx):
2905 assert self._use_cache, 'Indexing PagedList requires cache'
2906 if not isinstance(idx, int) or idx < 0:
2907 raise TypeError('indices must be non-negative integers')
2908 entries = self.getslice(idx, idx + 1)
2909 if not entries:
2910 raise self.IndexError()
2911 return entries[0]
2912
2913
2914 class OnDemandPagedList(PagedList):
2915 """Download pages until a page with less than maximum results"""
2916
2917 def _getslice(self, start, end):
2918 for pagenum in itertools.count(start // self._pagesize):
2919 firstid = pagenum * self._pagesize
2920 nextfirstid = pagenum * self._pagesize + self._pagesize
2921 if start >= nextfirstid:
2922 continue
2923
2924 startv = (
2925 start % self._pagesize
2926 if firstid <= start < nextfirstid
2927 else 0)
2928 endv = (
2929 ((end - 1) % self._pagesize) + 1
2930 if (end is not None and firstid <= end <= nextfirstid)
2931 else None)
2932
2933 try:
2934 page_results = self.getpage(pagenum)
2935 except Exception:
2936 self._pagecount = pagenum - 1
2937 raise
2938 if startv != 0 or endv is not None:
2939 page_results = page_results[startv:endv]
2940 yield from page_results
2941
2942 # A little optimization - if current page is not "full", ie. does
2943 # not contain page_size videos then we can assume that this page
2944 # is the last one - there are no more ids on further pages -
2945 # i.e. no need to query again.
2946 if len(page_results) + startv < self._pagesize:
2947 break
2948
2949 # If we got the whole page, but the next page is not interesting,
2950 # break out early as well
2951 if end == nextfirstid:
2952 break
2953
2954
2955 class InAdvancePagedList(PagedList):
2956 """PagedList with total number of pages known in advance"""
2957
2958 def __init__(self, pagefunc, pagecount, pagesize):
2959 PagedList.__init__(self, pagefunc, pagesize, True)
2960 self._pagecount = pagecount
2961
2962 def _getslice(self, start, end):
2963 start_page = start // self._pagesize
2964 end_page = self._pagecount if end is None else min(self._pagecount, end // self._pagesize + 1)
2965 skip_elems = start - start_page * self._pagesize
2966 only_more = None if end is None else end - start
2967 for pagenum in range(start_page, end_page):
2968 page_results = self.getpage(pagenum)
2969 if skip_elems:
2970 page_results = page_results[skip_elems:]
2971 skip_elems = None
2972 if only_more is not None:
2973 if len(page_results) < only_more:
2974 only_more -= len(page_results)
2975 else:
2976 yield from page_results[:only_more]
2977 break
2978 yield from page_results
2979
2980
2981 class PlaylistEntries:
2982 MissingEntry = object()
2983 is_exhausted = False
2984
2985 def __init__(self, ydl, info_dict):
2986 self.ydl = ydl
2987
2988 # _entries must be assigned now since infodict can change during iteration
2989 entries = info_dict.get('entries')
2990 if entries is None:
2991 raise EntryNotInPlaylist('There are no entries')
2992 elif isinstance(entries, list):
2993 self.is_exhausted = True
2994
2995 requested_entries = info_dict.get('requested_entries')
2996 self.is_incomplete = requested_entries is not None
2997 if self.is_incomplete:
2998 assert self.is_exhausted
2999 self._entries = [self.MissingEntry] * max(requested_entries or [0])
3000 for i, entry in zip(requested_entries, entries):
3001 self._entries[i - 1] = entry
3002 elif isinstance(entries, (list, PagedList, LazyList)):
3003 self._entries = entries
3004 else:
3005 self._entries = LazyList(entries)
3006
3007 PLAYLIST_ITEMS_RE = re.compile(r'''(?x)
3008 (?P<start>[+-]?\d+)?
3009 (?P<range>[:-]
3010 (?P<end>[+-]?\d+|inf(?:inite)?)?
3011 (?::(?P<step>[+-]?\d+))?
3012 )?''')
3013
3014 @classmethod
3015 def parse_playlist_items(cls, string):
3016 for segment in string.split(','):
3017 if not segment:
3018 raise ValueError('There is two or more consecutive commas')
3019 mobj = cls.PLAYLIST_ITEMS_RE.fullmatch(segment)
3020 if not mobj:
3021 raise ValueError(f'{segment!r} is not a valid specification')
3022 start, end, step, has_range = mobj.group('start', 'end', 'step', 'range')
3023 if int_or_none(step) == 0:
3024 raise ValueError(f'Step in {segment!r} cannot be zero')
3025 yield slice(int_or_none(start), float_or_none(end), int_or_none(step)) if has_range else int(start)
3026
3027 def get_requested_items(self):
3028 playlist_items = self.ydl.params.get('playlist_items')
3029 playlist_start = self.ydl.params.get('playliststart', 1)
3030 playlist_end = self.ydl.params.get('playlistend')
3031 # For backwards compatibility, interpret -1 as whole list
3032 if playlist_end in (-1, None):
3033 playlist_end = ''
3034 if not playlist_items:
3035 playlist_items = f'{playlist_start}:{playlist_end}'
3036 elif playlist_start != 1 or playlist_end:
3037 self.ydl.report_warning('Ignoring playliststart and playlistend because playlistitems was given', only_once=True)
3038
3039 for index in self.parse_playlist_items(playlist_items):
3040 for i, entry in self[index]:
3041 yield i, entry
3042 if not entry:
3043 continue
3044 try:
3045 # The item may have just been added to archive. Don't break due to it
3046 if not self.ydl.params.get('lazy_playlist'):
3047 # TODO: Add auto-generated fields
3048 self.ydl._match_entry(entry, incomplete=True, silent=True)
3049 except (ExistingVideoReached, RejectedVideoReached):
3050 return
3051
3052 def get_full_count(self):
3053 if self.is_exhausted and not self.is_incomplete:
3054 return len(self)
3055 elif isinstance(self._entries, InAdvancePagedList):
3056 if self._entries._pagesize == 1:
3057 return self._entries._pagecount
3058
3059 @functools.cached_property
3060 def _getter(self):
3061 if isinstance(self._entries, list):
3062 def get_entry(i):
3063 try:
3064 entry = self._entries[i]
3065 except IndexError:
3066 entry = self.MissingEntry
3067 if not self.is_incomplete:
3068 raise self.IndexError()
3069 if entry is self.MissingEntry:
3070 raise EntryNotInPlaylist(f'Entry {i + 1} cannot be found')
3071 return entry
3072 else:
3073 def get_entry(i):
3074 try:
3075 return type(self.ydl)._handle_extraction_exceptions(lambda _, i: self._entries[i])(self.ydl, i)
3076 except (LazyList.IndexError, PagedList.IndexError):
3077 raise self.IndexError()
3078 return get_entry
3079
3080 def __getitem__(self, idx):
3081 if isinstance(idx, int):
3082 idx = slice(idx, idx)
3083
3084 # NB: PlaylistEntries[1:10] => (0, 1, ... 9)
3085 step = 1 if idx.step is None else idx.step
3086 if idx.start is None:
3087 start = 0 if step > 0 else len(self) - 1
3088 else:
3089 start = idx.start - 1 if idx.start >= 0 else len(self) + idx.start
3090
3091 # NB: Do not call len(self) when idx == [:]
3092 if idx.stop is None:
3093 stop = 0 if step < 0 else float('inf')
3094 else:
3095 stop = idx.stop - 1 if idx.stop >= 0 else len(self) + idx.stop
3096 stop += [-1, 1][step > 0]
3097
3098 for i in frange(start, stop, step):
3099 if i < 0:
3100 continue
3101 try:
3102 entry = self._getter(i)
3103 except self.IndexError:
3104 self.is_exhausted = True
3105 if step > 0:
3106 break
3107 continue
3108 yield i + 1, entry
3109
3110 def __len__(self):
3111 return len(tuple(self[:]))
3112
3113 class IndexError(IndexError):
3114 pass
3115
3116
3117 def uppercase_escape(s):
3118 unicode_escape = codecs.getdecoder('unicode_escape')
3119 return re.sub(
3120 r'\\U[0-9a-fA-F]{8}',
3121 lambda m: unicode_escape(m.group(0))[0],
3122 s)
3123
3124
3125 def lowercase_escape(s):
3126 unicode_escape = codecs.getdecoder('unicode_escape')
3127 return re.sub(
3128 r'\\u[0-9a-fA-F]{4}',
3129 lambda m: unicode_escape(m.group(0))[0],
3130 s)
3131
3132
3133 def escape_rfc3986(s):
3134 """Escape non-ASCII characters as suggested by RFC 3986"""
3135 return urllib.parse.quote(s, b"%/;:@&=+$,!~*'()?#[]")
3136
3137
3138 def escape_url(url):
3139 """Escape URL as suggested by RFC 3986"""
3140 url_parsed = urllib.parse.urlparse(url)
3141 return url_parsed._replace(
3142 netloc=url_parsed.netloc.encode('idna').decode('ascii'),
3143 path=escape_rfc3986(url_parsed.path),
3144 params=escape_rfc3986(url_parsed.params),
3145 query=escape_rfc3986(url_parsed.query),
3146 fragment=escape_rfc3986(url_parsed.fragment)
3147 ).geturl()
3148
3149
3150 def parse_qs(url, **kwargs):
3151 return urllib.parse.parse_qs(urllib.parse.urlparse(url).query, **kwargs)
3152
3153
3154 def read_batch_urls(batch_fd):
3155 def fixup(url):
3156 if not isinstance(url, str):
3157 url = url.decode('utf-8', 'replace')
3158 BOM_UTF8 = ('\xef\xbb\xbf', '\ufeff')
3159 for bom in BOM_UTF8:
3160 if url.startswith(bom):
3161 url = url[len(bom):]
3162 url = url.lstrip()
3163 if not url or url.startswith(('#', ';', ']')):
3164 return False
3165 # "#" cannot be stripped out since it is part of the URI
3166 # However, it can be safely stripped out if following a whitespace
3167 return re.split(r'\s#', url, 1)[0].rstrip()
3168
3169 with contextlib.closing(batch_fd) as fd:
3170 return [url for url in map(fixup, fd) if url]
3171
3172
3173 def urlencode_postdata(*args, **kargs):
3174 return urllib.parse.urlencode(*args, **kargs).encode('ascii')
3175
3176
3177 def update_url(url, *, query_update=None, **kwargs):
3178 """Replace URL components specified by kwargs
3179 @param url str or parse url tuple
3180 @param query_update update query
3181 @returns str
3182 """
3183 if isinstance(url, str):
3184 if not kwargs and not query_update:
3185 return url
3186 else:
3187 url = urllib.parse.urlparse(url)
3188 if query_update:
3189 assert 'query' not in kwargs, 'query_update and query cannot be specified at the same time'
3190 kwargs['query'] = urllib.parse.urlencode({
3191 **urllib.parse.parse_qs(url.query),
3192 **query_update
3193 }, True)
3194 return urllib.parse.urlunparse(url._replace(**kwargs))
3195
3196
3197 def update_url_query(url, query):
3198 return update_url(url, query_update=query)
3199
3200
3201 def update_Request(req, url=None, data=None, headers=None, query=None):
3202 req_headers = req.headers.copy()
3203 req_headers.update(headers or {})
3204 req_data = data or req.data
3205 req_url = update_url_query(url or req.get_full_url(), query)
3206 req_get_method = req.get_method()
3207 if req_get_method == 'HEAD':
3208 req_type = HEADRequest
3209 elif req_get_method == 'PUT':
3210 req_type = PUTRequest
3211 else:
3212 req_type = urllib.request.Request
3213 new_req = req_type(
3214 req_url, data=req_data, headers=req_headers,
3215 origin_req_host=req.origin_req_host, unverifiable=req.unverifiable)
3216 if hasattr(req, 'timeout'):
3217 new_req.timeout = req.timeout
3218 return new_req
3219
3220
3221 def _multipart_encode_impl(data, boundary):
3222 content_type = 'multipart/form-data; boundary=%s' % boundary
3223
3224 out = b''
3225 for k, v in data.items():
3226 out += b'--' + boundary.encode('ascii') + b'\r\n'
3227 if isinstance(k, str):
3228 k = k.encode()
3229 if isinstance(v, str):
3230 v = v.encode()
3231 # RFC 2047 requires non-ASCII field names to be encoded, while RFC 7578
3232 # suggests sending UTF-8 directly. Firefox sends UTF-8, too
3233 content = b'Content-Disposition: form-data; name="' + k + b'"\r\n\r\n' + v + b'\r\n'
3234 if boundary.encode('ascii') in content:
3235 raise ValueError('Boundary overlaps with data')
3236 out += content
3237
3238 out += b'--' + boundary.encode('ascii') + b'--\r\n'
3239
3240 return out, content_type
3241
3242
3243 def multipart_encode(data, boundary=None):
3244 '''
3245 Encode a dict to RFC 7578-compliant form-data
3246
3247 data:
3248 A dict where keys and values can be either Unicode or bytes-like
3249 objects.
3250 boundary:
3251 If specified a Unicode object, it's used as the boundary. Otherwise
3252 a random boundary is generated.
3253
3254 Reference: https://tools.ietf.org/html/rfc7578
3255 '''
3256 has_specified_boundary = boundary is not None
3257
3258 while True:
3259 if boundary is None:
3260 boundary = '---------------' + str(random.randrange(0x0fffffff, 0xffffffff))
3261
3262 try:
3263 out, content_type = _multipart_encode_impl(data, boundary)
3264 break
3265 except ValueError:
3266 if has_specified_boundary:
3267 raise
3268 boundary = None
3269
3270 return out, content_type
3271
3272
3273 def variadic(x, allowed_types=(str, bytes, dict)):
3274 return x if isinstance(x, collections.abc.Iterable) and not isinstance(x, allowed_types) else (x,)
3275
3276
3277 def dict_get(d, key_or_keys, default=None, skip_false_values=True):
3278 for val in map(d.get, variadic(key_or_keys)):
3279 if val is not None and (val or not skip_false_values):
3280 return val
3281 return default
3282
3283
3284 def try_call(*funcs, expected_type=None, args=[], kwargs={}):
3285 for f in funcs:
3286 try:
3287 val = f(*args, **kwargs)
3288 except (AttributeError, KeyError, TypeError, IndexError, ValueError, ZeroDivisionError):
3289 pass
3290 else:
3291 if expected_type is None or isinstance(val, expected_type):
3292 return val
3293
3294
3295 def try_get(src, getter, expected_type=None):
3296 return try_call(*variadic(getter), args=(src,), expected_type=expected_type)
3297
3298
3299 def filter_dict(dct, cndn=lambda _, v: v is not None):
3300 return {k: v for k, v in dct.items() if cndn(k, v)}
3301
3302
3303 def merge_dicts(*dicts):
3304 merged = {}
3305 for a_dict in dicts:
3306 for k, v in a_dict.items():
3307 if (v is not None and k not in merged
3308 or isinstance(v, str) and merged[k] == ''):
3309 merged[k] = v
3310 return merged
3311
3312
3313 def encode_compat_str(string, encoding=preferredencoding(), errors='strict'):
3314 return string if isinstance(string, str) else str(string, encoding, errors)
3315
3316
3317 US_RATINGS = {
3318 'G': 0,
3319 'PG': 10,
3320 'PG-13': 13,
3321 'R': 16,
3322 'NC': 18,
3323 }
3324
3325
3326 TV_PARENTAL_GUIDELINES = {
3327 'TV-Y': 0,
3328 'TV-Y7': 7,
3329 'TV-G': 0,
3330 'TV-PG': 0,
3331 'TV-14': 14,
3332 'TV-MA': 17,
3333 }
3334
3335
3336 def parse_age_limit(s):
3337 # isinstance(False, int) is True. So type() must be used instead
3338 if type(s) is int: # noqa: E721
3339 return s if 0 <= s <= 21 else None
3340 elif not isinstance(s, str):
3341 return None
3342 m = re.match(r'^(?P<age>\d{1,2})\+?$', s)
3343 if m:
3344 return int(m.group('age'))
3345 s = s.upper()
3346 if s in US_RATINGS:
3347 return US_RATINGS[s]
3348 m = re.match(r'^TV[_-]?(%s)$' % '|'.join(k[3:] for k in TV_PARENTAL_GUIDELINES), s)
3349 if m:
3350 return TV_PARENTAL_GUIDELINES['TV-' + m.group(1)]
3351 return None
3352
3353
3354 def strip_jsonp(code):
3355 return re.sub(
3356 r'''(?sx)^
3357 (?:window\.)?(?P<func_name>[a-zA-Z0-9_.$]*)
3358 (?:\s*&&\s*(?P=func_name))?
3359 \s*\(\s*(?P<callback_data>.*)\);?
3360 \s*?(?://[^\n]*)*$''',
3361 r'\g<callback_data>', code)
3362
3363
3364 def js_to_json(code, vars={}, *, strict=False):
3365 # vars is a dict of var, val pairs to substitute
3366 STRING_QUOTES = '\'"'
3367 STRING_RE = '|'.join(rf'{q}(?:\\.|[^\\{q}])*{q}' for q in STRING_QUOTES)
3368 COMMENT_RE = r'/\*(?:(?!\*/).)*?\*/|//[^\n]*\n'
3369 SKIP_RE = fr'\s*(?:{COMMENT_RE})?\s*'
3370 INTEGER_TABLE = (
3371 (fr'(?s)^(0[xX][0-9a-fA-F]+){SKIP_RE}:?$', 16),
3372 (fr'(?s)^(0+[0-7]+){SKIP_RE}:?$', 8),
3373 )
3374
3375 def process_escape(match):
3376 JSON_PASSTHROUGH_ESCAPES = R'"\bfnrtu'
3377 escape = match.group(1) or match.group(2)
3378
3379 return (Rf'\{escape}' if escape in JSON_PASSTHROUGH_ESCAPES
3380 else R'\u00' if escape == 'x'
3381 else '' if escape == '\n'
3382 else escape)
3383
3384 def fix_kv(m):
3385 v = m.group(0)
3386 if v in ('true', 'false', 'null'):
3387 return v
3388 elif v in ('undefined', 'void 0'):
3389 return 'null'
3390 elif v.startswith('/*') or v.startswith('//') or v.startswith('!') or v == ',':
3391 return ''
3392
3393 if v[0] in STRING_QUOTES:
3394 escaped = re.sub(r'(?s)(")|\\(.)', process_escape, v[1:-1])
3395 return f'"{escaped}"'
3396
3397 for regex, base in INTEGER_TABLE:
3398 im = re.match(regex, v)
3399 if im:
3400 i = int(im.group(1), base)
3401 return f'"{i}":' if v.endswith(':') else str(i)
3402
3403 if v in vars:
3404 try:
3405 if not strict:
3406 json.loads(vars[v])
3407 except json.JSONDecodeError:
3408 return json.dumps(vars[v])
3409 else:
3410 return vars[v]
3411
3412 if not strict:
3413 return f'"{v}"'
3414
3415 raise ValueError(f'Unknown value: {v}')
3416
3417 def create_map(mobj):
3418 return json.dumps(dict(json.loads(js_to_json(mobj.group(1) or '[]', vars=vars))))
3419
3420 code = re.sub(r'new Map\((\[.*?\])?\)', create_map, code)
3421 if not strict:
3422 code = re.sub(r'new Date\((".+")\)', r'\g<1>', code)
3423 code = re.sub(r'new \w+\((.*?)\)', lambda m: json.dumps(m.group(0)), code)
3424 code = re.sub(r'parseInt\([^\d]+(\d+)[^\d]+\)', r'\1', code)
3425 code = re.sub(r'\(function\([^)]*\)\s*\{[^}]*\}\s*\)\s*\(\s*(["\'][^)]*["\'])\s*\)', r'\1', code)
3426
3427 return re.sub(rf'''(?sx)
3428 {STRING_RE}|
3429 {COMMENT_RE}|,(?={SKIP_RE}[\]}}])|
3430 void\s0|(?:(?<![0-9])[eE]|[a-df-zA-DF-Z_$])[.a-zA-Z_$0-9]*|
3431 \b(?:0[xX][0-9a-fA-F]+|0+[0-7]+)(?:{SKIP_RE}:)?|
3432 [0-9]+(?={SKIP_RE}:)|
3433 !+
3434 ''', fix_kv, code)
3435
3436
3437 def qualities(quality_ids):
3438 """ Get a numeric quality value out of a list of possible values """
3439 def q(qid):
3440 try:
3441 return quality_ids.index(qid)
3442 except ValueError:
3443 return -1
3444 return q
3445
3446
3447 POSTPROCESS_WHEN = ('pre_process', 'after_filter', 'video', 'before_dl', 'post_process', 'after_move', 'after_video', 'playlist')
3448
3449
3450 DEFAULT_OUTTMPL = {
3451 'default': '%(title)s [%(id)s].%(ext)s',
3452 'chapter': '%(title)s - %(section_number)03d %(section_title)s [%(id)s].%(ext)s',
3453 }
3454 OUTTMPL_TYPES = {
3455 'chapter': None,
3456 'subtitle': None,
3457 'thumbnail': None,
3458 'description': 'description',
3459 'annotation': 'annotations.xml',
3460 'infojson': 'info.json',
3461 'link': None,
3462 'pl_video': None,
3463 'pl_thumbnail': None,
3464 'pl_description': 'description',
3465 'pl_infojson': 'info.json',
3466 }
3467
3468 # As of [1] format syntax is:
3469 # %[mapping_key][conversion_flags][minimum_width][.precision][length_modifier]type
3470 # 1. https://docs.python.org/2/library/stdtypes.html#string-formatting
3471 STR_FORMAT_RE_TMPL = r'''(?x)
3472 (?<!%)(?P<prefix>(?:%%)*)
3473 %
3474 (?P<has_key>\((?P<key>{0})\))?
3475 (?P<format>
3476 (?P<conversion>[#0\-+ ]+)?
3477 (?P<min_width>\d+)?
3478 (?P<precision>\.\d+)?
3479 (?P<len_mod>[hlL])? # unused in python
3480 {1} # conversion type
3481 )
3482 '''
3483
3484
3485 STR_FORMAT_TYPES = 'diouxXeEfFgGcrs'
3486
3487
3488 def limit_length(s, length):
3489 """ Add ellipses to overly long strings """
3490 if s is None:
3491 return None
3492 ELLIPSES = '...'
3493 if len(s) > length:
3494 return s[:length - len(ELLIPSES)] + ELLIPSES
3495 return s
3496
3497
3498 def version_tuple(v):
3499 return tuple(int(e) for e in re.split(r'[-.]', v))
3500
3501
3502 def is_outdated_version(version, limit, assume_new=True):
3503 if not version:
3504 return not assume_new
3505 try:
3506 return version_tuple(version) < version_tuple(limit)
3507 except ValueError:
3508 return not assume_new
3509
3510
3511 def ytdl_is_updateable():
3512 """ Returns if yt-dlp can be updated with -U """
3513
3514 from .update import is_non_updateable
3515
3516 return not is_non_updateable()
3517
3518
3519 def args_to_str(args):
3520 # Get a short string representation for a subprocess command
3521 return ' '.join(compat_shlex_quote(a) for a in args)
3522
3523
3524 def error_to_compat_str(err):
3525 return str(err)
3526
3527
3528 def error_to_str(err):
3529 return f'{type(err).__name__}: {err}'
3530
3531
3532 def mimetype2ext(mt, default=NO_DEFAULT):
3533 if not isinstance(mt, str):
3534 if default is not NO_DEFAULT:
3535 return default
3536 return None
3537
3538 MAP = {
3539 # video
3540 '3gpp': '3gp',
3541 'mp2t': 'ts',
3542 'mp4': 'mp4',
3543 'mpeg': 'mpeg',
3544 'mpegurl': 'm3u8',
3545 'quicktime': 'mov',
3546 'webm': 'webm',
3547 'vp9': 'vp9',
3548 'x-flv': 'flv',
3549 'x-m4v': 'm4v',
3550 'x-matroska': 'mkv',
3551 'x-mng': 'mng',
3552 'x-mp4-fragmented': 'mp4',
3553 'x-ms-asf': 'asf',
3554 'x-ms-wmv': 'wmv',
3555 'x-msvideo': 'avi',
3556
3557 # application (streaming playlists)
3558 'dash+xml': 'mpd',
3559 'f4m+xml': 'f4m',
3560 'hds+xml': 'f4m',
3561 'vnd.apple.mpegurl': 'm3u8',
3562 'vnd.ms-sstr+xml': 'ism',
3563 'x-mpegurl': 'm3u8',
3564
3565 # audio
3566 'audio/mp4': 'm4a',
3567 # Per RFC 3003, audio/mpeg can be .mp1, .mp2 or .mp3.
3568 # Using .mp3 as it's the most popular one
3569 'audio/mpeg': 'mp3',
3570 'audio/webm': 'webm',
3571 'audio/x-matroska': 'mka',
3572 'audio/x-mpegurl': 'm3u',
3573 'midi': 'mid',
3574 'ogg': 'ogg',
3575 'wav': 'wav',
3576 'wave': 'wav',
3577 'x-aac': 'aac',
3578 'x-flac': 'flac',
3579 'x-m4a': 'm4a',
3580 'x-realaudio': 'ra',
3581 'x-wav': 'wav',
3582
3583 # image
3584 'avif': 'avif',
3585 'bmp': 'bmp',
3586 'gif': 'gif',
3587 'jpeg': 'jpg',
3588 'png': 'png',
3589 'svg+xml': 'svg',
3590 'tiff': 'tif',
3591 'vnd.wap.wbmp': 'wbmp',
3592 'webp': 'webp',
3593 'x-icon': 'ico',
3594 'x-jng': 'jng',
3595 'x-ms-bmp': 'bmp',
3596
3597 # caption
3598 'filmstrip+json': 'fs',
3599 'smptett+xml': 'tt',
3600 'ttaf+xml': 'dfxp',
3601 'ttml+xml': 'ttml',
3602 'x-ms-sami': 'sami',
3603
3604 # misc
3605 'gzip': 'gz',
3606 'json': 'json',
3607 'xml': 'xml',
3608 'zip': 'zip',
3609 }
3610
3611 mimetype = mt.partition(';')[0].strip().lower()
3612 _, _, subtype = mimetype.rpartition('/')
3613
3614 ext = traverse_obj(MAP, mimetype, subtype, subtype.rsplit('+')[-1])
3615 if ext:
3616 return ext
3617 elif default is not NO_DEFAULT:
3618 return default
3619 return subtype.replace('+', '.')
3620
3621
3622 def ext2mimetype(ext_or_url):
3623 if not ext_or_url:
3624 return None
3625 if '.' not in ext_or_url:
3626 ext_or_url = f'file.{ext_or_url}'
3627 return mimetypes.guess_type(ext_or_url)[0]
3628
3629
3630 def parse_codecs(codecs_str):
3631 # http://tools.ietf.org/html/rfc6381
3632 if not codecs_str:
3633 return {}
3634 split_codecs = list(filter(None, map(
3635 str.strip, codecs_str.strip().strip(',').split(','))))
3636 vcodec, acodec, scodec, hdr = None, None, None, None
3637 for full_codec in split_codecs:
3638 parts = re.sub(r'0+(?=\d)', '', full_codec).split('.')
3639 if parts[0] in ('avc1', 'avc2', 'avc3', 'avc4', 'vp9', 'vp8', 'hev1', 'hev2',
3640 'h263', 'h264', 'mp4v', 'hvc1', 'av1', 'theora', 'dvh1', 'dvhe'):
3641 if vcodec:
3642 continue
3643 vcodec = full_codec
3644 if parts[0] in ('dvh1', 'dvhe'):
3645 hdr = 'DV'
3646 elif parts[0] == 'av1' and traverse_obj(parts, 3) == '10':
3647 hdr = 'HDR10'
3648 elif parts[:2] == ['vp9', '2']:
3649 hdr = 'HDR10'
3650 elif parts[0] in ('flac', 'mp4a', 'opus', 'vorbis', 'mp3', 'aac', 'ac-4',
3651 'ac-3', 'ec-3', 'eac3', 'dtsc', 'dtse', 'dtsh', 'dtsl'):
3652 acodec = acodec or full_codec
3653 elif parts[0] in ('stpp', 'wvtt'):
3654 scodec = scodec or full_codec
3655 else:
3656 write_string(f'WARNING: Unknown codec {full_codec}\n')
3657 if vcodec or acodec or scodec:
3658 return {
3659 'vcodec': vcodec or 'none',
3660 'acodec': acodec or 'none',
3661 'dynamic_range': hdr,
3662 **({'scodec': scodec} if scodec is not None else {}),
3663 }
3664 elif len(split_codecs) == 2:
3665 return {
3666 'vcodec': split_codecs[0],
3667 'acodec': split_codecs[1],
3668 }
3669 return {}
3670
3671
3672 def get_compatible_ext(*, vcodecs, acodecs, vexts, aexts, preferences=None):
3673 assert len(vcodecs) == len(vexts) and len(acodecs) == len(aexts)
3674
3675 allow_mkv = not preferences or 'mkv' in preferences
3676
3677 if allow_mkv and max(len(acodecs), len(vcodecs)) > 1:
3678 return 'mkv' # TODO: any other format allows this?
3679
3680 # TODO: All codecs supported by parse_codecs isn't handled here
3681 COMPATIBLE_CODECS = {
3682 'mp4': {
3683 'av1', 'hevc', 'avc1', 'mp4a', 'ac-4', # fourcc (m3u8, mpd)
3684 'h264', 'aacl', 'ec-3', # Set in ISM
3685 },
3686 'webm': {
3687 'av1', 'vp9', 'vp8', 'opus', 'vrbs',
3688 'vp9x', 'vp8x', # in the webm spec
3689 },
3690 }
3691
3692 sanitize_codec = functools.partial(
3693 try_get, getter=lambda x: x[0].split('.')[0].replace('0', '').lower())
3694 vcodec, acodec = sanitize_codec(vcodecs), sanitize_codec(acodecs)
3695
3696 for ext in preferences or COMPATIBLE_CODECS.keys():
3697 codec_set = COMPATIBLE_CODECS.get(ext, set())
3698 if ext == 'mkv' or codec_set.issuperset((vcodec, acodec)):
3699 return ext
3700
3701 COMPATIBLE_EXTS = (
3702 {'mp3', 'mp4', 'm4a', 'm4p', 'm4b', 'm4r', 'm4v', 'ismv', 'isma', 'mov'},
3703 {'webm', 'weba'},
3704 )
3705 for ext in preferences or vexts:
3706 current_exts = {ext, *vexts, *aexts}
3707 if ext == 'mkv' or current_exts == {ext} or any(
3708 ext_sets.issuperset(current_exts) for ext_sets in COMPATIBLE_EXTS):
3709 return ext
3710 return 'mkv' if allow_mkv else preferences[-1]
3711
3712
3713 def urlhandle_detect_ext(url_handle, default=NO_DEFAULT):
3714 getheader = url_handle.headers.get
3715
3716 cd = getheader('Content-Disposition')
3717 if cd:
3718 m = re.match(r'attachment;\s*filename="(?P<filename>[^"]+)"', cd)
3719 if m:
3720 e = determine_ext(m.group('filename'), default_ext=None)
3721 if e:
3722 return e
3723
3724 meta_ext = getheader('x-amz-meta-name')
3725 if meta_ext:
3726 e = meta_ext.rpartition('.')[2]
3727 if e:
3728 return e
3729
3730 return mimetype2ext(getheader('Content-Type'), default=default)
3731
3732
3733 def encode_data_uri(data, mime_type):
3734 return 'data:%s;base64,%s' % (mime_type, base64.b64encode(data).decode('ascii'))
3735
3736
3737 def age_restricted(content_limit, age_limit):
3738 """ Returns True iff the content should be blocked """
3739
3740 if age_limit is None: # No limit set
3741 return False
3742 if content_limit is None:
3743 return False # Content available for everyone
3744 return age_limit < content_limit
3745
3746
3747 # List of known byte-order-marks (BOM)
3748 BOMS = [
3749 (b'\xef\xbb\xbf', 'utf-8'),
3750 (b'\x00\x00\xfe\xff', 'utf-32-be'),
3751 (b'\xff\xfe\x00\x00', 'utf-32-le'),
3752 (b'\xff\xfe', 'utf-16-le'),
3753 (b'\xfe\xff', 'utf-16-be'),
3754 ]
3755
3756
3757 def is_html(first_bytes):
3758 """ Detect whether a file contains HTML by examining its first bytes. """
3759
3760 encoding = 'utf-8'
3761 for bom, enc in BOMS:
3762 while first_bytes.startswith(bom):
3763 encoding, first_bytes = enc, first_bytes[len(bom):]
3764
3765 return re.match(r'^\s*<', first_bytes.decode(encoding, 'replace'))
3766
3767
3768 def determine_protocol(info_dict):
3769 protocol = info_dict.get('protocol')
3770 if protocol is not None:
3771 return protocol
3772
3773 url = sanitize_url(info_dict['url'])
3774 if url.startswith('rtmp'):
3775 return 'rtmp'
3776 elif url.startswith('mms'):
3777 return 'mms'
3778 elif url.startswith('rtsp'):
3779 return 'rtsp'
3780
3781 ext = determine_ext(url)
3782 if ext == 'm3u8':
3783 return 'm3u8' if info_dict.get('is_live') else 'm3u8_native'
3784 elif ext == 'f4m':
3785 return 'f4m'
3786
3787 return urllib.parse.urlparse(url).scheme
3788
3789
3790 def render_table(header_row, data, delim=False, extra_gap=0, hide_empty=False):
3791 """ Render a list of rows, each as a list of values.
3792 Text after a \t will be right aligned """
3793 def width(string):
3794 return len(remove_terminal_sequences(string).replace('\t', ''))
3795
3796 def get_max_lens(table):
3797 return [max(width(str(v)) for v in col) for col in zip(*table)]
3798
3799 def filter_using_list(row, filterArray):
3800 return [col for take, col in itertools.zip_longest(filterArray, row, fillvalue=True) if take]
3801
3802 max_lens = get_max_lens(data) if hide_empty else []
3803 header_row = filter_using_list(header_row, max_lens)
3804 data = [filter_using_list(row, max_lens) for row in data]
3805
3806 table = [header_row] + data
3807 max_lens = get_max_lens(table)
3808 extra_gap += 1
3809 if delim:
3810 table = [header_row, [delim * (ml + extra_gap) for ml in max_lens]] + data
3811 table[1][-1] = table[1][-1][:-extra_gap * len(delim)] # Remove extra_gap from end of delimiter
3812 for row in table:
3813 for pos, text in enumerate(map(str, row)):
3814 if '\t' in text:
3815 row[pos] = text.replace('\t', ' ' * (max_lens[pos] - width(text))) + ' ' * extra_gap
3816 else:
3817 row[pos] = text + ' ' * (max_lens[pos] - width(text) + extra_gap)
3818 ret = '\n'.join(''.join(row).rstrip() for row in table)
3819 return ret
3820
3821
3822 def _match_one(filter_part, dct, incomplete):
3823 # TODO: Generalize code with YoutubeDL._build_format_filter
3824 STRING_OPERATORS = {
3825 '*=': operator.contains,
3826 '^=': lambda attr, value: attr.startswith(value),
3827 '$=': lambda attr, value: attr.endswith(value),
3828 '~=': lambda attr, value: re.search(value, attr),
3829 }
3830 COMPARISON_OPERATORS = {
3831 **STRING_OPERATORS,
3832 '<=': operator.le, # "<=" must be defined above "<"
3833 '<': operator.lt,
3834 '>=': operator.ge,
3835 '>': operator.gt,
3836 '=': operator.eq,
3837 }
3838
3839 if isinstance(incomplete, bool):
3840 is_incomplete = lambda _: incomplete
3841 else:
3842 is_incomplete = lambda k: k in incomplete
3843
3844 operator_rex = re.compile(r'''(?x)
3845 (?P<key>[a-z_]+)
3846 \s*(?P<negation>!\s*)?(?P<op>%s)(?P<none_inclusive>\s*\?)?\s*
3847 (?:
3848 (?P<quote>["\'])(?P<quotedstrval>.+?)(?P=quote)|
3849 (?P<strval>.+?)
3850 )
3851 ''' % '|'.join(map(re.escape, COMPARISON_OPERATORS.keys())))
3852 m = operator_rex.fullmatch(filter_part.strip())
3853 if m:
3854 m = m.groupdict()
3855 unnegated_op = COMPARISON_OPERATORS[m['op']]
3856 if m['negation']:
3857 op = lambda attr, value: not unnegated_op(attr, value)
3858 else:
3859 op = unnegated_op
3860 comparison_value = m['quotedstrval'] or m['strval'] or m['intval']
3861 if m['quote']:
3862 comparison_value = comparison_value.replace(r'\%s' % m['quote'], m['quote'])
3863 actual_value = dct.get(m['key'])
3864 numeric_comparison = None
3865 if isinstance(actual_value, (int, float)):
3866 # If the original field is a string and matching comparisonvalue is
3867 # a number we should respect the origin of the original field
3868 # and process comparison value as a string (see
3869 # https://github.com/ytdl-org/youtube-dl/issues/11082)
3870 try:
3871 numeric_comparison = int(comparison_value)
3872 except ValueError:
3873 numeric_comparison = parse_filesize(comparison_value)
3874 if numeric_comparison is None:
3875 numeric_comparison = parse_filesize(f'{comparison_value}B')
3876 if numeric_comparison is None:
3877 numeric_comparison = parse_duration(comparison_value)
3878 if numeric_comparison is not None and m['op'] in STRING_OPERATORS:
3879 raise ValueError('Operator %s only supports string values!' % m['op'])
3880 if actual_value is None:
3881 return is_incomplete(m['key']) or m['none_inclusive']
3882 return op(actual_value, comparison_value if numeric_comparison is None else numeric_comparison)
3883
3884 UNARY_OPERATORS = {
3885 '': lambda v: (v is True) if isinstance(v, bool) else (v is not None),
3886 '!': lambda v: (v is False) if isinstance(v, bool) else (v is None),
3887 }
3888 operator_rex = re.compile(r'''(?x)
3889 (?P<op>%s)\s*(?P<key>[a-z_]+)
3890 ''' % '|'.join(map(re.escape, UNARY_OPERATORS.keys())))
3891 m = operator_rex.fullmatch(filter_part.strip())
3892 if m:
3893 op = UNARY_OPERATORS[m.group('op')]
3894 actual_value = dct.get(m.group('key'))
3895 if is_incomplete(m.group('key')) and actual_value is None:
3896 return True
3897 return op(actual_value)
3898
3899 raise ValueError('Invalid filter part %r' % filter_part)
3900
3901
3902 def match_str(filter_str, dct, incomplete=False):
3903 """ Filter a dictionary with a simple string syntax.
3904 @returns Whether the filter passes
3905 @param incomplete Set of keys that is expected to be missing from dct.
3906 Can be True/False to indicate all/none of the keys may be missing.
3907 All conditions on incomplete keys pass if the key is missing
3908 """
3909 return all(
3910 _match_one(filter_part.replace(r'\&', '&'), dct, incomplete)
3911 for filter_part in re.split(r'(?<!\\)&', filter_str))
3912
3913
3914 def match_filter_func(filters):
3915 if not filters:
3916 return None
3917 filters = set(variadic(filters))
3918
3919 interactive = '-' in filters
3920 if interactive:
3921 filters.remove('-')
3922
3923 def _match_func(info_dict, incomplete=False):
3924 if not filters or any(match_str(f, info_dict, incomplete) for f in filters):
3925 return NO_DEFAULT if interactive and not incomplete else None
3926 else:
3927 video_title = info_dict.get('title') or info_dict.get('id') or 'entry'
3928 filter_str = ') | ('.join(map(str.strip, filters))
3929 return f'{video_title} does not pass filter ({filter_str}), skipping ..'
3930 return _match_func
3931
3932
3933 class download_range_func:
3934 def __init__(self, chapters, ranges):
3935 self.chapters, self.ranges = chapters, ranges
3936
3937 def __call__(self, info_dict, ydl):
3938 if not self.ranges and not self.chapters:
3939 yield {}
3940
3941 warning = ('There are no chapters matching the regex' if info_dict.get('chapters')
3942 else 'Cannot match chapters since chapter information is unavailable')
3943 for regex in self.chapters or []:
3944 for i, chapter in enumerate(info_dict.get('chapters') or []):
3945 if re.search(regex, chapter['title']):
3946 warning = None
3947 yield {**chapter, 'index': i}
3948 if self.chapters and warning:
3949 ydl.to_screen(f'[info] {info_dict["id"]}: {warning}')
3950
3951 yield from ({'start_time': start, 'end_time': end} for start, end in self.ranges or [])
3952
3953 def __eq__(self, other):
3954 return (isinstance(other, download_range_func)
3955 and self.chapters == other.chapters and self.ranges == other.ranges)
3956
3957 def __repr__(self):
3958 return f'{__name__}.{type(self).__name__}({self.chapters}, {self.ranges})'
3959
3960
3961 def parse_dfxp_time_expr(time_expr):
3962 if not time_expr:
3963 return
3964
3965 mobj = re.match(rf'^(?P<time_offset>{NUMBER_RE})s?$', time_expr)
3966 if mobj:
3967 return float(mobj.group('time_offset'))
3968
3969 mobj = re.match(r'^(\d+):(\d\d):(\d\d(?:(?:\.|:)\d+)?)$', time_expr)
3970 if mobj:
3971 return 3600 * int(mobj.group(1)) + 60 * int(mobj.group(2)) + float(mobj.group(3).replace(':', '.'))
3972
3973
3974 def srt_subtitles_timecode(seconds):
3975 return '%02d:%02d:%02d,%03d' % timetuple_from_msec(seconds * 1000)
3976
3977
3978 def ass_subtitles_timecode(seconds):
3979 time = timetuple_from_msec(seconds * 1000)
3980 return '%01d:%02d:%02d.%02d' % (*time[:-1], time.milliseconds / 10)
3981
3982
3983 def dfxp2srt(dfxp_data):
3984 '''
3985 @param dfxp_data A bytes-like object containing DFXP data
3986 @returns A unicode object containing converted SRT data
3987 '''
3988 LEGACY_NAMESPACES = (
3989 (b'http://www.w3.org/ns/ttml', [
3990 b'http://www.w3.org/2004/11/ttaf1',
3991 b'http://www.w3.org/2006/04/ttaf1',
3992 b'http://www.w3.org/2006/10/ttaf1',
3993 ]),
3994 (b'http://www.w3.org/ns/ttml#styling', [
3995 b'http://www.w3.org/ns/ttml#style',
3996 ]),
3997 )
3998
3999 SUPPORTED_STYLING = [
4000 'color',
4001 'fontFamily',
4002 'fontSize',
4003 'fontStyle',
4004 'fontWeight',
4005 'textDecoration'
4006 ]
4007
4008 _x = functools.partial(xpath_with_ns, ns_map={
4009 'xml': 'http://www.w3.org/XML/1998/namespace',
4010 'ttml': 'http://www.w3.org/ns/ttml',
4011 'tts': 'http://www.w3.org/ns/ttml#styling',
4012 })
4013
4014 styles = {}
4015 default_style = {}
4016
4017 class TTMLPElementParser:
4018 _out = ''
4019 _unclosed_elements = []
4020 _applied_styles = []
4021
4022 def start(self, tag, attrib):
4023 if tag in (_x('ttml:br'), 'br'):
4024 self._out += '\n'
4025 else:
4026 unclosed_elements = []
4027 style = {}
4028 element_style_id = attrib.get('style')
4029 if default_style:
4030 style.update(default_style)
4031 if element_style_id:
4032 style.update(styles.get(element_style_id, {}))
4033 for prop in SUPPORTED_STYLING:
4034 prop_val = attrib.get(_x('tts:' + prop))
4035 if prop_val:
4036 style[prop] = prop_val
4037 if style:
4038 font = ''
4039 for k, v in sorted(style.items()):
4040 if self._applied_styles and self._applied_styles[-1].get(k) == v:
4041 continue
4042 if k == 'color':
4043 font += ' color="%s"' % v
4044 elif k == 'fontSize':
4045 font += ' size="%s"' % v
4046 elif k == 'fontFamily':
4047 font += ' face="%s"' % v
4048 elif k == 'fontWeight' and v == 'bold':
4049 self._out += '<b>'
4050 unclosed_elements.append('b')
4051 elif k == 'fontStyle' and v == 'italic':
4052 self._out += '<i>'
4053 unclosed_elements.append('i')
4054 elif k == 'textDecoration' and v == 'underline':
4055 self._out += '<u>'
4056 unclosed_elements.append('u')
4057 if font:
4058 self._out += '<font' + font + '>'
4059 unclosed_elements.append('font')
4060 applied_style = {}
4061 if self._applied_styles:
4062 applied_style.update(self._applied_styles[-1])
4063 applied_style.update(style)
4064 self._applied_styles.append(applied_style)
4065 self._unclosed_elements.append(unclosed_elements)
4066
4067 def end(self, tag):
4068 if tag not in (_x('ttml:br'), 'br'):
4069 unclosed_elements = self._unclosed_elements.pop()
4070 for element in reversed(unclosed_elements):
4071 self._out += '</%s>' % element
4072 if unclosed_elements and self._applied_styles:
4073 self._applied_styles.pop()
4074
4075 def data(self, data):
4076 self._out += data
4077
4078 def close(self):
4079 return self._out.strip()
4080
4081 def parse_node(node):
4082 target = TTMLPElementParser()
4083 parser = xml.etree.ElementTree.XMLParser(target=target)
4084 parser.feed(xml.etree.ElementTree.tostring(node))
4085 return parser.close()
4086
4087 for k, v in LEGACY_NAMESPACES:
4088 for ns in v:
4089 dfxp_data = dfxp_data.replace(ns, k)
4090
4091 dfxp = compat_etree_fromstring(dfxp_data)
4092 out = []
4093 paras = dfxp.findall(_x('.//ttml:p')) or dfxp.findall('.//p')
4094
4095 if not paras:
4096 raise ValueError('Invalid dfxp/TTML subtitle')
4097
4098 repeat = False
4099 while True:
4100 for style in dfxp.findall(_x('.//ttml:style')):
4101 style_id = style.get('id') or style.get(_x('xml:id'))
4102 if not style_id:
4103 continue
4104 parent_style_id = style.get('style')
4105 if parent_style_id:
4106 if parent_style_id not in styles:
4107 repeat = True
4108 continue
4109 styles[style_id] = styles[parent_style_id].copy()
4110 for prop in SUPPORTED_STYLING:
4111 prop_val = style.get(_x('tts:' + prop))
4112 if prop_val:
4113 styles.setdefault(style_id, {})[prop] = prop_val
4114 if repeat:
4115 repeat = False
4116 else:
4117 break
4118
4119 for p in ('body', 'div'):
4120 ele = xpath_element(dfxp, [_x('.//ttml:' + p), './/' + p])
4121 if ele is None:
4122 continue
4123 style = styles.get(ele.get('style'))
4124 if not style:
4125 continue
4126 default_style.update(style)
4127
4128 for para, index in zip(paras, itertools.count(1)):
4129 begin_time = parse_dfxp_time_expr(para.attrib.get('begin'))
4130 end_time = parse_dfxp_time_expr(para.attrib.get('end'))
4131 dur = parse_dfxp_time_expr(para.attrib.get('dur'))
4132 if begin_time is None:
4133 continue
4134 if not end_time:
4135 if not dur:
4136 continue
4137 end_time = begin_time + dur
4138 out.append('%d\n%s --> %s\n%s\n\n' % (
4139 index,
4140 srt_subtitles_timecode(begin_time),
4141 srt_subtitles_timecode(end_time),
4142 parse_node(para)))
4143
4144 return ''.join(out)
4145
4146
4147 def cli_option(params, command_option, param, separator=None):
4148 param = params.get(param)
4149 return ([] if param is None
4150 else [command_option, str(param)] if separator is None
4151 else [f'{command_option}{separator}{param}'])
4152
4153
4154 def cli_bool_option(params, command_option, param, true_value='true', false_value='false', separator=None):
4155 param = params.get(param)
4156 assert param in (True, False, None)
4157 return cli_option({True: true_value, False: false_value}, command_option, param, separator)
4158
4159
4160 def cli_valueless_option(params, command_option, param, expected_value=True):
4161 return [command_option] if params.get(param) == expected_value else []
4162
4163
4164 def cli_configuration_args(argdict, keys, default=[], use_compat=True):
4165 if isinstance(argdict, (list, tuple)): # for backward compatibility
4166 if use_compat:
4167 return argdict
4168 else:
4169 argdict = None
4170 if argdict is None:
4171 return default
4172 assert isinstance(argdict, dict)
4173
4174 assert isinstance(keys, (list, tuple))
4175 for key_list in keys:
4176 arg_list = list(filter(
4177 lambda x: x is not None,
4178 [argdict.get(key.lower()) for key in variadic(key_list)]))
4179 if arg_list:
4180 return [arg for args in arg_list for arg in args]
4181 return default
4182
4183
4184 def _configuration_args(main_key, argdict, exe, keys=None, default=[], use_compat=True):
4185 main_key, exe = main_key.lower(), exe.lower()
4186 root_key = exe if main_key == exe else f'{main_key}+{exe}'
4187 keys = [f'{root_key}{k}' for k in (keys or [''])]
4188 if root_key in keys:
4189 if main_key != exe:
4190 keys.append((main_key, exe))
4191 keys.append('default')
4192 else:
4193 use_compat = False
4194 return cli_configuration_args(argdict, keys, default, use_compat)
4195
4196
4197 class ISO639Utils:
4198 # See http://www.loc.gov/standards/iso639-2/ISO-639-2_utf-8.txt
4199 _lang_map = {
4200 'aa': 'aar',
4201 'ab': 'abk',
4202 'ae': 'ave',
4203 'af': 'afr',
4204 'ak': 'aka',
4205 'am': 'amh',
4206 'an': 'arg',
4207 'ar': 'ara',
4208 'as': 'asm',
4209 'av': 'ava',
4210 'ay': 'aym',
4211 'az': 'aze',
4212 'ba': 'bak',
4213 'be': 'bel',
4214 'bg': 'bul',
4215 'bh': 'bih',
4216 'bi': 'bis',
4217 'bm': 'bam',
4218 'bn': 'ben',
4219 'bo': 'bod',
4220 'br': 'bre',
4221 'bs': 'bos',
4222 'ca': 'cat',
4223 'ce': 'che',
4224 'ch': 'cha',
4225 'co': 'cos',
4226 'cr': 'cre',
4227 'cs': 'ces',
4228 'cu': 'chu',
4229 'cv': 'chv',
4230 'cy': 'cym',
4231 'da': 'dan',
4232 'de': 'deu',
4233 'dv': 'div',
4234 'dz': 'dzo',
4235 'ee': 'ewe',
4236 'el': 'ell',
4237 'en': 'eng',
4238 'eo': 'epo',
4239 'es': 'spa',
4240 'et': 'est',
4241 'eu': 'eus',
4242 'fa': 'fas',
4243 'ff': 'ful',
4244 'fi': 'fin',
4245 'fj': 'fij',
4246 'fo': 'fao',
4247 'fr': 'fra',
4248 'fy': 'fry',
4249 'ga': 'gle',
4250 'gd': 'gla',
4251 'gl': 'glg',
4252 'gn': 'grn',
4253 'gu': 'guj',
4254 'gv': 'glv',
4255 'ha': 'hau',
4256 'he': 'heb',
4257 'iw': 'heb', # Replaced by he in 1989 revision
4258 'hi': 'hin',
4259 'ho': 'hmo',
4260 'hr': 'hrv',
4261 'ht': 'hat',
4262 'hu': 'hun',
4263 'hy': 'hye',
4264 'hz': 'her',
4265 'ia': 'ina',
4266 'id': 'ind',
4267 'in': 'ind', # Replaced by id in 1989 revision
4268 'ie': 'ile',
4269 'ig': 'ibo',
4270 'ii': 'iii',
4271 'ik': 'ipk',
4272 'io': 'ido',
4273 'is': 'isl',
4274 'it': 'ita',
4275 'iu': 'iku',
4276 'ja': 'jpn',
4277 'jv': 'jav',
4278 'ka': 'kat',
4279 'kg': 'kon',
4280 'ki': 'kik',
4281 'kj': 'kua',
4282 'kk': 'kaz',
4283 'kl': 'kal',
4284 'km': 'khm',
4285 'kn': 'kan',
4286 'ko': 'kor',
4287 'kr': 'kau',
4288 'ks': 'kas',
4289 'ku': 'kur',
4290 'kv': 'kom',
4291 'kw': 'cor',
4292 'ky': 'kir',
4293 'la': 'lat',
4294 'lb': 'ltz',
4295 'lg': 'lug',
4296 'li': 'lim',
4297 'ln': 'lin',
4298 'lo': 'lao',
4299 'lt': 'lit',
4300 'lu': 'lub',
4301 'lv': 'lav',
4302 'mg': 'mlg',
4303 'mh': 'mah',
4304 'mi': 'mri',
4305 'mk': 'mkd',
4306 'ml': 'mal',
4307 'mn': 'mon',
4308 'mr': 'mar',
4309 'ms': 'msa',
4310 'mt': 'mlt',
4311 'my': 'mya',
4312 'na': 'nau',
4313 'nb': 'nob',
4314 'nd': 'nde',
4315 'ne': 'nep',
4316 'ng': 'ndo',
4317 'nl': 'nld',
4318 'nn': 'nno',
4319 'no': 'nor',
4320 'nr': 'nbl',
4321 'nv': 'nav',
4322 'ny': 'nya',
4323 'oc': 'oci',
4324 'oj': 'oji',
4325 'om': 'orm',
4326 'or': 'ori',
4327 'os': 'oss',
4328 'pa': 'pan',
4329 'pi': 'pli',
4330 'pl': 'pol',
4331 'ps': 'pus',
4332 'pt': 'por',
4333 'qu': 'que',
4334 'rm': 'roh',
4335 'rn': 'run',
4336 'ro': 'ron',
4337 'ru': 'rus',
4338 'rw': 'kin',
4339 'sa': 'san',
4340 'sc': 'srd',
4341 'sd': 'snd',
4342 'se': 'sme',
4343 'sg': 'sag',
4344 'si': 'sin',
4345 'sk': 'slk',
4346 'sl': 'slv',
4347 'sm': 'smo',
4348 'sn': 'sna',
4349 'so': 'som',
4350 'sq': 'sqi',
4351 'sr': 'srp',
4352 'ss': 'ssw',
4353 'st': 'sot',
4354 'su': 'sun',
4355 'sv': 'swe',
4356 'sw': 'swa',
4357 'ta': 'tam',
4358 'te': 'tel',
4359 'tg': 'tgk',
4360 'th': 'tha',
4361 'ti': 'tir',
4362 'tk': 'tuk',
4363 'tl': 'tgl',
4364 'tn': 'tsn',
4365 'to': 'ton',
4366 'tr': 'tur',
4367 'ts': 'tso',
4368 'tt': 'tat',
4369 'tw': 'twi',
4370 'ty': 'tah',
4371 'ug': 'uig',
4372 'uk': 'ukr',
4373 'ur': 'urd',
4374 'uz': 'uzb',
4375 've': 'ven',
4376 'vi': 'vie',
4377 'vo': 'vol',
4378 'wa': 'wln',
4379 'wo': 'wol',
4380 'xh': 'xho',
4381 'yi': 'yid',
4382 'ji': 'yid', # Replaced by yi in 1989 revision
4383 'yo': 'yor',
4384 'za': 'zha',
4385 'zh': 'zho',
4386 'zu': 'zul',
4387 }
4388
4389 @classmethod
4390 def short2long(cls, code):
4391 """Convert language code from ISO 639-1 to ISO 639-2/T"""
4392 return cls._lang_map.get(code[:2])
4393
4394 @classmethod
4395 def long2short(cls, code):
4396 """Convert language code from ISO 639-2/T to ISO 639-1"""
4397 for short_name, long_name in cls._lang_map.items():
4398 if long_name == code:
4399 return short_name
4400
4401
4402 class ISO3166Utils:
4403 # From http://data.okfn.org/data/core/country-list
4404 _country_map = {
4405 'AF': 'Afghanistan',
4406 'AX': 'Åland Islands',
4407 'AL': 'Albania',
4408 'DZ': 'Algeria',
4409 'AS': 'American Samoa',
4410 'AD': 'Andorra',
4411 'AO': 'Angola',
4412 'AI': 'Anguilla',
4413 'AQ': 'Antarctica',
4414 'AG': 'Antigua and Barbuda',
4415 'AR': 'Argentina',
4416 'AM': 'Armenia',
4417 'AW': 'Aruba',
4418 'AU': 'Australia',
4419 'AT': 'Austria',
4420 'AZ': 'Azerbaijan',
4421 'BS': 'Bahamas',
4422 'BH': 'Bahrain',
4423 'BD': 'Bangladesh',
4424 'BB': 'Barbados',
4425 'BY': 'Belarus',
4426 'BE': 'Belgium',
4427 'BZ': 'Belize',
4428 'BJ': 'Benin',
4429 'BM': 'Bermuda',
4430 'BT': 'Bhutan',
4431 'BO': 'Bolivia, Plurinational State of',
4432 'BQ': 'Bonaire, Sint Eustatius and Saba',
4433 'BA': 'Bosnia and Herzegovina',
4434 'BW': 'Botswana',
4435 'BV': 'Bouvet Island',
4436 'BR': 'Brazil',
4437 'IO': 'British Indian Ocean Territory',
4438 'BN': 'Brunei Darussalam',
4439 'BG': 'Bulgaria',
4440 'BF': 'Burkina Faso',
4441 'BI': 'Burundi',
4442 'KH': 'Cambodia',
4443 'CM': 'Cameroon',
4444 'CA': 'Canada',
4445 'CV': 'Cape Verde',
4446 'KY': 'Cayman Islands',
4447 'CF': 'Central African Republic',
4448 'TD': 'Chad',
4449 'CL': 'Chile',
4450 'CN': 'China',
4451 'CX': 'Christmas Island',
4452 'CC': 'Cocos (Keeling) Islands',
4453 'CO': 'Colombia',
4454 'KM': 'Comoros',
4455 'CG': 'Congo',
4456 'CD': 'Congo, the Democratic Republic of the',
4457 'CK': 'Cook Islands',
4458 'CR': 'Costa Rica',
4459 'CI': 'Côte d\'Ivoire',
4460 'HR': 'Croatia',
4461 'CU': 'Cuba',
4462 'CW': 'Curaçao',
4463 'CY': 'Cyprus',
4464 'CZ': 'Czech Republic',
4465 'DK': 'Denmark',
4466 'DJ': 'Djibouti',
4467 'DM': 'Dominica',
4468 'DO': 'Dominican Republic',
4469 'EC': 'Ecuador',
4470 'EG': 'Egypt',
4471 'SV': 'El Salvador',
4472 'GQ': 'Equatorial Guinea',
4473 'ER': 'Eritrea',
4474 'EE': 'Estonia',
4475 'ET': 'Ethiopia',
4476 'FK': 'Falkland Islands (Malvinas)',
4477 'FO': 'Faroe Islands',
4478 'FJ': 'Fiji',
4479 'FI': 'Finland',
4480 'FR': 'France',
4481 'GF': 'French Guiana',
4482 'PF': 'French Polynesia',
4483 'TF': 'French Southern Territories',
4484 'GA': 'Gabon',
4485 'GM': 'Gambia',
4486 'GE': 'Georgia',
4487 'DE': 'Germany',
4488 'GH': 'Ghana',
4489 'GI': 'Gibraltar',
4490 'GR': 'Greece',
4491 'GL': 'Greenland',
4492 'GD': 'Grenada',
4493 'GP': 'Guadeloupe',
4494 'GU': 'Guam',
4495 'GT': 'Guatemala',
4496 'GG': 'Guernsey',
4497 'GN': 'Guinea',
4498 'GW': 'Guinea-Bissau',
4499 'GY': 'Guyana',
4500 'HT': 'Haiti',
4501 'HM': 'Heard Island and McDonald Islands',
4502 'VA': 'Holy See (Vatican City State)',
4503 'HN': 'Honduras',
4504 'HK': 'Hong Kong',
4505 'HU': 'Hungary',
4506 'IS': 'Iceland',
4507 'IN': 'India',
4508 'ID': 'Indonesia',
4509 'IR': 'Iran, Islamic Republic of',
4510 'IQ': 'Iraq',
4511 'IE': 'Ireland',
4512 'IM': 'Isle of Man',
4513 'IL': 'Israel',
4514 'IT': 'Italy',
4515 'JM': 'Jamaica',
4516 'JP': 'Japan',
4517 'JE': 'Jersey',
4518 'JO': 'Jordan',
4519 'KZ': 'Kazakhstan',
4520 'KE': 'Kenya',
4521 'KI': 'Kiribati',
4522 'KP': 'Korea, Democratic People\'s Republic of',
4523 'KR': 'Korea, Republic of',
4524 'KW': 'Kuwait',
4525 'KG': 'Kyrgyzstan',
4526 'LA': 'Lao People\'s Democratic Republic',
4527 'LV': 'Latvia',
4528 'LB': 'Lebanon',
4529 'LS': 'Lesotho',
4530 'LR': 'Liberia',
4531 'LY': 'Libya',
4532 'LI': 'Liechtenstein',
4533 'LT': 'Lithuania',
4534 'LU': 'Luxembourg',
4535 'MO': 'Macao',
4536 'MK': 'Macedonia, the Former Yugoslav Republic of',
4537 'MG': 'Madagascar',
4538 'MW': 'Malawi',
4539 'MY': 'Malaysia',
4540 'MV': 'Maldives',
4541 'ML': 'Mali',
4542 'MT': 'Malta',
4543 'MH': 'Marshall Islands',
4544 'MQ': 'Martinique',
4545 'MR': 'Mauritania',
4546 'MU': 'Mauritius',
4547 'YT': 'Mayotte',
4548 'MX': 'Mexico',
4549 'FM': 'Micronesia, Federated States of',
4550 'MD': 'Moldova, Republic of',
4551 'MC': 'Monaco',
4552 'MN': 'Mongolia',
4553 'ME': 'Montenegro',
4554 'MS': 'Montserrat',
4555 'MA': 'Morocco',
4556 'MZ': 'Mozambique',
4557 'MM': 'Myanmar',
4558 'NA': 'Namibia',
4559 'NR': 'Nauru',
4560 'NP': 'Nepal',
4561 'NL': 'Netherlands',
4562 'NC': 'New Caledonia',
4563 'NZ': 'New Zealand',
4564 'NI': 'Nicaragua',
4565 'NE': 'Niger',
4566 'NG': 'Nigeria',
4567 'NU': 'Niue',
4568 'NF': 'Norfolk Island',
4569 'MP': 'Northern Mariana Islands',
4570 'NO': 'Norway',
4571 'OM': 'Oman',
4572 'PK': 'Pakistan',
4573 'PW': 'Palau',
4574 'PS': 'Palestine, State of',
4575 'PA': 'Panama',
4576 'PG': 'Papua New Guinea',
4577 'PY': 'Paraguay',
4578 'PE': 'Peru',
4579 'PH': 'Philippines',
4580 'PN': 'Pitcairn',
4581 'PL': 'Poland',
4582 'PT': 'Portugal',
4583 'PR': 'Puerto Rico',
4584 'QA': 'Qatar',
4585 'RE': 'Réunion',
4586 'RO': 'Romania',
4587 'RU': 'Russian Federation',
4588 'RW': 'Rwanda',
4589 'BL': 'Saint Barthélemy',
4590 'SH': 'Saint Helena, Ascension and Tristan da Cunha',
4591 'KN': 'Saint Kitts and Nevis',
4592 'LC': 'Saint Lucia',
4593 'MF': 'Saint Martin (French part)',
4594 'PM': 'Saint Pierre and Miquelon',
4595 'VC': 'Saint Vincent and the Grenadines',
4596 'WS': 'Samoa',
4597 'SM': 'San Marino',
4598 'ST': 'Sao Tome and Principe',
4599 'SA': 'Saudi Arabia',
4600 'SN': 'Senegal',
4601 'RS': 'Serbia',
4602 'SC': 'Seychelles',
4603 'SL': 'Sierra Leone',
4604 'SG': 'Singapore',
4605 'SX': 'Sint Maarten (Dutch part)',
4606 'SK': 'Slovakia',
4607 'SI': 'Slovenia',
4608 'SB': 'Solomon Islands',
4609 'SO': 'Somalia',
4610 'ZA': 'South Africa',
4611 'GS': 'South Georgia and the South Sandwich Islands',
4612 'SS': 'South Sudan',
4613 'ES': 'Spain',
4614 'LK': 'Sri Lanka',
4615 'SD': 'Sudan',
4616 'SR': 'Suriname',
4617 'SJ': 'Svalbard and Jan Mayen',
4618 'SZ': 'Swaziland',
4619 'SE': 'Sweden',
4620 'CH': 'Switzerland',
4621 'SY': 'Syrian Arab Republic',
4622 'TW': 'Taiwan, Province of China',
4623 'TJ': 'Tajikistan',
4624 'TZ': 'Tanzania, United Republic of',
4625 'TH': 'Thailand',
4626 'TL': 'Timor-Leste',
4627 'TG': 'Togo',
4628 'TK': 'Tokelau',
4629 'TO': 'Tonga',
4630 'TT': 'Trinidad and Tobago',
4631 'TN': 'Tunisia',
4632 'TR': 'Turkey',
4633 'TM': 'Turkmenistan',
4634 'TC': 'Turks and Caicos Islands',
4635 'TV': 'Tuvalu',
4636 'UG': 'Uganda',
4637 'UA': 'Ukraine',
4638 'AE': 'United Arab Emirates',
4639 'GB': 'United Kingdom',
4640 'US': 'United States',
4641 'UM': 'United States Minor Outlying Islands',
4642 'UY': 'Uruguay',
4643 'UZ': 'Uzbekistan',
4644 'VU': 'Vanuatu',
4645 'VE': 'Venezuela, Bolivarian Republic of',
4646 'VN': 'Viet Nam',
4647 'VG': 'Virgin Islands, British',
4648 'VI': 'Virgin Islands, U.S.',
4649 'WF': 'Wallis and Futuna',
4650 'EH': 'Western Sahara',
4651 'YE': 'Yemen',
4652 'ZM': 'Zambia',
4653 'ZW': 'Zimbabwe',
4654 # Not ISO 3166 codes, but used for IP blocks
4655 'AP': 'Asia/Pacific Region',
4656 'EU': 'Europe',
4657 }
4658
4659 @classmethod
4660 def short2full(cls, code):
4661 """Convert an ISO 3166-2 country code to the corresponding full name"""
4662 return cls._country_map.get(code.upper())
4663
4664
4665 class GeoUtils:
4666 # Major IPv4 address blocks per country
4667 _country_ip_map = {
4668 'AD': '46.172.224.0/19',
4669 'AE': '94.200.0.0/13',
4670 'AF': '149.54.0.0/17',
4671 'AG': '209.59.64.0/18',
4672 'AI': '204.14.248.0/21',
4673 'AL': '46.99.0.0/16',
4674 'AM': '46.70.0.0/15',
4675 'AO': '105.168.0.0/13',
4676 'AP': '182.50.184.0/21',
4677 'AQ': '23.154.160.0/24',
4678 'AR': '181.0.0.0/12',
4679 'AS': '202.70.112.0/20',
4680 'AT': '77.116.0.0/14',
4681 'AU': '1.128.0.0/11',
4682 'AW': '181.41.0.0/18',
4683 'AX': '185.217.4.0/22',
4684 'AZ': '5.197.0.0/16',
4685 'BA': '31.176.128.0/17',
4686 'BB': '65.48.128.0/17',
4687 'BD': '114.130.0.0/16',
4688 'BE': '57.0.0.0/8',
4689 'BF': '102.178.0.0/15',
4690 'BG': '95.42.0.0/15',
4691 'BH': '37.131.0.0/17',
4692 'BI': '154.117.192.0/18',
4693 'BJ': '137.255.0.0/16',
4694 'BL': '185.212.72.0/23',
4695 'BM': '196.12.64.0/18',
4696 'BN': '156.31.0.0/16',
4697 'BO': '161.56.0.0/16',
4698 'BQ': '161.0.80.0/20',
4699 'BR': '191.128.0.0/12',
4700 'BS': '24.51.64.0/18',
4701 'BT': '119.2.96.0/19',
4702 'BW': '168.167.0.0/16',
4703 'BY': '178.120.0.0/13',
4704 'BZ': '179.42.192.0/18',
4705 'CA': '99.224.0.0/11',
4706 'CD': '41.243.0.0/16',
4707 'CF': '197.242.176.0/21',
4708 'CG': '160.113.0.0/16',
4709 'CH': '85.0.0.0/13',
4710 'CI': '102.136.0.0/14',
4711 'CK': '202.65.32.0/19',
4712 'CL': '152.172.0.0/14',
4713 'CM': '102.244.0.0/14',
4714 'CN': '36.128.0.0/10',
4715 'CO': '181.240.0.0/12',
4716 'CR': '201.192.0.0/12',
4717 'CU': '152.206.0.0/15',
4718 'CV': '165.90.96.0/19',
4719 'CW': '190.88.128.0/17',
4720 'CY': '31.153.0.0/16',
4721 'CZ': '88.100.0.0/14',
4722 'DE': '53.0.0.0/8',
4723 'DJ': '197.241.0.0/17',
4724 'DK': '87.48.0.0/12',
4725 'DM': '192.243.48.0/20',
4726 'DO': '152.166.0.0/15',
4727 'DZ': '41.96.0.0/12',
4728 'EC': '186.68.0.0/15',
4729 'EE': '90.190.0.0/15',
4730 'EG': '156.160.0.0/11',
4731 'ER': '196.200.96.0/20',
4732 'ES': '88.0.0.0/11',
4733 'ET': '196.188.0.0/14',
4734 'EU': '2.16.0.0/13',
4735 'FI': '91.152.0.0/13',
4736 'FJ': '144.120.0.0/16',
4737 'FK': '80.73.208.0/21',
4738 'FM': '119.252.112.0/20',
4739 'FO': '88.85.32.0/19',
4740 'FR': '90.0.0.0/9',
4741 'GA': '41.158.0.0/15',
4742 'GB': '25.0.0.0/8',
4743 'GD': '74.122.88.0/21',
4744 'GE': '31.146.0.0/16',
4745 'GF': '161.22.64.0/18',
4746 'GG': '62.68.160.0/19',
4747 'GH': '154.160.0.0/12',
4748 'GI': '95.164.0.0/16',
4749 'GL': '88.83.0.0/19',
4750 'GM': '160.182.0.0/15',
4751 'GN': '197.149.192.0/18',
4752 'GP': '104.250.0.0/19',
4753 'GQ': '105.235.224.0/20',
4754 'GR': '94.64.0.0/13',
4755 'GT': '168.234.0.0/16',
4756 'GU': '168.123.0.0/16',
4757 'GW': '197.214.80.0/20',
4758 'GY': '181.41.64.0/18',
4759 'HK': '113.252.0.0/14',
4760 'HN': '181.210.0.0/16',
4761 'HR': '93.136.0.0/13',
4762 'HT': '148.102.128.0/17',
4763 'HU': '84.0.0.0/14',
4764 'ID': '39.192.0.0/10',
4765 'IE': '87.32.0.0/12',
4766 'IL': '79.176.0.0/13',
4767 'IM': '5.62.80.0/20',
4768 'IN': '117.192.0.0/10',
4769 'IO': '203.83.48.0/21',
4770 'IQ': '37.236.0.0/14',
4771 'IR': '2.176.0.0/12',
4772 'IS': '82.221.0.0/16',
4773 'IT': '79.0.0.0/10',
4774 'JE': '87.244.64.0/18',
4775 'JM': '72.27.0.0/17',
4776 'JO': '176.29.0.0/16',
4777 'JP': '133.0.0.0/8',
4778 'KE': '105.48.0.0/12',
4779 'KG': '158.181.128.0/17',
4780 'KH': '36.37.128.0/17',
4781 'KI': '103.25.140.0/22',
4782 'KM': '197.255.224.0/20',
4783 'KN': '198.167.192.0/19',
4784 'KP': '175.45.176.0/22',
4785 'KR': '175.192.0.0/10',
4786 'KW': '37.36.0.0/14',
4787 'KY': '64.96.0.0/15',
4788 'KZ': '2.72.0.0/13',
4789 'LA': '115.84.64.0/18',
4790 'LB': '178.135.0.0/16',
4791 'LC': '24.92.144.0/20',
4792 'LI': '82.117.0.0/19',
4793 'LK': '112.134.0.0/15',
4794 'LR': '102.183.0.0/16',
4795 'LS': '129.232.0.0/17',
4796 'LT': '78.56.0.0/13',
4797 'LU': '188.42.0.0/16',
4798 'LV': '46.109.0.0/16',
4799 'LY': '41.252.0.0/14',
4800 'MA': '105.128.0.0/11',
4801 'MC': '88.209.64.0/18',
4802 'MD': '37.246.0.0/16',
4803 'ME': '178.175.0.0/17',
4804 'MF': '74.112.232.0/21',
4805 'MG': '154.126.0.0/17',
4806 'MH': '117.103.88.0/21',
4807 'MK': '77.28.0.0/15',
4808 'ML': '154.118.128.0/18',
4809 'MM': '37.111.0.0/17',
4810 'MN': '49.0.128.0/17',
4811 'MO': '60.246.0.0/16',
4812 'MP': '202.88.64.0/20',
4813 'MQ': '109.203.224.0/19',
4814 'MR': '41.188.64.0/18',
4815 'MS': '208.90.112.0/22',
4816 'MT': '46.11.0.0/16',
4817 'MU': '105.16.0.0/12',
4818 'MV': '27.114.128.0/18',
4819 'MW': '102.70.0.0/15',
4820 'MX': '187.192.0.0/11',
4821 'MY': '175.136.0.0/13',
4822 'MZ': '197.218.0.0/15',
4823 'NA': '41.182.0.0/16',
4824 'NC': '101.101.0.0/18',
4825 'NE': '197.214.0.0/18',
4826 'NF': '203.17.240.0/22',
4827 'NG': '105.112.0.0/12',
4828 'NI': '186.76.0.0/15',
4829 'NL': '145.96.0.0/11',
4830 'NO': '84.208.0.0/13',
4831 'NP': '36.252.0.0/15',
4832 'NR': '203.98.224.0/19',
4833 'NU': '49.156.48.0/22',
4834 'NZ': '49.224.0.0/14',
4835 'OM': '5.36.0.0/15',
4836 'PA': '186.72.0.0/15',
4837 'PE': '186.160.0.0/14',
4838 'PF': '123.50.64.0/18',
4839 'PG': '124.240.192.0/19',
4840 'PH': '49.144.0.0/13',
4841 'PK': '39.32.0.0/11',
4842 'PL': '83.0.0.0/11',
4843 'PM': '70.36.0.0/20',
4844 'PR': '66.50.0.0/16',
4845 'PS': '188.161.0.0/16',
4846 'PT': '85.240.0.0/13',
4847 'PW': '202.124.224.0/20',
4848 'PY': '181.120.0.0/14',
4849 'QA': '37.210.0.0/15',
4850 'RE': '102.35.0.0/16',
4851 'RO': '79.112.0.0/13',
4852 'RS': '93.86.0.0/15',
4853 'RU': '5.136.0.0/13',
4854 'RW': '41.186.0.0/16',
4855 'SA': '188.48.0.0/13',
4856 'SB': '202.1.160.0/19',
4857 'SC': '154.192.0.0/11',
4858 'SD': '102.120.0.0/13',
4859 'SE': '78.64.0.0/12',
4860 'SG': '8.128.0.0/10',
4861 'SI': '188.196.0.0/14',
4862 'SK': '78.98.0.0/15',
4863 'SL': '102.143.0.0/17',
4864 'SM': '89.186.32.0/19',
4865 'SN': '41.82.0.0/15',
4866 'SO': '154.115.192.0/18',
4867 'SR': '186.179.128.0/17',
4868 'SS': '105.235.208.0/21',
4869 'ST': '197.159.160.0/19',
4870 'SV': '168.243.0.0/16',
4871 'SX': '190.102.0.0/20',
4872 'SY': '5.0.0.0/16',
4873 'SZ': '41.84.224.0/19',
4874 'TC': '65.255.48.0/20',
4875 'TD': '154.68.128.0/19',
4876 'TG': '196.168.0.0/14',
4877 'TH': '171.96.0.0/13',
4878 'TJ': '85.9.128.0/18',
4879 'TK': '27.96.24.0/21',
4880 'TL': '180.189.160.0/20',
4881 'TM': '95.85.96.0/19',
4882 'TN': '197.0.0.0/11',
4883 'TO': '175.176.144.0/21',
4884 'TR': '78.160.0.0/11',
4885 'TT': '186.44.0.0/15',
4886 'TV': '202.2.96.0/19',
4887 'TW': '120.96.0.0/11',
4888 'TZ': '156.156.0.0/14',
4889 'UA': '37.52.0.0/14',
4890 'UG': '102.80.0.0/13',
4891 'US': '6.0.0.0/8',
4892 'UY': '167.56.0.0/13',
4893 'UZ': '84.54.64.0/18',
4894 'VA': '212.77.0.0/19',
4895 'VC': '207.191.240.0/21',
4896 'VE': '186.88.0.0/13',
4897 'VG': '66.81.192.0/20',
4898 'VI': '146.226.0.0/16',
4899 'VN': '14.160.0.0/11',
4900 'VU': '202.80.32.0/20',
4901 'WF': '117.20.32.0/21',
4902 'WS': '202.4.32.0/19',
4903 'YE': '134.35.0.0/16',
4904 'YT': '41.242.116.0/22',
4905 'ZA': '41.0.0.0/11',
4906 'ZM': '102.144.0.0/13',
4907 'ZW': '102.177.192.0/18',
4908 }
4909
4910 @classmethod
4911 def random_ipv4(cls, code_or_block):
4912 if len(code_or_block) == 2:
4913 block = cls._country_ip_map.get(code_or_block.upper())
4914 if not block:
4915 return None
4916 else:
4917 block = code_or_block
4918 addr, preflen = block.split('/')
4919 addr_min = struct.unpack('!L', socket.inet_aton(addr))[0]
4920 addr_max = addr_min | (0xffffffff >> int(preflen))
4921 return str(socket.inet_ntoa(
4922 struct.pack('!L', random.randint(addr_min, addr_max))))
4923
4924
4925 class PerRequestProxyHandler(urllib.request.ProxyHandler):
4926 def __init__(self, proxies=None):
4927 # Set default handlers
4928 for type in ('http', 'https'):
4929 setattr(self, '%s_open' % type,
4930 lambda r, proxy='__noproxy__', type=type, meth=self.proxy_open:
4931 meth(r, proxy, type))
4932 urllib.request.ProxyHandler.__init__(self, proxies)
4933
4934 def proxy_open(self, req, proxy, type):
4935 req_proxy = req.headers.get('Ytdl-request-proxy')
4936 if req_proxy is not None:
4937 proxy = req_proxy
4938 del req.headers['Ytdl-request-proxy']
4939
4940 if proxy == '__noproxy__':
4941 return None # No Proxy
4942 if urllib.parse.urlparse(proxy).scheme.lower() in ('socks', 'socks4', 'socks4a', 'socks5'):
4943 req.add_header('Ytdl-socks-proxy', proxy)
4944 # yt-dlp's http/https handlers do wrapping the socket with socks
4945 return None
4946 return urllib.request.ProxyHandler.proxy_open(
4947 self, req, proxy, type)
4948
4949
4950 # Both long_to_bytes and bytes_to_long are adapted from PyCrypto, which is
4951 # released into Public Domain
4952 # https://github.com/dlitz/pycrypto/blob/master/lib/Crypto/Util/number.py#L387
4953
4954 def long_to_bytes(n, blocksize=0):
4955 """long_to_bytes(n:long, blocksize:int) : string
4956 Convert a long integer to a byte string.
4957
4958 If optional blocksize is given and greater than zero, pad the front of the
4959 byte string with binary zeros so that the length is a multiple of
4960 blocksize.
4961 """
4962 # after much testing, this algorithm was deemed to be the fastest
4963 s = b''
4964 n = int(n)
4965 while n > 0:
4966 s = struct.pack('>I', n & 0xffffffff) + s
4967 n = n >> 32
4968 # strip off leading zeros
4969 for i in range(len(s)):
4970 if s[i] != b'\000'[0]:
4971 break
4972 else:
4973 # only happens when n == 0
4974 s = b'\000'
4975 i = 0
4976 s = s[i:]
4977 # add back some pad bytes. this could be done more efficiently w.r.t. the
4978 # de-padding being done above, but sigh...
4979 if blocksize > 0 and len(s) % blocksize:
4980 s = (blocksize - len(s) % blocksize) * b'\000' + s
4981 return s
4982
4983
4984 def bytes_to_long(s):
4985 """bytes_to_long(string) : long
4986 Convert a byte string to a long integer.
4987
4988 This is (essentially) the inverse of long_to_bytes().
4989 """
4990 acc = 0
4991 length = len(s)
4992 if length % 4:
4993 extra = (4 - length % 4)
4994 s = b'\000' * extra + s
4995 length = length + extra
4996 for i in range(0, length, 4):
4997 acc = (acc << 32) + struct.unpack('>I', s[i:i + 4])[0]
4998 return acc
4999
5000
5001 def ohdave_rsa_encrypt(data, exponent, modulus):
5002 '''
5003 Implement OHDave's RSA algorithm. See http://www.ohdave.com/rsa/
5004
5005 Input:
5006 data: data to encrypt, bytes-like object
5007 exponent, modulus: parameter e and N of RSA algorithm, both integer
5008 Output: hex string of encrypted data
5009
5010 Limitation: supports one block encryption only
5011 '''
5012
5013 payload = int(binascii.hexlify(data[::-1]), 16)
5014 encrypted = pow(payload, exponent, modulus)
5015 return '%x' % encrypted
5016
5017
5018 def pkcs1pad(data, length):
5019 """
5020 Padding input data with PKCS#1 scheme
5021
5022 @param {int[]} data input data
5023 @param {int} length target length
5024 @returns {int[]} padded data
5025 """
5026 if len(data) > length - 11:
5027 raise ValueError('Input data too long for PKCS#1 padding')
5028
5029 pseudo_random = [random.randint(0, 254) for _ in range(length - len(data) - 3)]
5030 return [0, 2] + pseudo_random + [0] + data
5031
5032
5033 def _base_n_table(n, table):
5034 if not table and not n:
5035 raise ValueError('Either table or n must be specified')
5036 table = (table or '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ')[:n]
5037
5038 if n and n != len(table):
5039 raise ValueError(f'base {n} exceeds table length {len(table)}')
5040 return table
5041
5042
5043 def encode_base_n(num, n=None, table=None):
5044 """Convert given int to a base-n string"""
5045 table = _base_n_table(n, table)
5046 if not num:
5047 return table[0]
5048
5049 result, base = '', len(table)
5050 while num:
5051 result = table[num % base] + result
5052 num = num // base
5053 return result
5054
5055
5056 def decode_base_n(string, n=None, table=None):
5057 """Convert given base-n string to int"""
5058 table = {char: index for index, char in enumerate(_base_n_table(n, table))}
5059 result, base = 0, len(table)
5060 for char in string:
5061 result = result * base + table[char]
5062 return result
5063
5064
5065 def decode_base(value, digits):
5066 deprecation_warning(f'{__name__}.decode_base is deprecated and may be removed '
5067 f'in a future version. Use {__name__}.decode_base_n instead')
5068 return decode_base_n(value, table=digits)
5069
5070
5071 def decode_packed_codes(code):
5072 mobj = re.search(PACKED_CODES_RE, code)
5073 obfuscated_code, base, count, symbols = mobj.groups()
5074 base = int(base)
5075 count = int(count)
5076 symbols = symbols.split('|')
5077 symbol_table = {}
5078
5079 while count:
5080 count -= 1
5081 base_n_count = encode_base_n(count, base)
5082 symbol_table[base_n_count] = symbols[count] or base_n_count
5083
5084 return re.sub(
5085 r'\b(\w+)\b', lambda mobj: symbol_table[mobj.group(0)],
5086 obfuscated_code)
5087
5088
5089 def caesar(s, alphabet, shift):
5090 if shift == 0:
5091 return s
5092 l = len(alphabet)
5093 return ''.join(
5094 alphabet[(alphabet.index(c) + shift) % l] if c in alphabet else c
5095 for c in s)
5096
5097
5098 def rot47(s):
5099 return caesar(s, r'''!"#$%&'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\]^_`abcdefghijklmnopqrstuvwxyz{|}~''', 47)
5100
5101
5102 def parse_m3u8_attributes(attrib):
5103 info = {}
5104 for (key, val) in re.findall(r'(?P<key>[A-Z0-9-]+)=(?P<val>"[^"]+"|[^",]+)(?:,|$)', attrib):
5105 if val.startswith('"'):
5106 val = val[1:-1]
5107 info[key] = val
5108 return info
5109
5110
5111 def urshift(val, n):
5112 return val >> n if val >= 0 else (val + 0x100000000) >> n
5113
5114
5115 # Based on png2str() written by @gdkchan and improved by @yokrysty
5116 # Originally posted at https://github.com/ytdl-org/youtube-dl/issues/9706
5117 def decode_png(png_data):
5118 # Reference: https://www.w3.org/TR/PNG/
5119 header = png_data[8:]
5120
5121 if png_data[:8] != b'\x89PNG\x0d\x0a\x1a\x0a' or header[4:8] != b'IHDR':
5122 raise OSError('Not a valid PNG file.')
5123
5124 int_map = {1: '>B', 2: '>H', 4: '>I'}
5125 unpack_integer = lambda x: struct.unpack(int_map[len(x)], x)[0]
5126
5127 chunks = []
5128
5129 while header:
5130 length = unpack_integer(header[:4])
5131 header = header[4:]
5132
5133 chunk_type = header[:4]
5134 header = header[4:]
5135
5136 chunk_data = header[:length]
5137 header = header[length:]
5138
5139 header = header[4:] # Skip CRC
5140
5141 chunks.append({
5142 'type': chunk_type,
5143 'length': length,
5144 'data': chunk_data
5145 })
5146
5147 ihdr = chunks[0]['data']
5148
5149 width = unpack_integer(ihdr[:4])
5150 height = unpack_integer(ihdr[4:8])
5151
5152 idat = b''
5153
5154 for chunk in chunks:
5155 if chunk['type'] == b'IDAT':
5156 idat += chunk['data']
5157
5158 if not idat:
5159 raise OSError('Unable to read PNG data.')
5160
5161 decompressed_data = bytearray(zlib.decompress(idat))
5162
5163 stride = width * 3
5164 pixels = []
5165
5166 def _get_pixel(idx):
5167 x = idx % stride
5168 y = idx // stride
5169 return pixels[y][x]
5170
5171 for y in range(height):
5172 basePos = y * (1 + stride)
5173 filter_type = decompressed_data[basePos]
5174
5175 current_row = []
5176
5177 pixels.append(current_row)
5178
5179 for x in range(stride):
5180 color = decompressed_data[1 + basePos + x]
5181 basex = y * stride + x
5182 left = 0
5183 up = 0
5184
5185 if x > 2:
5186 left = _get_pixel(basex - 3)
5187 if y > 0:
5188 up = _get_pixel(basex - stride)
5189
5190 if filter_type == 1: # Sub
5191 color = (color + left) & 0xff
5192 elif filter_type == 2: # Up
5193 color = (color + up) & 0xff
5194 elif filter_type == 3: # Average
5195 color = (color + ((left + up) >> 1)) & 0xff
5196 elif filter_type == 4: # Paeth
5197 a = left
5198 b = up
5199 c = 0
5200
5201 if x > 2 and y > 0:
5202 c = _get_pixel(basex - stride - 3)
5203
5204 p = a + b - c
5205
5206 pa = abs(p - a)
5207 pb = abs(p - b)
5208 pc = abs(p - c)
5209
5210 if pa <= pb and pa <= pc:
5211 color = (color + a) & 0xff
5212 elif pb <= pc:
5213 color = (color + b) & 0xff
5214 else:
5215 color = (color + c) & 0xff
5216
5217 current_row.append(color)
5218
5219 return width, height, pixels
5220
5221
5222 def write_xattr(path, key, value):
5223 # Windows: Write xattrs to NTFS Alternate Data Streams:
5224 # http://en.wikipedia.org/wiki/NTFS#Alternate_data_streams_.28ADS.29
5225 if compat_os_name == 'nt':
5226 assert ':' not in key
5227 assert os.path.exists(path)
5228
5229 try:
5230 with open(f'{path}:{key}', 'wb') as f:
5231 f.write(value)
5232 except OSError as e:
5233 raise XAttrMetadataError(e.errno, e.strerror)
5234 return
5235
5236 # UNIX Method 1. Use xattrs/pyxattrs modules
5237
5238 setxattr = None
5239 if getattr(xattr, '_yt_dlp__identifier', None) == 'pyxattr':
5240 # Unicode arguments are not supported in pyxattr until version 0.5.0
5241 # See https://github.com/ytdl-org/youtube-dl/issues/5498
5242 if version_tuple(xattr.__version__) >= (0, 5, 0):
5243 setxattr = xattr.set
5244 elif xattr:
5245 setxattr = xattr.setxattr
5246
5247 if setxattr:
5248 try:
5249 setxattr(path, key, value)
5250 except OSError as e:
5251 raise XAttrMetadataError(e.errno, e.strerror)
5252 return
5253
5254 # UNIX Method 2. Use setfattr/xattr executables
5255 exe = ('setfattr' if check_executable('setfattr', ['--version'])
5256 else 'xattr' if check_executable('xattr', ['-h']) else None)
5257 if not exe:
5258 raise XAttrUnavailableError(
5259 'Couldn\'t find a tool to set the xattrs. Install either the python "xattr" or "pyxattr" modules or the '
5260 + ('"xattr" binary' if sys.platform != 'linux' else 'GNU "attr" package (which contains the "setfattr" tool)'))
5261
5262 value = value.decode()
5263 try:
5264 _, stderr, returncode = Popen.run(
5265 [exe, '-w', key, value, path] if exe == 'xattr' else [exe, '-n', key, '-v', value, path],
5266 text=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.PIPE)
5267 except OSError as e:
5268 raise XAttrMetadataError(e.errno, e.strerror)
5269 if returncode:
5270 raise XAttrMetadataError(returncode, stderr)
5271
5272
5273 def random_birthday(year_field, month_field, day_field):
5274 start_date = datetime.date(1950, 1, 1)
5275 end_date = datetime.date(1995, 12, 31)
5276 offset = random.randint(0, (end_date - start_date).days)
5277 random_date = start_date + datetime.timedelta(offset)
5278 return {
5279 year_field: str(random_date.year),
5280 month_field: str(random_date.month),
5281 day_field: str(random_date.day),
5282 }
5283
5284
5285 def find_available_port(interface=''):
5286 try:
5287 with socket.socket() as sock:
5288 sock.bind((interface, 0))
5289 return sock.getsockname()[1]
5290 except OSError:
5291 return None
5292
5293
5294 # Templates for internet shortcut files, which are plain text files.
5295 DOT_URL_LINK_TEMPLATE = '''\
5296 [InternetShortcut]
5297 URL=%(url)s
5298 '''
5299
5300 DOT_WEBLOC_LINK_TEMPLATE = '''\
5301 <?xml version="1.0" encoding="UTF-8"?>
5302 <!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
5303 <plist version="1.0">
5304 <dict>
5305 \t<key>URL</key>
5306 \t<string>%(url)s</string>
5307 </dict>
5308 </plist>
5309 '''
5310
5311 DOT_DESKTOP_LINK_TEMPLATE = '''\
5312 [Desktop Entry]
5313 Encoding=UTF-8
5314 Name=%(filename)s
5315 Type=Link
5316 URL=%(url)s
5317 Icon=text-html
5318 '''
5319
5320 LINK_TEMPLATES = {
5321 'url': DOT_URL_LINK_TEMPLATE,
5322 'desktop': DOT_DESKTOP_LINK_TEMPLATE,
5323 'webloc': DOT_WEBLOC_LINK_TEMPLATE,
5324 }
5325
5326
5327 def iri_to_uri(iri):
5328 """
5329 Converts an IRI (Internationalized Resource Identifier, allowing Unicode characters) to a URI (Uniform Resource Identifier, ASCII-only).
5330
5331 The function doesn't add an additional layer of escaping; e.g., it doesn't escape `%3C` as `%253C`. Instead, it percent-escapes characters with an underlying UTF-8 encoding *besides* those already escaped, leaving the URI intact.
5332 """
5333
5334 iri_parts = urllib.parse.urlparse(iri)
5335
5336 if '[' in iri_parts.netloc:
5337 raise ValueError('IPv6 URIs are not, yet, supported.')
5338 # Querying `.netloc`, when there's only one bracket, also raises a ValueError.
5339
5340 # The `safe` argument values, that the following code uses, contain the characters that should not be percent-encoded. Everything else but letters, digits and '_.-' will be percent-encoded with an underlying UTF-8 encoding. Everything already percent-encoded will be left as is.
5341
5342 net_location = ''
5343 if iri_parts.username:
5344 net_location += urllib.parse.quote(iri_parts.username, safe=r"!$%&'()*+,~")
5345 if iri_parts.password is not None:
5346 net_location += ':' + urllib.parse.quote(iri_parts.password, safe=r"!$%&'()*+,~")
5347 net_location += '@'
5348
5349 net_location += iri_parts.hostname.encode('idna').decode() # Punycode for Unicode hostnames.
5350 # The 'idna' encoding produces ASCII text.
5351 if iri_parts.port is not None and iri_parts.port != 80:
5352 net_location += ':' + str(iri_parts.port)
5353
5354 return urllib.parse.urlunparse(
5355 (iri_parts.scheme,
5356 net_location,
5357
5358 urllib.parse.quote_plus(iri_parts.path, safe=r"!$%&'()*+,/:;=@|~"),
5359
5360 # Unsure about the `safe` argument, since this is a legacy way of handling parameters.
5361 urllib.parse.quote_plus(iri_parts.params, safe=r"!$%&'()*+,/:;=@|~"),
5362
5363 # Not totally sure about the `safe` argument, since the source does not explicitly mention the query URI component.
5364 urllib.parse.quote_plus(iri_parts.query, safe=r"!$%&'()*+,/:;=?@{|}~"),
5365
5366 urllib.parse.quote_plus(iri_parts.fragment, safe=r"!#$%&'()*+,/:;=?@{|}~")))
5367
5368 # Source for `safe` arguments: https://url.spec.whatwg.org/#percent-encoded-bytes.
5369
5370
5371 def to_high_limit_path(path):
5372 if sys.platform in ['win32', 'cygwin']:
5373 # Work around MAX_PATH limitation on Windows. The maximum allowed length for the individual path segments may still be quite limited.
5374 return '\\\\?\\' + os.path.abspath(path)
5375
5376 return path
5377
5378
5379 def format_field(obj, field=None, template='%s', ignore=NO_DEFAULT, default='', func=IDENTITY):
5380 val = traverse_obj(obj, *variadic(field))
5381 if (not val and val != 0) if ignore is NO_DEFAULT else val in variadic(ignore):
5382 return default
5383 return template % func(val)
5384
5385
5386 def clean_podcast_url(url):
5387 return re.sub(r'''(?x)
5388 (?:
5389 (?:
5390 chtbl\.com/track|
5391 media\.blubrry\.com| # https://create.blubrry.com/resources/podcast-media-download-statistics/getting-started/
5392 play\.podtrac\.com
5393 )/[^/]+|
5394 (?:dts|www)\.podtrac\.com/(?:pts/)?redirect\.[0-9a-z]{3,4}| # http://analytics.podtrac.com/how-to-measure
5395 flex\.acast\.com|
5396 pd(?:
5397 cn\.co| # https://podcorn.com/analytics-prefix/
5398 st\.fm # https://podsights.com/docs/
5399 )/e
5400 )/''', '', url)
5401
5402
5403 _HEX_TABLE = '0123456789abcdef'
5404
5405
5406 def random_uuidv4():
5407 return re.sub(r'[xy]', lambda x: _HEX_TABLE[random.randint(0, 15)], 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx')
5408
5409
5410 def make_dir(path, to_screen=None):
5411 try:
5412 dn = os.path.dirname(path)
5413 if dn:
5414 os.makedirs(dn, exist_ok=True)
5415 return True
5416 except OSError as err:
5417 if callable(to_screen) is not None:
5418 to_screen('unable to create directory ' + error_to_compat_str(err))
5419 return False
5420
5421
5422 def get_executable_path():
5423 from .update import _get_variant_and_executable_path
5424
5425 return os.path.dirname(os.path.abspath(_get_variant_and_executable_path()[1]))
5426
5427
5428 def get_user_config_dirs(package_name):
5429 # .config (e.g. ~/.config/package_name)
5430 xdg_config_home = os.getenv('XDG_CONFIG_HOME') or compat_expanduser('~/.config')
5431 yield os.path.join(xdg_config_home, package_name)
5432
5433 # appdata (%APPDATA%/package_name)
5434 appdata_dir = os.getenv('appdata')
5435 if appdata_dir:
5436 yield os.path.join(appdata_dir, package_name)
5437
5438 # home (~/.package_name)
5439 yield os.path.join(compat_expanduser('~'), f'.{package_name}')
5440
5441
5442 def get_system_config_dirs(package_name):
5443 # /etc/package_name
5444 yield os.path.join('/etc', package_name)
5445
5446
5447 def traverse_obj(
5448 obj, *paths, default=NO_DEFAULT, expected_type=None, get_all=True,
5449 casesense=True, is_user_input=False, traverse_string=False):
5450 """
5451 Safely traverse nested `dict`s and `Sequence`s
5452
5453 >>> obj = [{}, {"key": "value"}]
5454 >>> traverse_obj(obj, (1, "key"))
5455 "value"
5456
5457 Each of the provided `paths` is tested and the first producing a valid result will be returned.
5458 The next path will also be tested if the path branched but no results could be found.
5459 Supported values for traversal are `Mapping`, `Sequence` and `re.Match`.
5460 Unhelpful values (`{}`, `None`) are treated as the absence of a value and discarded.
5461
5462 The paths will be wrapped in `variadic`, so that `'key'` is conveniently the same as `('key', )`.
5463
5464 The keys in the path can be one of:
5465 - `None`: Return the current object.
5466 - `set`: Requires the only item in the set to be a type or function,
5467 like `{type}`/`{func}`. If a `type`, returns only values
5468 of this type. If a function, returns `func(obj)`.
5469 - `str`/`int`: Return `obj[key]`. For `re.Match`, return `obj.group(key)`.
5470 - `slice`: Branch out and return all values in `obj[key]`.
5471 - `Ellipsis`: Branch out and return a list of all values.
5472 - `tuple`/`list`: Branch out and return a list of all matching values.
5473 Read as: `[traverse_obj(obj, branch) for branch in branches]`.
5474 - `function`: Branch out and return values filtered by the function.
5475 Read as: `[value for key, value in obj if function(key, value)]`.
5476 For `Sequence`s, `key` is the index of the value.
5477 For `re.Match`es, `key` is the group number (0 = full match)
5478 as well as additionally any group names, if given.
5479 - `dict` Transform the current object and return a matching dict.
5480 Read as: `{key: traverse_obj(obj, path) for key, path in dct.items()}`.
5481
5482 `tuple`, `list`, and `dict` all support nested paths and branches.
5483
5484 @params paths Paths which to traverse by.
5485 @param default Value to return if the paths do not match.
5486 If the last key in the path is a `dict`, it will apply to each value inside
5487 the dict instead, depth first. Try to avoid if using nested `dict` keys.
5488 @param expected_type If a `type`, only accept final values of this type.
5489 If any other callable, try to call the function on each result.
5490 If the last key in the path is a `dict`, it will apply to each value inside
5491 the dict instead, recursively. This does respect branching paths.
5492 @param get_all If `False`, return the first matching result, otherwise all matching ones.
5493 @param casesense If `False`, consider string dictionary keys as case insensitive.
5494
5495 The following are only meant to be used by YoutubeDL.prepare_outtmpl and are not part of the API
5496
5497 @param is_user_input Whether the keys are generated from user input.
5498 If `True` strings get converted to `int`/`slice` if needed.
5499 @param traverse_string Whether to traverse into objects as strings.
5500 If `True`, any non-compatible object will first be
5501 converted into a string and then traversed into.
5502 The return value of that path will be a string instead,
5503 not respecting any further branching.
5504
5505
5506 @returns The result of the object traversal.
5507 If successful, `get_all=True`, and the path branches at least once,
5508 then a list of results is returned instead.
5509 If no `default` is given and the last path branches, a `list` of results
5510 is always returned. If a path ends on a `dict` that result will always be a `dict`.
5511 """
5512 is_sequence = lambda x: isinstance(x, collections.abc.Sequence) and not isinstance(x, (str, bytes))
5513 casefold = lambda k: k.casefold() if isinstance(k, str) else k
5514
5515 if isinstance(expected_type, type):
5516 type_test = lambda val: val if isinstance(val, expected_type) else None
5517 else:
5518 type_test = lambda val: try_call(expected_type or IDENTITY, args=(val,))
5519
5520 def apply_key(key, obj, is_last):
5521 branching = False
5522 result = None
5523
5524 if obj is None and traverse_string:
5525 pass
5526
5527 elif key is None:
5528 result = obj
5529
5530 elif isinstance(key, set):
5531 assert len(key) == 1, 'Set should only be used to wrap a single item'
5532 item = next(iter(key))
5533 if isinstance(item, type):
5534 if isinstance(obj, item):
5535 result = obj
5536 else:
5537 result = try_call(item, args=(obj,))
5538
5539 elif isinstance(key, (list, tuple)):
5540 branching = True
5541 result = itertools.chain.from_iterable(
5542 apply_path(obj, branch, is_last)[0] for branch in key)
5543
5544 elif key is ...:
5545 branching = True
5546 if isinstance(obj, collections.abc.Mapping):
5547 result = obj.values()
5548 elif is_sequence(obj):
5549 result = obj
5550 elif isinstance(obj, re.Match):
5551 result = obj.groups()
5552 elif traverse_string:
5553 branching = False
5554 result = str(obj)
5555 else:
5556 result = ()
5557
5558 elif callable(key):
5559 branching = True
5560 if isinstance(obj, collections.abc.Mapping):
5561 iter_obj = obj.items()
5562 elif is_sequence(obj):
5563 iter_obj = enumerate(obj)
5564 elif isinstance(obj, re.Match):
5565 iter_obj = itertools.chain(
5566 enumerate((obj.group(), *obj.groups())),
5567 obj.groupdict().items())
5568 elif traverse_string:
5569 branching = False
5570 iter_obj = enumerate(str(obj))
5571 else:
5572 iter_obj = ()
5573
5574 result = (v for k, v in iter_obj if try_call(key, args=(k, v)))
5575 if not branching: # string traversal
5576 result = ''.join(result)
5577
5578 elif isinstance(key, dict):
5579 iter_obj = ((k, _traverse_obj(obj, v, False, is_last)) for k, v in key.items())
5580 result = {
5581 k: v if v is not None else default for k, v in iter_obj
5582 if v is not None or default is not NO_DEFAULT
5583 } or None
5584
5585 elif isinstance(obj, collections.abc.Mapping):
5586 result = (obj.get(key) if casesense or (key in obj) else
5587 next((v for k, v in obj.items() if casefold(k) == key), None))
5588
5589 elif isinstance(obj, re.Match):
5590 if isinstance(key, int) or casesense:
5591 with contextlib.suppress(IndexError):
5592 result = obj.group(key)
5593
5594 elif isinstance(key, str):
5595 result = next((v for k, v in obj.groupdict().items() if casefold(k) == key), None)
5596
5597 elif isinstance(key, (int, slice)):
5598 if is_sequence(obj):
5599 branching = isinstance(key, slice)
5600 with contextlib.suppress(IndexError):
5601 result = obj[key]
5602 elif traverse_string:
5603 with contextlib.suppress(IndexError):
5604 result = str(obj)[key]
5605
5606 return branching, result if branching else (result,)
5607
5608 def lazy_last(iterable):
5609 iterator = iter(iterable)
5610 prev = next(iterator, NO_DEFAULT)
5611 if prev is NO_DEFAULT:
5612 return
5613
5614 for item in iterator:
5615 yield False, prev
5616 prev = item
5617
5618 yield True, prev
5619
5620 def apply_path(start_obj, path, test_type):
5621 objs = (start_obj,)
5622 has_branched = False
5623
5624 key = None
5625 for last, key in lazy_last(variadic(path, (str, bytes, dict, set))):
5626 if is_user_input and isinstance(key, str):
5627 if key == ':':
5628 key = ...
5629 elif ':' in key:
5630 key = slice(*map(int_or_none, key.split(':')))
5631 elif int_or_none(key) is not None:
5632 key = int(key)
5633
5634 if not casesense and isinstance(key, str):
5635 key = key.casefold()
5636
5637 if __debug__ and callable(key):
5638 # Verify function signature
5639 inspect.signature(key).bind(None, None)
5640
5641 new_objs = []
5642 for obj in objs:
5643 branching, results = apply_key(key, obj, last)
5644 has_branched |= branching
5645 new_objs.append(results)
5646
5647 objs = itertools.chain.from_iterable(new_objs)
5648
5649 if test_type and not isinstance(key, (dict, list, tuple)):
5650 objs = map(type_test, objs)
5651
5652 return objs, has_branched, isinstance(key, dict)
5653
5654 def _traverse_obj(obj, path, allow_empty, test_type):
5655 results, has_branched, is_dict = apply_path(obj, path, test_type)
5656 results = LazyList(item for item in results if item not in (None, {}))
5657 if get_all and has_branched:
5658 if results:
5659 return results.exhaust()
5660 if allow_empty:
5661 return [] if default is NO_DEFAULT else default
5662 return None
5663
5664 return results[0] if results else {} if allow_empty and is_dict else None
5665
5666 for index, path in enumerate(paths, 1):
5667 result = _traverse_obj(obj, path, index == len(paths), True)
5668 if result is not None:
5669 return result
5670
5671 return None if default is NO_DEFAULT else default
5672
5673
5674 def traverse_dict(dictn, keys, casesense=True):
5675 deprecation_warning(f'"{__name__}.traverse_dict" is deprecated and may be removed '
5676 f'in a future version. Use "{__name__}.traverse_obj" instead')
5677 return traverse_obj(dictn, keys, casesense=casesense, is_user_input=True, traverse_string=True)
5678
5679
5680 def get_first(obj, keys, **kwargs):
5681 return traverse_obj(obj, (..., *variadic(keys)), **kwargs, get_all=False)
5682
5683
5684 def time_seconds(**kwargs):
5685 """
5686 Returns TZ-aware time in seconds since the epoch (1970-01-01T00:00:00Z)
5687 """
5688 return time.time() + datetime.timedelta(**kwargs).total_seconds()
5689
5690
5691 # create a JSON Web Signature (jws) with HS256 algorithm
5692 # the resulting format is in JWS Compact Serialization
5693 # implemented following JWT https://www.rfc-editor.org/rfc/rfc7519.html
5694 # implemented following JWS https://www.rfc-editor.org/rfc/rfc7515.html
5695 def jwt_encode_hs256(payload_data, key, headers={}):
5696 header_data = {
5697 'alg': 'HS256',
5698 'typ': 'JWT',
5699 }
5700 if headers:
5701 header_data.update(headers)
5702 header_b64 = base64.b64encode(json.dumps(header_data).encode())
5703 payload_b64 = base64.b64encode(json.dumps(payload_data).encode())
5704 h = hmac.new(key.encode(), header_b64 + b'.' + payload_b64, hashlib.sha256)
5705 signature_b64 = base64.b64encode(h.digest())
5706 token = header_b64 + b'.' + payload_b64 + b'.' + signature_b64
5707 return token
5708
5709
5710 # can be extended in future to verify the signature and parse header and return the algorithm used if it's not HS256
5711 def jwt_decode_hs256(jwt):
5712 header_b64, payload_b64, signature_b64 = jwt.split('.')
5713 # add trailing ='s that may have been stripped, superfluous ='s are ignored
5714 payload_data = json.loads(base64.urlsafe_b64decode(f'{payload_b64}==='))
5715 return payload_data
5716
5717
5718 WINDOWS_VT_MODE = False if compat_os_name == 'nt' else None
5719
5720
5721 @functools.cache
5722 def supports_terminal_sequences(stream):
5723 if compat_os_name == 'nt':
5724 if not WINDOWS_VT_MODE:
5725 return False
5726 elif not os.getenv('TERM'):
5727 return False
5728 try:
5729 return stream.isatty()
5730 except BaseException:
5731 return False
5732
5733
5734 def windows_enable_vt_mode():
5735 """Ref: https://bugs.python.org/issue30075 """
5736 if get_windows_version() < (10, 0, 10586):
5737 return
5738
5739 import ctypes
5740 import ctypes.wintypes
5741 import msvcrt
5742
5743 ENABLE_VIRTUAL_TERMINAL_PROCESSING = 0x0004
5744
5745 dll = ctypes.WinDLL('kernel32', use_last_error=False)
5746 handle = os.open('CONOUT$', os.O_RDWR)
5747 try:
5748 h_out = ctypes.wintypes.HANDLE(msvcrt.get_osfhandle(handle))
5749 dw_original_mode = ctypes.wintypes.DWORD()
5750 success = dll.GetConsoleMode(h_out, ctypes.byref(dw_original_mode))
5751 if not success:
5752 raise Exception('GetConsoleMode failed')
5753
5754 success = dll.SetConsoleMode(h_out, ctypes.wintypes.DWORD(
5755 dw_original_mode.value | ENABLE_VIRTUAL_TERMINAL_PROCESSING))
5756 if not success:
5757 raise Exception('SetConsoleMode failed')
5758 finally:
5759 os.close(handle)
5760
5761 global WINDOWS_VT_MODE
5762 WINDOWS_VT_MODE = True
5763 supports_terminal_sequences.cache_clear()
5764
5765
5766 _terminal_sequences_re = re.compile('\033\\[[^m]+m')
5767
5768
5769 def remove_terminal_sequences(string):
5770 return _terminal_sequences_re.sub('', string)
5771
5772
5773 def number_of_digits(number):
5774 return len('%d' % number)
5775
5776
5777 def join_nonempty(*values, delim='-', from_dict=None):
5778 if from_dict is not None:
5779 values = (traverse_obj(from_dict, variadic(v)) for v in values)
5780 return delim.join(map(str, filter(None, values)))
5781
5782
5783 def scale_thumbnails_to_max_format_width(formats, thumbnails, url_width_re):
5784 """
5785 Find the largest format dimensions in terms of video width and, for each thumbnail:
5786 * Modify the URL: Match the width with the provided regex and replace with the former width
5787 * Update dimensions
5788
5789 This function is useful with video services that scale the provided thumbnails on demand
5790 """
5791 _keys = ('width', 'height')
5792 max_dimensions = max(
5793 (tuple(format.get(k) or 0 for k in _keys) for format in formats),
5794 default=(0, 0))
5795 if not max_dimensions[0]:
5796 return thumbnails
5797 return [
5798 merge_dicts(
5799 {'url': re.sub(url_width_re, str(max_dimensions[0]), thumbnail['url'])},
5800 dict(zip(_keys, max_dimensions)), thumbnail)
5801 for thumbnail in thumbnails
5802 ]
5803
5804
5805 def parse_http_range(range):
5806 """ Parse value of "Range" or "Content-Range" HTTP header into tuple. """
5807 if not range:
5808 return None, None, None
5809 crg = re.search(r'bytes[ =](\d+)-(\d+)?(?:/(\d+))?', range)
5810 if not crg:
5811 return None, None, None
5812 return int(crg.group(1)), int_or_none(crg.group(2)), int_or_none(crg.group(3))
5813
5814
5815 def read_stdin(what):
5816 eof = 'Ctrl+Z' if compat_os_name == 'nt' else 'Ctrl+D'
5817 write_string(f'Reading {what} from STDIN - EOF ({eof}) to end:\n')
5818 return sys.stdin
5819
5820
5821 def determine_file_encoding(data):
5822 """
5823 Detect the text encoding used
5824 @returns (encoding, bytes to skip)
5825 """
5826
5827 # BOM marks are given priority over declarations
5828 for bom, enc in BOMS:
5829 if data.startswith(bom):
5830 return enc, len(bom)
5831
5832 # Strip off all null bytes to match even when UTF-16 or UTF-32 is used.
5833 # We ignore the endianness to get a good enough match
5834 data = data.replace(b'\0', b'')
5835 mobj = re.match(rb'(?m)^#\s*coding\s*:\s*(\S+)\s*$', data)
5836 return mobj.group(1).decode() if mobj else None, 0
5837
5838
5839 class Config:
5840 own_args = None
5841 parsed_args = None
5842 filename = None
5843 __initialized = False
5844
5845 def __init__(self, parser, label=None):
5846 self.parser, self.label = parser, label
5847 self._loaded_paths, self.configs = set(), []
5848
5849 def init(self, args=None, filename=None):
5850 assert not self.__initialized
5851 self.own_args, self.filename = args, filename
5852 return self.load_configs()
5853
5854 def load_configs(self):
5855 directory = ''
5856 if self.filename:
5857 location = os.path.realpath(self.filename)
5858 directory = os.path.dirname(location)
5859 if location in self._loaded_paths:
5860 return False
5861 self._loaded_paths.add(location)
5862
5863 self.__initialized = True
5864 opts, _ = self.parser.parse_known_args(self.own_args)
5865 self.parsed_args = self.own_args
5866 for location in opts.config_locations or []:
5867 if location == '-':
5868 if location in self._loaded_paths:
5869 continue
5870 self._loaded_paths.add(location)
5871 self.append_config(shlex.split(read_stdin('options'), comments=True), label='stdin')
5872 continue
5873 location = os.path.join(directory, expand_path(location))
5874 if os.path.isdir(location):
5875 location = os.path.join(location, 'yt-dlp.conf')
5876 if not os.path.exists(location):
5877 self.parser.error(f'config location {location} does not exist')
5878 self.append_config(self.read_file(location), location)
5879 return True
5880
5881 def __str__(self):
5882 label = join_nonempty(
5883 self.label, 'config', f'"{self.filename}"' if self.filename else '',
5884 delim=' ')
5885 return join_nonempty(
5886 self.own_args is not None and f'{label[0].upper()}{label[1:]}: {self.hide_login_info(self.own_args)}',
5887 *(f'\n{c}'.replace('\n', '\n| ')[1:] for c in self.configs),
5888 delim='\n')
5889
5890 @staticmethod
5891 def read_file(filename, default=[]):
5892 try:
5893 optionf = open(filename, 'rb')
5894 except OSError:
5895 return default # silently skip if file is not present
5896 try:
5897 enc, skip = determine_file_encoding(optionf.read(512))
5898 optionf.seek(skip, io.SEEK_SET)
5899 except OSError:
5900 enc = None # silently skip read errors
5901 try:
5902 # FIXME: https://github.com/ytdl-org/youtube-dl/commit/dfe5fa49aed02cf36ba9f743b11b0903554b5e56
5903 contents = optionf.read().decode(enc or preferredencoding())
5904 res = shlex.split(contents, comments=True)
5905 except Exception as err:
5906 raise ValueError(f'Unable to parse "{filename}": {err}')
5907 finally:
5908 optionf.close()
5909 return res
5910
5911 @staticmethod
5912 def hide_login_info(opts):
5913 PRIVATE_OPTS = {'-p', '--password', '-u', '--username', '--video-password', '--ap-password', '--ap-username'}
5914 eqre = re.compile('^(?P<key>' + ('|'.join(re.escape(po) for po in PRIVATE_OPTS)) + ')=.+$')
5915
5916 def _scrub_eq(o):
5917 m = eqre.match(o)
5918 if m:
5919 return m.group('key') + '=PRIVATE'
5920 else:
5921 return o
5922
5923 opts = list(map(_scrub_eq, opts))
5924 for idx, opt in enumerate(opts):
5925 if opt in PRIVATE_OPTS and idx + 1 < len(opts):
5926 opts[idx + 1] = 'PRIVATE'
5927 return opts
5928
5929 def append_config(self, *args, label=None):
5930 config = type(self)(self.parser, label)
5931 config._loaded_paths = self._loaded_paths
5932 if config.init(*args):
5933 self.configs.append(config)
5934
5935 @property
5936 def all_args(self):
5937 for config in reversed(self.configs):
5938 yield from config.all_args
5939 yield from self.parsed_args or []
5940
5941 def parse_known_args(self, **kwargs):
5942 return self.parser.parse_known_args(self.all_args, **kwargs)
5943
5944 def parse_args(self):
5945 return self.parser.parse_args(self.all_args)
5946
5947
5948 class WebSocketsWrapper:
5949 """Wraps websockets module to use in non-async scopes"""
5950 pool = None
5951
5952 def __init__(self, url, headers=None, connect=True):
5953 self.loop = asyncio.new_event_loop()
5954 # XXX: "loop" is deprecated
5955 self.conn = websockets.connect(
5956 url, extra_headers=headers, ping_interval=None,
5957 close_timeout=float('inf'), loop=self.loop, ping_timeout=float('inf'))
5958 if connect:
5959 self.__enter__()
5960 atexit.register(self.__exit__, None, None, None)
5961
5962 def __enter__(self):
5963 if not self.pool:
5964 self.pool = self.run_with_loop(self.conn.__aenter__(), self.loop)
5965 return self
5966
5967 def send(self, *args):
5968 self.run_with_loop(self.pool.send(*args), self.loop)
5969
5970 def recv(self, *args):
5971 return self.run_with_loop(self.pool.recv(*args), self.loop)
5972
5973 def __exit__(self, type, value, traceback):
5974 try:
5975 return self.run_with_loop(self.conn.__aexit__(type, value, traceback), self.loop)
5976 finally:
5977 self.loop.close()
5978 self._cancel_all_tasks(self.loop)
5979
5980 # taken from https://github.com/python/cpython/blob/3.9/Lib/asyncio/runners.py with modifications
5981 # for contributors: If there's any new library using asyncio needs to be run in non-async, move these function out of this class
5982 @staticmethod
5983 def run_with_loop(main, loop):
5984 if not asyncio.iscoroutine(main):
5985 raise ValueError(f'a coroutine was expected, got {main!r}')
5986
5987 try:
5988 return loop.run_until_complete(main)
5989 finally:
5990 loop.run_until_complete(loop.shutdown_asyncgens())
5991 if hasattr(loop, 'shutdown_default_executor'):
5992 loop.run_until_complete(loop.shutdown_default_executor())
5993
5994 @staticmethod
5995 def _cancel_all_tasks(loop):
5996 to_cancel = asyncio.all_tasks(loop)
5997
5998 if not to_cancel:
5999 return
6000
6001 for task in to_cancel:
6002 task.cancel()
6003
6004 # XXX: "loop" is removed in python 3.10+
6005 loop.run_until_complete(
6006 asyncio.gather(*to_cancel, loop=loop, return_exceptions=True))
6007
6008 for task in to_cancel:
6009 if task.cancelled():
6010 continue
6011 if task.exception() is not None:
6012 loop.call_exception_handler({
6013 'message': 'unhandled exception during asyncio.run() shutdown',
6014 'exception': task.exception(),
6015 'task': task,
6016 })
6017
6018
6019 def merge_headers(*dicts):
6020 """Merge dicts of http headers case insensitively, prioritizing the latter ones"""
6021 return {k.title(): v for k, v in itertools.chain.from_iterable(map(dict.items, dicts))}
6022
6023
6024 def cached_method(f):
6025 """Cache a method"""
6026 signature = inspect.signature(f)
6027
6028 @functools.wraps(f)
6029 def wrapper(self, *args, **kwargs):
6030 bound_args = signature.bind(self, *args, **kwargs)
6031 bound_args.apply_defaults()
6032 key = tuple(bound_args.arguments.values())[1:]
6033
6034 cache = vars(self).setdefault('_cached_method__cache', {}).setdefault(f.__name__, {})
6035 if key not in cache:
6036 cache[key] = f(self, *args, **kwargs)
6037 return cache[key]
6038 return wrapper
6039
6040
6041 class classproperty:
6042 """property access for class methods with optional caching"""
6043 def __new__(cls, func=None, *args, **kwargs):
6044 if not func:
6045 return functools.partial(cls, *args, **kwargs)
6046 return super().__new__(cls)
6047
6048 def __init__(self, func, *, cache=False):
6049 functools.update_wrapper(self, func)
6050 self.func = func
6051 self._cache = {} if cache else None
6052
6053 def __get__(self, _, cls):
6054 if self._cache is None:
6055 return self.func(cls)
6056 elif cls not in self._cache:
6057 self._cache[cls] = self.func(cls)
6058 return self._cache[cls]
6059
6060
6061 class function_with_repr:
6062 def __init__(self, func, repr_=None):
6063 functools.update_wrapper(self, func)
6064 self.func, self.__repr = func, repr_
6065
6066 def __call__(self, *args, **kwargs):
6067 return self.func(*args, **kwargs)
6068
6069 def __repr__(self):
6070 if self.__repr:
6071 return self.__repr
6072 return f'{self.func.__module__}.{self.func.__qualname__}'
6073
6074
6075 class Namespace(types.SimpleNamespace):
6076 """Immutable namespace"""
6077
6078 def __iter__(self):
6079 return iter(self.__dict__.values())
6080
6081 @property
6082 def items_(self):
6083 return self.__dict__.items()
6084
6085
6086 MEDIA_EXTENSIONS = Namespace(
6087 common_video=('avi', 'flv', 'mkv', 'mov', 'mp4', 'webm'),
6088 video=('3g2', '3gp', 'f4v', 'mk3d', 'divx', 'mpg', 'ogv', 'm4v', 'wmv'),
6089 common_audio=('aiff', 'alac', 'flac', 'm4a', 'mka', 'mp3', 'ogg', 'opus', 'wav'),
6090 audio=('aac', 'ape', 'asf', 'f4a', 'f4b', 'm4b', 'm4p', 'm4r', 'oga', 'ogx', 'spx', 'vorbis', 'wma', 'weba'),
6091 thumbnails=('jpg', 'png', 'webp'),
6092 storyboards=('mhtml', ),
6093 subtitles=('srt', 'vtt', 'ass', 'lrc'),
6094 manifests=('f4f', 'f4m', 'm3u8', 'smil', 'mpd'),
6095 )
6096 MEDIA_EXTENSIONS.video += MEDIA_EXTENSIONS.common_video
6097 MEDIA_EXTENSIONS.audio += MEDIA_EXTENSIONS.common_audio
6098
6099 KNOWN_EXTENSIONS = (*MEDIA_EXTENSIONS.video, *MEDIA_EXTENSIONS.audio, *MEDIA_EXTENSIONS.manifests)
6100
6101
6102 class RetryManager:
6103 """Usage:
6104 for retry in RetryManager(...):
6105 try:
6106 ...
6107 except SomeException as err:
6108 retry.error = err
6109 continue
6110 """
6111 attempt, _error = 0, None
6112
6113 def __init__(self, _retries, _error_callback, **kwargs):
6114 self.retries = _retries or 0
6115 self.error_callback = functools.partial(_error_callback, **kwargs)
6116
6117 def _should_retry(self):
6118 return self._error is not NO_DEFAULT and self.attempt <= self.retries
6119
6120 @property
6121 def error(self):
6122 if self._error is NO_DEFAULT:
6123 return None
6124 return self._error
6125
6126 @error.setter
6127 def error(self, value):
6128 self._error = value
6129
6130 def __iter__(self):
6131 while self._should_retry():
6132 self.error = NO_DEFAULT
6133 self.attempt += 1
6134 yield self
6135 if self.error:
6136 self.error_callback(self.error, self.attempt, self.retries)
6137
6138 @staticmethod
6139 def report_retry(e, count, retries, *, sleep_func, info, warn, error=None, suffix=None):
6140 """Utility function for reporting retries"""
6141 if count > retries:
6142 if error:
6143 return error(f'{e}. Giving up after {count - 1} retries') if count > 1 else error(str(e))
6144 raise e
6145
6146 if not count:
6147 return warn(e)
6148 elif isinstance(e, ExtractorError):
6149 e = remove_end(str_or_none(e.cause) or e.orig_msg, '.')
6150 warn(f'{e}. Retrying{format_field(suffix, None, " %s")} ({count}/{retries})...')
6151
6152 delay = float_or_none(sleep_func(n=count - 1)) if callable(sleep_func) else sleep_func
6153 if delay:
6154 info(f'Sleeping {delay:.2f} seconds ...')
6155 time.sleep(delay)
6156
6157
6158 def make_archive_id(ie, video_id):
6159 ie_key = ie if isinstance(ie, str) else ie.ie_key()
6160 return f'{ie_key.lower()} {video_id}'
6161
6162
6163 def truncate_string(s, left, right=0):
6164 assert left > 3 and right >= 0
6165 if s is None or len(s) <= left + right:
6166 return s
6167 return f'{s[:left-3]}...{s[-right:] if right else ""}'
6168
6169
6170 def orderedSet_from_options(options, alias_dict, *, use_regex=False, start=None):
6171 assert 'all' in alias_dict, '"all" alias is required'
6172 requested = list(start or [])
6173 for val in options:
6174 discard = val.startswith('-')
6175 if discard:
6176 val = val[1:]
6177
6178 if val in alias_dict:
6179 val = alias_dict[val] if not discard else [
6180 i[1:] if i.startswith('-') else f'-{i}' for i in alias_dict[val]]
6181 # NB: Do not allow regex in aliases for performance
6182 requested = orderedSet_from_options(val, alias_dict, start=requested)
6183 continue
6184
6185 current = (filter(re.compile(val, re.I).fullmatch, alias_dict['all']) if use_regex
6186 else [val] if val in alias_dict['all'] else None)
6187 if current is None:
6188 raise ValueError(val)
6189
6190 if discard:
6191 for item in current:
6192 while item in requested:
6193 requested.remove(item)
6194 else:
6195 requested.extend(current)
6196
6197 return orderedSet(requested)
6198
6199
6200 class FormatSorter:
6201 regex = r' *((?P<reverse>\+)?(?P<field>[a-zA-Z0-9_]+)((?P<separator>[~:])(?P<limit>.*?))?)? *$'
6202
6203 default = ('hidden', 'aud_or_vid', 'hasvid', 'ie_pref', 'lang', 'quality',
6204 'res', 'fps', 'hdr:12', 'vcodec:vp9.2', 'channels', 'acodec',
6205 'size', 'br', 'asr', 'proto', 'ext', 'hasaud', 'source', 'id') # These must not be aliases
6206 ytdl_default = ('hasaud', 'lang', 'quality', 'tbr', 'filesize', 'vbr',
6207 'height', 'width', 'proto', 'vext', 'abr', 'aext',
6208 'fps', 'fs_approx', 'source', 'id')
6209
6210 settings = {
6211 'vcodec': {'type': 'ordered', 'regex': True,
6212 'order': ['av0?1', 'vp0?9.2', 'vp0?9', '[hx]265|he?vc?', '[hx]264|avc', 'vp0?8', 'mp4v|h263', 'theora', '', None, 'none']},
6213 'acodec': {'type': 'ordered', 'regex': True,
6214 'order': ['[af]lac', 'wav|aiff', 'opus', 'vorbis|ogg', 'aac', 'mp?4a?', 'mp3', 'ac-?4', 'e-?a?c-?3', 'ac-?3', 'dts', '', None, 'none']},
6215 'hdr': {'type': 'ordered', 'regex': True, 'field': 'dynamic_range',
6216 'order': ['dv', '(hdr)?12', r'(hdr)?10\+', '(hdr)?10', 'hlg', '', 'sdr', None]},
6217 'proto': {'type': 'ordered', 'regex': True, 'field': 'protocol',
6218 'order': ['(ht|f)tps', '(ht|f)tp$', 'm3u8.*', '.*dash', 'websocket_frag', 'rtmpe?', '', 'mms|rtsp', 'ws|websocket', 'f4']},
6219 'vext': {'type': 'ordered', 'field': 'video_ext',
6220 'order': ('mp4', 'mov', 'webm', 'flv', '', 'none'),
6221 'order_free': ('webm', 'mp4', 'mov', 'flv', '', 'none')},
6222 'aext': {'type': 'ordered', 'regex': True, 'field': 'audio_ext',
6223 'order': ('m4a', 'aac', 'mp3', 'ogg', 'opus', 'web[am]', '', 'none'),
6224 'order_free': ('ogg', 'opus', 'web[am]', 'mp3', 'm4a', 'aac', '', 'none')},
6225 'hidden': {'visible': False, 'forced': True, 'type': 'extractor', 'max': -1000},
6226 'aud_or_vid': {'visible': False, 'forced': True, 'type': 'multiple',
6227 'field': ('vcodec', 'acodec'),
6228 'function': lambda it: int(any(v != 'none' for v in it))},
6229 'ie_pref': {'priority': True, 'type': 'extractor'},
6230 'hasvid': {'priority': True, 'field': 'vcodec', 'type': 'boolean', 'not_in_list': ('none',)},
6231 'hasaud': {'field': 'acodec', 'type': 'boolean', 'not_in_list': ('none',)},
6232 'lang': {'convert': 'float', 'field': 'language_preference', 'default': -1},
6233 'quality': {'convert': 'float', 'default': -1},
6234 'filesize': {'convert': 'bytes'},
6235 'fs_approx': {'convert': 'bytes', 'field': 'filesize_approx'},
6236 'id': {'convert': 'string', 'field': 'format_id'},
6237 'height': {'convert': 'float_none'},
6238 'width': {'convert': 'float_none'},
6239 'fps': {'convert': 'float_none'},
6240 'channels': {'convert': 'float_none', 'field': 'audio_channels'},
6241 'tbr': {'convert': 'float_none'},
6242 'vbr': {'convert': 'float_none'},
6243 'abr': {'convert': 'float_none'},
6244 'asr': {'convert': 'float_none'},
6245 'source': {'convert': 'float', 'field': 'source_preference', 'default': -1},
6246
6247 'codec': {'type': 'combined', 'field': ('vcodec', 'acodec')},
6248 'br': {'type': 'combined', 'field': ('tbr', 'vbr', 'abr'), 'same_limit': True},
6249 'size': {'type': 'combined', 'same_limit': True, 'field': ('filesize', 'fs_approx')},
6250 'ext': {'type': 'combined', 'field': ('vext', 'aext')},
6251 'res': {'type': 'multiple', 'field': ('height', 'width'),
6252 'function': lambda it: (lambda l: min(l) if l else 0)(tuple(filter(None, it)))},
6253
6254 # Actual field names
6255 'format_id': {'type': 'alias', 'field': 'id'},
6256 'preference': {'type': 'alias', 'field': 'ie_pref'},
6257 'language_preference': {'type': 'alias', 'field': 'lang'},
6258 'source_preference': {'type': 'alias', 'field': 'source'},
6259 'protocol': {'type': 'alias', 'field': 'proto'},
6260 'filesize_approx': {'type': 'alias', 'field': 'fs_approx'},
6261 'audio_channels': {'type': 'alias', 'field': 'channels'},
6262
6263 # Deprecated
6264 'dimension': {'type': 'alias', 'field': 'res', 'deprecated': True},
6265 'resolution': {'type': 'alias', 'field': 'res', 'deprecated': True},
6266 'extension': {'type': 'alias', 'field': 'ext', 'deprecated': True},
6267 'bitrate': {'type': 'alias', 'field': 'br', 'deprecated': True},
6268 'total_bitrate': {'type': 'alias', 'field': 'tbr', 'deprecated': True},
6269 'video_bitrate': {'type': 'alias', 'field': 'vbr', 'deprecated': True},
6270 'audio_bitrate': {'type': 'alias', 'field': 'abr', 'deprecated': True},
6271 'framerate': {'type': 'alias', 'field': 'fps', 'deprecated': True},
6272 'filesize_estimate': {'type': 'alias', 'field': 'size', 'deprecated': True},
6273 'samplerate': {'type': 'alias', 'field': 'asr', 'deprecated': True},
6274 'video_ext': {'type': 'alias', 'field': 'vext', 'deprecated': True},
6275 'audio_ext': {'type': 'alias', 'field': 'aext', 'deprecated': True},
6276 'video_codec': {'type': 'alias', 'field': 'vcodec', 'deprecated': True},
6277 'audio_codec': {'type': 'alias', 'field': 'acodec', 'deprecated': True},
6278 'video': {'type': 'alias', 'field': 'hasvid', 'deprecated': True},
6279 'has_video': {'type': 'alias', 'field': 'hasvid', 'deprecated': True},
6280 'audio': {'type': 'alias', 'field': 'hasaud', 'deprecated': True},
6281 'has_audio': {'type': 'alias', 'field': 'hasaud', 'deprecated': True},
6282 'extractor': {'type': 'alias', 'field': 'ie_pref', 'deprecated': True},
6283 'extractor_preference': {'type': 'alias', 'field': 'ie_pref', 'deprecated': True},
6284 }
6285
6286 def __init__(self, ydl, field_preference):
6287 self.ydl = ydl
6288 self._order = []
6289 self.evaluate_params(self.ydl.params, field_preference)
6290 if ydl.params.get('verbose'):
6291 self.print_verbose_info(self.ydl.write_debug)
6292
6293 def _get_field_setting(self, field, key):
6294 if field not in self.settings:
6295 if key in ('forced', 'priority'):
6296 return False
6297 self.ydl.deprecated_feature(f'Using arbitrary fields ({field}) for format sorting is '
6298 'deprecated and may be removed in a future version')
6299 self.settings[field] = {}
6300 propObj = self.settings[field]
6301 if key not in propObj:
6302 type = propObj.get('type')
6303 if key == 'field':
6304 default = 'preference' if type == 'extractor' else (field,) if type in ('combined', 'multiple') else field
6305 elif key == 'convert':
6306 default = 'order' if type == 'ordered' else 'float_string' if field else 'ignore'
6307 else:
6308 default = {'type': 'field', 'visible': True, 'order': [], 'not_in_list': (None,)}.get(key, None)
6309 propObj[key] = default
6310 return propObj[key]
6311
6312 def _resolve_field_value(self, field, value, convertNone=False):
6313 if value is None:
6314 if not convertNone:
6315 return None
6316 else:
6317 value = value.lower()
6318 conversion = self._get_field_setting(field, 'convert')
6319 if conversion == 'ignore':
6320 return None
6321 if conversion == 'string':
6322 return value
6323 elif conversion == 'float_none':
6324 return float_or_none(value)
6325 elif conversion == 'bytes':
6326 return parse_bytes(value)
6327 elif conversion == 'order':
6328 order_list = (self._use_free_order and self._get_field_setting(field, 'order_free')) or self._get_field_setting(field, 'order')
6329 use_regex = self._get_field_setting(field, 'regex')
6330 list_length = len(order_list)
6331 empty_pos = order_list.index('') if '' in order_list else list_length + 1
6332 if use_regex and value is not None:
6333 for i, regex in enumerate(order_list):
6334 if regex and re.match(regex, value):
6335 return list_length - i
6336 return list_length - empty_pos # not in list
6337 else: # not regex or value = None
6338 return list_length - (order_list.index(value) if value in order_list else empty_pos)
6339 else:
6340 if value.isnumeric():
6341 return float(value)
6342 else:
6343 self.settings[field]['convert'] = 'string'
6344 return value
6345
6346 def evaluate_params(self, params, sort_extractor):
6347 self._use_free_order = params.get('prefer_free_formats', False)
6348 self._sort_user = params.get('format_sort', [])
6349 self._sort_extractor = sort_extractor
6350
6351 def add_item(field, reverse, closest, limit_text):
6352 field = field.lower()
6353 if field in self._order:
6354 return
6355 self._order.append(field)
6356 limit = self._resolve_field_value(field, limit_text)
6357 data = {
6358 'reverse': reverse,
6359 'closest': False if limit is None else closest,
6360 'limit_text': limit_text,
6361 'limit': limit}
6362 if field in self.settings:
6363 self.settings[field].update(data)
6364 else:
6365 self.settings[field] = data
6366
6367 sort_list = (
6368 tuple(field for field in self.default if self._get_field_setting(field, 'forced'))
6369 + (tuple() if params.get('format_sort_force', False)
6370 else tuple(field for field in self.default if self._get_field_setting(field, 'priority')))
6371 + tuple(self._sort_user) + tuple(sort_extractor) + self.default)
6372
6373 for item in sort_list:
6374 match = re.match(self.regex, item)
6375 if match is None:
6376 raise ExtractorError('Invalid format sort string "%s" given by extractor' % item)
6377 field = match.group('field')
6378 if field is None:
6379 continue
6380 if self._get_field_setting(field, 'type') == 'alias':
6381 alias, field = field, self._get_field_setting(field, 'field')
6382 if self._get_field_setting(alias, 'deprecated'):
6383 self.ydl.deprecated_feature(f'Format sorting alias {alias} is deprecated and may '
6384 f'be removed in a future version. Please use {field} instead')
6385 reverse = match.group('reverse') is not None
6386 closest = match.group('separator') == '~'
6387 limit_text = match.group('limit')
6388
6389 has_limit = limit_text is not None
6390 has_multiple_fields = self._get_field_setting(field, 'type') == 'combined'
6391 has_multiple_limits = has_limit and has_multiple_fields and not self._get_field_setting(field, 'same_limit')
6392
6393 fields = self._get_field_setting(field, 'field') if has_multiple_fields else (field,)
6394 limits = limit_text.split(':') if has_multiple_limits else (limit_text,) if has_limit else tuple()
6395 limit_count = len(limits)
6396 for (i, f) in enumerate(fields):
6397 add_item(f, reverse, closest,
6398 limits[i] if i < limit_count
6399 else limits[0] if has_limit and not has_multiple_limits
6400 else None)
6401
6402 def print_verbose_info(self, write_debug):
6403 if self._sort_user:
6404 write_debug('Sort order given by user: %s' % ', '.join(self._sort_user))
6405 if self._sort_extractor:
6406 write_debug('Sort order given by extractor: %s' % ', '.join(self._sort_extractor))
6407 write_debug('Formats sorted by: %s' % ', '.join(['%s%s%s' % (
6408 '+' if self._get_field_setting(field, 'reverse') else '', field,
6409 '%s%s(%s)' % ('~' if self._get_field_setting(field, 'closest') else ':',
6410 self._get_field_setting(field, 'limit_text'),
6411 self._get_field_setting(field, 'limit'))
6412 if self._get_field_setting(field, 'limit_text') is not None else '')
6413 for field in self._order if self._get_field_setting(field, 'visible')]))
6414
6415 def _calculate_field_preference_from_value(self, format, field, type, value):
6416 reverse = self._get_field_setting(field, 'reverse')
6417 closest = self._get_field_setting(field, 'closest')
6418 limit = self._get_field_setting(field, 'limit')
6419
6420 if type == 'extractor':
6421 maximum = self._get_field_setting(field, 'max')
6422 if value is None or (maximum is not None and value >= maximum):
6423 value = -1
6424 elif type == 'boolean':
6425 in_list = self._get_field_setting(field, 'in_list')
6426 not_in_list = self._get_field_setting(field, 'not_in_list')
6427 value = 0 if ((in_list is None or value in in_list) and (not_in_list is None or value not in not_in_list)) else -1
6428 elif type == 'ordered':
6429 value = self._resolve_field_value(field, value, True)
6430
6431 # try to convert to number
6432 val_num = float_or_none(value, default=self._get_field_setting(field, 'default'))
6433 is_num = self._get_field_setting(field, 'convert') != 'string' and val_num is not None
6434 if is_num:
6435 value = val_num
6436
6437 return ((-10, 0) if value is None
6438 else (1, value, 0) if not is_num # if a field has mixed strings and numbers, strings are sorted higher
6439 else (0, -abs(value - limit), value - limit if reverse else limit - value) if closest
6440 else (0, value, 0) if not reverse and (limit is None or value <= limit)
6441 else (0, -value, 0) if limit is None or (reverse and value == limit) or value > limit
6442 else (-1, value, 0))
6443
6444 def _calculate_field_preference(self, format, field):
6445 type = self._get_field_setting(field, 'type') # extractor, boolean, ordered, field, multiple
6446 get_value = lambda f: format.get(self._get_field_setting(f, 'field'))
6447 if type == 'multiple':
6448 type = 'field' # Only 'field' is allowed in multiple for now
6449 actual_fields = self._get_field_setting(field, 'field')
6450
6451 value = self._get_field_setting(field, 'function')(get_value(f) for f in actual_fields)
6452 else:
6453 value = get_value(field)
6454 return self._calculate_field_preference_from_value(format, field, type, value)
6455
6456 def calculate_preference(self, format):
6457 # Determine missing protocol
6458 if not format.get('protocol'):
6459 format['protocol'] = determine_protocol(format)
6460
6461 # Determine missing ext
6462 if not format.get('ext') and 'url' in format:
6463 format['ext'] = determine_ext(format['url'])
6464 if format.get('vcodec') == 'none':
6465 format['audio_ext'] = format['ext'] if format.get('acodec') != 'none' else 'none'
6466 format['video_ext'] = 'none'
6467 else:
6468 format['video_ext'] = format['ext']
6469 format['audio_ext'] = 'none'
6470 # if format.get('preference') is None and format.get('ext') in ('f4f', 'f4m'): # Not supported?
6471 # format['preference'] = -1000
6472
6473 if format.get('preference') is None and format.get('ext') == 'flv' and re.match('[hx]265|he?vc?', format.get('vcodec') or ''):
6474 # HEVC-over-FLV is out-of-spec by FLV's original spec
6475 # ref. https://trac.ffmpeg.org/ticket/6389
6476 # ref. https://github.com/yt-dlp/yt-dlp/pull/5821
6477 format['preference'] = -100
6478
6479 # Determine missing bitrates
6480 if format.get('tbr') is None:
6481 if format.get('vbr') is not None and format.get('abr') is not None:
6482 format['tbr'] = format.get('vbr', 0) + format.get('abr', 0)
6483 else:
6484 if format.get('vcodec') != 'none' and format.get('vbr') is None:
6485 format['vbr'] = format.get('tbr') - format.get('abr', 0)
6486 if format.get('acodec') != 'none' and format.get('abr') is None:
6487 format['abr'] = format.get('tbr') - format.get('vbr', 0)
6488
6489 return tuple(self._calculate_field_preference(format, field) for field in self._order)
6490
6491
6492 # Deprecated
6493 has_certifi = bool(certifi)
6494 has_websockets = bool(websockets)
6495
6496
6497 def load_plugins(name, suffix, namespace):
6498 from .plugins import load_plugins
6499 ret = load_plugins(name, suffix)
6500 namespace.update(ret)
6501 return ret