]> jfr.im git - yt-dlp.git/blame - yt_dlp/utils.py
[youtube] Fix uploader for collaborative playlists (#3332)
[yt-dlp.git] / yt_dlp / utils.py
CommitLineData
cc52de43 1#!/usr/bin/env python3
dcdb292f 2# coding: utf-8
d77c3dfd 3
ecc0c5ee
PH
4from __future__ import unicode_literals
5
da42679b 6import asyncio
15dfb392 7import atexit
1e399778 8import base64
5bc880b9 9import binascii
912b38b4 10import calendar
676eb3f2 11import codecs
c380cc28 12import collections
62e609ab 13import contextlib
e3946f98 14import ctypes
c496ca96
PH
15import datetime
16import email.utils
0c265486 17import email.header
f45c185f 18import errno
be4a824d 19import functools
d77c3dfd 20import gzip
49fa4d9a
N
21import hashlib
22import hmac
019a94f7 23import importlib.util
03f9daab 24import io
79a2e94e 25import itertools
f4bfd65f 26import json
d77c3dfd 27import locale
02dbf93f 28import math
347de493 29import operator
d77c3dfd 30import os
c496ca96 31import platform
773f291d 32import random
d77c3dfd 33import re
c496ca96 34import socket
79a2e94e 35import ssl
1c088fa8 36import subprocess
d77c3dfd 37import sys
181c8655 38import tempfile
c380cc28 39import time
01951dda 40import traceback
bcf89ce6 41import xml.etree.ElementTree
d77c3dfd 42import zlib
2814f12b 43import mimetypes
d77c3dfd 44
8c25f81b 45from .compat import (
b4a3d461 46 compat_HTMLParseError,
8bb56eee 47 compat_HTMLParser,
201c1459 48 compat_HTTPError,
8f9312c3 49 compat_basestring,
4390d5ec 50 compat_brotli,
8c25f81b 51 compat_chr,
1bab3437 52 compat_cookiejar,
d7cd9a9e 53 compat_ctypes_WINFUNCTYPE,
36e6f62c 54 compat_etree_fromstring,
51098426 55 compat_expanduser,
8c25f81b 56 compat_html_entities,
55b2f099 57 compat_html_entities_html5,
be4a824d 58 compat_http_client,
42db58ec 59 compat_integer_types,
e29663c6 60 compat_numeric_types,
c86b6142 61 compat_kwargs,
efa97bdc 62 compat_os_name,
8c25f81b 63 compat_parse_qs,
06e57990 64 compat_shlex_split,
702ccf2d 65 compat_shlex_quote,
8c25f81b 66 compat_str,
edaa23f8 67 compat_struct_pack,
d3f8e038 68 compat_struct_unpack,
8c25f81b
PH
69 compat_urllib_error,
70 compat_urllib_parse,
15707c7e 71 compat_urllib_parse_urlencode,
8c25f81b 72 compat_urllib_parse_urlparse,
732044af 73 compat_urllib_parse_urlunparse,
74 compat_urllib_parse_quote,
75 compat_urllib_parse_quote_plus,
7581bfc9 76 compat_urllib_parse_unquote_plus,
8c25f81b
PH
77 compat_urllib_request,
78 compat_urlparse,
da42679b 79 compat_websockets,
810c10ba 80 compat_xpath,
8c25f81b 81)
4644ac55 82
71aff188
YCH
83from .socks import (
84 ProxyType,
85 sockssocket,
86)
87
d5820461 88try:
89 import certifi
90 has_certifi = True
91except ImportError:
92 has_certifi = False
93
4644ac55 94
51fb4995
YCH
95def register_socks_protocols():
96 # "Register" SOCKS protocols
d5ae6bb5
YCH
97 # In Python < 2.6.5, urlsplit() suffers from bug https://bugs.python.org/issue7904
98 # URLs with protocols not in urlparse.uses_netloc are not handled correctly
51fb4995
YCH
99 for scheme in ('socks', 'socks4', 'socks4a', 'socks5'):
100 if scheme not in compat_urlparse.uses_netloc:
101 compat_urlparse.uses_netloc.append(scheme)
102
103
468e2e92
FV
104# This is not clearly defined otherwise
105compiled_regex_type = type(re.compile(''))
106
f7a147e3
S
107
108def random_user_agent():
109 _USER_AGENT_TPL = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/%s Safari/537.36'
110 _CHROME_VERSIONS = (
19b4c74d 111 '90.0.4430.212',
112 '90.0.4430.24',
113 '90.0.4430.70',
114 '90.0.4430.72',
115 '90.0.4430.85',
116 '90.0.4430.93',
117 '91.0.4472.101',
118 '91.0.4472.106',
119 '91.0.4472.114',
120 '91.0.4472.124',
121 '91.0.4472.164',
122 '91.0.4472.19',
123 '91.0.4472.77',
124 '92.0.4515.107',
125 '92.0.4515.115',
126 '92.0.4515.131',
127 '92.0.4515.159',
128 '92.0.4515.43',
129 '93.0.4556.0',
130 '93.0.4577.15',
131 '93.0.4577.63',
132 '93.0.4577.82',
133 '94.0.4606.41',
134 '94.0.4606.54',
135 '94.0.4606.61',
136 '94.0.4606.71',
137 '94.0.4606.81',
138 '94.0.4606.85',
139 '95.0.4638.17',
140 '95.0.4638.50',
141 '95.0.4638.54',
142 '95.0.4638.69',
143 '95.0.4638.74',
144 '96.0.4664.18',
145 '96.0.4664.45',
146 '96.0.4664.55',
147 '96.0.4664.93',
148 '97.0.4692.20',
f7a147e3
S
149 )
150 return _USER_AGENT_TPL % random.choice(_CHROME_VERSIONS)
151
152
4390d5ec 153SUPPORTED_ENCODINGS = [
154 'gzip', 'deflate'
155]
156if compat_brotli:
157 SUPPORTED_ENCODINGS.append('br')
158
3e669f36 159std_headers = {
f7a147e3 160 'User-Agent': random_user_agent(),
59ae15a5 161 'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
59ae15a5 162 'Accept-Language': 'en-us,en;q=0.5',
b1156c1e 163 'Sec-Fetch-Mode': 'navigate',
3e669f36 164}
f427df17 165
5f6a1245 166
fb37eb25
S
167USER_AGENTS = {
168 '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',
169}
170
171
bf42a990
S
172NO_DEFAULT = object()
173
7105440c
YCH
174ENGLISH_MONTH_NAMES = [
175 'January', 'February', 'March', 'April', 'May', 'June',
176 'July', 'August', 'September', 'October', 'November', 'December']
177
f6717dec
S
178MONTH_NAMES = {
179 'en': ENGLISH_MONTH_NAMES,
180 'fr': [
3e4185c3
S
181 'janvier', 'février', 'mars', 'avril', 'mai', 'juin',
182 'juillet', 'août', 'septembre', 'octobre', 'novembre', 'décembre'],
f6717dec 183}
a942d6cb 184
a7aaa398
S
185KNOWN_EXTENSIONS = (
186 'mp4', 'm4a', 'm4p', 'm4b', 'm4r', 'm4v', 'aac',
187 'flv', 'f4v', 'f4a', 'f4b',
188 'webm', 'ogg', 'ogv', 'oga', 'ogx', 'spx', 'opus',
189 'mkv', 'mka', 'mk3d',
190 'avi', 'divx',
191 'mov',
192 'asf', 'wmv', 'wma',
193 '3gp', '3g2',
194 'mp3',
195 'flac',
196 'ape',
197 'wav',
198 'f4f', 'f4m', 'm3u8', 'smil')
199
c587cbb7 200# needed for sanitizing filenames in restricted mode
c8827027 201ACCENT_CHARS = dict(zip('ÂÃÄÀÁÅÆÇÈÉÊËÌÍÎÏÐÑÒÓÔÕÖŐØŒÙÚÛÜŰÝÞßàáâãäåæçèéêëìíîïðñòóôõöőøœùúûüűýþÿ',
fd35d8cd
JW
202 itertools.chain('AAAAAA', ['AE'], 'CEEEEIIIIDNOOOOOOO', ['OE'], 'UUUUUY', ['TH', 'ss'],
203 'aaaaaa', ['ae'], 'ceeeeiiiionooooooo', ['oe'], 'uuuuuy', ['th'], 'y')))
c587cbb7 204
46f59e89
S
205DATE_FORMATS = (
206 '%d %B %Y',
207 '%d %b %Y',
208 '%B %d %Y',
cb655f34
S
209 '%B %dst %Y',
210 '%B %dnd %Y',
9d30c213 211 '%B %drd %Y',
cb655f34 212 '%B %dth %Y',
46f59e89 213 '%b %d %Y',
cb655f34
S
214 '%b %dst %Y',
215 '%b %dnd %Y',
9d30c213 216 '%b %drd %Y',
cb655f34 217 '%b %dth %Y',
46f59e89
S
218 '%b %dst %Y %I:%M',
219 '%b %dnd %Y %I:%M',
9d30c213 220 '%b %drd %Y %I:%M',
46f59e89
S
221 '%b %dth %Y %I:%M',
222 '%Y %m %d',
223 '%Y-%m-%d',
bccdbd22 224 '%Y.%m.%d.',
46f59e89 225 '%Y/%m/%d',
81c13222 226 '%Y/%m/%d %H:%M',
46f59e89 227 '%Y/%m/%d %H:%M:%S',
1931a55e
THD
228 '%Y%m%d%H%M',
229 '%Y%m%d%H%M%S',
4f3fa23e 230 '%Y%m%d',
0c1c6f4b 231 '%Y-%m-%d %H:%M',
46f59e89
S
232 '%Y-%m-%d %H:%M:%S',
233 '%Y-%m-%d %H:%M:%S.%f',
5014558a 234 '%Y-%m-%d %H:%M:%S:%f',
46f59e89
S
235 '%d.%m.%Y %H:%M',
236 '%d.%m.%Y %H.%M',
237 '%Y-%m-%dT%H:%M:%SZ',
238 '%Y-%m-%dT%H:%M:%S.%fZ',
239 '%Y-%m-%dT%H:%M:%S.%f0Z',
240 '%Y-%m-%dT%H:%M:%S',
241 '%Y-%m-%dT%H:%M:%S.%f',
242 '%Y-%m-%dT%H:%M',
c6eed6b8
S
243 '%b %d %Y at %H:%M',
244 '%b %d %Y at %H:%M:%S',
b555ae9b
S
245 '%B %d %Y at %H:%M',
246 '%B %d %Y at %H:%M:%S',
a63d9bd0 247 '%H:%M %d-%b-%Y',
46f59e89
S
248)
249
250DATE_FORMATS_DAY_FIRST = list(DATE_FORMATS)
251DATE_FORMATS_DAY_FIRST.extend([
252 '%d-%m-%Y',
253 '%d.%m.%Y',
254 '%d.%m.%y',
255 '%d/%m/%Y',
256 '%d/%m/%y',
257 '%d/%m/%Y %H:%M:%S',
258])
259
260DATE_FORMATS_MONTH_FIRST = list(DATE_FORMATS)
261DATE_FORMATS_MONTH_FIRST.extend([
262 '%m-%d-%Y',
263 '%m.%d.%Y',
264 '%m/%d/%Y',
265 '%m/%d/%y',
266 '%m/%d/%Y %H:%M:%S',
267])
268
06b3fe29 269PACKED_CODES_RE = r"}\('(.+)',(\d+),(\d+),'([^']+)'\.split\('\|'\)"
22f5f5c6 270JSON_LD_RE = r'(?is)<script[^>]+type=(["\']?)application/ld\+json\1[^>]*>(?P<json_ld>.+?)</script>'
06b3fe29 271
7105440c 272
d77c3dfd 273def preferredencoding():
59ae15a5 274 """Get preferred encoding.
d77c3dfd 275
59ae15a5
PH
276 Returns the best encoding scheme for the system, based on
277 locale.getpreferredencoding() and some further tweaks.
278 """
279 try:
280 pref = locale.getpreferredencoding()
28e614de 281 'TEST'.encode(pref)
70a1165b 282 except Exception:
59ae15a5 283 pref = 'UTF-8'
bae611f2 284
59ae15a5 285 return pref
d77c3dfd 286
f4bfd65f 287
181c8655 288def write_json_file(obj, fn):
1394646a 289 """ Encode obj as JSON and write it to fn, atomically if possible """
181c8655 290
92120217 291 fn = encodeFilename(fn)
61ee5aeb 292 if sys.version_info < (3, 0) and sys.platform != 'win32':
ec5f6016
JMF
293 encoding = get_filesystem_encoding()
294 # os.path.basename returns a bytes object, but NamedTemporaryFile
295 # will fail if the filename contains non ascii characters unless we
296 # use a unicode object
297 path_basename = lambda f: os.path.basename(fn).decode(encoding)
298 # the same for os.path.dirname
299 path_dirname = lambda f: os.path.dirname(fn).decode(encoding)
300 else:
301 path_basename = os.path.basename
302 path_dirname = os.path.dirname
303
73159f99
S
304 args = {
305 'suffix': '.tmp',
ec5f6016
JMF
306 'prefix': path_basename(fn) + '.',
307 'dir': path_dirname(fn),
73159f99
S
308 'delete': False,
309 }
310
181c8655
PH
311 # In Python 2.x, json.dump expects a bytestream.
312 # In Python 3.x, it writes to a character stream
313 if sys.version_info < (3, 0):
73159f99 314 args['mode'] = 'wb'
181c8655 315 else:
73159f99
S
316 args.update({
317 'mode': 'w',
318 'encoding': 'utf-8',
319 })
320
c86b6142 321 tf = tempfile.NamedTemporaryFile(**compat_kwargs(args))
181c8655
PH
322
323 try:
324 with tf:
45d86abe 325 json.dump(obj, tf, ensure_ascii=False)
1394646a
IK
326 if sys.platform == 'win32':
327 # Need to remove existing file on Windows, else os.rename raises
328 # WindowsError or FileExistsError.
329 try:
330 os.unlink(fn)
331 except OSError:
332 pass
9cd5f54e
R
333 try:
334 mask = os.umask(0)
335 os.umask(mask)
336 os.chmod(tf.name, 0o666 & ~mask)
337 except OSError:
338 pass
181c8655 339 os.rename(tf.name, fn)
70a1165b 340 except Exception:
181c8655
PH
341 try:
342 os.remove(tf.name)
343 except OSError:
344 pass
345 raise
346
347
348if sys.version_info >= (2, 7):
ee114368 349 def find_xpath_attr(node, xpath, key, val=None):
59ae56fa 350 """ Find the xpath xpath[@key=val] """
5d2354f1 351 assert re.match(r'^[a-zA-Z_-]+$', key)
ee114368 352 expr = xpath + ('[@%s]' % key if val is None else "[@%s='%s']" % (key, val))
59ae56fa
PH
353 return node.find(expr)
354else:
ee114368 355 def find_xpath_attr(node, xpath, key, val=None):
810c10ba 356 for f in node.findall(compat_xpath(xpath)):
ee114368
S
357 if key not in f.attrib:
358 continue
359 if val is None or f.attrib.get(key) == val:
59ae56fa
PH
360 return f
361 return None
362
d7e66d39
JMF
363# On python2.6 the xml.etree.ElementTree.Element methods don't support
364# the namespace parameter
5f6a1245
JW
365
366
d7e66d39
JMF
367def xpath_with_ns(path, ns_map):
368 components = [c.split(':') for c in path.split('/')]
369 replaced = []
370 for c in components:
371 if len(c) == 1:
372 replaced.append(c[0])
373 else:
374 ns, tag = c
375 replaced.append('{%s}%s' % (ns_map[ns], tag))
376 return '/'.join(replaced)
377
d77c3dfd 378
a41fb80c 379def xpath_element(node, xpath, name=None, fatal=False, default=NO_DEFAULT):
578c0745 380 def _find_xpath(xpath):
810c10ba 381 return node.find(compat_xpath(xpath))
578c0745
S
382
383 if isinstance(xpath, (str, compat_str)):
384 n = _find_xpath(xpath)
385 else:
386 for xp in xpath:
387 n = _find_xpath(xp)
388 if n is not None:
389 break
d74bebd5 390
8e636da4 391 if n is None:
bf42a990
S
392 if default is not NO_DEFAULT:
393 return default
394 elif fatal:
bf0ff932
PH
395 name = xpath if name is None else name
396 raise ExtractorError('Could not find XML element %s' % name)
397 else:
398 return None
a41fb80c
S
399 return n
400
401
402def xpath_text(node, xpath, name=None, fatal=False, default=NO_DEFAULT):
8e636da4
S
403 n = xpath_element(node, xpath, name, fatal=fatal, default=default)
404 if n is None or n == default:
405 return n
406 if n.text is None:
407 if default is not NO_DEFAULT:
408 return default
409 elif fatal:
410 name = xpath if name is None else name
411 raise ExtractorError('Could not find XML element\'s text %s' % name)
412 else:
413 return None
414 return n.text
a41fb80c
S
415
416
417def xpath_attr(node, xpath, key, name=None, fatal=False, default=NO_DEFAULT):
418 n = find_xpath_attr(node, xpath, key)
419 if n is None:
420 if default is not NO_DEFAULT:
421 return default
422 elif fatal:
423 name = '%s[@%s]' % (xpath, key) if name is None else name
424 raise ExtractorError('Could not find XML attribute %s' % name)
425 else:
426 return None
427 return n.attrib[key]
bf0ff932
PH
428
429
9e6dd238 430def get_element_by_id(id, html):
43e8fafd 431 """Return the content of the tag with the specified ID in the passed HTML document"""
611c1dd9 432 return get_element_by_attribute('id', id, html)
43e8fafd 433
12ea2f30 434
6f32a0b5
ZM
435def get_element_html_by_id(id, html):
436 """Return the html of the tag with the specified ID in the passed HTML document"""
437 return get_element_html_by_attribute('id', id, html)
438
439
84c237fb 440def get_element_by_class(class_name, html):
2af12ad9
TC
441 """Return the content of the first tag with the specified class in the passed HTML document"""
442 retval = get_elements_by_class(class_name, html)
443 return retval[0] if retval else None
444
445
6f32a0b5
ZM
446def get_element_html_by_class(class_name, html):
447 """Return the html of the first tag with the specified class in the passed HTML document"""
448 retval = get_elements_html_by_class(class_name, html)
449 return retval[0] if retval else None
450
451
2af12ad9
TC
452def get_element_by_attribute(attribute, value, html, escape_value=True):
453 retval = get_elements_by_attribute(attribute, value, html, escape_value)
454 return retval[0] if retval else None
455
456
6f32a0b5
ZM
457def get_element_html_by_attribute(attribute, value, html, escape_value=True):
458 retval = get_elements_html_by_attribute(attribute, value, html, escape_value)
459 return retval[0] if retval else None
460
461
2af12ad9
TC
462def get_elements_by_class(class_name, html):
463 """Return the content of all tags with the specified class in the passed HTML document as a list"""
464 return get_elements_by_attribute(
84c237fb
YCH
465 'class', r'[^\'"]*\b%s\b[^\'"]*' % re.escape(class_name),
466 html, escape_value=False)
467
468
6f32a0b5
ZM
469def get_elements_html_by_class(class_name, html):
470 """Return the html of all tags with the specified class in the passed HTML document as a list"""
471 return get_elements_html_by_attribute(
472 'class', r'[^\'"]*\b%s\b[^\'"]*' % re.escape(class_name),
473 html, escape_value=False)
474
475
476def get_elements_by_attribute(*args, **kwargs):
43e8fafd 477 """Return the content of the tag with the specified attribute in the passed HTML document"""
6f32a0b5
ZM
478 return [content for content, _ in get_elements_text_and_html_by_attribute(*args, **kwargs)]
479
480
481def get_elements_html_by_attribute(*args, **kwargs):
482 """Return the html of the tag with the specified attribute in the passed HTML document"""
483 return [whole for _, whole in get_elements_text_and_html_by_attribute(*args, **kwargs)]
484
485
486def get_elements_text_and_html_by_attribute(attribute, value, html, escape_value=True):
487 """
488 Return the text (content) and the html (whole) of the tag with the specified
489 attribute in the passed HTML document
490 """
9e6dd238 491
0254f162
ZM
492 value_quote_optional = '' if re.match(r'''[\s"'`=<>]''', value) else '?'
493
84c237fb
YCH
494 value = re.escape(value) if escape_value else value
495
0254f162 496 partial_element_re = r'''(?x)
6f32a0b5 497 <(?P<tag>[a-zA-Z0-9:._-]+)
0254f162
ZM
498 (?:\s(?:[^>"']|"[^"]*"|'[^']*')*)?
499 \s%(attribute)s\s*=\s*(?P<_q>['"]%(vqo)s)(?-x:%(value)s)(?P=_q)
500 ''' % {'attribute': re.escape(attribute), 'value': value, 'vqo': value_quote_optional}
38285056 501
0254f162
ZM
502 for m in re.finditer(partial_element_re, html):
503 content, whole = get_element_text_and_html_by_tag(m.group('tag'), html[m.start():])
a921f407 504
0254f162
ZM
505 yield (
506 unescapeHTML(re.sub(r'^(?P<q>["\'])(?P<content>.*)(?P=q)$', r'\g<content>', content, flags=re.DOTALL)),
507 whole
508 )
a921f407 509
c5229f39 510
6f32a0b5
ZM
511class HTMLBreakOnClosingTagParser(compat_HTMLParser):
512 """
513 HTML parser which raises HTMLBreakOnClosingTagException upon reaching the
514 closing tag for the first opening tag it has encountered, and can be used
515 as a context manager
516 """
517
518 class HTMLBreakOnClosingTagException(Exception):
519 pass
520
521 def __init__(self):
522 self.tagstack = collections.deque()
523 compat_HTMLParser.__init__(self)
524
525 def __enter__(self):
526 return self
527
528 def __exit__(self, *_):
529 self.close()
530
531 def close(self):
532 # handle_endtag does not return upon raising HTMLBreakOnClosingTagException,
533 # so data remains buffered; we no longer have any interest in it, thus
534 # override this method to discard it
535 pass
536
537 def handle_starttag(self, tag, _):
538 self.tagstack.append(tag)
539
540 def handle_endtag(self, tag):
541 if not self.tagstack:
542 raise compat_HTMLParseError('no tags in the stack')
543 while self.tagstack:
544 inner_tag = self.tagstack.pop()
545 if inner_tag == tag:
546 break
547 else:
548 raise compat_HTMLParseError(f'matching opening tag for closing {tag} tag not found')
549 if not self.tagstack:
550 raise self.HTMLBreakOnClosingTagException()
551
552
553def get_element_text_and_html_by_tag(tag, html):
554 """
555 For the first element with the specified tag in the passed HTML document
556 return its' content (text) and the whole element (html)
557 """
558 def find_or_raise(haystack, needle, exc):
559 try:
560 return haystack.index(needle)
561 except ValueError:
562 raise exc
563 closing_tag = f'</{tag}>'
564 whole_start = find_or_raise(
565 html, f'<{tag}', compat_HTMLParseError(f'opening {tag} tag not found'))
566 content_start = find_or_raise(
567 html[whole_start:], '>', compat_HTMLParseError(f'malformed opening {tag} tag'))
568 content_start += whole_start + 1
569 with HTMLBreakOnClosingTagParser() as parser:
570 parser.feed(html[whole_start:content_start])
571 if not parser.tagstack or parser.tagstack[0] != tag:
572 raise compat_HTMLParseError(f'parser did not match opening {tag} tag')
573 offset = content_start
574 while offset < len(html):
575 next_closing_tag_start = find_or_raise(
576 html[offset:], closing_tag,
577 compat_HTMLParseError(f'closing {tag} tag not found'))
578 next_closing_tag_end = next_closing_tag_start + len(closing_tag)
579 try:
580 parser.feed(html[offset:offset + next_closing_tag_end])
581 offset += next_closing_tag_end
582 except HTMLBreakOnClosingTagParser.HTMLBreakOnClosingTagException:
583 return html[content_start:offset + next_closing_tag_start], \
584 html[whole_start:offset + next_closing_tag_end]
585 raise compat_HTMLParseError('unexpected end of html')
586
587
8bb56eee
BF
588class HTMLAttributeParser(compat_HTMLParser):
589 """Trivial HTML parser to gather the attributes for a single element"""
b6e0c7d2 590
8bb56eee 591 def __init__(self):
c5229f39 592 self.attrs = {}
8bb56eee
BF
593 compat_HTMLParser.__init__(self)
594
595 def handle_starttag(self, tag, attrs):
596 self.attrs = dict(attrs)
597
c5229f39 598
73673ccf
FF
599class HTMLListAttrsParser(compat_HTMLParser):
600 """HTML parser to gather the attributes for the elements of a list"""
601
602 def __init__(self):
603 compat_HTMLParser.__init__(self)
604 self.items = []
605 self._level = 0
606
607 def handle_starttag(self, tag, attrs):
608 if tag == 'li' and self._level == 0:
609 self.items.append(dict(attrs))
610 self._level += 1
611
612 def handle_endtag(self, tag):
613 self._level -= 1
614
615
8bb56eee
BF
616def extract_attributes(html_element):
617 """Given a string for an HTML element such as
618 <el
619 a="foo" B="bar" c="&98;az" d=boz
620 empty= noval entity="&amp;"
621 sq='"' dq="'"
622 >
623 Decode and return a dictionary of attributes.
624 {
625 'a': 'foo', 'b': 'bar', c: 'baz', d: 'boz',
626 'empty': '', 'noval': None, 'entity': '&',
627 'sq': '"', 'dq': '\''
628 }.
629 NB HTMLParser is stricter in Python 2.6 & 3.2 than in later versions,
630 but the cases in the unit test will work for all of 2.6, 2.7, 3.2-3.5.
631 """
632 parser = HTMLAttributeParser()
b4a3d461
S
633 try:
634 parser.feed(html_element)
635 parser.close()
636 # Older Python may throw HTMLParseError in case of malformed HTML
637 except compat_HTMLParseError:
638 pass
8bb56eee 639 return parser.attrs
9e6dd238 640
c5229f39 641
73673ccf
FF
642def parse_list(webpage):
643 """Given a string for an series of HTML <li> elements,
644 return a dictionary of their attributes"""
645 parser = HTMLListAttrsParser()
646 parser.feed(webpage)
647 parser.close()
648 return parser.items
649
650
9e6dd238 651def clean_html(html):
59ae15a5 652 """Clean an HTML snippet into a readable string"""
dd622d7c
PH
653
654 if html is None: # Convenience for sanitizing descriptions etc.
655 return html
656
49185227 657 html = re.sub(r'\s+', ' ', html)
658 html = re.sub(r'(?u)\s?<\s?br\s?/?\s?>\s?', '\n', html)
659 html = re.sub(r'(?u)<\s?/\s?p\s?>\s?<\s?p[^>]*>', '\n', html)
59ae15a5
PH
660 # Strip html tags
661 html = re.sub('<.*?>', '', html)
662 # Replace html entities
663 html = unescapeHTML(html)
7decf895 664 return html.strip()
9e6dd238
FV
665
666
d77c3dfd 667def sanitize_open(filename, open_mode):
59ae15a5
PH
668 """Try to open the given filename, and slightly tweak it if this fails.
669
670 Attempts to open the given filename. If this fails, it tries to change
671 the filename slightly, step by step, until it's either able to open it
672 or it fails and raises a final exception, like the standard open()
673 function.
674
675 It returns the tuple (stream, definitive_file_name).
676 """
0edb3e33 677 if filename == '-':
678 if sys.platform == 'win32':
679 import msvcrt
680 msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
681 return (sys.stdout.buffer if hasattr(sys.stdout, 'buffer') else sys.stdout, filename)
59ae15a5 682
0edb3e33 683 for attempt in range(2):
684 try:
685 try:
89737671 686 if sys.platform == 'win32':
b506289f 687 # FIXME: An exclusive lock also locks the file from being read.
688 # Since windows locks are mandatory, don't lock the file on windows (for now).
689 # Ref: https://github.com/yt-dlp/yt-dlp/issues/3124
89737671 690 raise LockingUnsupportedError()
0edb3e33 691 stream = locked_file(filename, open_mode, block=False).__enter__()
692 except LockingUnsupportedError:
693 stream = open(filename, open_mode)
694 return (stream, filename)
695 except (IOError, OSError) as err:
696 if attempt or err.errno in (errno.EACCES,):
697 raise
698 old_filename, filename = filename, sanitize_path(filename)
699 if old_filename == filename:
700 raise
d77c3dfd
FV
701
702
703def timeconvert(timestr):
59ae15a5
PH
704 """Convert RFC 2822 defined time string into system timestamp"""
705 timestamp = None
706 timetuple = email.utils.parsedate_tz(timestr)
707 if timetuple is not None:
708 timestamp = email.utils.mktime_tz(timetuple)
709 return timestamp
1c469a94 710
5f6a1245 711
5c3895ff 712def sanitize_filename(s, restricted=False, is_id=NO_DEFAULT):
59ae15a5 713 """Sanitizes a string so it could be used as part of a filename.
5c3895ff 714 @param restricted Use a stricter subset of allowed characters
715 @param is_id Whether this is an ID that should be kept unchanged if possible.
716 If unset, yt-dlp's new sanitization rules are in effect
59ae15a5 717 """
5c3895ff 718 if s == '':
719 return ''
720
59ae15a5 721 def replace_insane(char):
c587cbb7
AT
722 if restricted and char in ACCENT_CHARS:
723 return ACCENT_CHARS[char]
91dd88b9 724 elif not restricted and char == '\n':
5c3895ff 725 return '\0 '
91dd88b9 726 elif char == '?' or ord(char) < 32 or ord(char) == 127:
59ae15a5
PH
727 return ''
728 elif char == '"':
729 return '' if restricted else '\''
730 elif char == ':':
5c3895ff 731 return '\0_\0-' if restricted else '\0 \0-'
59ae15a5 732 elif char in '\\/|*<>':
5c3895ff 733 return '\0_'
734 if restricted and (char in '!&\'()[]{}$;`^,#' or char.isspace() or ord(char) > 127):
735 return '\0_'
59ae15a5
PH
736 return char
737
5c3895ff 738 s = re.sub(r'[0-9]+(?::[0-9]+)+', lambda m: m.group(0).replace(':', '_'), s) # Handle timestamps
28e614de 739 result = ''.join(map(replace_insane, s))
5c3895ff 740 if is_id is NO_DEFAULT:
741 result = re.sub('(\0.)(?:(?=\\1)..)+', r'\1', result) # Remove repeated substitute chars
742 STRIP_RE = '(?:\0.|[ _-])*'
743 result = re.sub(f'^\0.{STRIP_RE}|{STRIP_RE}\0.$', '', result) # Remove substitute chars from start/end
744 result = result.replace('\0', '') or '_'
745
796173d0
PH
746 if not is_id:
747 while '__' in result:
748 result = result.replace('__', '_')
749 result = result.strip('_')
750 # Common case of "Foreign band name - English song title"
751 if restricted and result.startswith('-_'):
752 result = result[2:]
5a42414b
PH
753 if result.startswith('-'):
754 result = '_' + result[len('-'):]
a7440261 755 result = result.lstrip('.')
796173d0
PH
756 if not result:
757 result = '_'
59ae15a5 758 return result
d77c3dfd 759
5f6a1245 760
c2934512 761def sanitize_path(s, force=False):
a2aaf4db 762 """Sanitizes and normalizes path on Windows"""
c2934512 763 if sys.platform == 'win32':
c4218ac3 764 force = False
c2934512 765 drive_or_unc, _ = os.path.splitdrive(s)
766 if sys.version_info < (2, 7) and not drive_or_unc:
767 drive_or_unc, _ = os.path.splitunc(s)
768 elif force:
769 drive_or_unc = ''
770 else:
a2aaf4db 771 return s
c2934512 772
be531ef1
S
773 norm_path = os.path.normpath(remove_start(s, drive_or_unc)).split(os.path.sep)
774 if drive_or_unc:
a2aaf4db
S
775 norm_path.pop(0)
776 sanitized_path = [
ec85ded8 777 path_part if path_part in ['.', '..'] else re.sub(r'(?:[/<>:"\|\\?\*]|[\s.]$)', '#', path_part)
a2aaf4db 778 for path_part in norm_path]
be531ef1
S
779 if drive_or_unc:
780 sanitized_path.insert(0, drive_or_unc + os.path.sep)
c4218ac3 781 elif force and s[0] == os.path.sep:
782 sanitized_path.insert(0, os.path.sep)
a2aaf4db
S
783 return os.path.join(*sanitized_path)
784
785
17bcc626 786def sanitize_url(url):
befa4708
S
787 # Prepend protocol-less URLs with `http:` scheme in order to mitigate
788 # the number of unwanted failures due to missing protocol
789 if url.startswith('//'):
790 return 'http:%s' % url
791 # Fix some common typos seen so far
792 COMMON_TYPOS = (
067aa17e 793 # https://github.com/ytdl-org/youtube-dl/issues/15649
befa4708
S
794 (r'^httpss://', r'https://'),
795 # https://bx1.be/lives/direct-tv/
796 (r'^rmtp([es]?)://', r'rtmp\1://'),
797 )
798 for mistake, fixup in COMMON_TYPOS:
799 if re.match(mistake, url):
800 return re.sub(mistake, fixup, url)
bc6b9bcd 801 return url
17bcc626
S
802
803
5435dcf9
HH
804def extract_basic_auth(url):
805 parts = compat_urlparse.urlsplit(url)
806 if parts.username is None:
807 return url, None
808 url = compat_urlparse.urlunsplit(parts._replace(netloc=(
809 parts.hostname if parts.port is None
810 else '%s:%d' % (parts.hostname, parts.port))))
811 auth_payload = base64.b64encode(
812 ('%s:%s' % (parts.username, parts.password or '')).encode('utf-8'))
813 return url, 'Basic ' + auth_payload.decode('utf-8')
814
815
67dda517 816def sanitized_Request(url, *args, **kwargs):
bc6b9bcd 817 url, auth_header = extract_basic_auth(escape_url(sanitize_url(url)))
5435dcf9
HH
818 if auth_header is not None:
819 headers = args[1] if len(args) >= 2 else kwargs.setdefault('headers', {})
820 headers['Authorization'] = auth_header
821 return compat_urllib_request.Request(url, *args, **kwargs)
67dda517
S
822
823
51098426
S
824def expand_path(s):
825 """Expand shell variables and ~"""
826 return os.path.expandvars(compat_expanduser(s))
827
828
d77c3dfd 829def orderedSet(iterable):
59ae15a5
PH
830 """ Remove all duplicates from the input iterable """
831 res = []
832 for el in iterable:
833 if el not in res:
834 res.append(el)
835 return res
d77c3dfd 836
912b38b4 837
55b2f099 838def _htmlentity_transform(entity_with_semicolon):
4e408e47 839 """Transforms an HTML entity to a character."""
55b2f099
YCH
840 entity = entity_with_semicolon[:-1]
841
4e408e47
PH
842 # Known non-numeric HTML entity
843 if entity in compat_html_entities.name2codepoint:
844 return compat_chr(compat_html_entities.name2codepoint[entity])
845
55b2f099
YCH
846 # TODO: HTML5 allows entities without a semicolon. For example,
847 # '&Eacuteric' should be decoded as 'Éric'.
848 if entity_with_semicolon in compat_html_entities_html5:
849 return compat_html_entities_html5[entity_with_semicolon]
850
91757b0f 851 mobj = re.match(r'#(x[0-9a-fA-F]+|[0-9]+)', entity)
4e408e47
PH
852 if mobj is not None:
853 numstr = mobj.group(1)
28e614de 854 if numstr.startswith('x'):
4e408e47 855 base = 16
28e614de 856 numstr = '0%s' % numstr
4e408e47
PH
857 else:
858 base = 10
067aa17e 859 # See https://github.com/ytdl-org/youtube-dl/issues/7518
7aefc49c
S
860 try:
861 return compat_chr(int(numstr, base))
862 except ValueError:
863 pass
4e408e47
PH
864
865 # Unknown entity in name, return its literal representation
7a3f0c00 866 return '&%s;' % entity
4e408e47
PH
867
868
d77c3dfd 869def unescapeHTML(s):
912b38b4
PH
870 if s is None:
871 return None
872 assert type(s) == compat_str
d77c3dfd 873
4e408e47 874 return re.sub(
95f3f7c2 875 r'&([^&;]+;)', lambda m: _htmlentity_transform(m.group(1)), s)
d77c3dfd 876
8bf48f23 877
cdb19aa4 878def escapeHTML(text):
879 return (
880 text
881 .replace('&', '&amp;')
882 .replace('<', '&lt;')
883 .replace('>', '&gt;')
884 .replace('"', '&quot;')
885 .replace("'", '&#39;')
886 )
887
888
f5b1bca9 889def process_communicate_or_kill(p, *args, **kwargs):
890 try:
891 return p.communicate(*args, **kwargs)
892 except BaseException: # Including KeyboardInterrupt
893 p.kill()
894 p.wait()
895 raise
896
897
d3c93ec2 898class Popen(subprocess.Popen):
899 if sys.platform == 'win32':
900 _startupinfo = subprocess.STARTUPINFO()
901 _startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
902 else:
903 _startupinfo = None
904
905 def __init__(self, *args, **kwargs):
906 super(Popen, self).__init__(*args, **kwargs, startupinfo=self._startupinfo)
907
908 def communicate_or_kill(self, *args, **kwargs):
909 return process_communicate_or_kill(self, *args, **kwargs)
910
911
aa49acd1
S
912def get_subprocess_encoding():
913 if sys.platform == 'win32' and sys.getwindowsversion()[0] >= 5:
914 # For subprocess calls, encode with locale encoding
915 # Refer to http://stackoverflow.com/a/9951851/35070
916 encoding = preferredencoding()
917 else:
918 encoding = sys.getfilesystemencoding()
919 if encoding is None:
920 encoding = 'utf-8'
921 return encoding
922
923
8bf48f23 924def encodeFilename(s, for_subprocess=False):
59ae15a5
PH
925 """
926 @param s The name of the file
927 """
d77c3dfd 928
8bf48f23 929 assert type(s) == compat_str
d77c3dfd 930
59ae15a5
PH
931 # Python 3 has a Unicode API
932 if sys.version_info >= (3, 0):
933 return s
0f00efed 934
aa49acd1
S
935 # Pass '' directly to use Unicode APIs on Windows 2000 and up
936 # (Detecting Windows NT 4 is tricky because 'major >= 4' would
937 # match Windows 9x series as well. Besides, NT 4 is obsolete.)
938 if not for_subprocess and sys.platform == 'win32' and sys.getwindowsversion()[0] >= 5:
939 return s
940
8ee239e9
YCH
941 # Jython assumes filenames are Unicode strings though reported as Python 2.x compatible
942 if sys.platform.startswith('java'):
943 return s
944
aa49acd1
S
945 return s.encode(get_subprocess_encoding(), 'ignore')
946
947
948def decodeFilename(b, for_subprocess=False):
949
950 if sys.version_info >= (3, 0):
951 return b
952
953 if not isinstance(b, bytes):
954 return b
955
956 return b.decode(get_subprocess_encoding(), 'ignore')
8bf48f23 957
f07b74fc
PH
958
959def encodeArgument(s):
960 if not isinstance(s, compat_str):
961 # Legacy code that uses byte strings
962 # Uncomment the following line after fixing all post processors
7af808a5 963 # assert False, 'Internal error: %r should be of type %r, is %r' % (s, compat_str, type(s))
f07b74fc
PH
964 s = s.decode('ascii')
965 return encodeFilename(s, True)
966
967
aa49acd1
S
968def decodeArgument(b):
969 return decodeFilename(b, True)
970
971
8271226a
PH
972def decodeOption(optval):
973 if optval is None:
974 return optval
975 if isinstance(optval, bytes):
976 optval = optval.decode(preferredencoding())
977
978 assert isinstance(optval, compat_str)
979 return optval
1c256f70 980
5f6a1245 981
aa7785f8 982_timetuple = collections.namedtuple('Time', ('hours', 'minutes', 'seconds', 'milliseconds'))
983
984
985def timetuple_from_msec(msec):
986 secs, msec = divmod(msec, 1000)
987 mins, secs = divmod(secs, 60)
988 hrs, mins = divmod(mins, 60)
989 return _timetuple(hrs, mins, secs, msec)
990
991
cdb19aa4 992def formatSeconds(secs, delim=':', msec=False):
aa7785f8 993 time = timetuple_from_msec(secs * 1000)
994 if time.hours:
995 ret = '%d%s%02d%s%02d' % (time.hours, delim, time.minutes, delim, time.seconds)
996 elif time.minutes:
997 ret = '%d%s%02d' % (time.minutes, delim, time.seconds)
4539dd30 998 else:
aa7785f8 999 ret = '%d' % time.seconds
1000 return '%s.%03d' % (ret, time.milliseconds) if msec else ret
4539dd30 1001
a0ddb8a2 1002
77562778 1003def _ssl_load_windows_store_certs(ssl_context, storename):
1004 # Code adapted from _load_windows_store_certs in https://github.com/python/cpython/blob/main/Lib/ssl.py
1005 try:
1006 certs = [cert for cert, encoding, trust in ssl.enum_certificates(storename)
1007 if encoding == 'x509_asn' and (
1008 trust is True or ssl.Purpose.SERVER_AUTH.oid in trust)]
1009 except PermissionError:
1010 return
1011 for cert in certs:
a2366922 1012 try:
77562778 1013 ssl_context.load_verify_locations(cadata=cert)
1014 except ssl.SSLError:
a2366922
PH
1015 pass
1016
77562778 1017
1018def make_HTTPS_handler(params, **kwargs):
1019 opts_check_certificate = not params.get('nocheckcertificate')
1020 context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
1021 context.check_hostname = opts_check_certificate
f81c62a6 1022 if params.get('legacyserverconnect'):
1023 context.options |= 4 # SSL_OP_LEGACY_SERVER_CONNECT
77562778 1024 context.verify_mode = ssl.CERT_REQUIRED if opts_check_certificate else ssl.CERT_NONE
1025 if opts_check_certificate:
d5820461 1026 if has_certifi and 'no-certifi' not in params.get('compat_opts', []):
1027 context.load_verify_locations(cafile=certifi.where())
1028 else:
1029 try:
1030 context.load_default_certs()
1031 # Work around the issue in load_default_certs when there are bad certificates. See:
1032 # https://github.com/yt-dlp/yt-dlp/issues/1060,
1033 # https://bugs.python.org/issue35665, https://bugs.python.org/issue45312
1034 except ssl.SSLError:
1035 # enum_certificates is not present in mingw python. See https://github.com/yt-dlp/yt-dlp/issues/1151
1036 if sys.platform == 'win32' and hasattr(ssl, 'enum_certificates'):
1037 # Create a new context to discard any certificates that were already loaded
1038 context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
1039 context.check_hostname, context.verify_mode = True, ssl.CERT_REQUIRED
1040 for storename in ('CA', 'ROOT'):
1041 _ssl_load_windows_store_certs(context, storename)
1042 context.set_default_verify_paths()
77562778 1043 return YoutubeDLHTTPSHandler(params, context=context, **kwargs)
ea6d901e 1044
732ea2f0 1045
5873d4cc 1046def bug_reports_message(before=';'):
a44ca5a4 1047 msg = ('please report this issue on https://github.com/yt-dlp/yt-dlp/issues?q= , '
592b7485 1048 'filling out the appropriate issue template. '
08d30158 1049 'Confirm you are on the latest version using yt-dlp -U')
5873d4cc
F
1050
1051 before = before.rstrip()
1052 if not before or before.endswith(('.', '!', '?')):
1053 msg = msg[0].title() + msg[1:]
1054
1055 return (before + ' ' if before else '') + msg
08f2a92c
JMF
1056
1057
bf5b9d85
PM
1058class YoutubeDLError(Exception):
1059 """Base exception for YoutubeDL errors."""
aa9369a2 1060 msg = None
1061
1062 def __init__(self, msg=None):
1063 if msg is not None:
1064 self.msg = msg
1065 elif self.msg is None:
1066 self.msg = type(self).__name__
1067 super().__init__(self.msg)
bf5b9d85
PM
1068
1069
3158150c 1070network_exceptions = [compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error]
1071if hasattr(ssl, 'CertificateError'):
1072 network_exceptions.append(ssl.CertificateError)
1073network_exceptions = tuple(network_exceptions)
1074
1075
bf5b9d85 1076class ExtractorError(YoutubeDLError):
1c256f70 1077 """Error during info extraction."""
5f6a1245 1078
1151c407 1079 def __init__(self, msg, tb=None, expected=False, cause=None, video_id=None, ie=None):
9a82b238 1080 """ tb, if given, is the original traceback (so that it can be printed out).
7a5c1cfe 1081 If expected is set, this is a normal error message and most likely not a bug in yt-dlp.
9a82b238 1082 """
3158150c 1083 if sys.exc_info()[0] in network_exceptions:
9a82b238 1084 expected = True
d5979c5d 1085
7265a219 1086 self.orig_msg = str(msg)
1c256f70 1087 self.traceback = tb
1151c407 1088 self.expected = expected
2eabb802 1089 self.cause = cause
d11271dd 1090 self.video_id = video_id
1151c407 1091 self.ie = ie
1092 self.exc_info = sys.exc_info() # preserve original exception
1093
1094 super(ExtractorError, self).__init__(''.join((
1095 format_field(ie, template='[%s] '),
1096 format_field(video_id, template='%s: '),
7265a219 1097 msg,
1151c407 1098 format_field(cause, template=' (caused by %r)'),
1099 '' if expected else bug_reports_message())))
1c256f70 1100
01951dda 1101 def format_traceback(self):
497d2fab 1102 return join_nonempty(
1103 self.traceback and ''.join(traceback.format_tb(self.traceback)),
e491d06d 1104 self.cause and ''.join(traceback.format_exception(None, self.cause, self.cause.__traceback__)[1:]),
497d2fab 1105 delim='\n') or None
01951dda 1106
1c256f70 1107
416c7fcb
PH
1108class UnsupportedError(ExtractorError):
1109 def __init__(self, url):
1110 super(UnsupportedError, self).__init__(
1111 'Unsupported URL: %s' % url, expected=True)
1112 self.url = url
1113
1114
55b3e45b
JMF
1115class RegexNotFoundError(ExtractorError):
1116 """Error when a regex didn't match"""
1117 pass
1118
1119
773f291d
S
1120class GeoRestrictedError(ExtractorError):
1121 """Geographic restriction Error exception.
1122
1123 This exception may be thrown when a video is not available from your
1124 geographic location due to geographic restrictions imposed by a website.
1125 """
b6e0c7d2 1126
0db3bae8 1127 def __init__(self, msg, countries=None, **kwargs):
1128 kwargs['expected'] = True
1129 super(GeoRestrictedError, self).__init__(msg, **kwargs)
773f291d
S
1130 self.countries = countries
1131
1132
bf5b9d85 1133class DownloadError(YoutubeDLError):
59ae15a5 1134 """Download Error exception.
d77c3dfd 1135
59ae15a5
PH
1136 This exception may be thrown by FileDownloader objects if they are not
1137 configured to continue on errors. They will contain the appropriate
1138 error message.
1139 """
5f6a1245 1140
8cc83b8d
FV
1141 def __init__(self, msg, exc_info=None):
1142 """ exc_info, if given, is the original exception that caused the trouble (as returned by sys.exc_info()). """
1143 super(DownloadError, self).__init__(msg)
1144 self.exc_info = exc_info
d77c3dfd
FV
1145
1146
498f5606 1147class EntryNotInPlaylist(YoutubeDLError):
1148 """Entry not in playlist exception.
1149
1150 This exception will be thrown by YoutubeDL when a requested entry
1151 is not found in the playlist info_dict
1152 """
aa9369a2 1153 msg = 'Entry not found in info'
498f5606 1154
1155
bf5b9d85 1156class SameFileError(YoutubeDLError):
59ae15a5 1157 """Same File exception.
d77c3dfd 1158
59ae15a5
PH
1159 This exception will be thrown by FileDownloader objects if they detect
1160 multiple files would have to be downloaded to the same file on disk.
1161 """
aa9369a2 1162 msg = 'Fixed output name but more than one file to download'
1163
1164 def __init__(self, filename=None):
1165 if filename is not None:
1166 self.msg += f': {filename}'
1167 super().__init__(self.msg)
d77c3dfd
FV
1168
1169
bf5b9d85 1170class PostProcessingError(YoutubeDLError):
59ae15a5 1171 """Post Processing exception.
d77c3dfd 1172
59ae15a5
PH
1173 This exception may be raised by PostProcessor's .run() method to
1174 indicate an error in the postprocessing task.
1175 """
5f6a1245 1176
5f6a1245 1177
48f79687 1178class DownloadCancelled(YoutubeDLError):
1179 """ Exception raised when the download queue should be interrupted """
1180 msg = 'The download was cancelled'
8b0d7497 1181
8b0d7497 1182
48f79687 1183class ExistingVideoReached(DownloadCancelled):
1184 """ --break-on-existing triggered """
1185 msg = 'Encountered a video that is already in the archive, stopping due to --break-on-existing'
8b0d7497 1186
48f79687 1187
1188class RejectedVideoReached(DownloadCancelled):
1189 """ --break-on-reject triggered """
1190 msg = 'Encountered a video that did not match filter, stopping due to --break-on-reject'
51d9739f 1191
1192
48f79687 1193class MaxDownloadsReached(DownloadCancelled):
59ae15a5 1194 """ --max-downloads limit has been reached. """
48f79687 1195 msg = 'Maximum number of downloads reached, stopping due to --max-downloads'
1196
1197
f2ebc5c7 1198class ReExtractInfo(YoutubeDLError):
1199 """ Video info needs to be re-extracted. """
1200
1201 def __init__(self, msg, expected=False):
1202 super().__init__(msg)
1203 self.expected = expected
1204
1205
1206class ThrottledDownload(ReExtractInfo):
48f79687 1207 """ Download speed below --throttled-rate. """
aa9369a2 1208 msg = 'The download speed is below throttle limit'
d77c3dfd 1209
43b22906 1210 def __init__(self):
1211 super().__init__(self.msg, expected=False)
f2ebc5c7 1212
d77c3dfd 1213
bf5b9d85 1214class UnavailableVideoError(YoutubeDLError):
59ae15a5 1215 """Unavailable Format exception.
d77c3dfd 1216
59ae15a5
PH
1217 This exception will be thrown when a video is requested
1218 in a format that is not available for that video.
1219 """
aa9369a2 1220 msg = 'Unable to download video'
1221
1222 def __init__(self, err=None):
1223 if err is not None:
1224 self.msg += f': {err}'
1225 super().__init__(self.msg)
d77c3dfd
FV
1226
1227
bf5b9d85 1228class ContentTooShortError(YoutubeDLError):
59ae15a5 1229 """Content Too Short exception.
d77c3dfd 1230
59ae15a5
PH
1231 This exception may be raised by FileDownloader objects when a file they
1232 download is too small for what the server announced first, indicating
1233 the connection was probably interrupted.
1234 """
d77c3dfd 1235
59ae15a5 1236 def __init__(self, downloaded, expected):
bf5b9d85
PM
1237 super(ContentTooShortError, self).__init__(
1238 'Downloaded {0} bytes, expected {1} bytes'.format(downloaded, expected)
1239 )
2c7ed247 1240 # Both in bytes
59ae15a5
PH
1241 self.downloaded = downloaded
1242 self.expected = expected
d77c3dfd 1243
5f6a1245 1244
bf5b9d85 1245class XAttrMetadataError(YoutubeDLError):
efa97bdc
YCH
1246 def __init__(self, code=None, msg='Unknown error'):
1247 super(XAttrMetadataError, self).__init__(msg)
1248 self.code = code
bd264412 1249 self.msg = msg
efa97bdc
YCH
1250
1251 # Parsing code and msg
3089bc74 1252 if (self.code in (errno.ENOSPC, errno.EDQUOT)
a0566bbf 1253 or 'No space left' in self.msg or 'Disk quota exceeded' in self.msg):
efa97bdc
YCH
1254 self.reason = 'NO_SPACE'
1255 elif self.code == errno.E2BIG or 'Argument list too long' in self.msg:
1256 self.reason = 'VALUE_TOO_LONG'
1257 else:
1258 self.reason = 'NOT_SUPPORTED'
1259
1260
bf5b9d85 1261class XAttrUnavailableError(YoutubeDLError):
efa97bdc
YCH
1262 pass
1263
1264
c5a59d93 1265def _create_http_connection(ydl_handler, http_class, is_https, *args, **kwargs):
e5e78797
S
1266 # Working around python 2 bug (see http://bugs.python.org/issue17849) by limiting
1267 # expected HTTP responses to meet HTTP/1.0 or later (see also
067aa17e 1268 # https://github.com/ytdl-org/youtube-dl/issues/6727)
e5e78797 1269 if sys.version_info < (3, 0):
65220c3b
S
1270 kwargs['strict'] = True
1271 hc = http_class(*args, **compat_kwargs(kwargs))
be4a824d 1272 source_address = ydl_handler._params.get('source_address')
8959018a 1273
be4a824d 1274 if source_address is not None:
8959018a
AU
1275 # This is to workaround _create_connection() from socket where it will try all
1276 # address data from getaddrinfo() including IPv6. This filters the result from
1277 # getaddrinfo() based on the source_address value.
1278 # This is based on the cpython socket.create_connection() function.
1279 # https://github.com/python/cpython/blob/master/Lib/socket.py#L691
1280 def _create_connection(address, timeout=socket._GLOBAL_DEFAULT_TIMEOUT, source_address=None):
1281 host, port = address
1282 err = None
1283 addrs = socket.getaddrinfo(host, port, 0, socket.SOCK_STREAM)
9e21e6d9
S
1284 af = socket.AF_INET if '.' in source_address[0] else socket.AF_INET6
1285 ip_addrs = [addr for addr in addrs if addr[0] == af]
1286 if addrs and not ip_addrs:
1287 ip_version = 'v4' if af == socket.AF_INET else 'v6'
1288 raise socket.error(
1289 "No remote IP%s addresses available for connect, can't use '%s' as source address"
1290 % (ip_version, source_address[0]))
8959018a
AU
1291 for res in ip_addrs:
1292 af, socktype, proto, canonname, sa = res
1293 sock = None
1294 try:
1295 sock = socket.socket(af, socktype, proto)
1296 if timeout is not socket._GLOBAL_DEFAULT_TIMEOUT:
1297 sock.settimeout(timeout)
1298 sock.bind(source_address)
1299 sock.connect(sa)
1300 err = None # Explicitly break reference cycle
1301 return sock
1302 except socket.error as _:
1303 err = _
1304 if sock is not None:
1305 sock.close()
1306 if err is not None:
1307 raise err
1308 else:
9e21e6d9
S
1309 raise socket.error('getaddrinfo returns an empty list')
1310 if hasattr(hc, '_create_connection'):
1311 hc._create_connection = _create_connection
be4a824d
PH
1312 sa = (source_address, 0)
1313 if hasattr(hc, 'source_address'): # Python 2.7+
1314 hc.source_address = sa
1315 else: # Python 2.6
1316 def _hc_connect(self, *args, **kwargs):
9e21e6d9 1317 sock = _create_connection(
be4a824d
PH
1318 (self.host, self.port), self.timeout, sa)
1319 if is_https:
d7932313
PH
1320 self.sock = ssl.wrap_socket(
1321 sock, self.key_file, self.cert_file,
1322 ssl_version=ssl.PROTOCOL_TLSv1)
be4a824d
PH
1323 else:
1324 self.sock = sock
1325 hc.connect = functools.partial(_hc_connect, hc)
1326
1327 return hc
1328
1329
87f0e62d 1330def handle_youtubedl_headers(headers):
992fc9d6
YCH
1331 filtered_headers = headers
1332
1333 if 'Youtubedl-no-compression' in filtered_headers:
1334 filtered_headers = dict((k, v) for k, v in filtered_headers.items() if k.lower() != 'accept-encoding')
87f0e62d 1335 del filtered_headers['Youtubedl-no-compression']
87f0e62d 1336
992fc9d6 1337 return filtered_headers
87f0e62d
YCH
1338
1339
acebc9cd 1340class YoutubeDLHandler(compat_urllib_request.HTTPHandler):
59ae15a5
PH
1341 """Handler for HTTP requests and responses.
1342
1343 This class, when installed with an OpenerDirector, automatically adds
1344 the standard headers to every HTTP request and handles gzipped and
1345 deflated responses from web servers. If compression is to be avoided in
1346 a particular request, the original request in the program code only has
0424ec30 1347 to include the HTTP header "Youtubedl-no-compression", which will be
59ae15a5
PH
1348 removed before making the real request.
1349
1350 Part of this code was copied from:
1351
1352 http://techknack.net/python-urllib2-handlers/
1353
1354 Andrew Rowls, the author of that code, agreed to release it to the
1355 public domain.
1356 """
1357
be4a824d
PH
1358 def __init__(self, params, *args, **kwargs):
1359 compat_urllib_request.HTTPHandler.__init__(self, *args, **kwargs)
1360 self._params = params
1361
1362 def http_open(self, req):
71aff188
YCH
1363 conn_class = compat_http_client.HTTPConnection
1364
1365 socks_proxy = req.headers.get('Ytdl-socks-proxy')
1366 if socks_proxy:
1367 conn_class = make_socks_conn_class(conn_class, socks_proxy)
1368 del req.headers['Ytdl-socks-proxy']
1369
be4a824d 1370 return self.do_open(functools.partial(
71aff188 1371 _create_http_connection, self, conn_class, False),
be4a824d
PH
1372 req)
1373
59ae15a5
PH
1374 @staticmethod
1375 def deflate(data):
fc2119f2 1376 if not data:
1377 return data
59ae15a5
PH
1378 try:
1379 return zlib.decompress(data, -zlib.MAX_WBITS)
1380 except zlib.error:
1381 return zlib.decompress(data)
1382
4390d5ec 1383 @staticmethod
1384 def brotli(data):
1385 if not data:
1386 return data
1387 return compat_brotli.decompress(data)
1388
acebc9cd 1389 def http_request(self, req):
51f267d9
S
1390 # According to RFC 3986, URLs can not contain non-ASCII characters, however this is not
1391 # always respected by websites, some tend to give out URLs with non percent-encoded
1392 # non-ASCII characters (see telemb.py, ard.py [#3412])
1393 # urllib chokes on URLs with non-ASCII characters (see http://bugs.python.org/issue3991)
1394 # To work around aforementioned issue we will replace request's original URL with
1395 # percent-encoded one
1396 # Since redirects are also affected (e.g. http://www.southpark.de/alle-episoden/s18e09)
1397 # the code of this workaround has been moved here from YoutubeDL.urlopen()
1398 url = req.get_full_url()
1399 url_escaped = escape_url(url)
1400
1401 # Substitute URL if any change after escaping
1402 if url != url_escaped:
15d260eb 1403 req = update_Request(req, url=url_escaped)
51f267d9 1404
8b7539d2 1405 for h, v in self._params.get('http_headers', std_headers).items():
3d5f7a39
JK
1406 # Capitalize is needed because of Python bug 2275: http://bugs.python.org/issue2275
1407 # The dict keys are capitalized because of this bug by urllib
1408 if h.capitalize() not in req.headers:
33ac271b 1409 req.add_header(h, v)
87f0e62d 1410
af14914b 1411 if 'Accept-encoding' not in req.headers:
1412 req.add_header('Accept-encoding', ', '.join(SUPPORTED_ENCODINGS))
1413
87f0e62d 1414 req.headers = handle_youtubedl_headers(req.headers)
989b4b2b
PH
1415
1416 if sys.version_info < (2, 7) and '#' in req.get_full_url():
1417 # Python 2.6 is brain-dead when it comes to fragments
1418 req._Request__original = req._Request__original.partition('#')[0]
1419 req._Request__r_type = req._Request__r_type.partition('#')[0]
1420
59ae15a5
PH
1421 return req
1422
acebc9cd 1423 def http_response(self, req, resp):
59ae15a5
PH
1424 old_resp = resp
1425 # gzip
1426 if resp.headers.get('Content-encoding', '') == 'gzip':
aa3e9507
PH
1427 content = resp.read()
1428 gz = gzip.GzipFile(fileobj=io.BytesIO(content), mode='rb')
1429 try:
1430 uncompressed = io.BytesIO(gz.read())
1431 except IOError as original_ioerror:
1432 # There may be junk add the end of the file
1433 # See http://stackoverflow.com/q/4928560/35070 for details
1434 for i in range(1, 1024):
1435 try:
1436 gz = gzip.GzipFile(fileobj=io.BytesIO(content[:-i]), mode='rb')
1437 uncompressed = io.BytesIO(gz.read())
1438 except IOError:
1439 continue
1440 break
1441 else:
1442 raise original_ioerror
b407d853 1443 resp = compat_urllib_request.addinfourl(uncompressed, old_resp.headers, old_resp.url, old_resp.code)
59ae15a5 1444 resp.msg = old_resp.msg
c047270c 1445 del resp.headers['Content-encoding']
59ae15a5
PH
1446 # deflate
1447 if resp.headers.get('Content-encoding', '') == 'deflate':
1448 gz = io.BytesIO(self.deflate(resp.read()))
b407d853 1449 resp = compat_urllib_request.addinfourl(gz, old_resp.headers, old_resp.url, old_resp.code)
59ae15a5 1450 resp.msg = old_resp.msg
c047270c 1451 del resp.headers['Content-encoding']
4390d5ec 1452 # brotli
1453 if resp.headers.get('Content-encoding', '') == 'br':
1454 resp = compat_urllib_request.addinfourl(
1455 io.BytesIO(self.brotli(resp.read())), old_resp.headers, old_resp.url, old_resp.code)
1456 resp.msg = old_resp.msg
1457 del resp.headers['Content-encoding']
ad729172 1458 # Percent-encode redirect URL of Location HTTP header to satisfy RFC 3986 (see
067aa17e 1459 # https://github.com/ytdl-org/youtube-dl/issues/6457).
5a4d9ddb
S
1460 if 300 <= resp.code < 400:
1461 location = resp.headers.get('Location')
1462 if location:
1463 # As of RFC 2616 default charset is iso-8859-1 that is respected by python 3
1464 if sys.version_info >= (3, 0):
1465 location = location.encode('iso-8859-1').decode('utf-8')
0ea59007
YCH
1466 else:
1467 location = location.decode('utf-8')
5a4d9ddb
S
1468 location_escaped = escape_url(location)
1469 if location != location_escaped:
1470 del resp.headers['Location']
9a4aec8b
YCH
1471 if sys.version_info < (3, 0):
1472 location_escaped = location_escaped.encode('utf-8')
5a4d9ddb 1473 resp.headers['Location'] = location_escaped
59ae15a5 1474 return resp
0f8d03f8 1475
acebc9cd
PH
1476 https_request = http_request
1477 https_response = http_response
bf50b038 1478
5de90176 1479
71aff188
YCH
1480def make_socks_conn_class(base_class, socks_proxy):
1481 assert issubclass(base_class, (
1482 compat_http_client.HTTPConnection, compat_http_client.HTTPSConnection))
1483
1484 url_components = compat_urlparse.urlparse(socks_proxy)
1485 if url_components.scheme.lower() == 'socks5':
1486 socks_type = ProxyType.SOCKS5
1487 elif url_components.scheme.lower() in ('socks', 'socks4'):
1488 socks_type = ProxyType.SOCKS4
51fb4995
YCH
1489 elif url_components.scheme.lower() == 'socks4a':
1490 socks_type = ProxyType.SOCKS4A
71aff188 1491
cdd94c2e
YCH
1492 def unquote_if_non_empty(s):
1493 if not s:
1494 return s
1495 return compat_urllib_parse_unquote_plus(s)
1496
71aff188
YCH
1497 proxy_args = (
1498 socks_type,
1499 url_components.hostname, url_components.port or 1080,
1500 True, # Remote DNS
cdd94c2e
YCH
1501 unquote_if_non_empty(url_components.username),
1502 unquote_if_non_empty(url_components.password),
71aff188
YCH
1503 )
1504
1505 class SocksConnection(base_class):
1506 def connect(self):
1507 self.sock = sockssocket()
1508 self.sock.setproxy(*proxy_args)
1509 if type(self.timeout) in (int, float):
1510 self.sock.settimeout(self.timeout)
1511 self.sock.connect((self.host, self.port))
1512
1513 if isinstance(self, compat_http_client.HTTPSConnection):
1514 if hasattr(self, '_context'): # Python > 2.6
1515 self.sock = self._context.wrap_socket(
1516 self.sock, server_hostname=self.host)
1517 else:
1518 self.sock = ssl.wrap_socket(self.sock)
1519
1520 return SocksConnection
1521
1522
be4a824d
PH
1523class YoutubeDLHTTPSHandler(compat_urllib_request.HTTPSHandler):
1524 def __init__(self, params, https_conn_class=None, *args, **kwargs):
1525 compat_urllib_request.HTTPSHandler.__init__(self, *args, **kwargs)
1526 self._https_conn_class = https_conn_class or compat_http_client.HTTPSConnection
1527 self._params = params
1528
1529 def https_open(self, req):
4f264c02 1530 kwargs = {}
71aff188
YCH
1531 conn_class = self._https_conn_class
1532
4f264c02
JMF
1533 if hasattr(self, '_context'): # python > 2.6
1534 kwargs['context'] = self._context
1535 if hasattr(self, '_check_hostname'): # python 3.x
1536 kwargs['check_hostname'] = self._check_hostname
71aff188
YCH
1537
1538 socks_proxy = req.headers.get('Ytdl-socks-proxy')
1539 if socks_proxy:
1540 conn_class = make_socks_conn_class(conn_class, socks_proxy)
1541 del req.headers['Ytdl-socks-proxy']
1542
be4a824d 1543 return self.do_open(functools.partial(
71aff188 1544 _create_http_connection, self, conn_class, True),
4f264c02 1545 req, **kwargs)
be4a824d
PH
1546
1547
1bab3437 1548class YoutubeDLCookieJar(compat_cookiejar.MozillaCookieJar):
f1a8511f
S
1549 """
1550 See [1] for cookie file format.
1551
1552 1. https://curl.haxx.se/docs/http-cookies.html
1553 """
e7e62441 1554 _HTTPONLY_PREFIX = '#HttpOnly_'
c380cc28
S
1555 _ENTRY_LEN = 7
1556 _HEADER = '''# Netscape HTTP Cookie File
7a5c1cfe 1557# This file is generated by yt-dlp. Do not edit.
c380cc28
S
1558
1559'''
1560 _CookieFileEntry = collections.namedtuple(
1561 'CookieFileEntry',
1562 ('domain_name', 'include_subdomains', 'path', 'https_only', 'expires_at', 'name', 'value'))
e7e62441 1563
1bab3437 1564 def save(self, filename=None, ignore_discard=False, ignore_expires=False):
c380cc28
S
1565 """
1566 Save cookies to a file.
1567
1568 Most of the code is taken from CPython 3.8 and slightly adapted
1569 to support cookie files with UTF-8 in both python 2 and 3.
1570 """
1571 if filename is None:
1572 if self.filename is not None:
1573 filename = self.filename
1574 else:
1575 raise ValueError(compat_cookiejar.MISSING_FILENAME_TEXT)
1576
1bab3437
S
1577 # Store session cookies with `expires` set to 0 instead of an empty
1578 # string
1579 for cookie in self:
1580 if cookie.expires is None:
1581 cookie.expires = 0
c380cc28
S
1582
1583 with io.open(filename, 'w', encoding='utf-8') as f:
1584 f.write(self._HEADER)
1585 now = time.time()
1586 for cookie in self:
1587 if not ignore_discard and cookie.discard:
1588 continue
1589 if not ignore_expires and cookie.is_expired(now):
1590 continue
1591 if cookie.secure:
1592 secure = 'TRUE'
1593 else:
1594 secure = 'FALSE'
1595 if cookie.domain.startswith('.'):
1596 initial_dot = 'TRUE'
1597 else:
1598 initial_dot = 'FALSE'
1599 if cookie.expires is not None:
1600 expires = compat_str(cookie.expires)
1601 else:
1602 expires = ''
1603 if cookie.value is None:
1604 # cookies.txt regards 'Set-Cookie: foo' as a cookie
1605 # with no name, whereas http.cookiejar regards it as a
1606 # cookie with no value.
1607 name = ''
1608 value = cookie.name
1609 else:
1610 name = cookie.name
1611 value = cookie.value
1612 f.write(
1613 '\t'.join([cookie.domain, initial_dot, cookie.path,
1614 secure, expires, name, value]) + '\n')
1bab3437
S
1615
1616 def load(self, filename=None, ignore_discard=False, ignore_expires=False):
e7e62441 1617 """Load cookies from a file."""
1618 if filename is None:
1619 if self.filename is not None:
1620 filename = self.filename
1621 else:
1622 raise ValueError(compat_cookiejar.MISSING_FILENAME_TEXT)
1623
c380cc28
S
1624 def prepare_line(line):
1625 if line.startswith(self._HTTPONLY_PREFIX):
1626 line = line[len(self._HTTPONLY_PREFIX):]
1627 # comments and empty lines are fine
1628 if line.startswith('#') or not line.strip():
1629 return line
1630 cookie_list = line.split('\t')
1631 if len(cookie_list) != self._ENTRY_LEN:
1632 raise compat_cookiejar.LoadError('invalid length %d' % len(cookie_list))
1633 cookie = self._CookieFileEntry(*cookie_list)
1634 if cookie.expires_at and not cookie.expires_at.isdigit():
1635 raise compat_cookiejar.LoadError('invalid expires at %s' % cookie.expires_at)
1636 return line
1637
e7e62441 1638 cf = io.StringIO()
c380cc28 1639 with io.open(filename, encoding='utf-8') as f:
e7e62441 1640 for line in f:
c380cc28
S
1641 try:
1642 cf.write(prepare_line(line))
1643 except compat_cookiejar.LoadError as e:
1644 write_string(
1645 'WARNING: skipping cookie file entry due to %s: %r\n'
1646 % (e, line), sys.stderr)
1647 continue
e7e62441 1648 cf.seek(0)
1649 self._really_load(cf, filename, ignore_discard, ignore_expires)
1bab3437
S
1650 # Session cookies are denoted by either `expires` field set to
1651 # an empty string or 0. MozillaCookieJar only recognizes the former
1652 # (see [1]). So we need force the latter to be recognized as session
1653 # cookies on our own.
1654 # Session cookies may be important for cookies-based authentication,
1655 # e.g. usually, when user does not check 'Remember me' check box while
1656 # logging in on a site, some important cookies are stored as session
1657 # cookies so that not recognizing them will result in failed login.
1658 # 1. https://bugs.python.org/issue17164
1659 for cookie in self:
1660 # Treat `expires=0` cookies as session cookies
1661 if cookie.expires == 0:
1662 cookie.expires = None
1663 cookie.discard = True
1664
1665
a6420bf5
S
1666class YoutubeDLCookieProcessor(compat_urllib_request.HTTPCookieProcessor):
1667 def __init__(self, cookiejar=None):
1668 compat_urllib_request.HTTPCookieProcessor.__init__(self, cookiejar)
1669
1670 def http_response(self, request, response):
1671 # Python 2 will choke on next HTTP request in row if there are non-ASCII
1672 # characters in Set-Cookie HTTP header of last response (see
067aa17e 1673 # https://github.com/ytdl-org/youtube-dl/issues/6769).
a6420bf5
S
1674 # In order to at least prevent crashing we will percent encode Set-Cookie
1675 # header before HTTPCookieProcessor starts processing it.
e28034c5
S
1676 # if sys.version_info < (3, 0) and response.headers:
1677 # for set_cookie_header in ('Set-Cookie', 'Set-Cookie2'):
1678 # set_cookie = response.headers.get(set_cookie_header)
1679 # if set_cookie:
1680 # set_cookie_escaped = compat_urllib_parse.quote(set_cookie, b"%/;:@&=+$,!~*'()?#[] ")
1681 # if set_cookie != set_cookie_escaped:
1682 # del response.headers[set_cookie_header]
1683 # response.headers[set_cookie_header] = set_cookie_escaped
a6420bf5
S
1684 return compat_urllib_request.HTTPCookieProcessor.http_response(self, request, response)
1685
f5fa042c 1686 https_request = compat_urllib_request.HTTPCookieProcessor.http_request
a6420bf5
S
1687 https_response = http_response
1688
1689
fca6dba8 1690class YoutubeDLRedirectHandler(compat_urllib_request.HTTPRedirectHandler):
201c1459 1691 """YoutubeDL redirect handler
1692
1693 The code is based on HTTPRedirectHandler implementation from CPython [1].
1694
1695 This redirect handler solves two issues:
1696 - ensures redirect URL is always unicode under python 2
1697 - introduces support for experimental HTTP response status code
1698 308 Permanent Redirect [2] used by some sites [3]
1699
1700 1. https://github.com/python/cpython/blob/master/Lib/urllib/request.py
1701 2. https://developer.mozilla.org/en-US/docs/Web/HTTP/Status/308
1702 3. https://github.com/ytdl-org/youtube-dl/issues/28768
1703 """
1704
1705 http_error_301 = http_error_303 = http_error_307 = http_error_308 = compat_urllib_request.HTTPRedirectHandler.http_error_302
1706
1707 def redirect_request(self, req, fp, code, msg, headers, newurl):
1708 """Return a Request or None in response to a redirect.
1709
1710 This is called by the http_error_30x methods when a
1711 redirection response is received. If a redirection should
1712 take place, return a new Request to allow http_error_30x to
1713 perform the redirect. Otherwise, raise HTTPError if no-one
1714 else should try to handle this url. Return None if you can't
1715 but another Handler might.
1716 """
1717 m = req.get_method()
1718 if (not (code in (301, 302, 303, 307, 308) and m in ("GET", "HEAD")
1719 or code in (301, 302, 303) and m == "POST")):
1720 raise compat_HTTPError(req.full_url, code, msg, headers, fp)
1721 # Strictly (according to RFC 2616), 301 or 302 in response to
1722 # a POST MUST NOT cause a redirection without confirmation
1723 # from the user (of urllib.request, in this case). In practice,
1724 # essentially all clients do redirect in this case, so we do
1725 # the same.
1726
1727 # On python 2 urlh.geturl() may sometimes return redirect URL
1728 # as byte string instead of unicode. This workaround allows
1729 # to force it always return unicode.
1730 if sys.version_info[0] < 3:
1731 newurl = compat_str(newurl)
1732
1733 # Be conciliant with URIs containing a space. This is mainly
1734 # redundant with the more complete encoding done in http_error_302(),
1735 # but it is kept for compatibility with other callers.
1736 newurl = newurl.replace(' ', '%20')
1737
1738 CONTENT_HEADERS = ("content-length", "content-type")
1739 # NB: don't use dict comprehension for python 2.6 compatibility
1740 newheaders = dict((k, v) for k, v in req.headers.items()
1741 if k.lower() not in CONTENT_HEADERS)
1742 return compat_urllib_request.Request(
1743 newurl, headers=newheaders, origin_req_host=req.origin_req_host,
1744 unverifiable=True)
fca6dba8
S
1745
1746
46f59e89
S
1747def extract_timezone(date_str):
1748 m = re.search(
f137e4c2 1749 r'''(?x)
1750 ^.{8,}? # >=8 char non-TZ prefix, if present
1751 (?P<tz>Z| # just the UTC Z, or
1752 (?:(?<=.\b\d{4}|\b\d{2}:\d\d)| # preceded by 4 digits or hh:mm or
1753 (?<!.\b[a-zA-Z]{3}|[a-zA-Z]{4}|..\b\d\d)) # not preceded by 3 alpha word or >= 4 alpha or 2 digits
1754 [ ]? # optional space
1755 (?P<sign>\+|-) # +/-
1756 (?P<hours>[0-9]{2}):?(?P<minutes>[0-9]{2}) # hh[:]mm
1757 $)
1758 ''', date_str)
46f59e89
S
1759 if not m:
1760 timezone = datetime.timedelta()
1761 else:
1762 date_str = date_str[:-len(m.group('tz'))]
1763 if not m.group('sign'):
1764 timezone = datetime.timedelta()
1765 else:
1766 sign = 1 if m.group('sign') == '+' else -1
1767 timezone = datetime.timedelta(
1768 hours=sign * int(m.group('hours')),
1769 minutes=sign * int(m.group('minutes')))
1770 return timezone, date_str
1771
1772
08b38d54 1773def parse_iso8601(date_str, delimiter='T', timezone=None):
912b38b4
PH
1774 """ Return a UNIX timestamp from the given date """
1775
1776 if date_str is None:
1777 return None
1778
52c3a6e4
S
1779 date_str = re.sub(r'\.[0-9]+', '', date_str)
1780
08b38d54 1781 if timezone is None:
46f59e89
S
1782 timezone, date_str = extract_timezone(date_str)
1783
52c3a6e4
S
1784 try:
1785 date_format = '%Y-%m-%d{0}%H:%M:%S'.format(delimiter)
1786 dt = datetime.datetime.strptime(date_str, date_format) - timezone
1787 return calendar.timegm(dt.timetuple())
1788 except ValueError:
1789 pass
912b38b4
PH
1790
1791
46f59e89
S
1792def date_formats(day_first=True):
1793 return DATE_FORMATS_DAY_FIRST if day_first else DATE_FORMATS_MONTH_FIRST
1794
1795
42bdd9d0 1796def unified_strdate(date_str, day_first=True):
bf50b038 1797 """Return a string with the date in the format YYYYMMDD"""
64e7ad60
PH
1798
1799 if date_str is None:
1800 return None
bf50b038 1801 upload_date = None
5f6a1245 1802 # Replace commas
026fcc04 1803 date_str = date_str.replace(',', ' ')
42bdd9d0 1804 # Remove AM/PM + timezone
9bb8e0a3 1805 date_str = re.sub(r'(?i)\s*(?:AM|PM)(?:\s+[A-Z]+)?', '', date_str)
46f59e89 1806 _, date_str = extract_timezone(date_str)
42bdd9d0 1807
46f59e89 1808 for expression in date_formats(day_first):
bf50b038
JMF
1809 try:
1810 upload_date = datetime.datetime.strptime(date_str, expression).strftime('%Y%m%d')
5de90176 1811 except ValueError:
bf50b038 1812 pass
42393ce2
PH
1813 if upload_date is None:
1814 timetuple = email.utils.parsedate_tz(date_str)
1815 if timetuple:
c6b9cf05
S
1816 try:
1817 upload_date = datetime.datetime(*timetuple[:6]).strftime('%Y%m%d')
1818 except ValueError:
1819 pass
6a750402
JMF
1820 if upload_date is not None:
1821 return compat_str(upload_date)
bf50b038 1822
5f6a1245 1823
46f59e89
S
1824def unified_timestamp(date_str, day_first=True):
1825 if date_str is None:
1826 return None
1827
2ae2ffda 1828 date_str = re.sub(r'[,|]', '', date_str)
46f59e89 1829
7dc2a74e 1830 pm_delta = 12 if re.search(r'(?i)PM', date_str) else 0
46f59e89
S
1831 timezone, date_str = extract_timezone(date_str)
1832
1833 # Remove AM/PM + timezone
1834 date_str = re.sub(r'(?i)\s*(?:AM|PM)(?:\s+[A-Z]+)?', '', date_str)
1835
deef3195
S
1836 # Remove unrecognized timezones from ISO 8601 alike timestamps
1837 m = re.search(r'\d{1,2}:\d{1,2}(?:\.\d+)?(?P<tz>\s*[A-Z]+)$', date_str)
1838 if m:
1839 date_str = date_str[:-len(m.group('tz'))]
1840
f226880c
PH
1841 # Python only supports microseconds, so remove nanoseconds
1842 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)
1843 if m:
1844 date_str = m.group(1)
1845
46f59e89
S
1846 for expression in date_formats(day_first):
1847 try:
7dc2a74e 1848 dt = datetime.datetime.strptime(date_str, expression) - timezone + datetime.timedelta(hours=pm_delta)
46f59e89
S
1849 return calendar.timegm(dt.timetuple())
1850 except ValueError:
1851 pass
1852 timetuple = email.utils.parsedate_tz(date_str)
1853 if timetuple:
7dc2a74e 1854 return calendar.timegm(timetuple) + pm_delta * 3600
46f59e89
S
1855
1856
28e614de 1857def determine_ext(url, default_ext='unknown_video'):
85750f89 1858 if url is None or '.' not in url:
f4776371 1859 return default_ext
9cb9a5df 1860 guess = url.partition('?')[0].rpartition('.')[2]
73e79f2a
PH
1861 if re.match(r'^[A-Za-z0-9]+$', guess):
1862 return guess
a7aaa398
S
1863 # Try extract ext from URLs like http://example.com/foo/bar.mp4/?download
1864 elif guess.rstrip('/') in KNOWN_EXTENSIONS:
9cb9a5df 1865 return guess.rstrip('/')
73e79f2a 1866 else:
cbdbb766 1867 return default_ext
73e79f2a 1868
5f6a1245 1869
824fa511
S
1870def subtitles_filename(filename, sub_lang, sub_format, expected_real_ext=None):
1871 return replace_extension(filename, sub_lang + '.' + sub_format, expected_real_ext)
d4051a8e 1872
5f6a1245 1873
9e62f283 1874def datetime_from_str(date_str, precision='auto', format='%Y%m%d'):
37254abc
JMF
1875 """
1876 Return a datetime object from a string in the format YYYYMMDD or
d49f8db3 1877 (now|today|yesterday|date)[+-][0-9](microsecond|second|minute|hour|day|week|month|year)(s)?
9e62f283 1878
1879 format: string date format used to return datetime object from
1880 precision: round the time portion of a datetime object.
1881 auto|microsecond|second|minute|hour|day.
1882 auto: round to the unit provided in date_str (if applicable).
1883 """
1884 auto_precision = False
1885 if precision == 'auto':
1886 auto_precision = True
1887 precision = 'microsecond'
396a76f7 1888 today = datetime_round(datetime.datetime.utcnow(), precision)
f8795e10 1889 if date_str in ('now', 'today'):
37254abc 1890 return today
f8795e10
PH
1891 if date_str == 'yesterday':
1892 return today - datetime.timedelta(days=1)
9e62f283 1893 match = re.match(
1894 r'(?P<start>.+)(?P<sign>[+-])(?P<time>\d+)(?P<unit>microsecond|second|minute|hour|day|week|month|year)(s)?',
1895 date_str)
37254abc 1896 if match is not None:
9e62f283 1897 start_time = datetime_from_str(match.group('start'), precision, format)
1898 time = int(match.group('time')) * (-1 if match.group('sign') == '-' else 1)
37254abc 1899 unit = match.group('unit')
9e62f283 1900 if unit == 'month' or unit == 'year':
1901 new_date = datetime_add_months(start_time, time * 12 if unit == 'year' else time)
37254abc 1902 unit = 'day'
9e62f283 1903 else:
1904 if unit == 'week':
1905 unit = 'day'
1906 time *= 7
1907 delta = datetime.timedelta(**{unit + 's': time})
1908 new_date = start_time + delta
1909 if auto_precision:
1910 return datetime_round(new_date, unit)
1911 return new_date
1912
1913 return datetime_round(datetime.datetime.strptime(date_str, format), precision)
1914
1915
d49f8db3 1916def date_from_str(date_str, format='%Y%m%d', strict=False):
9e62f283 1917 """
1918 Return a datetime object from a string in the format YYYYMMDD or
d49f8db3 1919 (now|today|yesterday|date)[+-][0-9](microsecond|second|minute|hour|day|week|month|year)(s)?
1920
1921 If "strict", only (now|today)[+-][0-9](day|week|month|year)(s)? is allowed
9e62f283 1922
1923 format: string date format used to return datetime object from
1924 """
d49f8db3 1925 if strict and not re.fullmatch(r'\d{8}|(now|today)[+-]\d+(day|week|month|year)(s)?', date_str):
1926 raise ValueError(f'Invalid date format {date_str}')
9e62f283 1927 return datetime_from_str(date_str, precision='microsecond', format=format).date()
1928
1929
1930def datetime_add_months(dt, months):
1931 """Increment/Decrement a datetime object by months."""
1932 month = dt.month + months - 1
1933 year = dt.year + month // 12
1934 month = month % 12 + 1
1935 day = min(dt.day, calendar.monthrange(year, month)[1])
1936 return dt.replace(year, month, day)
1937
1938
1939def datetime_round(dt, precision='day'):
1940 """
1941 Round a datetime object's time to a specific precision
1942 """
1943 if precision == 'microsecond':
1944 return dt
1945
1946 unit_seconds = {
1947 'day': 86400,
1948 'hour': 3600,
1949 'minute': 60,
1950 'second': 1,
1951 }
1952 roundto = lambda x, n: ((x + n / 2) // n) * n
1953 timestamp = calendar.timegm(dt.timetuple())
1954 return datetime.datetime.utcfromtimestamp(roundto(timestamp, unit_seconds[precision]))
5f6a1245
JW
1955
1956
e63fc1be 1957def hyphenate_date(date_str):
1958 """
1959 Convert a date in 'YYYYMMDD' format to 'YYYY-MM-DD' format"""
1960 match = re.match(r'^(\d\d\d\d)(\d\d)(\d\d)$', date_str)
1961 if match is not None:
1962 return '-'.join(match.groups())
1963 else:
1964 return date_str
1965
5f6a1245 1966
bd558525
JMF
1967class DateRange(object):
1968 """Represents a time interval between two dates"""
5f6a1245 1969
bd558525
JMF
1970 def __init__(self, start=None, end=None):
1971 """start and end must be strings in the format accepted by date"""
1972 if start is not None:
d49f8db3 1973 self.start = date_from_str(start, strict=True)
bd558525
JMF
1974 else:
1975 self.start = datetime.datetime.min.date()
1976 if end is not None:
d49f8db3 1977 self.end = date_from_str(end, strict=True)
bd558525
JMF
1978 else:
1979 self.end = datetime.datetime.max.date()
37254abc 1980 if self.start > self.end:
bd558525 1981 raise ValueError('Date range: "%s" , the start date must be before the end date' % self)
5f6a1245 1982
bd558525
JMF
1983 @classmethod
1984 def day(cls, day):
1985 """Returns a range that only contains the given day"""
5f6a1245
JW
1986 return cls(day, day)
1987
bd558525
JMF
1988 def __contains__(self, date):
1989 """Check if the date is in the range"""
37254abc
JMF
1990 if not isinstance(date, datetime.date):
1991 date = date_from_str(date)
1992 return self.start <= date <= self.end
5f6a1245 1993
bd558525 1994 def __str__(self):
5f6a1245 1995 return '%s - %s' % (self.start.isoformat(), self.end.isoformat())
c496ca96
PH
1996
1997
1998def platform_name():
1999 """ Returns the platform name as a compat_str """
2000 res = platform.platform()
2001 if isinstance(res, bytes):
2002 res = res.decode(preferredencoding())
2003
2004 assert isinstance(res, compat_str)
2005 return res
c257baff
PH
2006
2007
49fa4d9a
N
2008def get_windows_version():
2009 ''' Get Windows version. None if it's not running on Windows '''
2010 if compat_os_name == 'nt':
2011 return version_tuple(platform.win32_ver()[1])
2012 else:
2013 return None
2014
2015
b58ddb32
PH
2016def _windows_write_string(s, out):
2017 """ Returns True if the string was written using special methods,
2018 False if it has yet to be written out."""
2019 # Adapted from http://stackoverflow.com/a/3259271/35070
2020
b58ddb32
PH
2021 import ctypes.wintypes
2022
2023 WIN_OUTPUT_IDS = {
2024 1: -11,
2025 2: -12,
2026 }
2027
a383a98a
PH
2028 try:
2029 fileno = out.fileno()
2030 except AttributeError:
2031 # If the output stream doesn't have a fileno, it's virtual
2032 return False
aa42e873
PH
2033 except io.UnsupportedOperation:
2034 # Some strange Windows pseudo files?
2035 return False
b58ddb32
PH
2036 if fileno not in WIN_OUTPUT_IDS:
2037 return False
2038
d7cd9a9e 2039 GetStdHandle = compat_ctypes_WINFUNCTYPE(
b58ddb32 2040 ctypes.wintypes.HANDLE, ctypes.wintypes.DWORD)(
d7cd9a9e 2041 ('GetStdHandle', ctypes.windll.kernel32))
b58ddb32
PH
2042 h = GetStdHandle(WIN_OUTPUT_IDS[fileno])
2043
d7cd9a9e 2044 WriteConsoleW = compat_ctypes_WINFUNCTYPE(
b58ddb32
PH
2045 ctypes.wintypes.BOOL, ctypes.wintypes.HANDLE, ctypes.wintypes.LPWSTR,
2046 ctypes.wintypes.DWORD, ctypes.POINTER(ctypes.wintypes.DWORD),
d7cd9a9e 2047 ctypes.wintypes.LPVOID)(('WriteConsoleW', ctypes.windll.kernel32))
b58ddb32
PH
2048 written = ctypes.wintypes.DWORD(0)
2049
d7cd9a9e 2050 GetFileType = compat_ctypes_WINFUNCTYPE(ctypes.wintypes.DWORD, ctypes.wintypes.DWORD)(('GetFileType', ctypes.windll.kernel32))
b58ddb32
PH
2051 FILE_TYPE_CHAR = 0x0002
2052 FILE_TYPE_REMOTE = 0x8000
d7cd9a9e 2053 GetConsoleMode = compat_ctypes_WINFUNCTYPE(
b58ddb32
PH
2054 ctypes.wintypes.BOOL, ctypes.wintypes.HANDLE,
2055 ctypes.POINTER(ctypes.wintypes.DWORD))(
d7cd9a9e 2056 ('GetConsoleMode', ctypes.windll.kernel32))
b58ddb32
PH
2057 INVALID_HANDLE_VALUE = ctypes.wintypes.DWORD(-1).value
2058
2059 def not_a_console(handle):
2060 if handle == INVALID_HANDLE_VALUE or handle is None:
2061 return True
3089bc74
S
2062 return ((GetFileType(handle) & ~FILE_TYPE_REMOTE) != FILE_TYPE_CHAR
2063 or GetConsoleMode(handle, ctypes.byref(ctypes.wintypes.DWORD())) == 0)
b58ddb32
PH
2064
2065 if not_a_console(h):
2066 return False
2067
d1b9c912
PH
2068 def next_nonbmp_pos(s):
2069 try:
2070 return next(i for i, c in enumerate(s) if ord(c) > 0xffff)
2071 except StopIteration:
2072 return len(s)
2073
2074 while s:
2075 count = min(next_nonbmp_pos(s), 1024)
2076
b58ddb32 2077 ret = WriteConsoleW(
d1b9c912 2078 h, s, count if count else 2, ctypes.byref(written), None)
b58ddb32
PH
2079 if ret == 0:
2080 raise OSError('Failed to write string')
d1b9c912
PH
2081 if not count: # We just wrote a non-BMP character
2082 assert written.value == 2
2083 s = s[1:]
2084 else:
2085 assert written.value > 0
2086 s = s[written.value:]
b58ddb32
PH
2087 return True
2088
2089
734f90bb 2090def write_string(s, out=None, encoding=None):
7459e3a2
PH
2091 if out is None:
2092 out = sys.stderr
8bf48f23 2093 assert type(s) == compat_str
7459e3a2 2094
b58ddb32
PH
2095 if sys.platform == 'win32' and encoding is None and hasattr(out, 'fileno'):
2096 if _windows_write_string(s, out):
2097 return
2098
3089bc74
S
2099 if ('b' in getattr(out, 'mode', '')
2100 or sys.version_info[0] < 3): # Python 2 lies about mode of sys.stderr
104aa738
PH
2101 byt = s.encode(encoding or preferredencoding(), 'ignore')
2102 out.write(byt)
2103 elif hasattr(out, 'buffer'):
2104 enc = encoding or getattr(out, 'encoding', None) or preferredencoding()
2105 byt = s.encode(enc, 'ignore')
2106 out.buffer.write(byt)
2107 else:
8bf48f23 2108 out.write(s)
7459e3a2
PH
2109 out.flush()
2110
2111
48ea9cea
PH
2112def bytes_to_intlist(bs):
2113 if not bs:
2114 return []
2115 if isinstance(bs[0], int): # Python 3
2116 return list(bs)
2117 else:
2118 return [ord(c) for c in bs]
2119
c257baff 2120
cba892fa 2121def intlist_to_bytes(xs):
2122 if not xs:
2123 return b''
edaa23f8 2124 return compat_struct_pack('%dB' % len(xs), *xs)
c38b1e77
PH
2125
2126
0edb3e33 2127class LockingUnsupportedError(IOError):
2128 msg = 'File locking is not supported on this platform'
2129
2130 def __init__(self):
2131 super().__init__(self.msg)
2132
2133
c1c9a79c
PH
2134# Cross-platform file locking
2135if sys.platform == 'win32':
2136 import ctypes.wintypes
2137 import msvcrt
2138
2139 class OVERLAPPED(ctypes.Structure):
2140 _fields_ = [
2141 ('Internal', ctypes.wintypes.LPVOID),
2142 ('InternalHigh', ctypes.wintypes.LPVOID),
2143 ('Offset', ctypes.wintypes.DWORD),
2144 ('OffsetHigh', ctypes.wintypes.DWORD),
2145 ('hEvent', ctypes.wintypes.HANDLE),
2146 ]
2147
2148 kernel32 = ctypes.windll.kernel32
2149 LockFileEx = kernel32.LockFileEx
2150 LockFileEx.argtypes = [
2151 ctypes.wintypes.HANDLE, # hFile
2152 ctypes.wintypes.DWORD, # dwFlags
2153 ctypes.wintypes.DWORD, # dwReserved
2154 ctypes.wintypes.DWORD, # nNumberOfBytesToLockLow
2155 ctypes.wintypes.DWORD, # nNumberOfBytesToLockHigh
2156 ctypes.POINTER(OVERLAPPED) # Overlapped
2157 ]
2158 LockFileEx.restype = ctypes.wintypes.BOOL
2159 UnlockFileEx = kernel32.UnlockFileEx
2160 UnlockFileEx.argtypes = [
2161 ctypes.wintypes.HANDLE, # hFile
2162 ctypes.wintypes.DWORD, # dwReserved
2163 ctypes.wintypes.DWORD, # nNumberOfBytesToLockLow
2164 ctypes.wintypes.DWORD, # nNumberOfBytesToLockHigh
2165 ctypes.POINTER(OVERLAPPED) # Overlapped
2166 ]
2167 UnlockFileEx.restype = ctypes.wintypes.BOOL
2168 whole_low = 0xffffffff
2169 whole_high = 0x7fffffff
2170
747c0bd1 2171 def _lock_file(f, exclusive, block):
c1c9a79c
PH
2172 overlapped = OVERLAPPED()
2173 overlapped.Offset = 0
2174 overlapped.OffsetHigh = 0
2175 overlapped.hEvent = 0
2176 f._lock_file_overlapped_p = ctypes.pointer(overlapped)
747c0bd1 2177
2178 if not LockFileEx(msvcrt.get_osfhandle(f.fileno()),
2179 (0x2 if exclusive else 0x0) | (0x0 if block else 0x1),
2180 0, whole_low, whole_high, f._lock_file_overlapped_p):
2181 raise BlockingIOError('Locking file failed: %r' % ctypes.FormatError())
c1c9a79c
PH
2182
2183 def _unlock_file(f):
2184 assert f._lock_file_overlapped_p
2185 handle = msvcrt.get_osfhandle(f.fileno())
747c0bd1 2186 if not UnlockFileEx(handle, 0, whole_low, whole_high, f._lock_file_overlapped_p):
c1c9a79c
PH
2187 raise OSError('Unlocking file failed: %r' % ctypes.FormatError())
2188
2189else:
399a76e6
YCH
2190 try:
2191 import fcntl
c1c9a79c 2192
a3125791 2193 def _lock_file(f, exclusive, block):
b63837bc 2194 flags = fcntl.LOCK_EX if exclusive else fcntl.LOCK_SH
2195 if not block:
2196 flags |= fcntl.LOCK_NB
acea8d7c 2197 try:
b63837bc 2198 fcntl.flock(f, flags)
acea8d7c
JK
2199 except BlockingIOError:
2200 raise
2201 except OSError: # AOSP does not have flock()
b63837bc 2202 fcntl.lockf(f, flags)
c1c9a79c 2203
399a76e6 2204 def _unlock_file(f):
acea8d7c
JK
2205 try:
2206 fcntl.flock(f, fcntl.LOCK_UN)
2207 except OSError:
2208 fcntl.lockf(f, fcntl.LOCK_UN)
a3125791 2209
399a76e6 2210 except ImportError:
399a76e6 2211
a3125791 2212 def _lock_file(f, exclusive, block):
0edb3e33 2213 raise LockingUnsupportedError()
399a76e6
YCH
2214
2215 def _unlock_file(f):
0edb3e33 2216 raise LockingUnsupportedError()
c1c9a79c
PH
2217
2218
2219class locked_file(object):
0edb3e33 2220 locked = False
747c0bd1 2221
a3125791 2222 def __init__(self, filename, mode, block=True, encoding=None):
fcfa8853
JK
2223 if mode not in {'r', 'rb', 'a', 'ab', 'w', 'wb'}:
2224 raise NotImplementedError(mode)
2225 self.mode, self.block = mode, block
2226
2227 writable = any(f in mode for f in 'wax+')
2228 readable = any(f in mode for f in 'r+')
2229 flags = functools.reduce(operator.ior, (
2230 getattr(os, 'O_CLOEXEC', 0), # UNIX only
2231 getattr(os, 'O_BINARY', 0), # Windows only
2232 getattr(os, 'O_NOINHERIT', 0), # Windows only
2233 os.O_CREAT if writable else 0, # O_TRUNC only after locking
2234 os.O_APPEND if 'a' in mode else 0,
2235 os.O_EXCL if 'x' in mode else 0,
2236 os.O_RDONLY if not writable else os.O_RDWR if readable else os.O_WRONLY,
2237 ))
2238
2239 self.f = os.fdopen(os.open(filename, flags), mode, encoding=encoding)
c1c9a79c
PH
2240
2241 def __enter__(self):
a3125791 2242 exclusive = 'r' not in self.mode
c1c9a79c 2243 try:
a3125791 2244 _lock_file(self.f, exclusive, self.block)
0edb3e33 2245 self.locked = True
c1c9a79c
PH
2246 except IOError:
2247 self.f.close()
2248 raise
fcfa8853
JK
2249 if 'w' in self.mode:
2250 self.f.truncate()
c1c9a79c
PH
2251 return self
2252
0edb3e33 2253 def unlock(self):
2254 if not self.locked:
2255 return
c1c9a79c 2256 try:
0edb3e33 2257 _unlock_file(self.f)
c1c9a79c 2258 finally:
0edb3e33 2259 self.locked = False
c1c9a79c 2260
0edb3e33 2261 def __exit__(self, *_):
2262 try:
2263 self.unlock()
2264 finally:
2265 self.f.close()
4eb7f1d1 2266
0edb3e33 2267 open = __enter__
2268 close = __exit__
a3125791 2269
0edb3e33 2270 def __getattr__(self, attr):
2271 return getattr(self.f, attr)
a3125791 2272
0edb3e33 2273 def __iter__(self):
2274 return iter(self.f)
a3125791 2275
4eb7f1d1 2276
4644ac55
S
2277def get_filesystem_encoding():
2278 encoding = sys.getfilesystemencoding()
2279 return encoding if encoding is not None else 'utf-8'
2280
2281
4eb7f1d1 2282def shell_quote(args):
a6a173c2 2283 quoted_args = []
4644ac55 2284 encoding = get_filesystem_encoding()
a6a173c2
JMF
2285 for a in args:
2286 if isinstance(a, bytes):
2287 # We may get a filename encoded with 'encodeFilename'
2288 a = a.decode(encoding)
aefce8e6 2289 quoted_args.append(compat_shlex_quote(a))
28e614de 2290 return ' '.join(quoted_args)
9d4660ca
PH
2291
2292
2293def smuggle_url(url, data):
2294 """ Pass additional data in a URL for internal use. """
2295
81953d1a
RA
2296 url, idata = unsmuggle_url(url, {})
2297 data.update(idata)
15707c7e 2298 sdata = compat_urllib_parse_urlencode(
28e614de
PH
2299 {'__youtubedl_smuggle': json.dumps(data)})
2300 return url + '#' + sdata
9d4660ca
PH
2301
2302
79f82953 2303def unsmuggle_url(smug_url, default=None):
83e865a3 2304 if '#__youtubedl_smuggle' not in smug_url:
79f82953 2305 return smug_url, default
28e614de
PH
2306 url, _, sdata = smug_url.rpartition('#')
2307 jsond = compat_parse_qs(sdata)['__youtubedl_smuggle'][0]
9d4660ca
PH
2308 data = json.loads(jsond)
2309 return url, data
02dbf93f
PH
2310
2311
e0fd9573 2312def format_decimal_suffix(num, fmt='%d%s', *, factor=1000):
2313 """ Formats numbers with decimal sufixes like K, M, etc """
2314 num, factor = float_or_none(num), float(factor)
4c3f8c3f 2315 if num is None or num < 0:
e0fd9573 2316 return None
eeb2a770 2317 POSSIBLE_SUFFIXES = 'kMGTPEZY'
2318 exponent = 0 if num == 0 else min(int(math.log(num, factor)), len(POSSIBLE_SUFFIXES))
2319 suffix = ['', *POSSIBLE_SUFFIXES][exponent]
abbeeebc 2320 if factor == 1024:
2321 suffix = {'k': 'Ki', '': ''}.get(suffix, f'{suffix}i')
e0fd9573 2322 converted = num / (factor ** exponent)
abbeeebc 2323 return fmt % (converted, suffix)
e0fd9573 2324
2325
02dbf93f 2326def format_bytes(bytes):
f02d24d8 2327 return format_decimal_suffix(bytes, '%.2f%sB', factor=1024) or 'N/A'
f53c966a 2328
1c088fa8 2329
fb47597b
S
2330def lookup_unit_table(unit_table, s):
2331 units_re = '|'.join(re.escape(u) for u in unit_table)
2332 m = re.match(
782b1b5b 2333 r'(?P<num>[0-9]+(?:[,.][0-9]*)?)\s*(?P<unit>%s)\b' % units_re, s)
fb47597b
S
2334 if not m:
2335 return None
2336 num_str = m.group('num').replace(',', '.')
2337 mult = unit_table[m.group('unit')]
2338 return int(float(num_str) * mult)
2339
2340
be64b5b0
PH
2341def parse_filesize(s):
2342 if s is None:
2343 return None
2344
dfb1b146 2345 # The lower-case forms are of course incorrect and unofficial,
be64b5b0
PH
2346 # but we support those too
2347 _UNIT_TABLE = {
2348 'B': 1,
2349 'b': 1,
70852b47 2350 'bytes': 1,
be64b5b0
PH
2351 'KiB': 1024,
2352 'KB': 1000,
2353 'kB': 1024,
2354 'Kb': 1000,
13585d76 2355 'kb': 1000,
70852b47
YCH
2356 'kilobytes': 1000,
2357 'kibibytes': 1024,
be64b5b0
PH
2358 'MiB': 1024 ** 2,
2359 'MB': 1000 ** 2,
2360 'mB': 1024 ** 2,
2361 'Mb': 1000 ** 2,
13585d76 2362 'mb': 1000 ** 2,
70852b47
YCH
2363 'megabytes': 1000 ** 2,
2364 'mebibytes': 1024 ** 2,
be64b5b0
PH
2365 'GiB': 1024 ** 3,
2366 'GB': 1000 ** 3,
2367 'gB': 1024 ** 3,
2368 'Gb': 1000 ** 3,
13585d76 2369 'gb': 1000 ** 3,
70852b47
YCH
2370 'gigabytes': 1000 ** 3,
2371 'gibibytes': 1024 ** 3,
be64b5b0
PH
2372 'TiB': 1024 ** 4,
2373 'TB': 1000 ** 4,
2374 'tB': 1024 ** 4,
2375 'Tb': 1000 ** 4,
13585d76 2376 'tb': 1000 ** 4,
70852b47
YCH
2377 'terabytes': 1000 ** 4,
2378 'tebibytes': 1024 ** 4,
be64b5b0
PH
2379 'PiB': 1024 ** 5,
2380 'PB': 1000 ** 5,
2381 'pB': 1024 ** 5,
2382 'Pb': 1000 ** 5,
13585d76 2383 'pb': 1000 ** 5,
70852b47
YCH
2384 'petabytes': 1000 ** 5,
2385 'pebibytes': 1024 ** 5,
be64b5b0
PH
2386 'EiB': 1024 ** 6,
2387 'EB': 1000 ** 6,
2388 'eB': 1024 ** 6,
2389 'Eb': 1000 ** 6,
13585d76 2390 'eb': 1000 ** 6,
70852b47
YCH
2391 'exabytes': 1000 ** 6,
2392 'exbibytes': 1024 ** 6,
be64b5b0
PH
2393 'ZiB': 1024 ** 7,
2394 'ZB': 1000 ** 7,
2395 'zB': 1024 ** 7,
2396 'Zb': 1000 ** 7,
13585d76 2397 'zb': 1000 ** 7,
70852b47
YCH
2398 'zettabytes': 1000 ** 7,
2399 'zebibytes': 1024 ** 7,
be64b5b0
PH
2400 'YiB': 1024 ** 8,
2401 'YB': 1000 ** 8,
2402 'yB': 1024 ** 8,
2403 'Yb': 1000 ** 8,
13585d76 2404 'yb': 1000 ** 8,
70852b47
YCH
2405 'yottabytes': 1000 ** 8,
2406 'yobibytes': 1024 ** 8,
be64b5b0
PH
2407 }
2408
fb47597b
S
2409 return lookup_unit_table(_UNIT_TABLE, s)
2410
2411
2412def parse_count(s):
2413 if s is None:
be64b5b0
PH
2414 return None
2415
352d5da8 2416 s = re.sub(r'^[^\d]+\s', '', s).strip()
fb47597b
S
2417
2418 if re.match(r'^[\d,.]+$', s):
2419 return str_to_int(s)
2420
2421 _UNIT_TABLE = {
2422 'k': 1000,
2423 'K': 1000,
2424 'm': 1000 ** 2,
2425 'M': 1000 ** 2,
2426 'kk': 1000 ** 2,
2427 'KK': 1000 ** 2,
352d5da8 2428 'b': 1000 ** 3,
2429 'B': 1000 ** 3,
fb47597b 2430 }
be64b5b0 2431
352d5da8 2432 ret = lookup_unit_table(_UNIT_TABLE, s)
2433 if ret is not None:
2434 return ret
2435
2436 mobj = re.match(r'([\d,.]+)(?:$|\s)', s)
2437 if mobj:
2438 return str_to_int(mobj.group(1))
be64b5b0 2439
2f7ae819 2440
5d45484c 2441def parse_resolution(s, *, lenient=False):
b871d7e9
S
2442 if s is None:
2443 return {}
2444
5d45484c
LNO
2445 if lenient:
2446 mobj = re.search(r'(?P<w>\d+)\s*[xX×,]\s*(?P<h>\d+)', s)
2447 else:
2448 mobj = re.search(r'(?<![a-zA-Z0-9])(?P<w>\d+)\s*[xX×,]\s*(?P<h>\d+)(?![a-zA-Z0-9])', s)
b871d7e9
S
2449 if mobj:
2450 return {
2451 'width': int(mobj.group('w')),
2452 'height': int(mobj.group('h')),
2453 }
2454
17ec8bcf 2455 mobj = re.search(r'(?<![a-zA-Z0-9])(\d+)[pPiI](?![a-zA-Z0-9])', s)
b871d7e9
S
2456 if mobj:
2457 return {'height': int(mobj.group(1))}
2458
2459 mobj = re.search(r'\b([48])[kK]\b', s)
2460 if mobj:
2461 return {'height': int(mobj.group(1)) * 540}
2462
2463 return {}
2464
2465
0dc41787
S
2466def parse_bitrate(s):
2467 if not isinstance(s, compat_str):
2468 return
2469 mobj = re.search(r'\b(\d+)\s*kbps', s)
2470 if mobj:
2471 return int(mobj.group(1))
2472
2473
a942d6cb 2474def month_by_name(name, lang='en'):
caefb1de
PH
2475 """ Return the number of a month by (locale-independently) English name """
2476
f6717dec 2477 month_names = MONTH_NAMES.get(lang, MONTH_NAMES['en'])
a942d6cb 2478
caefb1de 2479 try:
f6717dec 2480 return month_names.index(name) + 1
7105440c
YCH
2481 except ValueError:
2482 return None
2483
2484
2485def month_by_abbreviation(abbrev):
2486 """ Return the number of a month by (locale-independently) English
2487 abbreviations """
2488
2489 try:
2490 return [s[:3] for s in ENGLISH_MONTH_NAMES].index(abbrev) + 1
caefb1de
PH
2491 except ValueError:
2492 return None
18258362
JMF
2493
2494
5aafe895 2495def fix_xml_ampersands(xml_str):
18258362 2496 """Replace all the '&' by '&amp;' in XML"""
5aafe895
PH
2497 return re.sub(
2498 r'&(?!amp;|lt;|gt;|apos;|quot;|#x[0-9a-fA-F]{,4};|#[0-9]{,4};)',
28e614de 2499 '&amp;',
5aafe895 2500 xml_str)
e3946f98
PH
2501
2502
2503def setproctitle(title):
8bf48f23 2504 assert isinstance(title, compat_str)
c1c05c67
YCH
2505
2506 # ctypes in Jython is not complete
2507 # http://bugs.jython.org/issue2148
2508 if sys.platform.startswith('java'):
2509 return
2510
e3946f98 2511 try:
611c1dd9 2512 libc = ctypes.cdll.LoadLibrary('libc.so.6')
e3946f98
PH
2513 except OSError:
2514 return
2f49bcd6
RC
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
6eefe533
PH
2520 title_bytes = title.encode('utf-8')
2521 buf = ctypes.create_string_buffer(len(title_bytes))
2522 buf.value = title_bytes
e3946f98 2523 try:
6eefe533 2524 libc.prctl(15, buf, 0, 0, 0)
e3946f98
PH
2525 except AttributeError:
2526 return # Strange libc, just skip this
d7dda168
PH
2527
2528
2529def remove_start(s, start):
46bc9b7d 2530 return s[len(start):] if s is not None and s.startswith(start) else s
29eb5174
PH
2531
2532
2b9faf55 2533def remove_end(s, end):
46bc9b7d 2534 return s[:-len(end)] if s is not None and s.endswith(end) else s
2b9faf55
PH
2535
2536
31b2051e
S
2537def 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
b6e0c7d2
U
2546def get_domain(url):
2547 domain = re.match(r'(?:https?:\/\/)?(?:www\.)?(?P<domain>[^\n\/]+\.[^\n\/]+)(?:\/(.*))?', url)
2548 return domain.group('domain') if domain else None
2549
2550
29eb5174 2551def url_basename(url):
9b8aaeed 2552 path = compat_urlparse.urlparse(url).path
28e614de 2553 return path.strip('/').split('/')[-1]
aa94a6d3
PH
2554
2555
02dc0a36
S
2556def base_url(url):
2557 return re.match(r'https?://[^?#&]+/', url).group()
2558
2559
e34c3361 2560def urljoin(base, path):
4b5de77b
S
2561 if isinstance(path, bytes):
2562 path = path.decode('utf-8')
e34c3361
S
2563 if not isinstance(path, compat_str) or not path:
2564 return None
fad4ceb5 2565 if re.match(r'^(?:[a-zA-Z][a-zA-Z0-9+-.]*:)?//', path):
e34c3361 2566 return path
4b5de77b
S
2567 if isinstance(base, bytes):
2568 base = base.decode('utf-8')
2569 if not isinstance(base, compat_str) or not re.match(
2570 r'^(?:https?:)?//', base):
e34c3361
S
2571 return None
2572 return compat_urlparse.urljoin(base, path)
2573
2574
aa94a6d3
PH
2575class HEADRequest(compat_urllib_request.Request):
2576 def get_method(self):
611c1dd9 2577 return 'HEAD'
7217e148
PH
2578
2579
95cf60e8
S
2580class PUTRequest(compat_urllib_request.Request):
2581 def get_method(self):
2582 return 'PUT'
2583
2584
9732d77e 2585def int_or_none(v, scale=1, default=None, get_attr=None, invscale=1):
9e907ebd 2586 if get_attr and v is not None:
2587 v = getattr(v, get_attr, None)
1812afb7
S
2588 try:
2589 return int(v) * invscale // scale
31c49255 2590 except (ValueError, TypeError, OverflowError):
af98f8ff 2591 return default
9732d77e 2592
9572013d 2593
40a90862
JMF
2594def str_or_none(v, default=None):
2595 return default if v is None else compat_str(v)
2596
9732d77e
PH
2597
2598def str_to_int(int_str):
48d4681e 2599 """ A more relaxed version of int_or_none """
42db58ec 2600 if isinstance(int_str, compat_integer_types):
348c6bf1 2601 return int_str
42db58ec
S
2602 elif isinstance(int_str, compat_str):
2603 int_str = re.sub(r'[,\.\+]', '', int_str)
2604 return int_or_none(int_str)
608d11f5
PH
2605
2606
9732d77e 2607def float_or_none(v, scale=1, invscale=1, default=None):
caf80631
S
2608 if v is None:
2609 return default
2610 try:
2611 return float(v) * invscale / scale
5e1271c5 2612 except (ValueError, TypeError):
caf80631 2613 return default
43f775e4
PH
2614
2615
c7e327c4
S
2616def bool_or_none(v, default=None):
2617 return v if isinstance(v, bool) else default
2618
2619
53cd37ba
S
2620def strip_or_none(v, default=None):
2621 return v.strip() if isinstance(v, compat_str) else default
b72b4431
S
2622
2623
af03000a
S
2624def url_or_none(url):
2625 if not url or not isinstance(url, compat_str):
2626 return None
2627 url = url.strip()
29f7c58a 2628 return url if re.match(r'^(?:(?:https?|rt(?:m(?:pt?[es]?|fp)|sp[su]?)|mms|ftps?):)?//', url) else None
af03000a
S
2629
2630
3e9b66d7
LNO
2631def request_to_url(req):
2632 if isinstance(req, compat_urllib_request.Request):
2633 return req.get_full_url()
2634 else:
2635 return req
2636
2637
e29663c6 2638def strftime_or_none(timestamp, date_format, default=None):
2639 datetime_object = None
2640 try:
2641 if isinstance(timestamp, compat_numeric_types): # unix timestamp
2642 datetime_object = datetime.datetime.utcfromtimestamp(timestamp)
2643 elif isinstance(timestamp, compat_str): # assume YYYYMMDD
2644 datetime_object = datetime.datetime.strptime(timestamp, '%Y%m%d')
2645 return datetime_object.strftime(date_format)
2646 except (ValueError, TypeError, AttributeError):
2647 return default
2648
2649
608d11f5 2650def parse_duration(s):
8f9312c3 2651 if not isinstance(s, compat_basestring):
608d11f5 2652 return None
ca7b3246 2653 s = s.strip()
38d79fd1 2654 if not s:
2655 return None
ca7b3246 2656
acaff495 2657 days, hours, mins, secs, ms = [None] * 5
8bd1c00b 2658 m = re.match(r'''(?x)
2659 (?P<before_secs>
2660 (?:(?:(?P<days>[0-9]+):)?(?P<hours>[0-9]+):)?(?P<mins>[0-9]+):)?
2661 (?P<secs>(?(before_secs)[0-9]{1,2}|[0-9]+))
2662 (?P<ms>[.:][0-9]+)?Z?$
2663 ''', s)
acaff495 2664 if m:
8bd1c00b 2665 days, hours, mins, secs, ms = m.group('days', 'hours', 'mins', 'secs', 'ms')
acaff495 2666 else:
2667 m = re.match(
056653bb
S
2668 r'''(?ix)(?:P?
2669 (?:
1c1b2f96 2670 [0-9]+\s*y(?:ears?)?,?\s*
056653bb
S
2671 )?
2672 (?:
1c1b2f96 2673 [0-9]+\s*m(?:onths?)?,?\s*
056653bb
S
2674 )?
2675 (?:
1c1b2f96 2676 [0-9]+\s*w(?:eeks?)?,?\s*
056653bb 2677 )?
8f4b58d7 2678 (?:
1c1b2f96 2679 (?P<days>[0-9]+)\s*d(?:ays?)?,?\s*
8f4b58d7 2680 )?
056653bb 2681 T)?
acaff495 2682 (?:
1c1b2f96 2683 (?P<hours>[0-9]+)\s*h(?:ours?)?,?\s*
acaff495 2684 )?
2685 (?:
1c1b2f96 2686 (?P<mins>[0-9]+)\s*m(?:in(?:ute)?s?)?,?\s*
acaff495 2687 )?
2688 (?:
2689 (?P<secs>[0-9]+)(?P<ms>\.[0-9]+)?\s*s(?:ec(?:ond)?s?)?\s*
15846398 2690 )?Z?$''', s)
acaff495 2691 if m:
2692 days, hours, mins, secs, ms = m.groups()
2693 else:
15846398 2694 m = re.match(r'(?i)(?:(?P<hours>[0-9.]+)\s*(?:hours?)|(?P<mins>[0-9.]+)\s*(?:mins?\.?|minutes?)\s*)Z?$', s)
acaff495 2695 if m:
2696 hours, mins = m.groups()
2697 else:
2698 return None
2699
2700 duration = 0
2701 if secs:
2702 duration += float(secs)
2703 if mins:
2704 duration += float(mins) * 60
2705 if hours:
2706 duration += float(hours) * 60 * 60
2707 if days:
2708 duration += float(days) * 24 * 60 * 60
2709 if ms:
8bd1c00b 2710 duration += float(ms.replace(':', '.'))
acaff495 2711 return duration
91d7d0b3
JMF
2712
2713
e65e4c88 2714def prepend_extension(filename, ext, expected_real_ext=None):
5f6a1245 2715 name, real_ext = os.path.splitext(filename)
e65e4c88
S
2716 return (
2717 '{0}.{1}{2}'.format(name, ext, real_ext)
2718 if not expected_real_ext or real_ext[1:] == expected_real_ext
2719 else '{0}.{1}'.format(filename, ext))
d70ad093
PH
2720
2721
b3ed15b7
S
2722def replace_extension(filename, ext, expected_real_ext=None):
2723 name, real_ext = os.path.splitext(filename)
2724 return '{0}.{1}'.format(
2725 name if not expected_real_ext or real_ext[1:] == expected_real_ext else filename,
2726 ext)
2727
2728
d70ad093
PH
2729def check_executable(exe, args=[]):
2730 """ Checks if the given binary is installed somewhere in PATH, and returns its name.
2731 args can be a list of arguments for a short output (like -version) """
2732 try:
d3c93ec2 2733 Popen([exe] + args, stdout=subprocess.PIPE, stderr=subprocess.PIPE).communicate_or_kill()
d70ad093
PH
2734 except OSError:
2735 return False
2736 return exe
b7ab0590
PH
2737
2738
8a7f68d0 2739def _get_exe_version_output(exe, args, *, to_screen=None):
2740 if to_screen:
2741 to_screen(f'Checking exe version: {shell_quote([exe] + args)}')
95807118 2742 try:
b64d04c1 2743 # STDIN should be redirected too. On UNIX-like systems, ffmpeg triggers
7a5c1cfe 2744 # SIGTTOU if yt-dlp is run in the background.
067aa17e 2745 # See https://github.com/ytdl-org/youtube-dl/issues/955#issuecomment-209789656
d3c93ec2 2746 out, _ = Popen(
2747 [encodeArgument(exe)] + args, stdin=subprocess.PIPE,
2748 stdout=subprocess.PIPE, stderr=subprocess.STDOUT).communicate_or_kill()
95807118
PH
2749 except OSError:
2750 return False
cae97f65
PH
2751 if isinstance(out, bytes): # Python 2.x
2752 out = out.decode('ascii', 'ignore')
9af98e17 2753 return out
cae97f65
PH
2754
2755
2756def detect_exe_version(output, version_re=None, unrecognized='present'):
2757 assert isinstance(output, compat_str)
2758 if version_re is None:
2759 version_re = r'version\s+([-0-9._a-zA-Z]+)'
2760 m = re.search(version_re, output)
95807118
PH
2761 if m:
2762 return m.group(1)
2763 else:
2764 return unrecognized
2765
2766
9af98e17 2767def get_exe_version(exe, args=['--version'],
2768 version_re=None, unrecognized='present'):
2769 """ Returns the version of the specified executable,
2770 or False if the executable is not present """
2771 out = _get_exe_version_output(exe, args)
2772 return detect_exe_version(out, version_re, unrecognized) if out else False
2773
2774
cb89cfc1 2775class LazyList(collections.abc.Sequence):
483336e7 2776 ''' Lazy immutable list from an iterable
2777 Note that slices of a LazyList are lists and not LazyList'''
2778
8e5fecc8 2779 class IndexError(IndexError):
2780 pass
2781
282f5709 2782 def __init__(self, iterable, *, reverse=False, _cache=None):
483336e7 2783 self.__iterable = iter(iterable)
282f5709 2784 self.__cache = [] if _cache is None else _cache
2785 self.__reversed = reverse
483336e7 2786
2787 def __iter__(self):
28419ca2 2788 if self.__reversed:
2789 # We need to consume the entire iterable to iterate in reverse
981052c9 2790 yield from self.exhaust()
28419ca2 2791 return
2792 yield from self.__cache
483336e7 2793 for item in self.__iterable:
2794 self.__cache.append(item)
2795 yield item
2796
981052c9 2797 def __exhaust(self):
483336e7 2798 self.__cache.extend(self.__iterable)
9f1a1c36 2799 # Discard the emptied iterable to make it pickle-able
2800 self.__iterable = []
28419ca2 2801 return self.__cache
2802
981052c9 2803 def exhaust(self):
2804 ''' Evaluate the entire iterable '''
2805 return self.__exhaust()[::-1 if self.__reversed else 1]
2806
28419ca2 2807 @staticmethod
981052c9 2808 def __reverse_index(x):
e0f2b4b4 2809 return None if x is None else -(x + 1)
483336e7 2810
2811 def __getitem__(self, idx):
2812 if isinstance(idx, slice):
28419ca2 2813 if self.__reversed:
e0f2b4b4 2814 idx = slice(self.__reverse_index(idx.start), self.__reverse_index(idx.stop), -(idx.step or 1))
2815 start, stop, step = idx.start, idx.stop, idx.step or 1
483336e7 2816 elif isinstance(idx, int):
28419ca2 2817 if self.__reversed:
981052c9 2818 idx = self.__reverse_index(idx)
e0f2b4b4 2819 start, stop, step = idx, idx, 0
483336e7 2820 else:
2821 raise TypeError('indices must be integers or slices')
e0f2b4b4 2822 if ((start or 0) < 0 or (stop or 0) < 0
2823 or (start is None and step < 0)
2824 or (stop is None and step > 0)):
483336e7 2825 # We need to consume the entire iterable to be able to slice from the end
2826 # Obviously, never use this with infinite iterables
8e5fecc8 2827 self.__exhaust()
2828 try:
2829 return self.__cache[idx]
2830 except IndexError as e:
2831 raise self.IndexError(e) from e
e0f2b4b4 2832 n = max(start or 0, stop or 0) - len(self.__cache) + 1
28419ca2 2833 if n > 0:
2834 self.__cache.extend(itertools.islice(self.__iterable, n))
8e5fecc8 2835 try:
2836 return self.__cache[idx]
2837 except IndexError as e:
2838 raise self.IndexError(e) from e
483336e7 2839
2840 def __bool__(self):
2841 try:
28419ca2 2842 self[-1] if self.__reversed else self[0]
8e5fecc8 2843 except self.IndexError:
483336e7 2844 return False
2845 return True
2846
2847 def __len__(self):
8e5fecc8 2848 self.__exhaust()
483336e7 2849 return len(self.__cache)
2850
282f5709 2851 def __reversed__(self):
2852 return type(self)(self.__iterable, reverse=not self.__reversed, _cache=self.__cache)
2853
2854 def __copy__(self):
2855 return type(self)(self.__iterable, reverse=self.__reversed, _cache=self.__cache)
2856
28419ca2 2857 def __repr__(self):
2858 # repr and str should mimic a list. So we exhaust the iterable
2859 return repr(self.exhaust())
2860
2861 def __str__(self):
2862 return repr(self.exhaust())
2863
483336e7 2864
7be9ccff 2865class PagedList:
c07a39ae 2866
2867 class IndexError(IndexError):
2868 pass
2869
dd26ced1
PH
2870 def __len__(self):
2871 # This is only useful for tests
2872 return len(self.getslice())
2873
7be9ccff 2874 def __init__(self, pagefunc, pagesize, use_cache=True):
2875 self._pagefunc = pagefunc
2876 self._pagesize = pagesize
f1d13090 2877 self._pagecount = float('inf')
7be9ccff 2878 self._use_cache = use_cache
2879 self._cache = {}
2880
2881 def getpage(self, pagenum):
d8cf8d97 2882 page_results = self._cache.get(pagenum)
2883 if page_results is None:
f1d13090 2884 page_results = [] if pagenum > self._pagecount else list(self._pagefunc(pagenum))
7be9ccff 2885 if self._use_cache:
2886 self._cache[pagenum] = page_results
2887 return page_results
2888
2889 def getslice(self, start=0, end=None):
2890 return list(self._getslice(start, end))
2891
2892 def _getslice(self, start, end):
55575225 2893 raise NotImplementedError('This method must be implemented by subclasses')
2894
2895 def __getitem__(self, idx):
f1d13090 2896 assert self._use_cache, 'Indexing PagedList requires cache'
55575225 2897 if not isinstance(idx, int) or idx < 0:
2898 raise TypeError('indices must be non-negative integers')
2899 entries = self.getslice(idx, idx + 1)
d8cf8d97 2900 if not entries:
c07a39ae 2901 raise self.IndexError()
d8cf8d97 2902 return entries[0]
55575225 2903
9c44d242
PH
2904
2905class OnDemandPagedList(PagedList):
a44ca5a4 2906 """Download pages until a page with less than maximum results"""
7be9ccff 2907 def _getslice(self, start, end):
b7ab0590
PH
2908 for pagenum in itertools.count(start // self._pagesize):
2909 firstid = pagenum * self._pagesize
2910 nextfirstid = pagenum * self._pagesize + self._pagesize
2911 if start >= nextfirstid:
2912 continue
2913
b7ab0590
PH
2914 startv = (
2915 start % self._pagesize
2916 if firstid <= start < nextfirstid
2917 else 0)
b7ab0590
PH
2918 endv = (
2919 ((end - 1) % self._pagesize) + 1
2920 if (end is not None and firstid <= end <= nextfirstid)
2921 else None)
2922
f1d13090 2923 try:
2924 page_results = self.getpage(pagenum)
2925 except Exception:
2926 self._pagecount = pagenum - 1
2927 raise
b7ab0590
PH
2928 if startv != 0 or endv is not None:
2929 page_results = page_results[startv:endv]
7be9ccff 2930 yield from page_results
b7ab0590
PH
2931
2932 # A little optimization - if current page is not "full", ie. does
2933 # not contain page_size videos then we can assume that this page
2934 # is the last one - there are no more ids on further pages -
2935 # i.e. no need to query again.
2936 if len(page_results) + startv < self._pagesize:
2937 break
2938
2939 # If we got the whole page, but the next page is not interesting,
2940 # break out early as well
2941 if end == nextfirstid:
2942 break
81c2f20b
PH
2943
2944
9c44d242 2945class InAdvancePagedList(PagedList):
a44ca5a4 2946 """PagedList with total number of pages known in advance"""
9c44d242 2947 def __init__(self, pagefunc, pagecount, pagesize):
7be9ccff 2948 PagedList.__init__(self, pagefunc, pagesize, True)
f1d13090 2949 self._pagecount = pagecount
9c44d242 2950
7be9ccff 2951 def _getslice(self, start, end):
9c44d242 2952 start_page = start // self._pagesize
d37707bd 2953 end_page = self._pagecount if end is None else min(self._pagecount, end // self._pagesize + 1)
9c44d242
PH
2954 skip_elems = start - start_page * self._pagesize
2955 only_more = None if end is None else end - start
2956 for pagenum in range(start_page, end_page):
7be9ccff 2957 page_results = self.getpage(pagenum)
9c44d242 2958 if skip_elems:
7be9ccff 2959 page_results = page_results[skip_elems:]
9c44d242
PH
2960 skip_elems = None
2961 if only_more is not None:
7be9ccff 2962 if len(page_results) < only_more:
2963 only_more -= len(page_results)
9c44d242 2964 else:
7be9ccff 2965 yield from page_results[:only_more]
9c44d242 2966 break
7be9ccff 2967 yield from page_results
9c44d242
PH
2968
2969
81c2f20b 2970def uppercase_escape(s):
676eb3f2 2971 unicode_escape = codecs.getdecoder('unicode_escape')
81c2f20b 2972 return re.sub(
a612753d 2973 r'\\U[0-9a-fA-F]{8}',
676eb3f2
PH
2974 lambda m: unicode_escape(m.group(0))[0],
2975 s)
0fe2ff78
YCH
2976
2977
2978def lowercase_escape(s):
2979 unicode_escape = codecs.getdecoder('unicode_escape')
2980 return re.sub(
2981 r'\\u[0-9a-fA-F]{4}',
2982 lambda m: unicode_escape(m.group(0))[0],
2983 s)
b53466e1 2984
d05cfe06
S
2985
2986def escape_rfc3986(s):
2987 """Escape non-ASCII characters as suggested by RFC 3986"""
8f9312c3 2988 if sys.version_info < (3, 0) and isinstance(s, compat_str):
d05cfe06 2989 s = s.encode('utf-8')
ecc0c5ee 2990 return compat_urllib_parse.quote(s, b"%/;:@&=+$,!~*'()?#[]")
d05cfe06
S
2991
2992
2993def escape_url(url):
2994 """Escape URL as suggested by RFC 3986"""
2995 url_parsed = compat_urllib_parse_urlparse(url)
2996 return url_parsed._replace(
efbed08d 2997 netloc=url_parsed.netloc.encode('idna').decode('ascii'),
d05cfe06
S
2998 path=escape_rfc3986(url_parsed.path),
2999 params=escape_rfc3986(url_parsed.params),
3000 query=escape_rfc3986(url_parsed.query),
3001 fragment=escape_rfc3986(url_parsed.fragment)
3002 ).geturl()
3003
62e609ab 3004
4dfbf869 3005def parse_qs(url):
3006 return compat_parse_qs(compat_urllib_parse_urlparse(url).query)
3007
3008
62e609ab
PH
3009def read_batch_urls(batch_fd):
3010 def fixup(url):
3011 if not isinstance(url, compat_str):
3012 url = url.decode('utf-8', 'replace')
8c04f0be 3013 BOM_UTF8 = ('\xef\xbb\xbf', '\ufeff')
3014 for bom in BOM_UTF8:
3015 if url.startswith(bom):
3016 url = url[len(bom):]
3017 url = url.lstrip()
3018 if not url or url.startswith(('#', ';', ']')):
62e609ab 3019 return False
8c04f0be 3020 # "#" cannot be stripped out since it is part of the URI
3021 # However, it can be safely stipped out if follwing a whitespace
3022 return re.split(r'\s#', url, 1)[0].rstrip()
62e609ab
PH
3023
3024 with contextlib.closing(batch_fd) as fd:
3025 return [url for url in map(fixup, fd) if url]
b74fa8cd
JMF
3026
3027
3028def urlencode_postdata(*args, **kargs):
15707c7e 3029 return compat_urllib_parse_urlencode(*args, **kargs).encode('ascii')
bcf89ce6
PH
3030
3031
38f9ef31 3032def update_url_query(url, query):
cacd9966
YCH
3033 if not query:
3034 return url
38f9ef31 3035 parsed_url = compat_urlparse.urlparse(url)
3036 qs = compat_parse_qs(parsed_url.query)
3037 qs.update(query)
3038 return compat_urlparse.urlunparse(parsed_url._replace(
15707c7e 3039 query=compat_urllib_parse_urlencode(qs, True)))
16392824 3040
8e60dc75 3041
ed0291d1
S
3042def update_Request(req, url=None, data=None, headers={}, query={}):
3043 req_headers = req.headers.copy()
3044 req_headers.update(headers)
3045 req_data = data or req.data
3046 req_url = update_url_query(url or req.get_full_url(), query)
95cf60e8
S
3047 req_get_method = req.get_method()
3048 if req_get_method == 'HEAD':
3049 req_type = HEADRequest
3050 elif req_get_method == 'PUT':
3051 req_type = PUTRequest
3052 else:
3053 req_type = compat_urllib_request.Request
ed0291d1
S
3054 new_req = req_type(
3055 req_url, data=req_data, headers=req_headers,
3056 origin_req_host=req.origin_req_host, unverifiable=req.unverifiable)
3057 if hasattr(req, 'timeout'):
3058 new_req.timeout = req.timeout
3059 return new_req
3060
3061
10c87c15 3062def _multipart_encode_impl(data, boundary):
0c265486
YCH
3063 content_type = 'multipart/form-data; boundary=%s' % boundary
3064
3065 out = b''
3066 for k, v in data.items():
3067 out += b'--' + boundary.encode('ascii') + b'\r\n'
3068 if isinstance(k, compat_str):
3069 k = k.encode('utf-8')
3070 if isinstance(v, compat_str):
3071 v = v.encode('utf-8')
3072 # RFC 2047 requires non-ASCII field names to be encoded, while RFC 7578
3073 # suggests sending UTF-8 directly. Firefox sends UTF-8, too
b2ad479d 3074 content = b'Content-Disposition: form-data; name="' + k + b'"\r\n\r\n' + v + b'\r\n'
0c265486
YCH
3075 if boundary.encode('ascii') in content:
3076 raise ValueError('Boundary overlaps with data')
3077 out += content
3078
3079 out += b'--' + boundary.encode('ascii') + b'--\r\n'
3080
3081 return out, content_type
3082
3083
3084def multipart_encode(data, boundary=None):
3085 '''
3086 Encode a dict to RFC 7578-compliant form-data
3087
3088 data:
3089 A dict where keys and values can be either Unicode or bytes-like
3090 objects.
3091 boundary:
3092 If specified a Unicode object, it's used as the boundary. Otherwise
3093 a random boundary is generated.
3094
3095 Reference: https://tools.ietf.org/html/rfc7578
3096 '''
3097 has_specified_boundary = boundary is not None
3098
3099 while True:
3100 if boundary is None:
3101 boundary = '---------------' + str(random.randrange(0x0fffffff, 0xffffffff))
3102
3103 try:
10c87c15 3104 out, content_type = _multipart_encode_impl(data, boundary)
0c265486
YCH
3105 break
3106 except ValueError:
3107 if has_specified_boundary:
3108 raise
3109 boundary = None
3110
3111 return out, content_type
3112
3113
86296ad2 3114def dict_get(d, key_or_keys, default=None, skip_false_values=True):
a44ca5a4 3115 for val in map(d.get, variadic(key_or_keys)):
3116 if val is not None and (val or not skip_false_values):
3117 return val
3118 return default
cbecc9b9
S
3119
3120
c4f60dd7 3121def try_call(*funcs, expected_type=None, args=[], kwargs={}):
3122 for f in funcs:
a32a9a7e 3123 try:
c4f60dd7 3124 val = f(*args, **kwargs)
3125 except (AttributeError, KeyError, TypeError, IndexError, ZeroDivisionError):
a32a9a7e
S
3126 pass
3127 else:
c4f60dd7 3128 if expected_type is None or isinstance(val, expected_type):
3129 return val
3130
3131
3132def try_get(src, getter, expected_type=None):
3133 return try_call(*variadic(getter), args=(src,), expected_type=expected_type)
329ca3be
S
3134
3135
90137ca4 3136def filter_dict(dct, cndn=lambda _, v: v is not None):
3137 return {k: v for k, v in dct.items() if cndn(k, v)}
3138
3139
6cc62232
S
3140def merge_dicts(*dicts):
3141 merged = {}
3142 for a_dict in dicts:
3143 for k, v in a_dict.items():
90137ca4 3144 if (v is not None and k not in merged
3145 or isinstance(v, str) and merged[k] == ''):
6cc62232
S
3146 merged[k] = v
3147 return merged
3148
3149
8e60dc75
S
3150def encode_compat_str(string, encoding=preferredencoding(), errors='strict'):
3151 return string if isinstance(string, compat_str) else compat_str(string, encoding, errors)
3152
16392824 3153
a1a530b0
PH
3154US_RATINGS = {
3155 'G': 0,
3156 'PG': 10,
3157 'PG-13': 13,
3158 'R': 16,
3159 'NC': 18,
3160}
fac55558
PH
3161
3162
a8795327 3163TV_PARENTAL_GUIDELINES = {
5a16c9d9
RA
3164 'TV-Y': 0,
3165 'TV-Y7': 7,
3166 'TV-G': 0,
3167 'TV-PG': 0,
3168 'TV-14': 14,
3169 'TV-MA': 17,
a8795327
S
3170}
3171
3172
146c80e2 3173def parse_age_limit(s):
a8795327
S
3174 if type(s) == int:
3175 return s if 0 <= s <= 21 else None
3176 if not isinstance(s, compat_basestring):
d838b1bd 3177 return None
146c80e2 3178 m = re.match(r'^(?P<age>\d{1,2})\+?$', s)
a8795327
S
3179 if m:
3180 return int(m.group('age'))
5c5fae6d 3181 s = s.upper()
a8795327
S
3182 if s in US_RATINGS:
3183 return US_RATINGS[s]
5a16c9d9 3184 m = re.match(r'^TV[_-]?(%s)$' % '|'.join(k[3:] for k in TV_PARENTAL_GUIDELINES), s)
b8361187 3185 if m:
5a16c9d9 3186 return TV_PARENTAL_GUIDELINES['TV-' + m.group(1)]
b8361187 3187 return None
146c80e2
S
3188
3189
fac55558 3190def strip_jsonp(code):
609a61e3 3191 return re.sub(
5552c9eb 3192 r'''(?sx)^
e9c671d5 3193 (?:window\.)?(?P<func_name>[a-zA-Z0-9_.$]*)
5552c9eb
YCH
3194 (?:\s*&&\s*(?P=func_name))?
3195 \s*\(\s*(?P<callback_data>.*)\);?
3196 \s*?(?://[^\n]*)*$''',
3197 r'\g<callback_data>', code)
478c2c61
PH
3198
3199
5c610515 3200def js_to_json(code, vars={}):
3201 # vars is a dict of var, val pairs to substitute
c843e685 3202 COMMENT_RE = r'/\*(?:(?!\*/).)*?\*/|//[^\n]*\n'
4195096e
S
3203 SKIP_RE = r'\s*(?:{comment})?\s*'.format(comment=COMMENT_RE)
3204 INTEGER_TABLE = (
3205 (r'(?s)^(0[xX][0-9a-fA-F]+){skip}:?$'.format(skip=SKIP_RE), 16),
3206 (r'(?s)^(0+[0-7]+){skip}:?$'.format(skip=SKIP_RE), 8),
3207 )
3208
e05f6939 3209 def fix_kv(m):
e7b6d122
PH
3210 v = m.group(0)
3211 if v in ('true', 'false', 'null'):
3212 return v
421ddcb8
C
3213 elif v in ('undefined', 'void 0'):
3214 return 'null'
8bdd16b4 3215 elif v.startswith('/*') or v.startswith('//') or v.startswith('!') or v == ',':
bd1e4844 3216 return ""
3217
3218 if v[0] in ("'", '"'):
3219 v = re.sub(r'(?s)\\.|"', lambda m: {
e7b6d122 3220 '"': '\\"',
bd1e4844 3221 "\\'": "'",
3222 '\\\n': '',
3223 '\\x': '\\u00',
3224 }.get(m.group(0), m.group(0)), v[1:-1])
8bdd16b4 3225 else:
3226 for regex, base in INTEGER_TABLE:
3227 im = re.match(regex, v)
3228 if im:
3229 i = int(im.group(1), base)
3230 return '"%d":' % i if v.endswith(':') else '%d' % i
89ac4a19 3231
5c610515 3232 if v in vars:
3233 return vars[v]
3234
e7b6d122 3235 return '"%s"' % v
e05f6939 3236
febff4c1
B
3237 code = re.sub(r'new Date\((".+")\)', r'\g<1>', code)
3238
bd1e4844 3239 return re.sub(r'''(?sx)
3240 "(?:[^"\\]*(?:\\\\|\\['"nurtbfx/\n]))*[^"\\]*"|
3241 '(?:[^'\\]*(?:\\\\|\\['"nurtbfx/\n]))*[^'\\]*'|
4195096e 3242 {comment}|,(?={skip}[\]}}])|
421ddcb8 3243 void\s0|(?:(?<![0-9])[eE]|[a-df-zA-DF-Z_$])[.a-zA-Z_$0-9]*|
4195096e 3244 \b(?:0[xX][0-9a-fA-F]+|0+[0-7]+)(?:{skip}:)?|
8bdd16b4 3245 [0-9]+(?={skip}:)|
3246 !+
4195096e 3247 '''.format(comment=COMMENT_RE, skip=SKIP_RE), fix_kv, code)
e05f6939
PH
3248
3249
478c2c61
PH
3250def qualities(quality_ids):
3251 """ Get a numeric quality value out of a list of possible values """
3252 def q(qid):
3253 try:
3254 return quality_ids.index(qid)
3255 except ValueError:
3256 return -1
3257 return q
3258
acd69589 3259
09b49e1f 3260POSTPROCESS_WHEN = {'pre_process', 'after_filter', 'before_dl', 'after_move', 'post_process', 'after_video', 'playlist'}
1e43a6f7 3261
3262
de6000d9 3263DEFAULT_OUTTMPL = {
3264 'default': '%(title)s [%(id)s].%(ext)s',
72755351 3265 'chapter': '%(title)s - %(section_number)03d %(section_title)s [%(id)s].%(ext)s',
de6000d9 3266}
3267OUTTMPL_TYPES = {
72755351 3268 'chapter': None,
de6000d9 3269 'subtitle': None,
3270 'thumbnail': None,
3271 'description': 'description',
3272 'annotation': 'annotations.xml',
3273 'infojson': 'info.json',
08438d2c 3274 'link': None,
3b603dbd 3275 'pl_video': None,
5112f26a 3276 'pl_thumbnail': None,
de6000d9 3277 'pl_description': 'description',
3278 'pl_infojson': 'info.json',
3279}
0a871f68 3280
143db31d 3281# As of [1] format syntax is:
3282# %[mapping_key][conversion_flags][minimum_width][.precision][length_modifier]type
3283# 1. https://docs.python.org/2/library/stdtypes.html#string-formatting
901130bb 3284STR_FORMAT_RE_TMPL = r'''(?x)
3285 (?<!%)(?P<prefix>(?:%%)*)
143db31d 3286 %
524e2e4f 3287 (?P<has_key>\((?P<key>{0})\))?
752cda38 3288 (?P<format>
524e2e4f 3289 (?P<conversion>[#0\-+ ]+)?
3290 (?P<min_width>\d+)?
3291 (?P<precision>\.\d+)?
3292 (?P<len_mod>[hlL])? # unused in python
901130bb 3293 {1} # conversion type
752cda38 3294 )
143db31d 3295'''
3296
7d1eb38a 3297
901130bb 3298STR_FORMAT_TYPES = 'diouxXeEfFgGcrs'
a020a0dc 3299
7d1eb38a 3300
a020a0dc
PH
3301def limit_length(s, length):
3302 """ Add ellipses to overly long strings """
3303 if s is None:
3304 return None
3305 ELLIPSES = '...'
3306 if len(s) > length:
3307 return s[:length - len(ELLIPSES)] + ELLIPSES
3308 return s
48844745
PH
3309
3310
3311def version_tuple(v):
5f9b8394 3312 return tuple(int(e) for e in re.split(r'[-.]', v))
48844745
PH
3313
3314
3315def is_outdated_version(version, limit, assume_new=True):
3316 if not version:
3317 return not assume_new
3318 try:
3319 return version_tuple(version) < version_tuple(limit)
3320 except ValueError:
3321 return not assume_new
732ea2f0
PH
3322
3323
3324def ytdl_is_updateable():
7a5c1cfe 3325 """ Returns if yt-dlp can be updated with -U """
735d865e 3326
5d535b4a 3327 from .update import is_non_updateable
732ea2f0 3328
5d535b4a 3329 return not is_non_updateable()
7d4111ed
PH
3330
3331
3332def args_to_str(args):
3333 # Get a short string representation for a subprocess command
702ccf2d 3334 return ' '.join(compat_shlex_quote(a) for a in args)
2ccd1b10
PH
3335
3336
9b9c5355 3337def error_to_compat_str(err):
fdae2358
S
3338 err_str = str(err)
3339 # On python 2 error byte string must be decoded with proper
3340 # encoding rather than ascii
3341 if sys.version_info[0] < 3:
3342 err_str = err_str.decode(preferredencoding())
3343 return err_str
3344
3345
a44ca5a4 3346def error_to_str(err):
3347 return f'{type(err).__name__}: {err}'
3348
3349
c460bdd5 3350def mimetype2ext(mt):
eb9ee194
S
3351 if mt is None:
3352 return None
3353
9359f3d4
F
3354 mt, _, params = mt.partition(';')
3355 mt = mt.strip()
3356
3357 FULL_MAP = {
765ac263 3358 'audio/mp4': 'm4a',
6c33d24b
YCH
3359 # Per RFC 3003, audio/mpeg can be .mp1, .mp2 or .mp3. Here use .mp3 as
3360 # it's the most popular one
3361 'audio/mpeg': 'mp3',
ba39289d 3362 'audio/x-wav': 'wav',
9359f3d4
F
3363 'audio/wav': 'wav',
3364 'audio/wave': 'wav',
3365 }
3366
3367 ext = FULL_MAP.get(mt)
765ac263
JMF
3368 if ext is not None:
3369 return ext
3370
9359f3d4 3371 SUBTYPE_MAP = {
f6861ec9 3372 '3gpp': '3gp',
cafcf657 3373 'smptett+xml': 'tt',
cafcf657 3374 'ttaf+xml': 'dfxp',
a0d8d704 3375 'ttml+xml': 'ttml',
f6861ec9 3376 'x-flv': 'flv',
a0d8d704 3377 'x-mp4-fragmented': 'mp4',
d4f05d47 3378 'x-ms-sami': 'sami',
a0d8d704 3379 'x-ms-wmv': 'wmv',
b4173f15
RA
3380 'mpegurl': 'm3u8',
3381 'x-mpegurl': 'm3u8',
3382 'vnd.apple.mpegurl': 'm3u8',
3383 'dash+xml': 'mpd',
b4173f15 3384 'f4m+xml': 'f4m',
f164b971 3385 'hds+xml': 'f4m',
e910fe2f 3386 'vnd.ms-sstr+xml': 'ism',
c2b2c7e1 3387 'quicktime': 'mov',
98ce1a3f 3388 'mp2t': 'ts',
39e7107d 3389 'x-wav': 'wav',
9359f3d4
F
3390 'filmstrip+json': 'fs',
3391 'svg+xml': 'svg',
3392 }
3393
3394 _, _, subtype = mt.rpartition('/')
3395 ext = SUBTYPE_MAP.get(subtype.lower())
3396 if ext is not None:
3397 return ext
3398
3399 SUFFIX_MAP = {
3400 'json': 'json',
3401 'xml': 'xml',
3402 'zip': 'zip',
3403 'gzip': 'gz',
3404 }
3405
3406 _, _, suffix = subtype.partition('+')
3407 ext = SUFFIX_MAP.get(suffix)
3408 if ext is not None:
3409 return ext
3410
3411 return subtype.replace('+', '.')
c460bdd5
PH
3412
3413
2814f12b
THD
3414def ext2mimetype(ext_or_url):
3415 if not ext_or_url:
3416 return None
3417 if '.' not in ext_or_url:
3418 ext_or_url = f'file.{ext_or_url}'
3419 return mimetypes.guess_type(ext_or_url)[0]
3420
3421
4f3c5e06 3422def parse_codecs(codecs_str):
3423 # http://tools.ietf.org/html/rfc6381
3424 if not codecs_str:
3425 return {}
a0566bbf 3426 split_codecs = list(filter(None, map(
dbf5416a 3427 str.strip, codecs_str.strip().strip(',').split(','))))
4afa3ec4 3428 vcodec, acodec, tcodec, hdr = None, None, None, None
a0566bbf 3429 for full_codec in split_codecs:
9bd979ca 3430 parts = full_codec.split('.')
3431 codec = parts[0].replace('0', '')
3432 if codec in ('avc1', 'avc2', 'avc3', 'avc4', 'vp9', 'vp8', 'hev1', 'hev2',
3433 'h263', 'h264', 'mp4v', 'hvc1', 'av1', 'theora', 'dvh1', 'dvhe'):
4f3c5e06 3434 if not vcodec:
b69fd25c 3435 vcodec = '.'.join(parts[:4]) if codec in ('vp9', 'av1', 'hvc1') else full_codec
176f1866 3436 if codec in ('dvh1', 'dvhe'):
3437 hdr = 'DV'
9bd979ca 3438 elif codec == 'av1' and len(parts) > 3 and parts[3] == '10':
3439 hdr = 'HDR10'
3440 elif full_codec.replace('0', '').startswith('vp9.2'):
176f1866 3441 hdr = 'HDR10'
b69fd25c 3442 elif codec in ('flac', 'mp4a', 'opus', 'vorbis', 'mp3', 'aac', 'ac-3', 'ec-3', 'eac3', 'dtsc', 'dtse', 'dtsh', 'dtsl'):
4f3c5e06 3443 if not acodec:
3444 acodec = full_codec
4afa3ec4
F
3445 elif codec in ('stpp', 'wvtt',):
3446 if not tcodec:
3447 tcodec = full_codec
4f3c5e06 3448 else:
60f5c9fb 3449 write_string('WARNING: Unknown codec %s\n' % full_codec, sys.stderr)
4afa3ec4 3450 if vcodec or acodec or tcodec:
4f3c5e06 3451 return {
3452 'vcodec': vcodec or 'none',
3453 'acodec': acodec or 'none',
176f1866 3454 'dynamic_range': hdr,
4afa3ec4 3455 **({'tcodec': tcodec} if tcodec is not None else {}),
4f3c5e06 3456 }
b69fd25c 3457 elif len(split_codecs) == 2:
3458 return {
3459 'vcodec': split_codecs[0],
3460 'acodec': split_codecs[1],
3461 }
4f3c5e06 3462 return {}
3463
3464
2ccd1b10 3465def urlhandle_detect_ext(url_handle):
79298173 3466 getheader = url_handle.headers.get
2ccd1b10 3467
b55ee18f
PH
3468 cd = getheader('Content-Disposition')
3469 if cd:
3470 m = re.match(r'attachment;\s*filename="(?P<filename>[^"]+)"', cd)
3471 if m:
3472 e = determine_ext(m.group('filename'), default_ext=None)
3473 if e:
3474 return e
3475
c460bdd5 3476 return mimetype2ext(getheader('Content-Type'))
05900629
PH
3477
3478
1e399778
YCH
3479def encode_data_uri(data, mime_type):
3480 return 'data:%s;base64,%s' % (mime_type, base64.b64encode(data).decode('ascii'))
3481
3482
05900629 3483def age_restricted(content_limit, age_limit):
6ec6cb4e 3484 """ Returns True iff the content should be blocked """
05900629
PH
3485
3486 if age_limit is None: # No limit set
3487 return False
3488 if content_limit is None:
3489 return False # Content available for everyone
3490 return age_limit < content_limit
61ca9a80
PH
3491
3492
3493def is_html(first_bytes):
3494 """ Detect whether a file contains HTML by examining its first bytes. """
3495
3496 BOMS = [
3497 (b'\xef\xbb\xbf', 'utf-8'),
3498 (b'\x00\x00\xfe\xff', 'utf-32-be'),
3499 (b'\xff\xfe\x00\x00', 'utf-32-le'),
3500 (b'\xff\xfe', 'utf-16-le'),
3501 (b'\xfe\xff', 'utf-16-be'),
3502 ]
3503 for bom, enc in BOMS:
3504 if first_bytes.startswith(bom):
3505 s = first_bytes[len(bom):].decode(enc, 'replace')
3506 break
3507 else:
3508 s = first_bytes.decode('utf-8', 'replace')
3509
3510 return re.match(r'^\s*<', s)
a055469f
PH
3511
3512
3513def determine_protocol(info_dict):
3514 protocol = info_dict.get('protocol')
3515 if protocol is not None:
3516 return protocol
3517
7de837a5 3518 url = sanitize_url(info_dict['url'])
a055469f
PH
3519 if url.startswith('rtmp'):
3520 return 'rtmp'
3521 elif url.startswith('mms'):
3522 return 'mms'
3523 elif url.startswith('rtsp'):
3524 return 'rtsp'
3525
3526 ext = determine_ext(url)
3527 if ext == 'm3u8':
3528 return 'm3u8'
3529 elif ext == 'f4m':
3530 return 'f4m'
3531
3532 return compat_urllib_parse_urlparse(url).scheme
cfb56d1a
PH
3533
3534
c5e3f849 3535def render_table(header_row, data, delim=False, extra_gap=0, hide_empty=False):
3536 """ Render a list of rows, each as a list of values.
3537 Text after a \t will be right aligned """
ec11a9f4 3538 def width(string):
c5e3f849 3539 return len(remove_terminal_sequences(string).replace('\t', ''))
76d321f6 3540
3541 def get_max_lens(table):
ec11a9f4 3542 return [max(width(str(v)) for v in col) for col in zip(*table)]
76d321f6 3543
3544 def filter_using_list(row, filterArray):
d16df59d 3545 return [col for take, col in itertools.zip_longest(filterArray, row, fillvalue=True) if take]
76d321f6 3546
d16df59d 3547 max_lens = get_max_lens(data) if hide_empty else []
3548 header_row = filter_using_list(header_row, max_lens)
3549 data = [filter_using_list(row, max_lens) for row in data]
76d321f6 3550
cfb56d1a 3551 table = [header_row] + data
76d321f6 3552 max_lens = get_max_lens(table)
c5e3f849 3553 extra_gap += 1
76d321f6 3554 if delim:
c5e3f849 3555 table = [header_row, [delim * (ml + extra_gap) for ml in max_lens]] + data
1ed7953a 3556 table[1][-1] = table[1][-1][:-extra_gap * len(delim)] # Remove extra_gap from end of delimiter
ec11a9f4 3557 for row in table:
3558 for pos, text in enumerate(map(str, row)):
c5e3f849 3559 if '\t' in text:
3560 row[pos] = text.replace('\t', ' ' * (max_lens[pos] - width(text))) + ' ' * extra_gap
3561 else:
3562 row[pos] = text + ' ' * (max_lens[pos] - width(text) + extra_gap)
3563 ret = '\n'.join(''.join(row).rstrip() for row in table)
ec11a9f4 3564 return ret
347de493
PH
3565
3566
8f18aca8 3567def _match_one(filter_part, dct, incomplete):
77b87f05 3568 # TODO: Generalize code with YoutubeDL._build_format_filter
a047eeb6 3569 STRING_OPERATORS = {
3570 '*=': operator.contains,
3571 '^=': lambda attr, value: attr.startswith(value),
3572 '$=': lambda attr, value: attr.endswith(value),
3573 '~=': lambda attr, value: re.search(value, attr),
3574 }
347de493 3575 COMPARISON_OPERATORS = {
a047eeb6 3576 **STRING_OPERATORS,
3577 '<=': operator.le, # "<=" must be defined above "<"
347de493 3578 '<': operator.lt,
347de493 3579 '>=': operator.ge,
a047eeb6 3580 '>': operator.gt,
347de493 3581 '=': operator.eq,
347de493 3582 }
a047eeb6 3583
6db9c4d5 3584 if isinstance(incomplete, bool):
3585 is_incomplete = lambda _: incomplete
3586 else:
3587 is_incomplete = lambda k: k in incomplete
3588
347de493
PH
3589 operator_rex = re.compile(r'''(?x)\s*
3590 (?P<key>[a-z_]+)
77b87f05 3591 \s*(?P<negation>!\s*)?(?P<op>%s)(?P<none_inclusive>\s*\?)?\s*
347de493 3592 (?:
a047eeb6 3593 (?P<quote>["\'])(?P<quotedstrval>.+?)(?P=quote)|
3594 (?P<strval>.+?)
347de493
PH
3595 )
3596 \s*$
3597 ''' % '|'.join(map(re.escape, COMPARISON_OPERATORS.keys())))
3598 m = operator_rex.search(filter_part)
3599 if m:
18f96d12 3600 m = m.groupdict()
3601 unnegated_op = COMPARISON_OPERATORS[m['op']]
3602 if m['negation']:
77b87f05
MT
3603 op = lambda attr, value: not unnegated_op(attr, value)
3604 else:
3605 op = unnegated_op
18f96d12 3606 comparison_value = m['quotedstrval'] or m['strval'] or m['intval']
3607 if m['quote']:
3608 comparison_value = comparison_value.replace(r'\%s' % m['quote'], m['quote'])
3609 actual_value = dct.get(m['key'])
3610 numeric_comparison = None
3611 if isinstance(actual_value, compat_numeric_types):
e5a088dc
S
3612 # If the original field is a string and matching comparisonvalue is
3613 # a number we should respect the origin of the original field
3614 # and process comparison value as a string (see
18f96d12 3615 # https://github.com/ytdl-org/youtube-dl/issues/11082)
347de493 3616 try:
18f96d12 3617 numeric_comparison = int(comparison_value)
347de493 3618 except ValueError:
18f96d12 3619 numeric_comparison = parse_filesize(comparison_value)
3620 if numeric_comparison is None:
3621 numeric_comparison = parse_filesize(f'{comparison_value}B')
3622 if numeric_comparison is None:
3623 numeric_comparison = parse_duration(comparison_value)
3624 if numeric_comparison is not None and m['op'] in STRING_OPERATORS:
3625 raise ValueError('Operator %s only supports string values!' % m['op'])
347de493 3626 if actual_value is None:
6db9c4d5 3627 return is_incomplete(m['key']) or m['none_inclusive']
18f96d12 3628 return op(actual_value, comparison_value if numeric_comparison is None else numeric_comparison)
347de493
PH
3629
3630 UNARY_OPERATORS = {
1cc47c66
S
3631 '': lambda v: (v is True) if isinstance(v, bool) else (v is not None),
3632 '!': lambda v: (v is False) if isinstance(v, bool) else (v is None),
347de493
PH
3633 }
3634 operator_rex = re.compile(r'''(?x)\s*
3635 (?P<op>%s)\s*(?P<key>[a-z_]+)
3636 \s*$
3637 ''' % '|'.join(map(re.escape, UNARY_OPERATORS.keys())))
3638 m = operator_rex.search(filter_part)
3639 if m:
3640 op = UNARY_OPERATORS[m.group('op')]
3641 actual_value = dct.get(m.group('key'))
6db9c4d5 3642 if is_incomplete(m.group('key')) and actual_value is None:
8f18aca8 3643 return True
347de493
PH
3644 return op(actual_value)
3645
3646 raise ValueError('Invalid filter part %r' % filter_part)
3647
3648
8f18aca8 3649def match_str(filter_str, dct, incomplete=False):
6db9c4d5 3650 """ Filter a dictionary with a simple string syntax.
3651 @returns Whether the filter passes
3652 @param incomplete Set of keys that is expected to be missing from dct.
3653 Can be True/False to indicate all/none of the keys may be missing.
3654 All conditions on incomplete keys pass if the key is missing
8f18aca8 3655 """
347de493 3656 return all(
8f18aca8 3657 _match_one(filter_part.replace(r'\&', '&'), dct, incomplete)
a047eeb6 3658 for filter_part in re.split(r'(?<!\\)&', filter_str))
347de493
PH
3659
3660
b1a7cd05 3661def match_filter_func(filters):
3662 if not filters:
d1b5f70b 3663 return None
b1a7cd05 3664 filters = variadic(filters)
d1b5f70b 3665
8f18aca8 3666 def _match_func(info_dict, *args, **kwargs):
b1a7cd05 3667 if any(match_str(f, info_dict, *args, **kwargs) for f in filters):
347de493
PH
3668 return None
3669 else:
b1a7cd05 3670 video_title = info_dict.get('title') or info_dict.get('id') or 'video'
3671 filter_str = ') | ('.join(map(str.strip, filters))
3672 return f'{video_title} does not pass filter ({filter_str}), skipping ..'
347de493 3673 return _match_func
91410c9b
PH
3674
3675
bf6427d2
YCH
3676def parse_dfxp_time_expr(time_expr):
3677 if not time_expr:
d631d5f9 3678 return
bf6427d2
YCH
3679
3680 mobj = re.match(r'^(?P<time_offset>\d+(?:\.\d+)?)s?$', time_expr)
3681 if mobj:
3682 return float(mobj.group('time_offset'))
3683
db2fe38b 3684 mobj = re.match(r'^(\d+):(\d\d):(\d\d(?:(?:\.|:)\d+)?)$', time_expr)
bf6427d2 3685 if mobj:
db2fe38b 3686 return 3600 * int(mobj.group(1)) + 60 * int(mobj.group(2)) + float(mobj.group(3).replace(':', '.'))
bf6427d2
YCH
3687
3688
c1c924ab 3689def srt_subtitles_timecode(seconds):
aa7785f8 3690 return '%02d:%02d:%02d,%03d' % timetuple_from_msec(seconds * 1000)
3691
3692
3693def ass_subtitles_timecode(seconds):
3694 time = timetuple_from_msec(seconds * 1000)
3695 return '%01d:%02d:%02d.%02d' % (*time[:-1], time.milliseconds / 10)
bf6427d2
YCH
3696
3697
3698def dfxp2srt(dfxp_data):
3869028f
YCH
3699 '''
3700 @param dfxp_data A bytes-like object containing DFXP data
3701 @returns A unicode object containing converted SRT data
3702 '''
5b995f71 3703 LEGACY_NAMESPACES = (
3869028f
YCH
3704 (b'http://www.w3.org/ns/ttml', [
3705 b'http://www.w3.org/2004/11/ttaf1',
3706 b'http://www.w3.org/2006/04/ttaf1',
3707 b'http://www.w3.org/2006/10/ttaf1',
5b995f71 3708 ]),
3869028f
YCH
3709 (b'http://www.w3.org/ns/ttml#styling', [
3710 b'http://www.w3.org/ns/ttml#style',
5b995f71
RA
3711 ]),
3712 )
3713
3714 SUPPORTED_STYLING = [
3715 'color',
3716 'fontFamily',
3717 'fontSize',
3718 'fontStyle',
3719 'fontWeight',
3720 'textDecoration'
3721 ]
3722
4e335771 3723 _x = functools.partial(xpath_with_ns, ns_map={
261f4730 3724 'xml': 'http://www.w3.org/XML/1998/namespace',
4e335771 3725 'ttml': 'http://www.w3.org/ns/ttml',
5b995f71 3726 'tts': 'http://www.w3.org/ns/ttml#styling',
4e335771 3727 })
bf6427d2 3728
5b995f71
RA
3729 styles = {}
3730 default_style = {}
3731
87de7069 3732 class TTMLPElementParser(object):
5b995f71
RA
3733 _out = ''
3734 _unclosed_elements = []
3735 _applied_styles = []
bf6427d2 3736
2b14cb56 3737 def start(self, tag, attrib):
5b995f71
RA
3738 if tag in (_x('ttml:br'), 'br'):
3739 self._out += '\n'
3740 else:
3741 unclosed_elements = []
3742 style = {}
3743 element_style_id = attrib.get('style')
3744 if default_style:
3745 style.update(default_style)
3746 if element_style_id:
3747 style.update(styles.get(element_style_id, {}))
3748 for prop in SUPPORTED_STYLING:
3749 prop_val = attrib.get(_x('tts:' + prop))
3750 if prop_val:
3751 style[prop] = prop_val
3752 if style:
3753 font = ''
3754 for k, v in sorted(style.items()):
3755 if self._applied_styles and self._applied_styles[-1].get(k) == v:
3756 continue
3757 if k == 'color':
3758 font += ' color="%s"' % v
3759 elif k == 'fontSize':
3760 font += ' size="%s"' % v
3761 elif k == 'fontFamily':
3762 font += ' face="%s"' % v
3763 elif k == 'fontWeight' and v == 'bold':
3764 self._out += '<b>'
3765 unclosed_elements.append('b')
3766 elif k == 'fontStyle' and v == 'italic':
3767 self._out += '<i>'
3768 unclosed_elements.append('i')
3769 elif k == 'textDecoration' and v == 'underline':
3770 self._out += '<u>'
3771 unclosed_elements.append('u')
3772 if font:
3773 self._out += '<font' + font + '>'
3774 unclosed_elements.append('font')
3775 applied_style = {}
3776 if self._applied_styles:
3777 applied_style.update(self._applied_styles[-1])
3778 applied_style.update(style)
3779 self._applied_styles.append(applied_style)
3780 self._unclosed_elements.append(unclosed_elements)
bf6427d2 3781
2b14cb56 3782 def end(self, tag):
5b995f71
RA
3783 if tag not in (_x('ttml:br'), 'br'):
3784 unclosed_elements = self._unclosed_elements.pop()
3785 for element in reversed(unclosed_elements):
3786 self._out += '</%s>' % element
3787 if unclosed_elements and self._applied_styles:
3788 self._applied_styles.pop()
bf6427d2 3789
2b14cb56 3790 def data(self, data):
5b995f71 3791 self._out += data
2b14cb56 3792
3793 def close(self):
5b995f71 3794 return self._out.strip()
2b14cb56 3795
3796 def parse_node(node):
3797 target = TTMLPElementParser()
3798 parser = xml.etree.ElementTree.XMLParser(target=target)
3799 parser.feed(xml.etree.ElementTree.tostring(node))
3800 return parser.close()
bf6427d2 3801
5b995f71
RA
3802 for k, v in LEGACY_NAMESPACES:
3803 for ns in v:
3804 dfxp_data = dfxp_data.replace(ns, k)
3805
3869028f 3806 dfxp = compat_etree_fromstring(dfxp_data)
bf6427d2 3807 out = []
5b995f71 3808 paras = dfxp.findall(_x('.//ttml:p')) or dfxp.findall('.//p')
1b0427e6
YCH
3809
3810 if not paras:
3811 raise ValueError('Invalid dfxp/TTML subtitle')
bf6427d2 3812
5b995f71
RA
3813 repeat = False
3814 while True:
3815 for style in dfxp.findall(_x('.//ttml:style')):
261f4730
RA
3816 style_id = style.get('id') or style.get(_x('xml:id'))
3817 if not style_id:
3818 continue
5b995f71
RA
3819 parent_style_id = style.get('style')
3820 if parent_style_id:
3821 if parent_style_id not in styles:
3822 repeat = True
3823 continue
3824 styles[style_id] = styles[parent_style_id].copy()
3825 for prop in SUPPORTED_STYLING:
3826 prop_val = style.get(_x('tts:' + prop))
3827 if prop_val:
3828 styles.setdefault(style_id, {})[prop] = prop_val
3829 if repeat:
3830 repeat = False
3831 else:
3832 break
3833
3834 for p in ('body', 'div'):
3835 ele = xpath_element(dfxp, [_x('.//ttml:' + p), './/' + p])
3836 if ele is None:
3837 continue
3838 style = styles.get(ele.get('style'))
3839 if not style:
3840 continue
3841 default_style.update(style)
3842
bf6427d2 3843 for para, index in zip(paras, itertools.count(1)):
d631d5f9 3844 begin_time = parse_dfxp_time_expr(para.attrib.get('begin'))
7dff0363 3845 end_time = parse_dfxp_time_expr(para.attrib.get('end'))
d631d5f9
YCH
3846 dur = parse_dfxp_time_expr(para.attrib.get('dur'))
3847 if begin_time is None:
3848 continue
7dff0363 3849 if not end_time:
d631d5f9
YCH
3850 if not dur:
3851 continue
3852 end_time = begin_time + dur
bf6427d2
YCH
3853 out.append('%d\n%s --> %s\n%s\n\n' % (
3854 index,
c1c924ab
YCH
3855 srt_subtitles_timecode(begin_time),
3856 srt_subtitles_timecode(end_time),
bf6427d2
YCH
3857 parse_node(para)))
3858
3859 return ''.join(out)
3860
3861
66e289ba
S
3862def cli_option(params, command_option, param):
3863 param = params.get(param)
98e698f1
RA
3864 if param:
3865 param = compat_str(param)
66e289ba
S
3866 return [command_option, param] if param is not None else []
3867
3868
3869def cli_bool_option(params, command_option, param, true_value='true', false_value='false', separator=None):
3870 param = params.get(param)
5b232f46
S
3871 if param is None:
3872 return []
66e289ba
S
3873 assert isinstance(param, bool)
3874 if separator:
3875 return [command_option + separator + (true_value if param else false_value)]
3876 return [command_option, true_value if param else false_value]
3877
3878
3879def cli_valueless_option(params, command_option, param, expected_value=True):
3880 param = params.get(param)
3881 return [command_option] if param == expected_value else []
3882
3883
e92caff5 3884def cli_configuration_args(argdict, keys, default=[], use_compat=True):
eab9b2bc 3885 if isinstance(argdict, (list, tuple)): # for backward compatibility
e92caff5 3886 if use_compat:
5b1ecbb3 3887 return argdict
3888 else:
3889 argdict = None
eab9b2bc 3890 if argdict is None:
5b1ecbb3 3891 return default
eab9b2bc 3892 assert isinstance(argdict, dict)
3893
e92caff5 3894 assert isinstance(keys, (list, tuple))
3895 for key_list in keys:
e92caff5 3896 arg_list = list(filter(
3897 lambda x: x is not None,
6606817a 3898 [argdict.get(key.lower()) for key in variadic(key_list)]))
e92caff5 3899 if arg_list:
3900 return [arg for args in arg_list for arg in args]
3901 return default
66e289ba 3902
6251555f 3903
330690a2 3904def _configuration_args(main_key, argdict, exe, keys=None, default=[], use_compat=True):
3905 main_key, exe = main_key.lower(), exe.lower()
3906 root_key = exe if main_key == exe else f'{main_key}+{exe}'
3907 keys = [f'{root_key}{k}' for k in (keys or [''])]
3908 if root_key in keys:
3909 if main_key != exe:
3910 keys.append((main_key, exe))
3911 keys.append('default')
3912 else:
3913 use_compat = False
3914 return cli_configuration_args(argdict, keys, default, use_compat)
3915
66e289ba 3916
39672624
YCH
3917class ISO639Utils(object):
3918 # See http://www.loc.gov/standards/iso639-2/ISO-639-2_utf-8.txt
3919 _lang_map = {
3920 'aa': 'aar',
3921 'ab': 'abk',
3922 'ae': 'ave',
3923 'af': 'afr',
3924 'ak': 'aka',
3925 'am': 'amh',
3926 'an': 'arg',
3927 'ar': 'ara',
3928 'as': 'asm',
3929 'av': 'ava',
3930 'ay': 'aym',
3931 'az': 'aze',
3932 'ba': 'bak',
3933 'be': 'bel',
3934 'bg': 'bul',
3935 'bh': 'bih',
3936 'bi': 'bis',
3937 'bm': 'bam',
3938 'bn': 'ben',
3939 'bo': 'bod',
3940 'br': 'bre',
3941 'bs': 'bos',
3942 'ca': 'cat',
3943 'ce': 'che',
3944 'ch': 'cha',
3945 'co': 'cos',
3946 'cr': 'cre',
3947 'cs': 'ces',
3948 'cu': 'chu',
3949 'cv': 'chv',
3950 'cy': 'cym',
3951 'da': 'dan',
3952 'de': 'deu',
3953 'dv': 'div',
3954 'dz': 'dzo',
3955 'ee': 'ewe',
3956 'el': 'ell',
3957 'en': 'eng',
3958 'eo': 'epo',
3959 'es': 'spa',
3960 'et': 'est',
3961 'eu': 'eus',
3962 'fa': 'fas',
3963 'ff': 'ful',
3964 'fi': 'fin',
3965 'fj': 'fij',
3966 'fo': 'fao',
3967 'fr': 'fra',
3968 'fy': 'fry',
3969 'ga': 'gle',
3970 'gd': 'gla',
3971 'gl': 'glg',
3972 'gn': 'grn',
3973 'gu': 'guj',
3974 'gv': 'glv',
3975 'ha': 'hau',
3976 'he': 'heb',
b7acc835 3977 'iw': 'heb', # Replaced by he in 1989 revision
39672624
YCH
3978 'hi': 'hin',
3979 'ho': 'hmo',
3980 'hr': 'hrv',
3981 'ht': 'hat',
3982 'hu': 'hun',
3983 'hy': 'hye',
3984 'hz': 'her',
3985 'ia': 'ina',
3986 'id': 'ind',
b7acc835 3987 'in': 'ind', # Replaced by id in 1989 revision
39672624
YCH
3988 'ie': 'ile',
3989 'ig': 'ibo',
3990 'ii': 'iii',
3991 'ik': 'ipk',
3992 'io': 'ido',
3993 'is': 'isl',
3994 'it': 'ita',
3995 'iu': 'iku',
3996 'ja': 'jpn',
3997 'jv': 'jav',
3998 'ka': 'kat',
3999 'kg': 'kon',
4000 'ki': 'kik',
4001 'kj': 'kua',
4002 'kk': 'kaz',
4003 'kl': 'kal',
4004 'km': 'khm',
4005 'kn': 'kan',
4006 'ko': 'kor',
4007 'kr': 'kau',
4008 'ks': 'kas',
4009 'ku': 'kur',
4010 'kv': 'kom',
4011 'kw': 'cor',
4012 'ky': 'kir',
4013 'la': 'lat',
4014 'lb': 'ltz',
4015 'lg': 'lug',
4016 'li': 'lim',
4017 'ln': 'lin',
4018 'lo': 'lao',
4019 'lt': 'lit',
4020 'lu': 'lub',
4021 'lv': 'lav',
4022 'mg': 'mlg',
4023 'mh': 'mah',
4024 'mi': 'mri',
4025 'mk': 'mkd',
4026 'ml': 'mal',
4027 'mn': 'mon',
4028 'mr': 'mar',
4029 'ms': 'msa',
4030 'mt': 'mlt',
4031 'my': 'mya',
4032 'na': 'nau',
4033 'nb': 'nob',
4034 'nd': 'nde',
4035 'ne': 'nep',
4036 'ng': 'ndo',
4037 'nl': 'nld',
4038 'nn': 'nno',
4039 'no': 'nor',
4040 'nr': 'nbl',
4041 'nv': 'nav',
4042 'ny': 'nya',
4043 'oc': 'oci',
4044 'oj': 'oji',
4045 'om': 'orm',
4046 'or': 'ori',
4047 'os': 'oss',
4048 'pa': 'pan',
4049 'pi': 'pli',
4050 'pl': 'pol',
4051 'ps': 'pus',
4052 'pt': 'por',
4053 'qu': 'que',
4054 'rm': 'roh',
4055 'rn': 'run',
4056 'ro': 'ron',
4057 'ru': 'rus',
4058 'rw': 'kin',
4059 'sa': 'san',
4060 'sc': 'srd',
4061 'sd': 'snd',
4062 'se': 'sme',
4063 'sg': 'sag',
4064 'si': 'sin',
4065 'sk': 'slk',
4066 'sl': 'slv',
4067 'sm': 'smo',
4068 'sn': 'sna',
4069 'so': 'som',
4070 'sq': 'sqi',
4071 'sr': 'srp',
4072 'ss': 'ssw',
4073 'st': 'sot',
4074 'su': 'sun',
4075 'sv': 'swe',
4076 'sw': 'swa',
4077 'ta': 'tam',
4078 'te': 'tel',
4079 'tg': 'tgk',
4080 'th': 'tha',
4081 'ti': 'tir',
4082 'tk': 'tuk',
4083 'tl': 'tgl',
4084 'tn': 'tsn',
4085 'to': 'ton',
4086 'tr': 'tur',
4087 'ts': 'tso',
4088 'tt': 'tat',
4089 'tw': 'twi',
4090 'ty': 'tah',
4091 'ug': 'uig',
4092 'uk': 'ukr',
4093 'ur': 'urd',
4094 'uz': 'uzb',
4095 've': 'ven',
4096 'vi': 'vie',
4097 'vo': 'vol',
4098 'wa': 'wln',
4099 'wo': 'wol',
4100 'xh': 'xho',
4101 'yi': 'yid',
e9a50fba 4102 'ji': 'yid', # Replaced by yi in 1989 revision
39672624
YCH
4103 'yo': 'yor',
4104 'za': 'zha',
4105 'zh': 'zho',
4106 'zu': 'zul',
4107 }
4108
4109 @classmethod
4110 def short2long(cls, code):
4111 """Convert language code from ISO 639-1 to ISO 639-2/T"""
4112 return cls._lang_map.get(code[:2])
4113
4114 @classmethod
4115 def long2short(cls, code):
4116 """Convert language code from ISO 639-2/T to ISO 639-1"""
4117 for short_name, long_name in cls._lang_map.items():
4118 if long_name == code:
4119 return short_name
4120
4121
4eb10f66
YCH
4122class ISO3166Utils(object):
4123 # From http://data.okfn.org/data/core/country-list
4124 _country_map = {
4125 'AF': 'Afghanistan',
4126 'AX': 'Åland Islands',
4127 'AL': 'Albania',
4128 'DZ': 'Algeria',
4129 'AS': 'American Samoa',
4130 'AD': 'Andorra',
4131 'AO': 'Angola',
4132 'AI': 'Anguilla',
4133 'AQ': 'Antarctica',
4134 'AG': 'Antigua and Barbuda',
4135 'AR': 'Argentina',
4136 'AM': 'Armenia',
4137 'AW': 'Aruba',
4138 'AU': 'Australia',
4139 'AT': 'Austria',
4140 'AZ': 'Azerbaijan',
4141 'BS': 'Bahamas',
4142 'BH': 'Bahrain',
4143 'BD': 'Bangladesh',
4144 'BB': 'Barbados',
4145 'BY': 'Belarus',
4146 'BE': 'Belgium',
4147 'BZ': 'Belize',
4148 'BJ': 'Benin',
4149 'BM': 'Bermuda',
4150 'BT': 'Bhutan',
4151 'BO': 'Bolivia, Plurinational State of',
4152 'BQ': 'Bonaire, Sint Eustatius and Saba',
4153 'BA': 'Bosnia and Herzegovina',
4154 'BW': 'Botswana',
4155 'BV': 'Bouvet Island',
4156 'BR': 'Brazil',
4157 'IO': 'British Indian Ocean Territory',
4158 'BN': 'Brunei Darussalam',
4159 'BG': 'Bulgaria',
4160 'BF': 'Burkina Faso',
4161 'BI': 'Burundi',
4162 'KH': 'Cambodia',
4163 'CM': 'Cameroon',
4164 'CA': 'Canada',
4165 'CV': 'Cape Verde',
4166 'KY': 'Cayman Islands',
4167 'CF': 'Central African Republic',
4168 'TD': 'Chad',
4169 'CL': 'Chile',
4170 'CN': 'China',
4171 'CX': 'Christmas Island',
4172 'CC': 'Cocos (Keeling) Islands',
4173 'CO': 'Colombia',
4174 'KM': 'Comoros',
4175 'CG': 'Congo',
4176 'CD': 'Congo, the Democratic Republic of the',
4177 'CK': 'Cook Islands',
4178 'CR': 'Costa Rica',
4179 'CI': 'Côte d\'Ivoire',
4180 'HR': 'Croatia',
4181 'CU': 'Cuba',
4182 'CW': 'Curaçao',
4183 'CY': 'Cyprus',
4184 'CZ': 'Czech Republic',
4185 'DK': 'Denmark',
4186 'DJ': 'Djibouti',
4187 'DM': 'Dominica',
4188 'DO': 'Dominican Republic',
4189 'EC': 'Ecuador',
4190 'EG': 'Egypt',
4191 'SV': 'El Salvador',
4192 'GQ': 'Equatorial Guinea',
4193 'ER': 'Eritrea',
4194 'EE': 'Estonia',
4195 'ET': 'Ethiopia',
4196 'FK': 'Falkland Islands (Malvinas)',
4197 'FO': 'Faroe Islands',
4198 'FJ': 'Fiji',
4199 'FI': 'Finland',
4200 'FR': 'France',
4201 'GF': 'French Guiana',
4202 'PF': 'French Polynesia',
4203 'TF': 'French Southern Territories',
4204 'GA': 'Gabon',
4205 'GM': 'Gambia',
4206 'GE': 'Georgia',
4207 'DE': 'Germany',
4208 'GH': 'Ghana',
4209 'GI': 'Gibraltar',
4210 'GR': 'Greece',
4211 'GL': 'Greenland',
4212 'GD': 'Grenada',
4213 'GP': 'Guadeloupe',
4214 'GU': 'Guam',
4215 'GT': 'Guatemala',
4216 'GG': 'Guernsey',
4217 'GN': 'Guinea',
4218 'GW': 'Guinea-Bissau',
4219 'GY': 'Guyana',
4220 'HT': 'Haiti',
4221 'HM': 'Heard Island and McDonald Islands',
4222 'VA': 'Holy See (Vatican City State)',
4223 'HN': 'Honduras',
4224 'HK': 'Hong Kong',
4225 'HU': 'Hungary',
4226 'IS': 'Iceland',
4227 'IN': 'India',
4228 'ID': 'Indonesia',
4229 'IR': 'Iran, Islamic Republic of',
4230 'IQ': 'Iraq',
4231 'IE': 'Ireland',
4232 'IM': 'Isle of Man',
4233 'IL': 'Israel',
4234 'IT': 'Italy',
4235 'JM': 'Jamaica',
4236 'JP': 'Japan',
4237 'JE': 'Jersey',
4238 'JO': 'Jordan',
4239 'KZ': 'Kazakhstan',
4240 'KE': 'Kenya',
4241 'KI': 'Kiribati',
4242 'KP': 'Korea, Democratic People\'s Republic of',
4243 'KR': 'Korea, Republic of',
4244 'KW': 'Kuwait',
4245 'KG': 'Kyrgyzstan',
4246 'LA': 'Lao People\'s Democratic Republic',
4247 'LV': 'Latvia',
4248 'LB': 'Lebanon',
4249 'LS': 'Lesotho',
4250 'LR': 'Liberia',
4251 'LY': 'Libya',
4252 'LI': 'Liechtenstein',
4253 'LT': 'Lithuania',
4254 'LU': 'Luxembourg',
4255 'MO': 'Macao',
4256 'MK': 'Macedonia, the Former Yugoslav Republic of',
4257 'MG': 'Madagascar',
4258 'MW': 'Malawi',
4259 'MY': 'Malaysia',
4260 'MV': 'Maldives',
4261 'ML': 'Mali',
4262 'MT': 'Malta',
4263 'MH': 'Marshall Islands',
4264 'MQ': 'Martinique',
4265 'MR': 'Mauritania',
4266 'MU': 'Mauritius',
4267 'YT': 'Mayotte',
4268 'MX': 'Mexico',
4269 'FM': 'Micronesia, Federated States of',
4270 'MD': 'Moldova, Republic of',
4271 'MC': 'Monaco',
4272 'MN': 'Mongolia',
4273 'ME': 'Montenegro',
4274 'MS': 'Montserrat',
4275 'MA': 'Morocco',
4276 'MZ': 'Mozambique',
4277 'MM': 'Myanmar',
4278 'NA': 'Namibia',
4279 'NR': 'Nauru',
4280 'NP': 'Nepal',
4281 'NL': 'Netherlands',
4282 'NC': 'New Caledonia',
4283 'NZ': 'New Zealand',
4284 'NI': 'Nicaragua',
4285 'NE': 'Niger',
4286 'NG': 'Nigeria',
4287 'NU': 'Niue',
4288 'NF': 'Norfolk Island',
4289 'MP': 'Northern Mariana Islands',
4290 'NO': 'Norway',
4291 'OM': 'Oman',
4292 'PK': 'Pakistan',
4293 'PW': 'Palau',
4294 'PS': 'Palestine, State of',
4295 'PA': 'Panama',
4296 'PG': 'Papua New Guinea',
4297 'PY': 'Paraguay',
4298 'PE': 'Peru',
4299 'PH': 'Philippines',
4300 'PN': 'Pitcairn',
4301 'PL': 'Poland',
4302 'PT': 'Portugal',
4303 'PR': 'Puerto Rico',
4304 'QA': 'Qatar',
4305 'RE': 'Réunion',
4306 'RO': 'Romania',
4307 'RU': 'Russian Federation',
4308 'RW': 'Rwanda',
4309 'BL': 'Saint Barthélemy',
4310 'SH': 'Saint Helena, Ascension and Tristan da Cunha',
4311 'KN': 'Saint Kitts and Nevis',
4312 'LC': 'Saint Lucia',
4313 'MF': 'Saint Martin (French part)',
4314 'PM': 'Saint Pierre and Miquelon',
4315 'VC': 'Saint Vincent and the Grenadines',
4316 'WS': 'Samoa',
4317 'SM': 'San Marino',
4318 'ST': 'Sao Tome and Principe',
4319 'SA': 'Saudi Arabia',
4320 'SN': 'Senegal',
4321 'RS': 'Serbia',
4322 'SC': 'Seychelles',
4323 'SL': 'Sierra Leone',
4324 'SG': 'Singapore',
4325 'SX': 'Sint Maarten (Dutch part)',
4326 'SK': 'Slovakia',
4327 'SI': 'Slovenia',
4328 'SB': 'Solomon Islands',
4329 'SO': 'Somalia',
4330 'ZA': 'South Africa',
4331 'GS': 'South Georgia and the South Sandwich Islands',
4332 'SS': 'South Sudan',
4333 'ES': 'Spain',
4334 'LK': 'Sri Lanka',
4335 'SD': 'Sudan',
4336 'SR': 'Suriname',
4337 'SJ': 'Svalbard and Jan Mayen',
4338 'SZ': 'Swaziland',
4339 'SE': 'Sweden',
4340 'CH': 'Switzerland',
4341 'SY': 'Syrian Arab Republic',
4342 'TW': 'Taiwan, Province of China',
4343 'TJ': 'Tajikistan',
4344 'TZ': 'Tanzania, United Republic of',
4345 'TH': 'Thailand',
4346 'TL': 'Timor-Leste',
4347 'TG': 'Togo',
4348 'TK': 'Tokelau',
4349 'TO': 'Tonga',
4350 'TT': 'Trinidad and Tobago',
4351 'TN': 'Tunisia',
4352 'TR': 'Turkey',
4353 'TM': 'Turkmenistan',
4354 'TC': 'Turks and Caicos Islands',
4355 'TV': 'Tuvalu',
4356 'UG': 'Uganda',
4357 'UA': 'Ukraine',
4358 'AE': 'United Arab Emirates',
4359 'GB': 'United Kingdom',
4360 'US': 'United States',
4361 'UM': 'United States Minor Outlying Islands',
4362 'UY': 'Uruguay',
4363 'UZ': 'Uzbekistan',
4364 'VU': 'Vanuatu',
4365 'VE': 'Venezuela, Bolivarian Republic of',
4366 'VN': 'Viet Nam',
4367 'VG': 'Virgin Islands, British',
4368 'VI': 'Virgin Islands, U.S.',
4369 'WF': 'Wallis and Futuna',
4370 'EH': 'Western Sahara',
4371 'YE': 'Yemen',
4372 'ZM': 'Zambia',
4373 'ZW': 'Zimbabwe',
4374 }
4375
4376 @classmethod
4377 def short2full(cls, code):
4378 """Convert an ISO 3166-2 country code to the corresponding full name"""
4379 return cls._country_map.get(code.upper())
4380
4381
773f291d
S
4382class GeoUtils(object):
4383 # Major IPv4 address blocks per country
4384 _country_ip_map = {
53896ca5 4385 'AD': '46.172.224.0/19',
773f291d
S
4386 'AE': '94.200.0.0/13',
4387 'AF': '149.54.0.0/17',
4388 'AG': '209.59.64.0/18',
4389 'AI': '204.14.248.0/21',
4390 'AL': '46.99.0.0/16',
4391 'AM': '46.70.0.0/15',
4392 'AO': '105.168.0.0/13',
53896ca5
S
4393 'AP': '182.50.184.0/21',
4394 'AQ': '23.154.160.0/24',
773f291d
S
4395 'AR': '181.0.0.0/12',
4396 'AS': '202.70.112.0/20',
53896ca5 4397 'AT': '77.116.0.0/14',
773f291d
S
4398 'AU': '1.128.0.0/11',
4399 'AW': '181.41.0.0/18',
53896ca5
S
4400 'AX': '185.217.4.0/22',
4401 'AZ': '5.197.0.0/16',
773f291d
S
4402 'BA': '31.176.128.0/17',
4403 'BB': '65.48.128.0/17',
4404 'BD': '114.130.0.0/16',
4405 'BE': '57.0.0.0/8',
53896ca5 4406 'BF': '102.178.0.0/15',
773f291d
S
4407 'BG': '95.42.0.0/15',
4408 'BH': '37.131.0.0/17',
4409 'BI': '154.117.192.0/18',
4410 'BJ': '137.255.0.0/16',
53896ca5 4411 'BL': '185.212.72.0/23',
773f291d
S
4412 'BM': '196.12.64.0/18',
4413 'BN': '156.31.0.0/16',
4414 'BO': '161.56.0.0/16',
4415 'BQ': '161.0.80.0/20',
53896ca5 4416 'BR': '191.128.0.0/12',
773f291d
S
4417 'BS': '24.51.64.0/18',
4418 'BT': '119.2.96.0/19',
4419 'BW': '168.167.0.0/16',
4420 'BY': '178.120.0.0/13',
4421 'BZ': '179.42.192.0/18',
4422 'CA': '99.224.0.0/11',
4423 'CD': '41.243.0.0/16',
53896ca5
S
4424 'CF': '197.242.176.0/21',
4425 'CG': '160.113.0.0/16',
773f291d 4426 'CH': '85.0.0.0/13',
53896ca5 4427 'CI': '102.136.0.0/14',
773f291d
S
4428 'CK': '202.65.32.0/19',
4429 'CL': '152.172.0.0/14',
53896ca5 4430 'CM': '102.244.0.0/14',
773f291d
S
4431 'CN': '36.128.0.0/10',
4432 'CO': '181.240.0.0/12',
4433 'CR': '201.192.0.0/12',
4434 'CU': '152.206.0.0/15',
4435 'CV': '165.90.96.0/19',
4436 'CW': '190.88.128.0/17',
53896ca5 4437 'CY': '31.153.0.0/16',
773f291d
S
4438 'CZ': '88.100.0.0/14',
4439 'DE': '53.0.0.0/8',
4440 'DJ': '197.241.0.0/17',
4441 'DK': '87.48.0.0/12',
4442 'DM': '192.243.48.0/20',
4443 'DO': '152.166.0.0/15',
4444 'DZ': '41.96.0.0/12',
4445 'EC': '186.68.0.0/15',
4446 'EE': '90.190.0.0/15',
4447 'EG': '156.160.0.0/11',
4448 'ER': '196.200.96.0/20',
4449 'ES': '88.0.0.0/11',
4450 'ET': '196.188.0.0/14',
4451 'EU': '2.16.0.0/13',
4452 'FI': '91.152.0.0/13',
4453 'FJ': '144.120.0.0/16',
53896ca5 4454 'FK': '80.73.208.0/21',
773f291d
S
4455 'FM': '119.252.112.0/20',
4456 'FO': '88.85.32.0/19',
4457 'FR': '90.0.0.0/9',
4458 'GA': '41.158.0.0/15',
4459 'GB': '25.0.0.0/8',
4460 'GD': '74.122.88.0/21',
4461 'GE': '31.146.0.0/16',
4462 'GF': '161.22.64.0/18',
4463 'GG': '62.68.160.0/19',
53896ca5
S
4464 'GH': '154.160.0.0/12',
4465 'GI': '95.164.0.0/16',
773f291d
S
4466 'GL': '88.83.0.0/19',
4467 'GM': '160.182.0.0/15',
4468 'GN': '197.149.192.0/18',
4469 'GP': '104.250.0.0/19',
4470 'GQ': '105.235.224.0/20',
4471 'GR': '94.64.0.0/13',
4472 'GT': '168.234.0.0/16',
4473 'GU': '168.123.0.0/16',
4474 'GW': '197.214.80.0/20',
4475 'GY': '181.41.64.0/18',
4476 'HK': '113.252.0.0/14',
4477 'HN': '181.210.0.0/16',
4478 'HR': '93.136.0.0/13',
4479 'HT': '148.102.128.0/17',
4480 'HU': '84.0.0.0/14',
4481 'ID': '39.192.0.0/10',
4482 'IE': '87.32.0.0/12',
4483 'IL': '79.176.0.0/13',
4484 'IM': '5.62.80.0/20',
4485 'IN': '117.192.0.0/10',
4486 'IO': '203.83.48.0/21',
4487 'IQ': '37.236.0.0/14',
4488 'IR': '2.176.0.0/12',
4489 'IS': '82.221.0.0/16',
4490 'IT': '79.0.0.0/10',
4491 'JE': '87.244.64.0/18',
4492 'JM': '72.27.0.0/17',
4493 'JO': '176.29.0.0/16',
53896ca5 4494 'JP': '133.0.0.0/8',
773f291d
S
4495 'KE': '105.48.0.0/12',
4496 'KG': '158.181.128.0/17',
4497 'KH': '36.37.128.0/17',
4498 'KI': '103.25.140.0/22',
4499 'KM': '197.255.224.0/20',
53896ca5 4500 'KN': '198.167.192.0/19',
773f291d
S
4501 'KP': '175.45.176.0/22',
4502 'KR': '175.192.0.0/10',
4503 'KW': '37.36.0.0/14',
4504 'KY': '64.96.0.0/15',
4505 'KZ': '2.72.0.0/13',
4506 'LA': '115.84.64.0/18',
4507 'LB': '178.135.0.0/16',
53896ca5 4508 'LC': '24.92.144.0/20',
773f291d
S
4509 'LI': '82.117.0.0/19',
4510 'LK': '112.134.0.0/15',
53896ca5 4511 'LR': '102.183.0.0/16',
773f291d
S
4512 'LS': '129.232.0.0/17',
4513 'LT': '78.56.0.0/13',
4514 'LU': '188.42.0.0/16',
4515 'LV': '46.109.0.0/16',
4516 'LY': '41.252.0.0/14',
4517 'MA': '105.128.0.0/11',
4518 'MC': '88.209.64.0/18',
4519 'MD': '37.246.0.0/16',
4520 'ME': '178.175.0.0/17',
4521 'MF': '74.112.232.0/21',
4522 'MG': '154.126.0.0/17',
4523 'MH': '117.103.88.0/21',
4524 'MK': '77.28.0.0/15',
4525 'ML': '154.118.128.0/18',
4526 'MM': '37.111.0.0/17',
4527 'MN': '49.0.128.0/17',
4528 'MO': '60.246.0.0/16',
4529 'MP': '202.88.64.0/20',
4530 'MQ': '109.203.224.0/19',
4531 'MR': '41.188.64.0/18',
4532 'MS': '208.90.112.0/22',
4533 'MT': '46.11.0.0/16',
4534 'MU': '105.16.0.0/12',
4535 'MV': '27.114.128.0/18',
53896ca5 4536 'MW': '102.70.0.0/15',
773f291d
S
4537 'MX': '187.192.0.0/11',
4538 'MY': '175.136.0.0/13',
4539 'MZ': '197.218.0.0/15',
4540 'NA': '41.182.0.0/16',
4541 'NC': '101.101.0.0/18',
4542 'NE': '197.214.0.0/18',
4543 'NF': '203.17.240.0/22',
4544 'NG': '105.112.0.0/12',
4545 'NI': '186.76.0.0/15',
4546 'NL': '145.96.0.0/11',
4547 'NO': '84.208.0.0/13',
4548 'NP': '36.252.0.0/15',
4549 'NR': '203.98.224.0/19',
4550 'NU': '49.156.48.0/22',
4551 'NZ': '49.224.0.0/14',
4552 'OM': '5.36.0.0/15',
4553 'PA': '186.72.0.0/15',
4554 'PE': '186.160.0.0/14',
4555 'PF': '123.50.64.0/18',
4556 'PG': '124.240.192.0/19',
4557 'PH': '49.144.0.0/13',
4558 'PK': '39.32.0.0/11',
4559 'PL': '83.0.0.0/11',
4560 'PM': '70.36.0.0/20',
4561 'PR': '66.50.0.0/16',
4562 'PS': '188.161.0.0/16',
4563 'PT': '85.240.0.0/13',
4564 'PW': '202.124.224.0/20',
4565 'PY': '181.120.0.0/14',
4566 'QA': '37.210.0.0/15',
53896ca5 4567 'RE': '102.35.0.0/16',
773f291d 4568 'RO': '79.112.0.0/13',
53896ca5 4569 'RS': '93.86.0.0/15',
773f291d 4570 'RU': '5.136.0.0/13',
53896ca5 4571 'RW': '41.186.0.0/16',
773f291d
S
4572 'SA': '188.48.0.0/13',
4573 'SB': '202.1.160.0/19',
4574 'SC': '154.192.0.0/11',
53896ca5 4575 'SD': '102.120.0.0/13',
773f291d 4576 'SE': '78.64.0.0/12',
53896ca5 4577 'SG': '8.128.0.0/10',
773f291d
S
4578 'SI': '188.196.0.0/14',
4579 'SK': '78.98.0.0/15',
53896ca5 4580 'SL': '102.143.0.0/17',
773f291d
S
4581 'SM': '89.186.32.0/19',
4582 'SN': '41.82.0.0/15',
53896ca5 4583 'SO': '154.115.192.0/18',
773f291d
S
4584 'SR': '186.179.128.0/17',
4585 'SS': '105.235.208.0/21',
4586 'ST': '197.159.160.0/19',
4587 'SV': '168.243.0.0/16',
4588 'SX': '190.102.0.0/20',
4589 'SY': '5.0.0.0/16',
4590 'SZ': '41.84.224.0/19',
4591 'TC': '65.255.48.0/20',
4592 'TD': '154.68.128.0/19',
4593 'TG': '196.168.0.0/14',
4594 'TH': '171.96.0.0/13',
4595 'TJ': '85.9.128.0/18',
4596 'TK': '27.96.24.0/21',
4597 'TL': '180.189.160.0/20',
4598 'TM': '95.85.96.0/19',
4599 'TN': '197.0.0.0/11',
4600 'TO': '175.176.144.0/21',
4601 'TR': '78.160.0.0/11',
4602 'TT': '186.44.0.0/15',
4603 'TV': '202.2.96.0/19',
4604 'TW': '120.96.0.0/11',
4605 'TZ': '156.156.0.0/14',
53896ca5
S
4606 'UA': '37.52.0.0/14',
4607 'UG': '102.80.0.0/13',
4608 'US': '6.0.0.0/8',
773f291d 4609 'UY': '167.56.0.0/13',
53896ca5 4610 'UZ': '84.54.64.0/18',
773f291d 4611 'VA': '212.77.0.0/19',
53896ca5 4612 'VC': '207.191.240.0/21',
773f291d 4613 'VE': '186.88.0.0/13',
53896ca5 4614 'VG': '66.81.192.0/20',
773f291d
S
4615 'VI': '146.226.0.0/16',
4616 'VN': '14.160.0.0/11',
4617 'VU': '202.80.32.0/20',
4618 'WF': '117.20.32.0/21',
4619 'WS': '202.4.32.0/19',
4620 'YE': '134.35.0.0/16',
4621 'YT': '41.242.116.0/22',
4622 'ZA': '41.0.0.0/11',
53896ca5
S
4623 'ZM': '102.144.0.0/13',
4624 'ZW': '102.177.192.0/18',
773f291d
S
4625 }
4626
4627 @classmethod
5f95927a
S
4628 def random_ipv4(cls, code_or_block):
4629 if len(code_or_block) == 2:
4630 block = cls._country_ip_map.get(code_or_block.upper())
4631 if not block:
4632 return None
4633 else:
4634 block = code_or_block
773f291d
S
4635 addr, preflen = block.split('/')
4636 addr_min = compat_struct_unpack('!L', socket.inet_aton(addr))[0]
4637 addr_max = addr_min | (0xffffffff >> int(preflen))
18a0defa 4638 return compat_str(socket.inet_ntoa(
4248dad9 4639 compat_struct_pack('!L', random.randint(addr_min, addr_max))))
773f291d
S
4640
4641
91410c9b 4642class PerRequestProxyHandler(compat_urllib_request.ProxyHandler):
2461f79d
PH
4643 def __init__(self, proxies=None):
4644 # Set default handlers
4645 for type in ('http', 'https'):
4646 setattr(self, '%s_open' % type,
4647 lambda r, proxy='__noproxy__', type=type, meth=self.proxy_open:
4648 meth(r, proxy, type))
38e87f6c 4649 compat_urllib_request.ProxyHandler.__init__(self, proxies)
2461f79d 4650
91410c9b 4651 def proxy_open(self, req, proxy, type):
2461f79d 4652 req_proxy = req.headers.get('Ytdl-request-proxy')
91410c9b
PH
4653 if req_proxy is not None:
4654 proxy = req_proxy
2461f79d
PH
4655 del req.headers['Ytdl-request-proxy']
4656
4657 if proxy == '__noproxy__':
4658 return None # No Proxy
51fb4995 4659 if compat_urlparse.urlparse(proxy).scheme.lower() in ('socks', 'socks4', 'socks4a', 'socks5'):
71aff188 4660 req.add_header('Ytdl-socks-proxy', proxy)
7a5c1cfe 4661 # yt-dlp's http/https handlers do wrapping the socket with socks
71aff188 4662 return None
91410c9b
PH
4663 return compat_urllib_request.ProxyHandler.proxy_open(
4664 self, req, proxy, type)
5bc880b9
YCH
4665
4666
0a5445dd
YCH
4667# Both long_to_bytes and bytes_to_long are adapted from PyCrypto, which is
4668# released into Public Domain
4669# https://github.com/dlitz/pycrypto/blob/master/lib/Crypto/Util/number.py#L387
4670
4671def long_to_bytes(n, blocksize=0):
4672 """long_to_bytes(n:long, blocksize:int) : string
4673 Convert a long integer to a byte string.
4674
4675 If optional blocksize is given and greater than zero, pad the front of the
4676 byte string with binary zeros so that the length is a multiple of
4677 blocksize.
4678 """
4679 # after much testing, this algorithm was deemed to be the fastest
4680 s = b''
4681 n = int(n)
4682 while n > 0:
4683 s = compat_struct_pack('>I', n & 0xffffffff) + s
4684 n = n >> 32
4685 # strip off leading zeros
4686 for i in range(len(s)):
4687 if s[i] != b'\000'[0]:
4688 break
4689 else:
4690 # only happens when n == 0
4691 s = b'\000'
4692 i = 0
4693 s = s[i:]
4694 # add back some pad bytes. this could be done more efficiently w.r.t. the
4695 # de-padding being done above, but sigh...
4696 if blocksize > 0 and len(s) % blocksize:
4697 s = (blocksize - len(s) % blocksize) * b'\000' + s
4698 return s
4699
4700
4701def bytes_to_long(s):
4702 """bytes_to_long(string) : long
4703 Convert a byte string to a long integer.
4704
4705 This is (essentially) the inverse of long_to_bytes().
4706 """
4707 acc = 0
4708 length = len(s)
4709 if length % 4:
4710 extra = (4 - length % 4)
4711 s = b'\000' * extra + s
4712 length = length + extra
4713 for i in range(0, length, 4):
4714 acc = (acc << 32) + compat_struct_unpack('>I', s[i:i + 4])[0]
4715 return acc
4716
4717
5bc880b9
YCH
4718def ohdave_rsa_encrypt(data, exponent, modulus):
4719 '''
4720 Implement OHDave's RSA algorithm. See http://www.ohdave.com/rsa/
4721
4722 Input:
4723 data: data to encrypt, bytes-like object
4724 exponent, modulus: parameter e and N of RSA algorithm, both integer
4725 Output: hex string of encrypted data
4726
4727 Limitation: supports one block encryption only
4728 '''
4729
4730 payload = int(binascii.hexlify(data[::-1]), 16)
4731 encrypted = pow(payload, exponent, modulus)
4732 return '%x' % encrypted
81bdc8fd
YCH
4733
4734
f48409c7
YCH
4735def pkcs1pad(data, length):
4736 """
4737 Padding input data with PKCS#1 scheme
4738
4739 @param {int[]} data input data
4740 @param {int} length target length
4741 @returns {int[]} padded data
4742 """
4743 if len(data) > length - 11:
4744 raise ValueError('Input data too long for PKCS#1 padding')
4745
4746 pseudo_random = [random.randint(0, 254) for _ in range(length - len(data) - 3)]
4747 return [0, 2] + pseudo_random + [0] + data
4748
4749
5eb6bdce 4750def encode_base_n(num, n, table=None):
59f898b7 4751 FULL_TABLE = '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
59f898b7
YCH
4752 if not table:
4753 table = FULL_TABLE[:n]
4754
5eb6bdce
YCH
4755 if n > len(table):
4756 raise ValueError('base %d exceeds table length %d' % (n, len(table)))
4757
4758 if num == 0:
4759 return table[0]
4760
81bdc8fd
YCH
4761 ret = ''
4762 while num:
4763 ret = table[num % n] + ret
4764 num = num // n
4765 return ret
f52354a8
YCH
4766
4767
4768def decode_packed_codes(code):
06b3fe29 4769 mobj = re.search(PACKED_CODES_RE, code)
a0566bbf 4770 obfuscated_code, base, count, symbols = mobj.groups()
f52354a8
YCH
4771 base = int(base)
4772 count = int(count)
4773 symbols = symbols.split('|')
4774 symbol_table = {}
4775
4776 while count:
4777 count -= 1
5eb6bdce 4778 base_n_count = encode_base_n(count, base)
f52354a8
YCH
4779 symbol_table[base_n_count] = symbols[count] or base_n_count
4780
4781 return re.sub(
4782 r'\b(\w+)\b', lambda mobj: symbol_table[mobj.group(0)],
a0566bbf 4783 obfuscated_code)
e154c651 4784
4785
1ced2221
S
4786def caesar(s, alphabet, shift):
4787 if shift == 0:
4788 return s
4789 l = len(alphabet)
4790 return ''.join(
4791 alphabet[(alphabet.index(c) + shift) % l] if c in alphabet else c
4792 for c in s)
4793
4794
4795def rot47(s):
4796 return caesar(s, r'''!"#$%&'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\]^_`abcdefghijklmnopqrstuvwxyz{|}~''', 47)
4797
4798
e154c651 4799def parse_m3u8_attributes(attrib):
4800 info = {}
4801 for (key, val) in re.findall(r'(?P<key>[A-Z0-9-]+)=(?P<val>"[^"]+"|[^",]+)(?:,|$)', attrib):
4802 if val.startswith('"'):
4803 val = val[1:-1]
4804 info[key] = val
4805 return info
1143535d
YCH
4806
4807
4808def urshift(val, n):
4809 return val >> n if val >= 0 else (val + 0x100000000) >> n
d3f8e038
YCH
4810
4811
4812# Based on png2str() written by @gdkchan and improved by @yokrysty
067aa17e 4813# Originally posted at https://github.com/ytdl-org/youtube-dl/issues/9706
d3f8e038
YCH
4814def decode_png(png_data):
4815 # Reference: https://www.w3.org/TR/PNG/
4816 header = png_data[8:]
4817
4818 if png_data[:8] != b'\x89PNG\x0d\x0a\x1a\x0a' or header[4:8] != b'IHDR':
4819 raise IOError('Not a valid PNG file.')
4820
4821 int_map = {1: '>B', 2: '>H', 4: '>I'}
4822 unpack_integer = lambda x: compat_struct_unpack(int_map[len(x)], x)[0]
4823
4824 chunks = []
4825
4826 while header:
4827 length = unpack_integer(header[:4])
4828 header = header[4:]
4829
4830 chunk_type = header[:4]
4831 header = header[4:]
4832
4833 chunk_data = header[:length]
4834 header = header[length:]
4835
4836 header = header[4:] # Skip CRC
4837
4838 chunks.append({
4839 'type': chunk_type,
4840 'length': length,
4841 'data': chunk_data
4842 })
4843
4844 ihdr = chunks[0]['data']
4845
4846 width = unpack_integer(ihdr[:4])
4847 height = unpack_integer(ihdr[4:8])
4848
4849 idat = b''
4850
4851 for chunk in chunks:
4852 if chunk['type'] == b'IDAT':
4853 idat += chunk['data']
4854
4855 if not idat:
4856 raise IOError('Unable to read PNG data.')
4857
4858 decompressed_data = bytearray(zlib.decompress(idat))
4859
4860 stride = width * 3
4861 pixels = []
4862
4863 def _get_pixel(idx):
4864 x = idx % stride
4865 y = idx // stride
4866 return pixels[y][x]
4867
4868 for y in range(height):
4869 basePos = y * (1 + stride)
4870 filter_type = decompressed_data[basePos]
4871
4872 current_row = []
4873
4874 pixels.append(current_row)
4875
4876 for x in range(stride):
4877 color = decompressed_data[1 + basePos + x]
4878 basex = y * stride + x
4879 left = 0
4880 up = 0
4881
4882 if x > 2:
4883 left = _get_pixel(basex - 3)
4884 if y > 0:
4885 up = _get_pixel(basex - stride)
4886
4887 if filter_type == 1: # Sub
4888 color = (color + left) & 0xff
4889 elif filter_type == 2: # Up
4890 color = (color + up) & 0xff
4891 elif filter_type == 3: # Average
4892 color = (color + ((left + up) >> 1)) & 0xff
4893 elif filter_type == 4: # Paeth
4894 a = left
4895 b = up
4896 c = 0
4897
4898 if x > 2 and y > 0:
4899 c = _get_pixel(basex - stride - 3)
4900
4901 p = a + b - c
4902
4903 pa = abs(p - a)
4904 pb = abs(p - b)
4905 pc = abs(p - c)
4906
4907 if pa <= pb and pa <= pc:
4908 color = (color + a) & 0xff
4909 elif pb <= pc:
4910 color = (color + b) & 0xff
4911 else:
4912 color = (color + c) & 0xff
4913
4914 current_row.append(color)
4915
4916 return width, height, pixels
efa97bdc
YCH
4917
4918
4919def write_xattr(path, key, value):
4920 # This mess below finds the best xattr tool for the job
4921 try:
4922 # try the pyxattr module...
4923 import xattr
4924
53a7e3d2
YCH
4925 if hasattr(xattr, 'set'): # pyxattr
4926 # Unicode arguments are not supported in python-pyxattr until
4927 # version 0.5.0
067aa17e 4928 # See https://github.com/ytdl-org/youtube-dl/issues/5498
53a7e3d2
YCH
4929 pyxattr_required_version = '0.5.0'
4930 if version_tuple(xattr.__version__) < version_tuple(pyxattr_required_version):
4931 # TODO: fallback to CLI tools
4932 raise XAttrUnavailableError(
4933 'python-pyxattr is detected but is too old. '
7a5c1cfe 4934 'yt-dlp requires %s or above while your version is %s. '
53a7e3d2
YCH
4935 'Falling back to other xattr implementations' % (
4936 pyxattr_required_version, xattr.__version__))
4937
4938 setxattr = xattr.set
4939 else: # xattr
4940 setxattr = xattr.setxattr
efa97bdc
YCH
4941
4942 try:
53a7e3d2 4943 setxattr(path, key, value)
efa97bdc
YCH
4944 except EnvironmentError as e:
4945 raise XAttrMetadataError(e.errno, e.strerror)
4946
4947 except ImportError:
4948 if compat_os_name == 'nt':
4949 # Write xattrs to NTFS Alternate Data Streams:
4950 # http://en.wikipedia.org/wiki/NTFS#Alternate_data_streams_.28ADS.29
4951 assert ':' not in key
4952 assert os.path.exists(path)
4953
4954 ads_fn = path + ':' + key
4955 try:
4956 with open(ads_fn, 'wb') as f:
4957 f.write(value)
4958 except EnvironmentError as e:
4959 raise XAttrMetadataError(e.errno, e.strerror)
4960 else:
4961 user_has_setfattr = check_executable('setfattr', ['--version'])
4962 user_has_xattr = check_executable('xattr', ['-h'])
4963
4964 if user_has_setfattr or user_has_xattr:
4965
4966 value = value.decode('utf-8')
4967 if user_has_setfattr:
4968 executable = 'setfattr'
4969 opts = ['-n', key, '-v', value]
4970 elif user_has_xattr:
4971 executable = 'xattr'
4972 opts = ['-w', key, value]
4973
3089bc74
S
4974 cmd = ([encodeFilename(executable, True)]
4975 + [encodeArgument(o) for o in opts]
4976 + [encodeFilename(path, True)])
efa97bdc
YCH
4977
4978 try:
d3c93ec2 4979 p = Popen(
efa97bdc
YCH
4980 cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.PIPE)
4981 except EnvironmentError as e:
4982 raise XAttrMetadataError(e.errno, e.strerror)
d3c93ec2 4983 stdout, stderr = p.communicate_or_kill()
efa97bdc
YCH
4984 stderr = stderr.decode('utf-8', 'replace')
4985 if p.returncode != 0:
4986 raise XAttrMetadataError(p.returncode, stderr)
4987
4988 else:
4989 # On Unix, and can't find pyxattr, setfattr, or xattr.
4990 if sys.platform.startswith('linux'):
4991 raise XAttrUnavailableError(
4992 "Couldn't find a tool to set the xattrs. "
4993 "Install either the python 'pyxattr' or 'xattr' "
4994 "modules, or the GNU 'attr' package "
4995 "(which contains the 'setfattr' tool).")
4996 else:
4997 raise XAttrUnavailableError(
4998 "Couldn't find a tool to set the xattrs. "
4999 "Install either the python 'xattr' module, "
5000 "or the 'xattr' binary.")
0c265486
YCH
5001
5002
5003def random_birthday(year_field, month_field, day_field):
aa374bc7
AS
5004 start_date = datetime.date(1950, 1, 1)
5005 end_date = datetime.date(1995, 12, 31)
5006 offset = random.randint(0, (end_date - start_date).days)
5007 random_date = start_date + datetime.timedelta(offset)
0c265486 5008 return {
aa374bc7
AS
5009 year_field: str(random_date.year),
5010 month_field: str(random_date.month),
5011 day_field: str(random_date.day),
0c265486 5012 }
732044af 5013
c76eb41b 5014
732044af 5015# Templates for internet shortcut files, which are plain text files.
5016DOT_URL_LINK_TEMPLATE = '''
5017[InternetShortcut]
5018URL=%(url)s
5019'''.lstrip()
5020
5021DOT_WEBLOC_LINK_TEMPLATE = '''
5022<?xml version="1.0" encoding="UTF-8"?>
5023<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
5024<plist version="1.0">
5025<dict>
5026\t<key>URL</key>
5027\t<string>%(url)s</string>
5028</dict>
5029</plist>
5030'''.lstrip()
5031
5032DOT_DESKTOP_LINK_TEMPLATE = '''
5033[Desktop Entry]
5034Encoding=UTF-8
5035Name=%(filename)s
5036Type=Link
5037URL=%(url)s
5038Icon=text-html
5039'''.lstrip()
5040
08438d2c 5041LINK_TEMPLATES = {
5042 'url': DOT_URL_LINK_TEMPLATE,
5043 'desktop': DOT_DESKTOP_LINK_TEMPLATE,
5044 'webloc': DOT_WEBLOC_LINK_TEMPLATE,
5045}
5046
732044af 5047
5048def iri_to_uri(iri):
5049 """
5050 Converts an IRI (Internationalized Resource Identifier, allowing Unicode characters) to a URI (Uniform Resource Identifier, ASCII-only).
5051
5052 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.
5053 """
5054
5055 iri_parts = compat_urllib_parse_urlparse(iri)
5056
5057 if '[' in iri_parts.netloc:
5058 raise ValueError('IPv6 URIs are not, yet, supported.')
5059 # Querying `.netloc`, when there's only one bracket, also raises a ValueError.
5060
5061 # 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.
5062
5063 net_location = ''
5064 if iri_parts.username:
5065 net_location += compat_urllib_parse_quote(iri_parts.username, safe=r"!$%&'()*+,~")
5066 if iri_parts.password is not None:
5067 net_location += ':' + compat_urllib_parse_quote(iri_parts.password, safe=r"!$%&'()*+,~")
5068 net_location += '@'
5069
5070 net_location += iri_parts.hostname.encode('idna').decode('utf-8') # Punycode for Unicode hostnames.
5071 # The 'idna' encoding produces ASCII text.
5072 if iri_parts.port is not None and iri_parts.port != 80:
5073 net_location += ':' + str(iri_parts.port)
5074
5075 return compat_urllib_parse_urlunparse(
5076 (iri_parts.scheme,
5077 net_location,
5078
5079 compat_urllib_parse_quote_plus(iri_parts.path, safe=r"!$%&'()*+,/:;=@|~"),
5080
5081 # Unsure about the `safe` argument, since this is a legacy way of handling parameters.
5082 compat_urllib_parse_quote_plus(iri_parts.params, safe=r"!$%&'()*+,/:;=@|~"),
5083
5084 # Not totally sure about the `safe` argument, since the source does not explicitly mention the query URI component.
5085 compat_urllib_parse_quote_plus(iri_parts.query, safe=r"!$%&'()*+,/:;=?@{|}~"),
5086
5087 compat_urllib_parse_quote_plus(iri_parts.fragment, safe=r"!#$%&'()*+,/:;=?@{|}~")))
5088
5089 # Source for `safe` arguments: https://url.spec.whatwg.org/#percent-encoded-bytes.
5090
5091
5092def to_high_limit_path(path):
5093 if sys.platform in ['win32', 'cygwin']:
5094 # Work around MAX_PATH limitation on Windows. The maximum allowed length for the individual path segments may still be quite limited.
5095 return r'\\?\ '.rstrip() + os.path.abspath(path)
5096
5097 return path
76d321f6 5098
c76eb41b 5099
b868936c 5100def format_field(obj, field=None, template='%s', ignore=(None, ''), default='', func=None):
e0ddbd02 5101 val = traverse_obj(obj, *variadic(field))
5102 if val in ignore:
5103 return default
5104 return template % (func(val) if func else val)
00dd0cd5 5105
5106
5107def clean_podcast_url(url):
5108 return re.sub(r'''(?x)
5109 (?:
5110 (?:
5111 chtbl\.com/track|
5112 media\.blubrry\.com| # https://create.blubrry.com/resources/podcast-media-download-statistics/getting-started/
5113 play\.podtrac\.com
5114 )/[^/]+|
5115 (?:dts|www)\.podtrac\.com/(?:pts/)?redirect\.[0-9a-z]{3,4}| # http://analytics.podtrac.com/how-to-measure
5116 flex\.acast\.com|
5117 pd(?:
5118 cn\.co| # https://podcorn.com/analytics-prefix/
5119 st\.fm # https://podsights.com/docs/
5120 )/e
5121 )/''', '', url)
ffcb8191
THD
5122
5123
5124_HEX_TABLE = '0123456789abcdef'
5125
5126
5127def random_uuidv4():
5128 return re.sub(r'[xy]', lambda x: _HEX_TABLE[random.randint(0, 15)], 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx')
0202b52a 5129
5130
5131def make_dir(path, to_screen=None):
5132 try:
5133 dn = os.path.dirname(path)
5134 if dn and not os.path.exists(dn):
5135 os.makedirs(dn)
5136 return True
5137 except (OSError, IOError) as err:
5138 if callable(to_screen) is not None:
5139 to_screen('unable to create directory ' + error_to_compat_str(err))
5140 return False
f74980cb 5141
5142
5143def get_executable_path():
c552ae88 5144 from zipimport import zipimporter
5145 if hasattr(sys, 'frozen'): # Running from PyInstaller
5146 path = os.path.dirname(sys.executable)
5147 elif isinstance(globals().get('__loader__'), zipimporter): # Running from ZIP
5148 path = os.path.join(os.path.dirname(__file__), '../..')
5149 else:
5150 path = os.path.join(os.path.dirname(__file__), '..')
f74980cb 5151 return os.path.abspath(path)
5152
5153
2f567473 5154def load_plugins(name, suffix, namespace):
3ae5e797 5155 classes = {}
f74980cb 5156 try:
019a94f7
ÁS
5157 plugins_spec = importlib.util.spec_from_file_location(
5158 name, os.path.join(get_executable_path(), 'ytdlp_plugins', name, '__init__.py'))
5159 plugins = importlib.util.module_from_spec(plugins_spec)
5160 sys.modules[plugins_spec.name] = plugins
5161 plugins_spec.loader.exec_module(plugins)
f74980cb 5162 for name in dir(plugins):
2f567473 5163 if name in namespace:
5164 continue
5165 if not name.endswith(suffix):
f74980cb 5166 continue
5167 klass = getattr(plugins, name)
3ae5e797 5168 classes[name] = namespace[name] = klass
019a94f7 5169 except FileNotFoundError:
f74980cb 5170 pass
f74980cb 5171 return classes
06167fbb 5172
5173
325ebc17 5174def traverse_obj(
352d63fd 5175 obj, *path_list, default=None, expected_type=None, get_all=True,
325ebc17 5176 casesense=True, is_user_input=False, traverse_string=False):
324ad820 5177 ''' Traverse nested list/dict/tuple
8f334380 5178 @param path_list A list of paths which are checked one by one.
5179 Each path is a list of keys where each key is a string,
1797b073 5180 a function, a tuple of strings/None or "...".
e6f868a6 5181 When a fuction is given, it takes the key and value as arguments
5182 and returns whether the key matches or not. When a tuple is given,
8f334380 5183 all the keys given in the tuple are traversed, and
5184 "..." traverses all the keys in the object
1797b073 5185 "None" returns the object without traversal
325ebc17 5186 @param default Default value to return
352d63fd 5187 @param expected_type Only accept final value of this type (Can also be any callable)
5188 @param get_all Return all the values obtained from a path or only the first one
324ad820 5189 @param casesense Whether to consider dictionary keys as case sensitive
5190 @param is_user_input Whether the keys are generated from user input. If True,
5191 strings are converted to int/slice if necessary
5192 @param traverse_string Whether to traverse inside strings. If True, any
5193 non-compatible object will also be converted into a string
8f334380 5194 # TODO: Write tests
324ad820 5195 '''
325ebc17 5196 if not casesense:
dbf5416a 5197 _lower = lambda k: (k.lower() if isinstance(k, str) else k)
8f334380 5198 path_list = (map(_lower, variadic(path)) for path in path_list)
5199
5200 def _traverse_obj(obj, path, _current_depth=0):
5201 nonlocal depth
5202 path = tuple(variadic(path))
5203 for i, key in enumerate(path):
1797b073 5204 if None in (key, obj):
5205 return obj
8f334380 5206 if isinstance(key, (list, tuple)):
5207 obj = [_traverse_obj(obj, sub_key, _current_depth) for sub_key in key]
5208 key = ...
5209 if key is ...:
5210 obj = (obj.values() if isinstance(obj, dict)
5211 else obj if isinstance(obj, (list, tuple, LazyList))
5212 else str(obj) if traverse_string else [])
5213 _current_depth += 1
5214 depth = max(depth, _current_depth)
5215 return [_traverse_obj(inner_obj, path[i + 1:], _current_depth) for inner_obj in obj]
2614f646 5216 elif callable(key):
5217 if isinstance(obj, (list, tuple, LazyList)):
5218 obj = enumerate(obj)
5219 elif isinstance(obj, dict):
5220 obj = obj.items()
5221 else:
5222 if not traverse_string:
5223 return None
5224 obj = str(obj)
5225 _current_depth += 1
5226 depth = max(depth, _current_depth)
e6f868a6 5227 return [_traverse_obj(v, path[i + 1:], _current_depth) for k, v in obj if try_call(key, args=(k, v))]
575e17a1 5228 elif isinstance(obj, dict) and not (is_user_input and key == ':'):
325ebc17 5229 obj = (obj.get(key) if casesense or (key in obj)
5230 else next((v for k, v in obj.items() if _lower(k) == key), None))
5231 else:
5232 if is_user_input:
5233 key = (int_or_none(key) if ':' not in key
5234 else slice(*map(int_or_none, key.split(':'))))
8f334380 5235 if key == slice(None):
575e17a1 5236 return _traverse_obj(obj, (..., *path[i + 1:]), _current_depth)
325ebc17 5237 if not isinstance(key, (int, slice)):
9fea350f 5238 return None
8f334380 5239 if not isinstance(obj, (list, tuple, LazyList)):
325ebc17 5240 if not traverse_string:
5241 return None
5242 obj = str(obj)
5243 try:
5244 obj = obj[key]
5245 except IndexError:
324ad820 5246 return None
325ebc17 5247 return obj
5248
352d63fd 5249 if isinstance(expected_type, type):
5250 type_test = lambda val: val if isinstance(val, expected_type) else None
5251 elif expected_type is not None:
5252 type_test = expected_type
5253 else:
5254 type_test = lambda val: val
5255
8f334380 5256 for path in path_list:
5257 depth = 0
5258 val = _traverse_obj(obj, path)
325ebc17 5259 if val is not None:
8f334380 5260 if depth:
5261 for _ in range(depth - 1):
6586bca9 5262 val = itertools.chain.from_iterable(v for v in val if v is not None)
352d63fd 5263 val = [v for v in map(type_test, val) if v is not None]
8f334380 5264 if val:
352d63fd 5265 return val if get_all else val[0]
5266 else:
5267 val = type_test(val)
5268 if val is not None:
8f334380 5269 return val
325ebc17 5270 return default
324ad820 5271
5272
5273def traverse_dict(dictn, keys, casesense=True):
ee8dd27a 5274 write_string('DeprecationWarning: yt_dlp.utils.traverse_dict is deprecated '
5275 'and may be removed in a future version. Use yt_dlp.utils.traverse_obj instead')
5276 return traverse_obj(dictn, keys, casesense=casesense, is_user_input=True, traverse_string=True)
6606817a 5277
5278
ff91cf74 5279def get_first(obj, keys, **kwargs):
5280 return traverse_obj(obj, (..., *variadic(keys)), **kwargs, get_all=False)
5281
5282
4b4b7f74 5283def variadic(x, allowed_types=(str, bytes, dict)):
cb89cfc1 5284 return x if isinstance(x, collections.abc.Iterable) and not isinstance(x, allowed_types) else (x,)
bd50a52b
THD
5285
5286
3e9b66d7
LNO
5287def decode_base(value, digits):
5288 # This will convert given base-x string to scalar (long or int)
5289 table = {char: index for index, char in enumerate(digits)}
5290 result = 0
5291 base = len(digits)
5292 for chr in value:
5293 result *= base
5294 result += table[chr]
5295 return result
5296
5297
5298def time_seconds(**kwargs):
5299 t = datetime.datetime.now(datetime.timezone(datetime.timedelta(**kwargs)))
5300 return t.timestamp()
5301
5302
49fa4d9a
N
5303# create a JSON Web Signature (jws) with HS256 algorithm
5304# the resulting format is in JWS Compact Serialization
5305# implemented following JWT https://www.rfc-editor.org/rfc/rfc7519.html
5306# implemented following JWS https://www.rfc-editor.org/rfc/rfc7515.html
5307def jwt_encode_hs256(payload_data, key, headers={}):
5308 header_data = {
5309 'alg': 'HS256',
5310 'typ': 'JWT',
5311 }
5312 if headers:
5313 header_data.update(headers)
5314 header_b64 = base64.b64encode(json.dumps(header_data).encode('utf-8'))
5315 payload_b64 = base64.b64encode(json.dumps(payload_data).encode('utf-8'))
5316 h = hmac.new(key.encode('utf-8'), header_b64 + b'.' + payload_b64, hashlib.sha256)
5317 signature_b64 = base64.b64encode(h.digest())
5318 token = header_b64 + b'.' + payload_b64 + b'.' + signature_b64
5319 return token
819e0531 5320
5321
16b0d7e6 5322# can be extended in future to verify the signature and parse header and return the algorithm used if it's not HS256
5323def jwt_decode_hs256(jwt):
5324 header_b64, payload_b64, signature_b64 = jwt.split('.')
5325 payload_data = json.loads(base64.urlsafe_b64decode(payload_b64))
5326 return payload_data
5327
5328
819e0531 5329def supports_terminal_sequences(stream):
5330 if compat_os_name == 'nt':
e3c7d495 5331 from .compat import WINDOWS_VT_MODE # Must be imported locally
5332 if not WINDOWS_VT_MODE or get_windows_version() < (10, 0, 10586):
819e0531 5333 return False
5334 elif not os.getenv('TERM'):
5335 return False
5336 try:
5337 return stream.isatty()
5338 except BaseException:
5339 return False
5340
5341
ec11a9f4 5342_terminal_sequences_re = re.compile('\033\\[[^m]+m')
5343
5344
5345def remove_terminal_sequences(string):
5346 return _terminal_sequences_re.sub('', string)
5347
5348
5349def number_of_digits(number):
5350 return len('%d' % number)
34921b43 5351
5352
5353def join_nonempty(*values, delim='-', from_dict=None):
5354 if from_dict is not None:
c586f9e8 5355 values = map(from_dict.get, values)
34921b43 5356 return delim.join(map(str, filter(None, values)))
06e57990 5357
5358
27231526
ZM
5359def scale_thumbnails_to_max_format_width(formats, thumbnails, url_width_re):
5360 """
5361 Find the largest format dimensions in terms of video width and, for each thumbnail:
5362 * Modify the URL: Match the width with the provided regex and replace with the former width
5363 * Update dimensions
5364
5365 This function is useful with video services that scale the provided thumbnails on demand
5366 """
5367 _keys = ('width', 'height')
5368 max_dimensions = max(
5369 [tuple(format.get(k) or 0 for k in _keys) for format in formats],
5370 default=(0, 0))
5371 if not max_dimensions[0]:
5372 return thumbnails
5373 return [
5374 merge_dicts(
5375 {'url': re.sub(url_width_re, str(max_dimensions[0]), thumbnail['url'])},
5376 dict(zip(_keys, max_dimensions)), thumbnail)
5377 for thumbnail in thumbnails
5378 ]
5379
5380
93c8410d
LNO
5381def parse_http_range(range):
5382 """ Parse value of "Range" or "Content-Range" HTTP header into tuple. """
5383 if not range:
5384 return None, None, None
5385 crg = re.search(r'bytes[ =](\d+)-(\d+)?(?:/(\d+))?', range)
5386 if not crg:
5387 return None, None, None
5388 return int(crg.group(1)), int_or_none(crg.group(2)), int_or_none(crg.group(3))
5389
5390
06e57990 5391class Config:
5392 own_args = None
5393 filename = None
5394 __initialized = False
5395
5396 def __init__(self, parser, label=None):
5397 self._parser, self.label = parser, label
5398 self._loaded_paths, self.configs = set(), []
5399
5400 def init(self, args=None, filename=None):
5401 assert not self.__initialized
65662dff 5402 directory = ''
06e57990 5403 if filename:
5404 location = os.path.realpath(filename)
65662dff 5405 directory = os.path.dirname(location)
06e57990 5406 if location in self._loaded_paths:
5407 return False
5408 self._loaded_paths.add(location)
5409
5410 self.__initialized = True
5411 self.own_args, self.filename = args, filename
5412 for location in self._parser.parse_args(args)[0].config_locations or []:
65662dff 5413 location = os.path.join(directory, expand_path(location))
06e57990 5414 if os.path.isdir(location):
5415 location = os.path.join(location, 'yt-dlp.conf')
5416 if not os.path.exists(location):
5417 self._parser.error(f'config location {location} does not exist')
5418 self.append_config(self.read_file(location), location)
5419 return True
5420
5421 def __str__(self):
5422 label = join_nonempty(
5423 self.label, 'config', f'"{self.filename}"' if self.filename else '',
5424 delim=' ')
5425 return join_nonempty(
5426 self.own_args is not None and f'{label[0].upper()}{label[1:]}: {self.hide_login_info(self.own_args)}',
5427 *(f'\n{c}'.replace('\n', '\n| ')[1:] for c in self.configs),
5428 delim='\n')
5429
5430 @staticmethod
5431 def read_file(filename, default=[]):
5432 try:
5433 optionf = open(filename)
5434 except IOError:
5435 return default # silently skip if file is not present
5436 try:
5437 # FIXME: https://github.com/ytdl-org/youtube-dl/commit/dfe5fa49aed02cf36ba9f743b11b0903554b5e56
5438 contents = optionf.read()
5439 if sys.version_info < (3,):
5440 contents = contents.decode(preferredencoding())
5441 res = compat_shlex_split(contents, comments=True)
5442 finally:
5443 optionf.close()
5444 return res
5445
5446 @staticmethod
5447 def hide_login_info(opts):
5448 PRIVATE_OPTS = set(['-p', '--password', '-u', '--username', '--video-password', '--ap-password', '--ap-username'])
5449 eqre = re.compile('^(?P<key>' + ('|'.join(re.escape(po) for po in PRIVATE_OPTS)) + ')=.+$')
5450
5451 def _scrub_eq(o):
5452 m = eqre.match(o)
5453 if m:
5454 return m.group('key') + '=PRIVATE'
5455 else:
5456 return o
5457
5458 opts = list(map(_scrub_eq, opts))
5459 for idx, opt in enumerate(opts):
5460 if opt in PRIVATE_OPTS and idx + 1 < len(opts):
5461 opts[idx + 1] = 'PRIVATE'
5462 return opts
5463
5464 def append_config(self, *args, label=None):
5465 config = type(self)(self._parser, label)
5466 config._loaded_paths = self._loaded_paths
5467 if config.init(*args):
5468 self.configs.append(config)
5469
5470 @property
5471 def all_args(self):
5472 for config in reversed(self.configs):
5473 yield from config.all_args
5474 yield from self.own_args or []
5475
5476 def parse_args(self):
5477 return self._parser.parse_args(list(self.all_args))
da42679b
LNO
5478
5479
5480class WebSocketsWrapper():
5481 """Wraps websockets module to use in non-async scopes"""
5482
3cea3edd 5483 def __init__(self, url, headers=None, connect=True):
da42679b
LNO
5484 self.loop = asyncio.events.new_event_loop()
5485 self.conn = compat_websockets.connect(
5486 url, extra_headers=headers, ping_interval=None,
5487 close_timeout=float('inf'), loop=self.loop, ping_timeout=float('inf'))
3cea3edd
LNO
5488 if connect:
5489 self.__enter__()
15dfb392 5490 atexit.register(self.__exit__, None, None, None)
da42679b
LNO
5491
5492 def __enter__(self):
3cea3edd
LNO
5493 if not self.pool:
5494 self.pool = self.run_with_loop(self.conn.__aenter__(), self.loop)
da42679b
LNO
5495 return self
5496
5497 def send(self, *args):
5498 self.run_with_loop(self.pool.send(*args), self.loop)
5499
5500 def recv(self, *args):
5501 return self.run_with_loop(self.pool.recv(*args), self.loop)
5502
5503 def __exit__(self, type, value, traceback):
5504 try:
5505 return self.run_with_loop(self.conn.__aexit__(type, value, traceback), self.loop)
5506 finally:
5507 self.loop.close()
15dfb392 5508 self._cancel_all_tasks(self.loop)
da42679b
LNO
5509
5510 # taken from https://github.com/python/cpython/blob/3.9/Lib/asyncio/runners.py with modifications
5511 # for contributors: If there's any new library using asyncio needs to be run in non-async, move these function out of this class
5512 @staticmethod
5513 def run_with_loop(main, loop):
5514 if not asyncio.coroutines.iscoroutine(main):
5515 raise ValueError(f'a coroutine was expected, got {main!r}')
5516
5517 try:
5518 return loop.run_until_complete(main)
5519 finally:
5520 loop.run_until_complete(loop.shutdown_asyncgens())
5521 if hasattr(loop, 'shutdown_default_executor'):
5522 loop.run_until_complete(loop.shutdown_default_executor())
5523
5524 @staticmethod
5525 def _cancel_all_tasks(loop):
5526 to_cancel = asyncio.tasks.all_tasks(loop)
5527
5528 if not to_cancel:
5529 return
5530
5531 for task in to_cancel:
5532 task.cancel()
5533
5534 loop.run_until_complete(
5535 asyncio.tasks.gather(*to_cancel, loop=loop, return_exceptions=True))
5536
5537 for task in to_cancel:
5538 if task.cancelled():
5539 continue
5540 if task.exception() is not None:
5541 loop.call_exception_handler({
5542 'message': 'unhandled exception during asyncio.run() shutdown',
5543 'exception': task.exception(),
5544 'task': task,
5545 })
5546
5547
5548has_websockets = bool(compat_websockets)
8b7539d2 5549
5550
5551def merge_headers(*dicts):
08d30158 5552 """Merge dicts of http headers case insensitively, prioritizing the latter ones"""
76aa9913 5553 return {k.title(): v for k, v in itertools.chain.from_iterable(map(dict.items, dicts))}
28787f16 5554
5555
5556class classproperty:
5557 def __init__(self, f):
5558 self.f = f
5559
5560 def __get__(self, _, cls):
5561 return self.f(cls)