]> jfr.im git - yt-dlp.git/blame - youtube_dl/utils.py
[nbc:news] Remove unnecessary compat_str
[yt-dlp.git] / youtube_dl / utils.py
CommitLineData
d77c3dfd
FV
1#!/usr/bin/env python
2# -*- coding: utf-8 -*-
3
ecc0c5ee
PH
4from __future__ import unicode_literals
5
1e399778 6import base64
912b38b4 7import calendar
676eb3f2 8import codecs
62e609ab 9import contextlib
e3946f98 10import ctypes
c496ca96
PH
11import datetime
12import email.utils
f45c185f 13import errno
be4a824d 14import functools
d77c3dfd 15import gzip
b7ab0590 16import itertools
03f9daab 17import io
f4bfd65f 18import json
d77c3dfd 19import locale
02dbf93f 20import math
347de493 21import operator
d77c3dfd 22import os
4eb7f1d1 23import pipes
c496ca96 24import platform
d77c3dfd 25import re
13ebea79 26import ssl
c496ca96 27import socket
b53466e1 28import struct
1c088fa8 29import subprocess
d77c3dfd 30import sys
181c8655 31import tempfile
01951dda 32import traceback
bcf89ce6 33import xml.etree.ElementTree
d77c3dfd 34import zlib
d77c3dfd 35
8c25f81b 36from .compat import (
8f9312c3 37 compat_basestring,
8c25f81b 38 compat_chr,
36e6f62c 39 compat_etree_fromstring,
8c25f81b 40 compat_html_entities,
be4a824d 41 compat_http_client,
c86b6142 42 compat_kwargs,
8c25f81b 43 compat_parse_qs,
be4a824d 44 compat_socket_create_connection,
8c25f81b
PH
45 compat_str,
46 compat_urllib_error,
47 compat_urllib_parse,
48 compat_urllib_parse_urlparse,
49 compat_urllib_request,
50 compat_urlparse,
7d4111ed 51 shlex_quote,
8c25f81b 52)
4644ac55
S
53
54
468e2e92
FV
55# This is not clearly defined otherwise
56compiled_regex_type = type(re.compile(''))
57
3e669f36 58std_headers = {
18313934 59 'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64; rv:10.0) Gecko/20150101 Firefox/20.0 (Chrome)',
59ae15a5
PH
60 'Accept-Charset': 'ISO-8859-1,utf-8;q=0.7,*;q=0.7',
61 'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
62 'Accept-Encoding': 'gzip, deflate',
63 'Accept-Language': 'en-us,en;q=0.5',
3e669f36 64}
f427df17 65
5f6a1245 66
bf42a990
S
67NO_DEFAULT = object()
68
7105440c
YCH
69ENGLISH_MONTH_NAMES = [
70 'January', 'February', 'March', 'April', 'May', 'June',
71 'July', 'August', 'September', 'October', 'November', 'December']
72
73
d77c3dfd 74def preferredencoding():
59ae15a5 75 """Get preferred encoding.
d77c3dfd 76
59ae15a5
PH
77 Returns the best encoding scheme for the system, based on
78 locale.getpreferredencoding() and some further tweaks.
79 """
80 try:
81 pref = locale.getpreferredencoding()
28e614de 82 'TEST'.encode(pref)
70a1165b 83 except Exception:
59ae15a5 84 pref = 'UTF-8'
bae611f2 85
59ae15a5 86 return pref
d77c3dfd 87
f4bfd65f 88
181c8655 89def write_json_file(obj, fn):
1394646a 90 """ Encode obj as JSON and write it to fn, atomically if possible """
181c8655 91
92120217 92 fn = encodeFilename(fn)
61ee5aeb 93 if sys.version_info < (3, 0) and sys.platform != 'win32':
ec5f6016
JMF
94 encoding = get_filesystem_encoding()
95 # os.path.basename returns a bytes object, but NamedTemporaryFile
96 # will fail if the filename contains non ascii characters unless we
97 # use a unicode object
98 path_basename = lambda f: os.path.basename(fn).decode(encoding)
99 # the same for os.path.dirname
100 path_dirname = lambda f: os.path.dirname(fn).decode(encoding)
101 else:
102 path_basename = os.path.basename
103 path_dirname = os.path.dirname
104
73159f99
S
105 args = {
106 'suffix': '.tmp',
ec5f6016
JMF
107 'prefix': path_basename(fn) + '.',
108 'dir': path_dirname(fn),
73159f99
S
109 'delete': False,
110 }
111
181c8655
PH
112 # In Python 2.x, json.dump expects a bytestream.
113 # In Python 3.x, it writes to a character stream
114 if sys.version_info < (3, 0):
73159f99 115 args['mode'] = 'wb'
181c8655 116 else:
73159f99
S
117 args.update({
118 'mode': 'w',
119 'encoding': 'utf-8',
120 })
121
c86b6142 122 tf = tempfile.NamedTemporaryFile(**compat_kwargs(args))
181c8655
PH
123
124 try:
125 with tf:
126 json.dump(obj, tf)
1394646a
IK
127 if sys.platform == 'win32':
128 # Need to remove existing file on Windows, else os.rename raises
129 # WindowsError or FileExistsError.
130 try:
131 os.unlink(fn)
132 except OSError:
133 pass
181c8655 134 os.rename(tf.name, fn)
70a1165b 135 except Exception:
181c8655
PH
136 try:
137 os.remove(tf.name)
138 except OSError:
139 pass
140 raise
141
142
143if sys.version_info >= (2, 7):
ee114368 144 def find_xpath_attr(node, xpath, key, val=None):
59ae56fa 145 """ Find the xpath xpath[@key=val] """
5d2354f1 146 assert re.match(r'^[a-zA-Z_-]+$', key)
ee114368
S
147 if val:
148 assert re.match(r'^[a-zA-Z0-9@\s:._-]*$', val)
149 expr = xpath + ('[@%s]' % key if val is None else "[@%s='%s']" % (key, val))
59ae56fa
PH
150 return node.find(expr)
151else:
ee114368 152 def find_xpath_attr(node, xpath, key, val=None):
4eefbfdb
PH
153 # Here comes the crazy part: In 2.6, if the xpath is a unicode,
154 # .//node does not match if a node is a direct child of . !
8f9312c3 155 if isinstance(xpath, compat_str):
4eefbfdb
PH
156 xpath = xpath.encode('ascii')
157
59ae56fa 158 for f in node.findall(xpath):
ee114368
S
159 if key not in f.attrib:
160 continue
161 if val is None or f.attrib.get(key) == val:
59ae56fa
PH
162 return f
163 return None
164
d7e66d39
JMF
165# On python2.6 the xml.etree.ElementTree.Element methods don't support
166# the namespace parameter
5f6a1245
JW
167
168
d7e66d39
JMF
169def xpath_with_ns(path, ns_map):
170 components = [c.split(':') for c in path.split('/')]
171 replaced = []
172 for c in components:
173 if len(c) == 1:
174 replaced.append(c[0])
175 else:
176 ns, tag = c
177 replaced.append('{%s}%s' % (ns_map[ns], tag))
178 return '/'.join(replaced)
179
d77c3dfd 180
a41fb80c 181def xpath_element(node, xpath, name=None, fatal=False, default=NO_DEFAULT):
578c0745
S
182 def _find_xpath(xpath):
183 if sys.version_info < (2, 7): # Crazy 2.6
184 xpath = xpath.encode('ascii')
185 return node.find(xpath)
186
187 if isinstance(xpath, (str, compat_str)):
188 n = _find_xpath(xpath)
189 else:
190 for xp in xpath:
191 n = _find_xpath(xp)
192 if n is not None:
193 break
d74bebd5 194
8e636da4 195 if n is None:
bf42a990
S
196 if default is not NO_DEFAULT:
197 return default
198 elif fatal:
bf0ff932
PH
199 name = xpath if name is None else name
200 raise ExtractorError('Could not find XML element %s' % name)
201 else:
202 return None
a41fb80c
S
203 return n
204
205
206def xpath_text(node, xpath, name=None, fatal=False, default=NO_DEFAULT):
8e636da4
S
207 n = xpath_element(node, xpath, name, fatal=fatal, default=default)
208 if n is None or n == default:
209 return n
210 if n.text is None:
211 if default is not NO_DEFAULT:
212 return default
213 elif fatal:
214 name = xpath if name is None else name
215 raise ExtractorError('Could not find XML element\'s text %s' % name)
216 else:
217 return None
218 return n.text
a41fb80c
S
219
220
221def xpath_attr(node, xpath, key, name=None, fatal=False, default=NO_DEFAULT):
222 n = find_xpath_attr(node, xpath, key)
223 if n is None:
224 if default is not NO_DEFAULT:
225 return default
226 elif fatal:
227 name = '%s[@%s]' % (xpath, key) if name is None else name
228 raise ExtractorError('Could not find XML attribute %s' % name)
229 else:
230 return None
231 return n.attrib[key]
bf0ff932
PH
232
233
9e6dd238 234def get_element_by_id(id, html):
43e8fafd
ND
235 """Return the content of the tag with the specified ID in the passed HTML document"""
236 return get_element_by_attribute("id", id, html)
237
12ea2f30 238
43e8fafd
ND
239def get_element_by_attribute(attribute, value, html):
240 """Return the content of the tag with the specified attribute in the passed HTML document"""
9e6dd238 241
38285056
PH
242 m = re.search(r'''(?xs)
243 <([a-zA-Z0-9:._-]+)
244 (?:\s+[a-zA-Z0-9:._-]+(?:=[a-zA-Z0-9:._-]+|="[^"]+"|='[^']+'))*?
245 \s+%s=['"]?%s['"]?
246 (?:\s+[a-zA-Z0-9:._-]+(?:=[a-zA-Z0-9:._-]+|="[^"]+"|='[^']+'))*?
247 \s*>
248 (?P<content>.*?)
249 </\1>
250 ''' % (re.escape(attribute), re.escape(value)), html)
251
252 if not m:
253 return None
254 res = m.group('content')
255
256 if res.startswith('"') or res.startswith("'"):
257 res = res[1:-1]
a921f407 258
38285056 259 return unescapeHTML(res)
a921f407 260
9e6dd238
FV
261
262def clean_html(html):
59ae15a5 263 """Clean an HTML snippet into a readable string"""
dd622d7c
PH
264
265 if html is None: # Convenience for sanitizing descriptions etc.
266 return html
267
59ae15a5
PH
268 # Newline vs <br />
269 html = html.replace('\n', ' ')
6b3aef80
FV
270 html = re.sub(r'\s*<\s*br\s*/?\s*>\s*', '\n', html)
271 html = re.sub(r'<\s*/\s*p\s*>\s*<\s*p[^>]*>', '\n', html)
59ae15a5
PH
272 # Strip html tags
273 html = re.sub('<.*?>', '', html)
274 # Replace html entities
275 html = unescapeHTML(html)
7decf895 276 return html.strip()
9e6dd238
FV
277
278
d77c3dfd 279def sanitize_open(filename, open_mode):
59ae15a5
PH
280 """Try to open the given filename, and slightly tweak it if this fails.
281
282 Attempts to open the given filename. If this fails, it tries to change
283 the filename slightly, step by step, until it's either able to open it
284 or it fails and raises a final exception, like the standard open()
285 function.
286
287 It returns the tuple (stream, definitive_file_name).
288 """
289 try:
28e614de 290 if filename == '-':
59ae15a5
PH
291 if sys.platform == 'win32':
292 import msvcrt
293 msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
898280a0 294 return (sys.stdout.buffer if hasattr(sys.stdout, 'buffer') else sys.stdout, filename)
59ae15a5
PH
295 stream = open(encodeFilename(filename), open_mode)
296 return (stream, filename)
297 except (IOError, OSError) as err:
f45c185f
PH
298 if err.errno in (errno.EACCES,):
299 raise
59ae15a5 300
f45c185f 301 # In case of error, try to remove win32 forbidden chars
d55de57b 302 alt_filename = sanitize_path(filename)
f45c185f
PH
303 if alt_filename == filename:
304 raise
305 else:
306 # An exception here should be caught in the caller
d55de57b 307 stream = open(encodeFilename(alt_filename), open_mode)
f45c185f 308 return (stream, alt_filename)
d77c3dfd
FV
309
310
311def timeconvert(timestr):
59ae15a5
PH
312 """Convert RFC 2822 defined time string into system timestamp"""
313 timestamp = None
314 timetuple = email.utils.parsedate_tz(timestr)
315 if timetuple is not None:
316 timestamp = email.utils.mktime_tz(timetuple)
317 return timestamp
1c469a94 318
5f6a1245 319
796173d0 320def sanitize_filename(s, restricted=False, is_id=False):
59ae15a5
PH
321 """Sanitizes a string so it could be used as part of a filename.
322 If restricted is set, use a stricter subset of allowed characters.
796173d0 323 Set is_id if this is not an arbitrary string, but an ID that should be kept if possible
59ae15a5
PH
324 """
325 def replace_insane(char):
326 if char == '?' or ord(char) < 32 or ord(char) == 127:
327 return ''
328 elif char == '"':
329 return '' if restricted else '\''
330 elif char == ':':
331 return '_-' if restricted else ' -'
332 elif char in '\\/|*<>':
333 return '_'
627dcfff 334 if restricted and (char in '!&\'()[]{}$;`^,#' or char.isspace()):
59ae15a5
PH
335 return '_'
336 if restricted and ord(char) > 127:
337 return '_'
338 return char
339
2aeb06d6
PH
340 # Handle timestamps
341 s = re.sub(r'[0-9]+(?::[0-9]+)+', lambda m: m.group(0).replace(':', '_'), s)
28e614de 342 result = ''.join(map(replace_insane, s))
796173d0
PH
343 if not is_id:
344 while '__' in result:
345 result = result.replace('__', '_')
346 result = result.strip('_')
347 # Common case of "Foreign band name - English song title"
348 if restricted and result.startswith('-_'):
349 result = result[2:]
5a42414b
PH
350 if result.startswith('-'):
351 result = '_' + result[len('-'):]
a7440261 352 result = result.lstrip('.')
796173d0
PH
353 if not result:
354 result = '_'
59ae15a5 355 return result
d77c3dfd 356
5f6a1245 357
a2aaf4db
S
358def sanitize_path(s):
359 """Sanitizes and normalizes path on Windows"""
360 if sys.platform != 'win32':
361 return s
be531ef1
S
362 drive_or_unc, _ = os.path.splitdrive(s)
363 if sys.version_info < (2, 7) and not drive_or_unc:
364 drive_or_unc, _ = os.path.splitunc(s)
365 norm_path = os.path.normpath(remove_start(s, drive_or_unc)).split(os.path.sep)
366 if drive_or_unc:
a2aaf4db
S
367 norm_path.pop(0)
368 sanitized_path = [
c90d16cf 369 path_part if path_part in ['.', '..'] else re.sub('(?:[/<>:"\\|\\\\?\\*]|[\s.]$)', '#', path_part)
a2aaf4db 370 for path_part in norm_path]
be531ef1
S
371 if drive_or_unc:
372 sanitized_path.insert(0, drive_or_unc + os.path.sep)
a2aaf4db
S
373 return os.path.join(*sanitized_path)
374
375
67dda517
S
376# Prepend protocol-less URLs with `http:` scheme in order to mitigate the number of
377# unwanted failures due to missing protocol
378def sanitized_Request(url, *args, **kwargs):
379 return compat_urllib_request.Request(
380 'http:%s' % url if url.startswith('//') else url, *args, **kwargs)
381
382
d77c3dfd 383def orderedSet(iterable):
59ae15a5
PH
384 """ Remove all duplicates from the input iterable """
385 res = []
386 for el in iterable:
387 if el not in res:
388 res.append(el)
389 return res
d77c3dfd 390
912b38b4 391
4e408e47
PH
392def _htmlentity_transform(entity):
393 """Transforms an HTML entity to a character."""
394 # Known non-numeric HTML entity
395 if entity in compat_html_entities.name2codepoint:
396 return compat_chr(compat_html_entities.name2codepoint[entity])
397
91757b0f 398 mobj = re.match(r'#(x[0-9a-fA-F]+|[0-9]+)', entity)
4e408e47
PH
399 if mobj is not None:
400 numstr = mobj.group(1)
28e614de 401 if numstr.startswith('x'):
4e408e47 402 base = 16
28e614de 403 numstr = '0%s' % numstr
4e408e47
PH
404 else:
405 base = 10
7aefc49c
S
406 # See https://github.com/rg3/youtube-dl/issues/7518
407 try:
408 return compat_chr(int(numstr, base))
409 except ValueError:
410 pass
4e408e47
PH
411
412 # Unknown entity in name, return its literal representation
7a3f0c00 413 return '&%s;' % entity
4e408e47
PH
414
415
d77c3dfd 416def unescapeHTML(s):
912b38b4
PH
417 if s is None:
418 return None
419 assert type(s) == compat_str
d77c3dfd 420
4e408e47
PH
421 return re.sub(
422 r'&([^;]+);', lambda m: _htmlentity_transform(m.group(1)), s)
d77c3dfd 423
8bf48f23 424
aa49acd1
S
425def get_subprocess_encoding():
426 if sys.platform == 'win32' and sys.getwindowsversion()[0] >= 5:
427 # For subprocess calls, encode with locale encoding
428 # Refer to http://stackoverflow.com/a/9951851/35070
429 encoding = preferredencoding()
430 else:
431 encoding = sys.getfilesystemencoding()
432 if encoding is None:
433 encoding = 'utf-8'
434 return encoding
435
436
8bf48f23 437def encodeFilename(s, for_subprocess=False):
59ae15a5
PH
438 """
439 @param s The name of the file
440 """
d77c3dfd 441
8bf48f23 442 assert type(s) == compat_str
d77c3dfd 443
59ae15a5
PH
444 # Python 3 has a Unicode API
445 if sys.version_info >= (3, 0):
446 return s
0f00efed 447
aa49acd1
S
448 # Pass '' directly to use Unicode APIs on Windows 2000 and up
449 # (Detecting Windows NT 4 is tricky because 'major >= 4' would
450 # match Windows 9x series as well. Besides, NT 4 is obsolete.)
451 if not for_subprocess and sys.platform == 'win32' and sys.getwindowsversion()[0] >= 5:
452 return s
453
454 return s.encode(get_subprocess_encoding(), 'ignore')
455
456
457def decodeFilename(b, for_subprocess=False):
458
459 if sys.version_info >= (3, 0):
460 return b
461
462 if not isinstance(b, bytes):
463 return b
464
465 return b.decode(get_subprocess_encoding(), 'ignore')
8bf48f23 466
f07b74fc
PH
467
468def encodeArgument(s):
469 if not isinstance(s, compat_str):
470 # Legacy code that uses byte strings
471 # Uncomment the following line after fixing all post processors
7af808a5 472 # assert False, 'Internal error: %r should be of type %r, is %r' % (s, compat_str, type(s))
f07b74fc
PH
473 s = s.decode('ascii')
474 return encodeFilename(s, True)
475
476
aa49acd1
S
477def decodeArgument(b):
478 return decodeFilename(b, True)
479
480
8271226a
PH
481def decodeOption(optval):
482 if optval is None:
483 return optval
484 if isinstance(optval, bytes):
485 optval = optval.decode(preferredencoding())
486
487 assert isinstance(optval, compat_str)
488 return optval
1c256f70 489
5f6a1245 490
4539dd30
PH
491def formatSeconds(secs):
492 if secs > 3600:
493 return '%d:%02d:%02d' % (secs // 3600, (secs % 3600) // 60, secs % 60)
494 elif secs > 60:
495 return '%d:%02d' % (secs // 60, secs % 60)
496 else:
497 return '%d' % secs
498
a0ddb8a2 499
be4a824d
PH
500def make_HTTPS_handler(params, **kwargs):
501 opts_no_check_certificate = params.get('nocheckcertificate', False)
0db261ba 502 if hasattr(ssl, 'create_default_context'): # Python >= 3.4 or 2.7.9
be5f2c19 503 context = ssl.create_default_context(ssl.Purpose.SERVER_AUTH)
0db261ba 504 if opts_no_check_certificate:
be5f2c19 505 context.check_hostname = False
0db261ba 506 context.verify_mode = ssl.CERT_NONE
a2366922 507 try:
be4a824d 508 return YoutubeDLHTTPSHandler(params, context=context, **kwargs)
a2366922
PH
509 except TypeError:
510 # Python 2.7.8
511 # (create_default_context present but HTTPSHandler has no context=)
512 pass
513
514 if sys.version_info < (3, 2):
d7932313 515 return YoutubeDLHTTPSHandler(params, **kwargs)
aa37e3d4 516 else: # Python < 3.4
d7932313 517 context = ssl.SSLContext(ssl.PROTOCOL_TLSv1)
ea6d901e 518 context.verify_mode = (ssl.CERT_NONE
dca08720 519 if opts_no_check_certificate
ea6d901e 520 else ssl.CERT_REQUIRED)
303b479e 521 context.set_default_verify_paths()
be4a824d 522 return YoutubeDLHTTPSHandler(params, context=context, **kwargs)
ea6d901e 523
732ea2f0 524
08f2a92c
JMF
525def bug_reports_message():
526 if ytdl_is_updateable():
527 update_cmd = 'type youtube-dl -U to update'
528 else:
529 update_cmd = 'see https://yt-dl.org/update on how to update'
530 msg = '; please report this issue on https://yt-dl.org/bug .'
531 msg += ' Make sure you are using the latest version; %s.' % update_cmd
532 msg += ' Be sure to call youtube-dl with the --verbose flag and include its complete output.'
533 return msg
534
535
1c256f70
PH
536class ExtractorError(Exception):
537 """Error during info extraction."""
5f6a1245 538
d11271dd 539 def __init__(self, msg, tb=None, expected=False, cause=None, video_id=None):
9a82b238
PH
540 """ tb, if given, is the original traceback (so that it can be printed out).
541 If expected is set, this is a normal error message and most likely not a bug in youtube-dl.
542 """
543
544 if sys.exc_info()[0] in (compat_urllib_error.URLError, socket.timeout, UnavailableVideoError):
545 expected = True
d11271dd
PH
546 if video_id is not None:
547 msg = video_id + ': ' + msg
410f3e73 548 if cause:
28e614de 549 msg += ' (caused by %r)' % cause
9a82b238 550 if not expected:
08f2a92c 551 msg += bug_reports_message()
1c256f70 552 super(ExtractorError, self).__init__(msg)
d5979c5d 553
1c256f70 554 self.traceback = tb
8cc83b8d 555 self.exc_info = sys.exc_info() # preserve original exception
2eabb802 556 self.cause = cause
d11271dd 557 self.video_id = video_id
1c256f70 558
01951dda
PH
559 def format_traceback(self):
560 if self.traceback is None:
561 return None
28e614de 562 return ''.join(traceback.format_tb(self.traceback))
01951dda 563
1c256f70 564
416c7fcb
PH
565class UnsupportedError(ExtractorError):
566 def __init__(self, url):
567 super(UnsupportedError, self).__init__(
568 'Unsupported URL: %s' % url, expected=True)
569 self.url = url
570
571
55b3e45b
JMF
572class RegexNotFoundError(ExtractorError):
573 """Error when a regex didn't match"""
574 pass
575
576
d77c3dfd 577class DownloadError(Exception):
59ae15a5 578 """Download Error exception.
d77c3dfd 579
59ae15a5
PH
580 This exception may be thrown by FileDownloader objects if they are not
581 configured to continue on errors. They will contain the appropriate
582 error message.
583 """
5f6a1245 584
8cc83b8d
FV
585 def __init__(self, msg, exc_info=None):
586 """ exc_info, if given, is the original exception that caused the trouble (as returned by sys.exc_info()). """
587 super(DownloadError, self).__init__(msg)
588 self.exc_info = exc_info
d77c3dfd
FV
589
590
591class SameFileError(Exception):
59ae15a5 592 """Same File exception.
d77c3dfd 593
59ae15a5
PH
594 This exception will be thrown by FileDownloader objects if they detect
595 multiple files would have to be downloaded to the same file on disk.
596 """
597 pass
d77c3dfd
FV
598
599
600class PostProcessingError(Exception):
59ae15a5 601 """Post Processing exception.
d77c3dfd 602
59ae15a5
PH
603 This exception may be raised by PostProcessor's .run() method to
604 indicate an error in the postprocessing task.
605 """
5f6a1245 606
7851b379
PH
607 def __init__(self, msg):
608 self.msg = msg
d77c3dfd 609
5f6a1245 610
d77c3dfd 611class MaxDownloadsReached(Exception):
59ae15a5
PH
612 """ --max-downloads limit has been reached. """
613 pass
d77c3dfd
FV
614
615
616class UnavailableVideoError(Exception):
59ae15a5 617 """Unavailable Format exception.
d77c3dfd 618
59ae15a5
PH
619 This exception will be thrown when a video is requested
620 in a format that is not available for that video.
621 """
622 pass
d77c3dfd
FV
623
624
625class ContentTooShortError(Exception):
59ae15a5 626 """Content Too Short exception.
d77c3dfd 627
59ae15a5
PH
628 This exception may be raised by FileDownloader objects when a file they
629 download is too small for what the server announced first, indicating
630 the connection was probably interrupted.
631 """
d77c3dfd 632
59ae15a5 633 def __init__(self, downloaded, expected):
2c7ed247 634 # Both in bytes
59ae15a5
PH
635 self.downloaded = downloaded
636 self.expected = expected
d77c3dfd 637
5f6a1245 638
c5a59d93 639def _create_http_connection(ydl_handler, http_class, is_https, *args, **kwargs):
e5e78797
S
640 # Working around python 2 bug (see http://bugs.python.org/issue17849) by limiting
641 # expected HTTP responses to meet HTTP/1.0 or later (see also
642 # https://github.com/rg3/youtube-dl/issues/6727)
643 if sys.version_info < (3, 0):
5a1a2e94 644 kwargs[b'strict'] = True
be4a824d
PH
645 hc = http_class(*args, **kwargs)
646 source_address = ydl_handler._params.get('source_address')
647 if source_address is not None:
648 sa = (source_address, 0)
649 if hasattr(hc, 'source_address'): # Python 2.7+
650 hc.source_address = sa
651 else: # Python 2.6
652 def _hc_connect(self, *args, **kwargs):
653 sock = compat_socket_create_connection(
654 (self.host, self.port), self.timeout, sa)
655 if is_https:
d7932313
PH
656 self.sock = ssl.wrap_socket(
657 sock, self.key_file, self.cert_file,
658 ssl_version=ssl.PROTOCOL_TLSv1)
be4a824d
PH
659 else:
660 self.sock = sock
661 hc.connect = functools.partial(_hc_connect, hc)
662
663 return hc
664
665
87f0e62d 666def handle_youtubedl_headers(headers):
992fc9d6
YCH
667 filtered_headers = headers
668
669 if 'Youtubedl-no-compression' in filtered_headers:
670 filtered_headers = dict((k, v) for k, v in filtered_headers.items() if k.lower() != 'accept-encoding')
87f0e62d 671 del filtered_headers['Youtubedl-no-compression']
87f0e62d 672
992fc9d6 673 return filtered_headers
87f0e62d
YCH
674
675
acebc9cd 676class YoutubeDLHandler(compat_urllib_request.HTTPHandler):
59ae15a5
PH
677 """Handler for HTTP requests and responses.
678
679 This class, when installed with an OpenerDirector, automatically adds
680 the standard headers to every HTTP request and handles gzipped and
681 deflated responses from web servers. If compression is to be avoided in
682 a particular request, the original request in the program code only has
0424ec30 683 to include the HTTP header "Youtubedl-no-compression", which will be
59ae15a5
PH
684 removed before making the real request.
685
686 Part of this code was copied from:
687
688 http://techknack.net/python-urllib2-handlers/
689
690 Andrew Rowls, the author of that code, agreed to release it to the
691 public domain.
692 """
693
be4a824d
PH
694 def __init__(self, params, *args, **kwargs):
695 compat_urllib_request.HTTPHandler.__init__(self, *args, **kwargs)
696 self._params = params
697
698 def http_open(self, req):
699 return self.do_open(functools.partial(
c5a59d93 700 _create_http_connection, self, compat_http_client.HTTPConnection, False),
be4a824d
PH
701 req)
702
59ae15a5
PH
703 @staticmethod
704 def deflate(data):
705 try:
706 return zlib.decompress(data, -zlib.MAX_WBITS)
707 except zlib.error:
708 return zlib.decompress(data)
709
710 @staticmethod
711 def addinfourl_wrapper(stream, headers, url, code):
712 if hasattr(compat_urllib_request.addinfourl, 'getcode'):
713 return compat_urllib_request.addinfourl(stream, headers, url, code)
714 ret = compat_urllib_request.addinfourl(stream, headers, url)
715 ret.code = code
716 return ret
717
acebc9cd 718 def http_request(self, req):
51f267d9
S
719 # According to RFC 3986, URLs can not contain non-ASCII characters, however this is not
720 # always respected by websites, some tend to give out URLs with non percent-encoded
721 # non-ASCII characters (see telemb.py, ard.py [#3412])
722 # urllib chokes on URLs with non-ASCII characters (see http://bugs.python.org/issue3991)
723 # To work around aforementioned issue we will replace request's original URL with
724 # percent-encoded one
725 # Since redirects are also affected (e.g. http://www.southpark.de/alle-episoden/s18e09)
726 # the code of this workaround has been moved here from YoutubeDL.urlopen()
727 url = req.get_full_url()
728 url_escaped = escape_url(url)
729
730 # Substitute URL if any change after escaping
731 if url != url_escaped:
732 req_type = HEADRequest if req.get_method() == 'HEAD' else compat_urllib_request.Request
733 new_req = req_type(
734 url_escaped, data=req.data, headers=req.headers,
735 origin_req_host=req.origin_req_host, unverifiable=req.unverifiable)
736 new_req.timeout = req.timeout
737 req = new_req
738
33ac271b 739 for h, v in std_headers.items():
3d5f7a39
JK
740 # Capitalize is needed because of Python bug 2275: http://bugs.python.org/issue2275
741 # The dict keys are capitalized because of this bug by urllib
742 if h.capitalize() not in req.headers:
33ac271b 743 req.add_header(h, v)
87f0e62d
YCH
744
745 req.headers = handle_youtubedl_headers(req.headers)
989b4b2b
PH
746
747 if sys.version_info < (2, 7) and '#' in req.get_full_url():
748 # Python 2.6 is brain-dead when it comes to fragments
749 req._Request__original = req._Request__original.partition('#')[0]
750 req._Request__r_type = req._Request__r_type.partition('#')[0]
751
59ae15a5
PH
752 return req
753
acebc9cd 754 def http_response(self, req, resp):
59ae15a5
PH
755 old_resp = resp
756 # gzip
757 if resp.headers.get('Content-encoding', '') == 'gzip':
aa3e9507
PH
758 content = resp.read()
759 gz = gzip.GzipFile(fileobj=io.BytesIO(content), mode='rb')
760 try:
761 uncompressed = io.BytesIO(gz.read())
762 except IOError as original_ioerror:
763 # There may be junk add the end of the file
764 # See http://stackoverflow.com/q/4928560/35070 for details
765 for i in range(1, 1024):
766 try:
767 gz = gzip.GzipFile(fileobj=io.BytesIO(content[:-i]), mode='rb')
768 uncompressed = io.BytesIO(gz.read())
769 except IOError:
770 continue
771 break
772 else:
773 raise original_ioerror
774 resp = self.addinfourl_wrapper(uncompressed, old_resp.headers, old_resp.url, old_resp.code)
59ae15a5
PH
775 resp.msg = old_resp.msg
776 # deflate
777 if resp.headers.get('Content-encoding', '') == 'deflate':
778 gz = io.BytesIO(self.deflate(resp.read()))
779 resp = self.addinfourl_wrapper(gz, old_resp.headers, old_resp.url, old_resp.code)
780 resp.msg = old_resp.msg
ad729172
S
781 # Percent-encode redirect URL of Location HTTP header to satisfy RFC 3986 (see
782 # https://github.com/rg3/youtube-dl/issues/6457).
5a4d9ddb
S
783 if 300 <= resp.code < 400:
784 location = resp.headers.get('Location')
785 if location:
786 # As of RFC 2616 default charset is iso-8859-1 that is respected by python 3
787 if sys.version_info >= (3, 0):
788 location = location.encode('iso-8859-1').decode('utf-8')
789 location_escaped = escape_url(location)
790 if location != location_escaped:
791 del resp.headers['Location']
792 resp.headers['Location'] = location_escaped
59ae15a5 793 return resp
0f8d03f8 794
acebc9cd
PH
795 https_request = http_request
796 https_response = http_response
bf50b038 797
5de90176 798
be4a824d
PH
799class YoutubeDLHTTPSHandler(compat_urllib_request.HTTPSHandler):
800 def __init__(self, params, https_conn_class=None, *args, **kwargs):
801 compat_urllib_request.HTTPSHandler.__init__(self, *args, **kwargs)
802 self._https_conn_class = https_conn_class or compat_http_client.HTTPSConnection
803 self._params = params
804
805 def https_open(self, req):
4f264c02
JMF
806 kwargs = {}
807 if hasattr(self, '_context'): # python > 2.6
808 kwargs['context'] = self._context
809 if hasattr(self, '_check_hostname'): # python 3.x
810 kwargs['check_hostname'] = self._check_hostname
be4a824d
PH
811 return self.do_open(functools.partial(
812 _create_http_connection, self, self._https_conn_class, True),
4f264c02 813 req, **kwargs)
be4a824d
PH
814
815
a6420bf5
S
816class YoutubeDLCookieProcessor(compat_urllib_request.HTTPCookieProcessor):
817 def __init__(self, cookiejar=None):
818 compat_urllib_request.HTTPCookieProcessor.__init__(self, cookiejar)
819
820 def http_response(self, request, response):
821 # Python 2 will choke on next HTTP request in row if there are non-ASCII
822 # characters in Set-Cookie HTTP header of last response (see
823 # https://github.com/rg3/youtube-dl/issues/6769).
824 # In order to at least prevent crashing we will percent encode Set-Cookie
825 # header before HTTPCookieProcessor starts processing it.
e28034c5
S
826 # if sys.version_info < (3, 0) and response.headers:
827 # for set_cookie_header in ('Set-Cookie', 'Set-Cookie2'):
828 # set_cookie = response.headers.get(set_cookie_header)
829 # if set_cookie:
830 # set_cookie_escaped = compat_urllib_parse.quote(set_cookie, b"%/;:@&=+$,!~*'()?#[] ")
831 # if set_cookie != set_cookie_escaped:
832 # del response.headers[set_cookie_header]
833 # response.headers[set_cookie_header] = set_cookie_escaped
a6420bf5
S
834 return compat_urllib_request.HTTPCookieProcessor.http_response(self, request, response)
835
836 https_request = compat_urllib_request.HTTPCookieProcessor.http_request
837 https_response = http_response
838
839
08b38d54 840def parse_iso8601(date_str, delimiter='T', timezone=None):
912b38b4
PH
841 """ Return a UNIX timestamp from the given date """
842
843 if date_str is None:
844 return None
845
52c3a6e4
S
846 date_str = re.sub(r'\.[0-9]+', '', date_str)
847
08b38d54
PH
848 if timezone is None:
849 m = re.search(
52c3a6e4 850 r'(?:Z$| ?(?P<sign>\+|-)(?P<hours>[0-9]{2}):?(?P<minutes>[0-9]{2})$)',
08b38d54
PH
851 date_str)
852 if not m:
912b38b4
PH
853 timezone = datetime.timedelta()
854 else:
08b38d54
PH
855 date_str = date_str[:-len(m.group(0))]
856 if not m.group('sign'):
857 timezone = datetime.timedelta()
858 else:
859 sign = 1 if m.group('sign') == '+' else -1
860 timezone = datetime.timedelta(
861 hours=sign * int(m.group('hours')),
862 minutes=sign * int(m.group('minutes')))
52c3a6e4
S
863 try:
864 date_format = '%Y-%m-%d{0}%H:%M:%S'.format(delimiter)
865 dt = datetime.datetime.strptime(date_str, date_format) - timezone
866 return calendar.timegm(dt.timetuple())
867 except ValueError:
868 pass
912b38b4
PH
869
870
42bdd9d0 871def unified_strdate(date_str, day_first=True):
bf50b038 872 """Return a string with the date in the format YYYYMMDD"""
64e7ad60
PH
873
874 if date_str is None:
875 return None
bf50b038 876 upload_date = None
5f6a1245 877 # Replace commas
026fcc04 878 date_str = date_str.replace(',', ' ')
bf50b038 879 # %z (UTC offset) is only supported in python>=3.2
15ac8413
S
880 if not re.match(r'^[0-9]{1,2}-[0-9]{1,2}-[0-9]{4}$', date_str):
881 date_str = re.sub(r' ?(\+|-)[0-9]{2}:?[0-9]{2}$', '', date_str)
42bdd9d0 882 # Remove AM/PM + timezone
9bb8e0a3 883 date_str = re.sub(r'(?i)\s*(?:AM|PM)(?:\s+[A-Z]+)?', '', date_str)
42bdd9d0 884
19e1d359
JMF
885 format_expressions = [
886 '%d %B %Y',
0f99566c 887 '%d %b %Y',
19e1d359
JMF
888 '%B %d %Y',
889 '%b %d %Y',
78ff59d0
PP
890 '%b %dst %Y %I:%M%p',
891 '%b %dnd %Y %I:%M%p',
892 '%b %dth %Y %I:%M%p',
a69801e2 893 '%Y %m %d',
19e1d359 894 '%Y-%m-%d',
fe556f1b 895 '%Y/%m/%d',
19e1d359 896 '%Y/%m/%d %H:%M:%S',
5d73273f 897 '%Y-%m-%d %H:%M:%S',
e9be9a6a 898 '%Y-%m-%d %H:%M:%S.%f',
19e1d359 899 '%d.%m.%Y %H:%M',
b047de6f 900 '%d.%m.%Y %H.%M',
19e1d359 901 '%Y-%m-%dT%H:%M:%SZ',
59040888
PH
902 '%Y-%m-%dT%H:%M:%S.%fZ',
903 '%Y-%m-%dT%H:%M:%S.%f0Z',
2e1fa03b 904 '%Y-%m-%dT%H:%M:%S',
7ff5d5c2 905 '%Y-%m-%dT%H:%M:%S.%f',
5de90176 906 '%Y-%m-%dT%H:%M',
19e1d359 907 ]
42bdd9d0
PH
908 if day_first:
909 format_expressions.extend([
79c21abb 910 '%d-%m-%Y',
776dc399
S
911 '%d.%m.%Y',
912 '%d/%m/%Y',
913 '%d/%m/%y',
42bdd9d0
PH
914 '%d/%m/%Y %H:%M:%S',
915 ])
916 else:
917 format_expressions.extend([
79c21abb 918 '%m-%d-%Y',
776dc399
S
919 '%m.%d.%Y',
920 '%m/%d/%Y',
921 '%m/%d/%y',
42bdd9d0
PH
922 '%m/%d/%Y %H:%M:%S',
923 ])
bf50b038
JMF
924 for expression in format_expressions:
925 try:
926 upload_date = datetime.datetime.strptime(date_str, expression).strftime('%Y%m%d')
5de90176 927 except ValueError:
bf50b038 928 pass
42393ce2
PH
929 if upload_date is None:
930 timetuple = email.utils.parsedate_tz(date_str)
931 if timetuple:
932 upload_date = datetime.datetime(*timetuple[:6]).strftime('%Y%m%d')
6a750402
JMF
933 if upload_date is not None:
934 return compat_str(upload_date)
bf50b038 935
5f6a1245 936
28e614de 937def determine_ext(url, default_ext='unknown_video'):
f4776371
S
938 if url is None:
939 return default_ext
9cb9a5df 940 guess = url.partition('?')[0].rpartition('.')[2]
73e79f2a
PH
941 if re.match(r'^[A-Za-z0-9]+$', guess):
942 return guess
9cb9a5df
S
943 elif guess.rstrip('/') in (
944 'mp4', 'm4a', 'm4p', 'm4b', 'm4r', 'm4v', 'aac',
945 'flv', 'f4v', 'f4a', 'f4b',
946 'webm', 'ogg', 'ogv', 'oga', 'ogx', 'spx', 'opus',
947 'mkv', 'mka', 'mk3d',
948 'avi', 'divx',
949 'mov',
950 'asf', 'wmv', 'wma',
951 '3gp', '3g2',
952 'mp3',
953 'flac',
954 'ape',
955 'wav',
956 'f4f', 'f4m', 'm3u8', 'smil'):
957 return guess.rstrip('/')
73e79f2a 958 else:
cbdbb766 959 return default_ext
73e79f2a 960
5f6a1245 961
d4051a8e 962def subtitles_filename(filename, sub_lang, sub_format):
28e614de 963 return filename.rsplit('.', 1)[0] + '.' + sub_lang + '.' + sub_format
d4051a8e 964
5f6a1245 965
bd558525 966def date_from_str(date_str):
37254abc
JMF
967 """
968 Return a datetime object from a string in the format YYYYMMDD or
969 (now|today)[+-][0-9](day|week|month|year)(s)?"""
970 today = datetime.date.today()
f8795e10 971 if date_str in ('now', 'today'):
37254abc 972 return today
f8795e10
PH
973 if date_str == 'yesterday':
974 return today - datetime.timedelta(days=1)
37254abc
JMF
975 match = re.match('(now|today)(?P<sign>[+-])(?P<time>\d+)(?P<unit>day|week|month|year)(s)?', date_str)
976 if match is not None:
977 sign = match.group('sign')
978 time = int(match.group('time'))
979 if sign == '-':
980 time = -time
981 unit = match.group('unit')
5f6a1245 982 # A bad aproximation?
37254abc
JMF
983 if unit == 'month':
984 unit = 'day'
985 time *= 30
986 elif unit == 'year':
987 unit = 'day'
988 time *= 365
989 unit += 's'
990 delta = datetime.timedelta(**{unit: time})
991 return today + delta
bd558525 992 return datetime.datetime.strptime(date_str, "%Y%m%d").date()
5f6a1245
JW
993
994
e63fc1be 995def hyphenate_date(date_str):
996 """
997 Convert a date in 'YYYYMMDD' format to 'YYYY-MM-DD' format"""
998 match = re.match(r'^(\d\d\d\d)(\d\d)(\d\d)$', date_str)
999 if match is not None:
1000 return '-'.join(match.groups())
1001 else:
1002 return date_str
1003
5f6a1245 1004
bd558525
JMF
1005class DateRange(object):
1006 """Represents a time interval between two dates"""
5f6a1245 1007
bd558525
JMF
1008 def __init__(self, start=None, end=None):
1009 """start and end must be strings in the format accepted by date"""
1010 if start is not None:
1011 self.start = date_from_str(start)
1012 else:
1013 self.start = datetime.datetime.min.date()
1014 if end is not None:
1015 self.end = date_from_str(end)
1016 else:
1017 self.end = datetime.datetime.max.date()
37254abc 1018 if self.start > self.end:
bd558525 1019 raise ValueError('Date range: "%s" , the start date must be before the end date' % self)
5f6a1245 1020
bd558525
JMF
1021 @classmethod
1022 def day(cls, day):
1023 """Returns a range that only contains the given day"""
5f6a1245
JW
1024 return cls(day, day)
1025
bd558525
JMF
1026 def __contains__(self, date):
1027 """Check if the date is in the range"""
37254abc
JMF
1028 if not isinstance(date, datetime.date):
1029 date = date_from_str(date)
1030 return self.start <= date <= self.end
5f6a1245 1031
bd558525 1032 def __str__(self):
5f6a1245 1033 return '%s - %s' % (self.start.isoformat(), self.end.isoformat())
c496ca96
PH
1034
1035
1036def platform_name():
1037 """ Returns the platform name as a compat_str """
1038 res = platform.platform()
1039 if isinstance(res, bytes):
1040 res = res.decode(preferredencoding())
1041
1042 assert isinstance(res, compat_str)
1043 return res
c257baff
PH
1044
1045
b58ddb32
PH
1046def _windows_write_string(s, out):
1047 """ Returns True if the string was written using special methods,
1048 False if it has yet to be written out."""
1049 # Adapted from http://stackoverflow.com/a/3259271/35070
1050
1051 import ctypes
1052 import ctypes.wintypes
1053
1054 WIN_OUTPUT_IDS = {
1055 1: -11,
1056 2: -12,
1057 }
1058
a383a98a
PH
1059 try:
1060 fileno = out.fileno()
1061 except AttributeError:
1062 # If the output stream doesn't have a fileno, it's virtual
1063 return False
aa42e873
PH
1064 except io.UnsupportedOperation:
1065 # Some strange Windows pseudo files?
1066 return False
b58ddb32
PH
1067 if fileno not in WIN_OUTPUT_IDS:
1068 return False
1069
e2f89ec7 1070 GetStdHandle = ctypes.WINFUNCTYPE(
b58ddb32 1071 ctypes.wintypes.HANDLE, ctypes.wintypes.DWORD)(
6ac4e806 1072 (b"GetStdHandle", ctypes.windll.kernel32))
b58ddb32
PH
1073 h = GetStdHandle(WIN_OUTPUT_IDS[fileno])
1074
e2f89ec7 1075 WriteConsoleW = ctypes.WINFUNCTYPE(
b58ddb32
PH
1076 ctypes.wintypes.BOOL, ctypes.wintypes.HANDLE, ctypes.wintypes.LPWSTR,
1077 ctypes.wintypes.DWORD, ctypes.POINTER(ctypes.wintypes.DWORD),
6ac4e806 1078 ctypes.wintypes.LPVOID)((b"WriteConsoleW", ctypes.windll.kernel32))
b58ddb32
PH
1079 written = ctypes.wintypes.DWORD(0)
1080
6ac4e806 1081 GetFileType = ctypes.WINFUNCTYPE(ctypes.wintypes.DWORD, ctypes.wintypes.DWORD)((b"GetFileType", ctypes.windll.kernel32))
b58ddb32
PH
1082 FILE_TYPE_CHAR = 0x0002
1083 FILE_TYPE_REMOTE = 0x8000
e2f89ec7 1084 GetConsoleMode = ctypes.WINFUNCTYPE(
b58ddb32
PH
1085 ctypes.wintypes.BOOL, ctypes.wintypes.HANDLE,
1086 ctypes.POINTER(ctypes.wintypes.DWORD))(
6ac4e806 1087 (b"GetConsoleMode", ctypes.windll.kernel32))
b58ddb32
PH
1088 INVALID_HANDLE_VALUE = ctypes.wintypes.DWORD(-1).value
1089
1090 def not_a_console(handle):
1091 if handle == INVALID_HANDLE_VALUE or handle is None:
1092 return True
8fb3ac36
PH
1093 return ((GetFileType(handle) & ~FILE_TYPE_REMOTE) != FILE_TYPE_CHAR or
1094 GetConsoleMode(handle, ctypes.byref(ctypes.wintypes.DWORD())) == 0)
b58ddb32
PH
1095
1096 if not_a_console(h):
1097 return False
1098
d1b9c912
PH
1099 def next_nonbmp_pos(s):
1100 try:
1101 return next(i for i, c in enumerate(s) if ord(c) > 0xffff)
1102 except StopIteration:
1103 return len(s)
1104
1105 while s:
1106 count = min(next_nonbmp_pos(s), 1024)
1107
b58ddb32 1108 ret = WriteConsoleW(
d1b9c912 1109 h, s, count if count else 2, ctypes.byref(written), None)
b58ddb32
PH
1110 if ret == 0:
1111 raise OSError('Failed to write string')
d1b9c912
PH
1112 if not count: # We just wrote a non-BMP character
1113 assert written.value == 2
1114 s = s[1:]
1115 else:
1116 assert written.value > 0
1117 s = s[written.value:]
b58ddb32
PH
1118 return True
1119
1120
734f90bb 1121def write_string(s, out=None, encoding=None):
7459e3a2
PH
1122 if out is None:
1123 out = sys.stderr
8bf48f23 1124 assert type(s) == compat_str
7459e3a2 1125
b58ddb32
PH
1126 if sys.platform == 'win32' and encoding is None and hasattr(out, 'fileno'):
1127 if _windows_write_string(s, out):
1128 return
1129
7459e3a2
PH
1130 if ('b' in getattr(out, 'mode', '') or
1131 sys.version_info[0] < 3): # Python 2 lies about mode of sys.stderr
104aa738
PH
1132 byt = s.encode(encoding or preferredencoding(), 'ignore')
1133 out.write(byt)
1134 elif hasattr(out, 'buffer'):
1135 enc = encoding or getattr(out, 'encoding', None) or preferredencoding()
1136 byt = s.encode(enc, 'ignore')
1137 out.buffer.write(byt)
1138 else:
8bf48f23 1139 out.write(s)
7459e3a2
PH
1140 out.flush()
1141
1142
48ea9cea
PH
1143def bytes_to_intlist(bs):
1144 if not bs:
1145 return []
1146 if isinstance(bs[0], int): # Python 3
1147 return list(bs)
1148 else:
1149 return [ord(c) for c in bs]
1150
c257baff 1151
cba892fa 1152def intlist_to_bytes(xs):
1153 if not xs:
1154 return b''
eb4157fd 1155 return struct_pack('%dB' % len(xs), *xs)
c38b1e77
PH
1156
1157
c1c9a79c
PH
1158# Cross-platform file locking
1159if sys.platform == 'win32':
1160 import ctypes.wintypes
1161 import msvcrt
1162
1163 class OVERLAPPED(ctypes.Structure):
1164 _fields_ = [
1165 ('Internal', ctypes.wintypes.LPVOID),
1166 ('InternalHigh', ctypes.wintypes.LPVOID),
1167 ('Offset', ctypes.wintypes.DWORD),
1168 ('OffsetHigh', ctypes.wintypes.DWORD),
1169 ('hEvent', ctypes.wintypes.HANDLE),
1170 ]
1171
1172 kernel32 = ctypes.windll.kernel32
1173 LockFileEx = kernel32.LockFileEx
1174 LockFileEx.argtypes = [
1175 ctypes.wintypes.HANDLE, # hFile
1176 ctypes.wintypes.DWORD, # dwFlags
1177 ctypes.wintypes.DWORD, # dwReserved
1178 ctypes.wintypes.DWORD, # nNumberOfBytesToLockLow
1179 ctypes.wintypes.DWORD, # nNumberOfBytesToLockHigh
1180 ctypes.POINTER(OVERLAPPED) # Overlapped
1181 ]
1182 LockFileEx.restype = ctypes.wintypes.BOOL
1183 UnlockFileEx = kernel32.UnlockFileEx
1184 UnlockFileEx.argtypes = [
1185 ctypes.wintypes.HANDLE, # hFile
1186 ctypes.wintypes.DWORD, # dwReserved
1187 ctypes.wintypes.DWORD, # nNumberOfBytesToLockLow
1188 ctypes.wintypes.DWORD, # nNumberOfBytesToLockHigh
1189 ctypes.POINTER(OVERLAPPED) # Overlapped
1190 ]
1191 UnlockFileEx.restype = ctypes.wintypes.BOOL
1192 whole_low = 0xffffffff
1193 whole_high = 0x7fffffff
1194
1195 def _lock_file(f, exclusive):
1196 overlapped = OVERLAPPED()
1197 overlapped.Offset = 0
1198 overlapped.OffsetHigh = 0
1199 overlapped.hEvent = 0
1200 f._lock_file_overlapped_p = ctypes.pointer(overlapped)
1201 handle = msvcrt.get_osfhandle(f.fileno())
1202 if not LockFileEx(handle, 0x2 if exclusive else 0x0, 0,
1203 whole_low, whole_high, f._lock_file_overlapped_p):
1204 raise OSError('Locking file failed: %r' % ctypes.FormatError())
1205
1206 def _unlock_file(f):
1207 assert f._lock_file_overlapped_p
1208 handle = msvcrt.get_osfhandle(f.fileno())
1209 if not UnlockFileEx(handle, 0,
1210 whole_low, whole_high, f._lock_file_overlapped_p):
1211 raise OSError('Unlocking file failed: %r' % ctypes.FormatError())
1212
1213else:
1214 import fcntl
1215
1216 def _lock_file(f, exclusive):
2582bebe 1217 fcntl.flock(f, fcntl.LOCK_EX if exclusive else fcntl.LOCK_SH)
c1c9a79c
PH
1218
1219 def _unlock_file(f):
2582bebe 1220 fcntl.flock(f, fcntl.LOCK_UN)
c1c9a79c
PH
1221
1222
1223class locked_file(object):
1224 def __init__(self, filename, mode, encoding=None):
1225 assert mode in ['r', 'a', 'w']
1226 self.f = io.open(filename, mode, encoding=encoding)
1227 self.mode = mode
1228
1229 def __enter__(self):
1230 exclusive = self.mode != 'r'
1231 try:
1232 _lock_file(self.f, exclusive)
1233 except IOError:
1234 self.f.close()
1235 raise
1236 return self
1237
1238 def __exit__(self, etype, value, traceback):
1239 try:
1240 _unlock_file(self.f)
1241 finally:
1242 self.f.close()
1243
1244 def __iter__(self):
1245 return iter(self.f)
1246
1247 def write(self, *args):
1248 return self.f.write(*args)
1249
1250 def read(self, *args):
1251 return self.f.read(*args)
4eb7f1d1
JMF
1252
1253
4644ac55
S
1254def get_filesystem_encoding():
1255 encoding = sys.getfilesystemencoding()
1256 return encoding if encoding is not None else 'utf-8'
1257
1258
4eb7f1d1 1259def shell_quote(args):
a6a173c2 1260 quoted_args = []
4644ac55 1261 encoding = get_filesystem_encoding()
a6a173c2
JMF
1262 for a in args:
1263 if isinstance(a, bytes):
1264 # We may get a filename encoded with 'encodeFilename'
1265 a = a.decode(encoding)
1266 quoted_args.append(pipes.quote(a))
28e614de 1267 return ' '.join(quoted_args)
9d4660ca
PH
1268
1269
1270def smuggle_url(url, data):
1271 """ Pass additional data in a URL for internal use. """
1272
1273 sdata = compat_urllib_parse.urlencode(
28e614de
PH
1274 {'__youtubedl_smuggle': json.dumps(data)})
1275 return url + '#' + sdata
9d4660ca
PH
1276
1277
79f82953 1278def unsmuggle_url(smug_url, default=None):
83e865a3 1279 if '#__youtubedl_smuggle' not in smug_url:
79f82953 1280 return smug_url, default
28e614de
PH
1281 url, _, sdata = smug_url.rpartition('#')
1282 jsond = compat_parse_qs(sdata)['__youtubedl_smuggle'][0]
9d4660ca
PH
1283 data = json.loads(jsond)
1284 return url, data
02dbf93f
PH
1285
1286
02dbf93f
PH
1287def format_bytes(bytes):
1288 if bytes is None:
28e614de 1289 return 'N/A'
02dbf93f
PH
1290 if type(bytes) is str:
1291 bytes = float(bytes)
1292 if bytes == 0.0:
1293 exponent = 0
1294 else:
1295 exponent = int(math.log(bytes, 1024.0))
28e614de 1296 suffix = ['B', 'KiB', 'MiB', 'GiB', 'TiB', 'PiB', 'EiB', 'ZiB', 'YiB'][exponent]
02dbf93f 1297 converted = float(bytes) / float(1024 ** exponent)
28e614de 1298 return '%.2f%s' % (converted, suffix)
f53c966a 1299
1c088fa8 1300
be64b5b0
PH
1301def parse_filesize(s):
1302 if s is None:
1303 return None
1304
1305 # The lower-case forms are of course incorrect and inofficial,
1306 # but we support those too
1307 _UNIT_TABLE = {
1308 'B': 1,
1309 'b': 1,
1310 'KiB': 1024,
1311 'KB': 1000,
1312 'kB': 1024,
1313 'Kb': 1000,
1314 'MiB': 1024 ** 2,
1315 'MB': 1000 ** 2,
1316 'mB': 1024 ** 2,
1317 'Mb': 1000 ** 2,
1318 'GiB': 1024 ** 3,
1319 'GB': 1000 ** 3,
1320 'gB': 1024 ** 3,
1321 'Gb': 1000 ** 3,
1322 'TiB': 1024 ** 4,
1323 'TB': 1000 ** 4,
1324 'tB': 1024 ** 4,
1325 'Tb': 1000 ** 4,
1326 'PiB': 1024 ** 5,
1327 'PB': 1000 ** 5,
1328 'pB': 1024 ** 5,
1329 'Pb': 1000 ** 5,
1330 'EiB': 1024 ** 6,
1331 'EB': 1000 ** 6,
1332 'eB': 1024 ** 6,
1333 'Eb': 1000 ** 6,
1334 'ZiB': 1024 ** 7,
1335 'ZB': 1000 ** 7,
1336 'zB': 1024 ** 7,
1337 'Zb': 1000 ** 7,
1338 'YiB': 1024 ** 8,
1339 'YB': 1000 ** 8,
1340 'yB': 1024 ** 8,
1341 'Yb': 1000 ** 8,
1342 }
1343
1344 units_re = '|'.join(re.escape(u) for u in _UNIT_TABLE)
4349c07d
PH
1345 m = re.match(
1346 r'(?P<num>[0-9]+(?:[,.][0-9]*)?)\s*(?P<unit>%s)' % units_re, s)
be64b5b0
PH
1347 if not m:
1348 return None
1349
4349c07d
PH
1350 num_str = m.group('num').replace(',', '.')
1351 mult = _UNIT_TABLE[m.group('unit')]
1352 return int(float(num_str) * mult)
be64b5b0
PH
1353
1354
caefb1de
PH
1355def month_by_name(name):
1356 """ Return the number of a month by (locale-independently) English name """
1357
caefb1de 1358 try:
7105440c
YCH
1359 return ENGLISH_MONTH_NAMES.index(name) + 1
1360 except ValueError:
1361 return None
1362
1363
1364def month_by_abbreviation(abbrev):
1365 """ Return the number of a month by (locale-independently) English
1366 abbreviations """
1367
1368 try:
1369 return [s[:3] for s in ENGLISH_MONTH_NAMES].index(abbrev) + 1
caefb1de
PH
1370 except ValueError:
1371 return None
18258362
JMF
1372
1373
5aafe895 1374def fix_xml_ampersands(xml_str):
18258362 1375 """Replace all the '&' by '&amp;' in XML"""
5aafe895
PH
1376 return re.sub(
1377 r'&(?!amp;|lt;|gt;|apos;|quot;|#x[0-9a-fA-F]{,4};|#[0-9]{,4};)',
28e614de 1378 '&amp;',
5aafe895 1379 xml_str)
e3946f98
PH
1380
1381
1382def setproctitle(title):
8bf48f23 1383 assert isinstance(title, compat_str)
e3946f98
PH
1384 try:
1385 libc = ctypes.cdll.LoadLibrary("libc.so.6")
1386 except OSError:
1387 return
6eefe533
PH
1388 title_bytes = title.encode('utf-8')
1389 buf = ctypes.create_string_buffer(len(title_bytes))
1390 buf.value = title_bytes
e3946f98 1391 try:
6eefe533 1392 libc.prctl(15, buf, 0, 0, 0)
e3946f98
PH
1393 except AttributeError:
1394 return # Strange libc, just skip this
d7dda168
PH
1395
1396
1397def remove_start(s, start):
1398 if s.startswith(start):
1399 return s[len(start):]
1400 return s
29eb5174
PH
1401
1402
2b9faf55
PH
1403def remove_end(s, end):
1404 if s.endswith(end):
1405 return s[:-len(end)]
1406 return s
1407
1408
31b2051e
S
1409def remove_quotes(s):
1410 if s is None or len(s) < 2:
1411 return s
1412 for quote in ('"', "'", ):
1413 if s[0] == quote and s[-1] == quote:
1414 return s[1:-1]
1415 return s
1416
1417
29eb5174 1418def url_basename(url):
9b8aaeed 1419 path = compat_urlparse.urlparse(url).path
28e614de 1420 return path.strip('/').split('/')[-1]
aa94a6d3
PH
1421
1422
1423class HEADRequest(compat_urllib_request.Request):
1424 def get_method(self):
1425 return "HEAD"
7217e148
PH
1426
1427
9732d77e 1428def int_or_none(v, scale=1, default=None, get_attr=None, invscale=1):
28746fbd
PH
1429 if get_attr:
1430 if v is not None:
1431 v = getattr(v, get_attr, None)
9572013d
PH
1432 if v == '':
1433 v = None
1812afb7
S
1434 if v is None:
1435 return default
1436 try:
1437 return int(v) * invscale // scale
1438 except ValueError:
af98f8ff 1439 return default
9732d77e 1440
9572013d 1441
40a90862
JMF
1442def str_or_none(v, default=None):
1443 return default if v is None else compat_str(v)
1444
9732d77e
PH
1445
1446def str_to_int(int_str):
48d4681e 1447 """ A more relaxed version of int_or_none """
9732d77e
PH
1448 if int_str is None:
1449 return None
28e614de 1450 int_str = re.sub(r'[,\.\+]', '', int_str)
9732d77e 1451 return int(int_str)
608d11f5
PH
1452
1453
9732d77e 1454def float_or_none(v, scale=1, invscale=1, default=None):
caf80631
S
1455 if v is None:
1456 return default
1457 try:
1458 return float(v) * invscale / scale
1459 except ValueError:
1460 return default
43f775e4
PH
1461
1462
608d11f5 1463def parse_duration(s):
8f9312c3 1464 if not isinstance(s, compat_basestring):
608d11f5
PH
1465 return None
1466
ca7b3246
S
1467 s = s.strip()
1468
608d11f5 1469 m = re.match(
9d22a7df 1470 r'''(?ix)(?:P?T)?
e8df5cee 1471 (?:
9c29bc69 1472 (?P<only_mins>[0-9.]+)\s*(?:mins?\.?|minutes?)\s*|
e8df5cee
PH
1473 (?P<only_hours>[0-9.]+)\s*(?:hours?)|
1474
9c29bc69 1475 \s*(?P<hours_reversed>[0-9]+)\s*(?:[:h]|hours?)\s*(?P<mins_reversed>[0-9]+)\s*(?:[:m]|mins?\.?|minutes?)\s*|
6a68bb57 1476 (?:
8f4b58d7
PH
1477 (?:
1478 (?:(?P<days>[0-9]+)\s*(?:[:d]|days?)\s*)?
1479 (?P<hours>[0-9]+)\s*(?:[:h]|hours?)\s*
1480 )?
6a68bb57
PH
1481 (?P<mins>[0-9]+)\s*(?:[:m]|mins?|minutes?)\s*
1482 )?
e8df5cee
PH
1483 (?P<secs>[0-9]+)(?P<ms>\.[0-9]+)?\s*(?:s|secs?|seconds?)?
1484 )$''', s)
608d11f5
PH
1485 if not m:
1486 return None
e8df5cee
PH
1487 res = 0
1488 if m.group('only_mins'):
1489 return float_or_none(m.group('only_mins'), invscale=60)
1490 if m.group('only_hours'):
1491 return float_or_none(m.group('only_hours'), invscale=60 * 60)
1492 if m.group('secs'):
1493 res += int(m.group('secs'))
3e675fab
PH
1494 if m.group('mins_reversed'):
1495 res += int(m.group('mins_reversed')) * 60
608d11f5
PH
1496 if m.group('mins'):
1497 res += int(m.group('mins')) * 60
e8df5cee
PH
1498 if m.group('hours'):
1499 res += int(m.group('hours')) * 60 * 60
3e675fab
PH
1500 if m.group('hours_reversed'):
1501 res += int(m.group('hours_reversed')) * 60 * 60
8f4b58d7
PH
1502 if m.group('days'):
1503 res += int(m.group('days')) * 24 * 60 * 60
7adcbe75
PH
1504 if m.group('ms'):
1505 res += float(m.group('ms'))
608d11f5 1506 return res
91d7d0b3
JMF
1507
1508
e65e4c88 1509def prepend_extension(filename, ext, expected_real_ext=None):
5f6a1245 1510 name, real_ext = os.path.splitext(filename)
e65e4c88
S
1511 return (
1512 '{0}.{1}{2}'.format(name, ext, real_ext)
1513 if not expected_real_ext or real_ext[1:] == expected_real_ext
1514 else '{0}.{1}'.format(filename, ext))
d70ad093
PH
1515
1516
b3ed15b7
S
1517def replace_extension(filename, ext, expected_real_ext=None):
1518 name, real_ext = os.path.splitext(filename)
1519 return '{0}.{1}'.format(
1520 name if not expected_real_ext or real_ext[1:] == expected_real_ext else filename,
1521 ext)
1522
1523
d70ad093
PH
1524def check_executable(exe, args=[]):
1525 """ Checks if the given binary is installed somewhere in PATH, and returns its name.
1526 args can be a list of arguments for a short output (like -version) """
1527 try:
1528 subprocess.Popen([exe] + args, stdout=subprocess.PIPE, stderr=subprocess.PIPE).communicate()
1529 except OSError:
1530 return False
1531 return exe
b7ab0590
PH
1532
1533
95807118 1534def get_exe_version(exe, args=['--version'],
cae97f65 1535 version_re=None, unrecognized='present'):
95807118
PH
1536 """ Returns the version of the specified executable,
1537 or False if the executable is not present """
1538 try:
cae97f65 1539 out, _ = subprocess.Popen(
54116803 1540 [encodeArgument(exe)] + args,
95807118
PH
1541 stdout=subprocess.PIPE, stderr=subprocess.STDOUT).communicate()
1542 except OSError:
1543 return False
cae97f65
PH
1544 if isinstance(out, bytes): # Python 2.x
1545 out = out.decode('ascii', 'ignore')
1546 return detect_exe_version(out, version_re, unrecognized)
1547
1548
1549def detect_exe_version(output, version_re=None, unrecognized='present'):
1550 assert isinstance(output, compat_str)
1551 if version_re is None:
1552 version_re = r'version\s+([-0-9._a-zA-Z]+)'
1553 m = re.search(version_re, output)
95807118
PH
1554 if m:
1555 return m.group(1)
1556 else:
1557 return unrecognized
1558
1559
b7ab0590 1560class PagedList(object):
dd26ced1
PH
1561 def __len__(self):
1562 # This is only useful for tests
1563 return len(self.getslice())
1564
9c44d242
PH
1565
1566class OnDemandPagedList(PagedList):
1567 def __init__(self, pagefunc, pagesize):
1568 self._pagefunc = pagefunc
1569 self._pagesize = pagesize
1570
b7ab0590
PH
1571 def getslice(self, start=0, end=None):
1572 res = []
1573 for pagenum in itertools.count(start // self._pagesize):
1574 firstid = pagenum * self._pagesize
1575 nextfirstid = pagenum * self._pagesize + self._pagesize
1576 if start >= nextfirstid:
1577 continue
1578
1579 page_results = list(self._pagefunc(pagenum))
1580
1581 startv = (
1582 start % self._pagesize
1583 if firstid <= start < nextfirstid
1584 else 0)
1585
1586 endv = (
1587 ((end - 1) % self._pagesize) + 1
1588 if (end is not None and firstid <= end <= nextfirstid)
1589 else None)
1590
1591 if startv != 0 or endv is not None:
1592 page_results = page_results[startv:endv]
1593 res.extend(page_results)
1594
1595 # A little optimization - if current page is not "full", ie. does
1596 # not contain page_size videos then we can assume that this page
1597 # is the last one - there are no more ids on further pages -
1598 # i.e. no need to query again.
1599 if len(page_results) + startv < self._pagesize:
1600 break
1601
1602 # If we got the whole page, but the next page is not interesting,
1603 # break out early as well
1604 if end == nextfirstid:
1605 break
1606 return res
81c2f20b
PH
1607
1608
9c44d242
PH
1609class InAdvancePagedList(PagedList):
1610 def __init__(self, pagefunc, pagecount, pagesize):
1611 self._pagefunc = pagefunc
1612 self._pagecount = pagecount
1613 self._pagesize = pagesize
1614
1615 def getslice(self, start=0, end=None):
1616 res = []
1617 start_page = start // self._pagesize
1618 end_page = (
1619 self._pagecount if end is None else (end // self._pagesize + 1))
1620 skip_elems = start - start_page * self._pagesize
1621 only_more = None if end is None else end - start
1622 for pagenum in range(start_page, end_page):
1623 page = list(self._pagefunc(pagenum))
1624 if skip_elems:
1625 page = page[skip_elems:]
1626 skip_elems = None
1627 if only_more is not None:
1628 if len(page) < only_more:
1629 only_more -= len(page)
1630 else:
1631 page = page[:only_more]
1632 res.extend(page)
1633 break
1634 res.extend(page)
1635 return res
1636
1637
81c2f20b 1638def uppercase_escape(s):
676eb3f2 1639 unicode_escape = codecs.getdecoder('unicode_escape')
81c2f20b 1640 return re.sub(
a612753d 1641 r'\\U[0-9a-fA-F]{8}',
676eb3f2
PH
1642 lambda m: unicode_escape(m.group(0))[0],
1643 s)
0fe2ff78
YCH
1644
1645
1646def lowercase_escape(s):
1647 unicode_escape = codecs.getdecoder('unicode_escape')
1648 return re.sub(
1649 r'\\u[0-9a-fA-F]{4}',
1650 lambda m: unicode_escape(m.group(0))[0],
1651 s)
b53466e1 1652
d05cfe06
S
1653
1654def escape_rfc3986(s):
1655 """Escape non-ASCII characters as suggested by RFC 3986"""
8f9312c3 1656 if sys.version_info < (3, 0) and isinstance(s, compat_str):
d05cfe06 1657 s = s.encode('utf-8')
ecc0c5ee 1658 return compat_urllib_parse.quote(s, b"%/;:@&=+$,!~*'()?#[]")
d05cfe06
S
1659
1660
1661def escape_url(url):
1662 """Escape URL as suggested by RFC 3986"""
1663 url_parsed = compat_urllib_parse_urlparse(url)
1664 return url_parsed._replace(
1665 path=escape_rfc3986(url_parsed.path),
1666 params=escape_rfc3986(url_parsed.params),
1667 query=escape_rfc3986(url_parsed.query),
1668 fragment=escape_rfc3986(url_parsed.fragment)
1669 ).geturl()
1670
b53466e1 1671try:
28e614de 1672 struct.pack('!I', 0)
b53466e1
PH
1673except TypeError:
1674 # In Python 2.6 (and some 2.7 versions), struct requires a bytes argument
1675 def struct_pack(spec, *args):
1676 if isinstance(spec, compat_str):
1677 spec = spec.encode('ascii')
1678 return struct.pack(spec, *args)
1679
1680 def struct_unpack(spec, *args):
1681 if isinstance(spec, compat_str):
1682 spec = spec.encode('ascii')
1683 return struct.unpack(spec, *args)
1684else:
1685 struct_pack = struct.pack
1686 struct_unpack = struct.unpack
62e609ab
PH
1687
1688
1689def read_batch_urls(batch_fd):
1690 def fixup(url):
1691 if not isinstance(url, compat_str):
1692 url = url.decode('utf-8', 'replace')
28e614de 1693 BOM_UTF8 = '\xef\xbb\xbf'
62e609ab
PH
1694 if url.startswith(BOM_UTF8):
1695 url = url[len(BOM_UTF8):]
1696 url = url.strip()
1697 if url.startswith(('#', ';', ']')):
1698 return False
1699 return url
1700
1701 with contextlib.closing(batch_fd) as fd:
1702 return [url for url in map(fixup, fd) if url]
b74fa8cd
JMF
1703
1704
1705def urlencode_postdata(*args, **kargs):
1706 return compat_urllib_parse.urlencode(*args, **kargs).encode('ascii')
bcf89ce6
PH
1707
1708
16392824 1709def encode_dict(d, encoding='utf-8'):
7e1f5447
S
1710 def encode(v):
1711 return v.encode(encoding) if isinstance(v, compat_basestring) else v
1712 return dict((encode(k), encode(v)) for k, v in d.items())
16392824 1713
8e60dc75
S
1714
1715def encode_compat_str(string, encoding=preferredencoding(), errors='strict'):
1716 return string if isinstance(string, compat_str) else compat_str(string, encoding, errors)
1717
16392824 1718
a1a530b0
PH
1719US_RATINGS = {
1720 'G': 0,
1721 'PG': 10,
1722 'PG-13': 13,
1723 'R': 16,
1724 'NC': 18,
1725}
fac55558
PH
1726
1727
146c80e2
S
1728def parse_age_limit(s):
1729 if s is None:
d838b1bd 1730 return None
146c80e2 1731 m = re.match(r'^(?P<age>\d{1,2})\+?$', s)
d838b1bd 1732 return int(m.group('age')) if m else US_RATINGS.get(s, None)
146c80e2
S
1733
1734
fac55558 1735def strip_jsonp(code):
609a61e3
PH
1736 return re.sub(
1737 r'(?s)^[a-zA-Z0-9_]+\s*\(\s*(.*)\);?\s*?(?://[^\n]*)*$', r'\1', code)
478c2c61
PH
1738
1739
e05f6939
PH
1740def js_to_json(code):
1741 def fix_kv(m):
e7b6d122
PH
1742 v = m.group(0)
1743 if v in ('true', 'false', 'null'):
1744 return v
1745 if v.startswith('"'):
d01949dc
S
1746 v = re.sub(r"\\'", "'", v[1:-1])
1747 elif v.startswith("'"):
e7b6d122
PH
1748 v = v[1:-1]
1749 v = re.sub(r"\\\\|\\'|\"", lambda m: {
1750 '\\\\': '\\\\',
1751 "\\'": "'",
1752 '"': '\\"',
1753 }[m.group(0)], v)
1754 return '"%s"' % v
e05f6939
PH
1755
1756 res = re.sub(r'''(?x)
d305dd73
PH
1757 "(?:[^"\\]*(?:\\\\|\\['"nu]))*[^"\\]*"|
1758 '(?:[^'\\]*(?:\\\\|\\['"nu]))*[^'\\]*'|
8f4b58d7 1759 [a-zA-Z_][.a-zA-Z_0-9]*
e05f6939 1760 ''', fix_kv, code)
ba9e68f4 1761 res = re.sub(r',(\s*[\]}])', lambda m: m.group(1), res)
e05f6939
PH
1762 return res
1763
1764
478c2c61
PH
1765def qualities(quality_ids):
1766 """ Get a numeric quality value out of a list of possible values """
1767 def q(qid):
1768 try:
1769 return quality_ids.index(qid)
1770 except ValueError:
1771 return -1
1772 return q
1773
acd69589
PH
1774
1775DEFAULT_OUTTMPL = '%(title)s-%(id)s.%(ext)s'
0a871f68 1776
a020a0dc
PH
1777
1778def limit_length(s, length):
1779 """ Add ellipses to overly long strings """
1780 if s is None:
1781 return None
1782 ELLIPSES = '...'
1783 if len(s) > length:
1784 return s[:length - len(ELLIPSES)] + ELLIPSES
1785 return s
48844745
PH
1786
1787
1788def version_tuple(v):
5f9b8394 1789 return tuple(int(e) for e in re.split(r'[-.]', v))
48844745
PH
1790
1791
1792def is_outdated_version(version, limit, assume_new=True):
1793 if not version:
1794 return not assume_new
1795 try:
1796 return version_tuple(version) < version_tuple(limit)
1797 except ValueError:
1798 return not assume_new
732ea2f0
PH
1799
1800
1801def ytdl_is_updateable():
1802 """ Returns if youtube-dl can be updated with -U """
1803 from zipimport import zipimporter
1804
1805 return isinstance(globals().get('__loader__'), zipimporter) or hasattr(sys, 'frozen')
7d4111ed
PH
1806
1807
1808def args_to_str(args):
1809 # Get a short string representation for a subprocess command
1810 return ' '.join(shlex_quote(a) for a in args)
2ccd1b10
PH
1811
1812
fdae2358
S
1813def error_to_str(err):
1814 err_str = str(err)
1815 # On python 2 error byte string must be decoded with proper
1816 # encoding rather than ascii
1817 if sys.version_info[0] < 3:
1818 err_str = err_str.decode(preferredencoding())
1819 return err_str
1820
1821
c460bdd5
PH
1822def mimetype2ext(mt):
1823 _, _, res = mt.rpartition('/')
1824
1825 return {
1826 'x-ms-wmv': 'wmv',
1827 'x-mp4-fragmented': 'mp4',
ecee5724 1828 'ttml+xml': 'ttml',
c460bdd5
PH
1829 }.get(res, res)
1830
1831
2ccd1b10
PH
1832def urlhandle_detect_ext(url_handle):
1833 try:
1834 url_handle.headers
1835 getheader = lambda h: url_handle.headers[h]
1836 except AttributeError: # Python < 3
1837 getheader = url_handle.info().getheader
1838
b55ee18f
PH
1839 cd = getheader('Content-Disposition')
1840 if cd:
1841 m = re.match(r'attachment;\s*filename="(?P<filename>[^"]+)"', cd)
1842 if m:
1843 e = determine_ext(m.group('filename'), default_ext=None)
1844 if e:
1845 return e
1846
c460bdd5 1847 return mimetype2ext(getheader('Content-Type'))
05900629
PH
1848
1849
1e399778
YCH
1850def encode_data_uri(data, mime_type):
1851 return 'data:%s;base64,%s' % (mime_type, base64.b64encode(data).decode('ascii'))
1852
1853
05900629
PH
1854def age_restricted(content_limit, age_limit):
1855 """ Returns True iff the content should be blocked """
1856
1857 if age_limit is None: # No limit set
1858 return False
1859 if content_limit is None:
1860 return False # Content available for everyone
1861 return age_limit < content_limit
61ca9a80
PH
1862
1863
1864def is_html(first_bytes):
1865 """ Detect whether a file contains HTML by examining its first bytes. """
1866
1867 BOMS = [
1868 (b'\xef\xbb\xbf', 'utf-8'),
1869 (b'\x00\x00\xfe\xff', 'utf-32-be'),
1870 (b'\xff\xfe\x00\x00', 'utf-32-le'),
1871 (b'\xff\xfe', 'utf-16-le'),
1872 (b'\xfe\xff', 'utf-16-be'),
1873 ]
1874 for bom, enc in BOMS:
1875 if first_bytes.startswith(bom):
1876 s = first_bytes[len(bom):].decode(enc, 'replace')
1877 break
1878 else:
1879 s = first_bytes.decode('utf-8', 'replace')
1880
1881 return re.match(r'^\s*<', s)
a055469f
PH
1882
1883
1884def determine_protocol(info_dict):
1885 protocol = info_dict.get('protocol')
1886 if protocol is not None:
1887 return protocol
1888
1889 url = info_dict['url']
1890 if url.startswith('rtmp'):
1891 return 'rtmp'
1892 elif url.startswith('mms'):
1893 return 'mms'
1894 elif url.startswith('rtsp'):
1895 return 'rtsp'
1896
1897 ext = determine_ext(url)
1898 if ext == 'm3u8':
1899 return 'm3u8'
1900 elif ext == 'f4m':
1901 return 'f4m'
1902
1903 return compat_urllib_parse_urlparse(url).scheme
cfb56d1a
PH
1904
1905
1906def render_table(header_row, data):
1907 """ Render a list of rows, each as a list of values """
1908 table = [header_row] + data
1909 max_lens = [max(len(compat_str(v)) for v in col) for col in zip(*table)]
1910 format_str = ' '.join('%-' + compat_str(ml + 1) + 's' for ml in max_lens[:-1]) + '%s'
1911 return '\n'.join(format_str % tuple(row) for row in table)
347de493
PH
1912
1913
1914def _match_one(filter_part, dct):
1915 COMPARISON_OPERATORS = {
1916 '<': operator.lt,
1917 '<=': operator.le,
1918 '>': operator.gt,
1919 '>=': operator.ge,
1920 '=': operator.eq,
1921 '!=': operator.ne,
1922 }
1923 operator_rex = re.compile(r'''(?x)\s*
1924 (?P<key>[a-z_]+)
1925 \s*(?P<op>%s)(?P<none_inclusive>\s*\?)?\s*
1926 (?:
1927 (?P<intval>[0-9.]+(?:[kKmMgGtTpPeEzZyY]i?[Bb]?)?)|
1928 (?P<strval>(?![0-9.])[a-z0-9A-Z]*)
1929 )
1930 \s*$
1931 ''' % '|'.join(map(re.escape, COMPARISON_OPERATORS.keys())))
1932 m = operator_rex.search(filter_part)
1933 if m:
1934 op = COMPARISON_OPERATORS[m.group('op')]
1935 if m.group('strval') is not None:
1936 if m.group('op') not in ('=', '!='):
1937 raise ValueError(
1938 'Operator %s does not support string values!' % m.group('op'))
1939 comparison_value = m.group('strval')
1940 else:
1941 try:
1942 comparison_value = int(m.group('intval'))
1943 except ValueError:
1944 comparison_value = parse_filesize(m.group('intval'))
1945 if comparison_value is None:
1946 comparison_value = parse_filesize(m.group('intval') + 'B')
1947 if comparison_value is None:
1948 raise ValueError(
1949 'Invalid integer value %r in filter part %r' % (
1950 m.group('intval'), filter_part))
1951 actual_value = dct.get(m.group('key'))
1952 if actual_value is None:
1953 return m.group('none_inclusive')
1954 return op(actual_value, comparison_value)
1955
1956 UNARY_OPERATORS = {
1957 '': lambda v: v is not None,
1958 '!': lambda v: v is None,
1959 }
1960 operator_rex = re.compile(r'''(?x)\s*
1961 (?P<op>%s)\s*(?P<key>[a-z_]+)
1962 \s*$
1963 ''' % '|'.join(map(re.escape, UNARY_OPERATORS.keys())))
1964 m = operator_rex.search(filter_part)
1965 if m:
1966 op = UNARY_OPERATORS[m.group('op')]
1967 actual_value = dct.get(m.group('key'))
1968 return op(actual_value)
1969
1970 raise ValueError('Invalid filter part %r' % filter_part)
1971
1972
1973def match_str(filter_str, dct):
1974 """ Filter a dictionary with a simple string syntax. Returns True (=passes filter) or false """
1975
1976 return all(
1977 _match_one(filter_part, dct) for filter_part in filter_str.split('&'))
1978
1979
1980def match_filter_func(filter_str):
1981 def _match_func(info_dict):
1982 if match_str(filter_str, info_dict):
1983 return None
1984 else:
1985 video_title = info_dict.get('title', info_dict.get('id', 'video'))
1986 return '%s does not pass filter %s, skipping ..' % (video_title, filter_str)
1987 return _match_func
91410c9b
PH
1988
1989
bf6427d2
YCH
1990def parse_dfxp_time_expr(time_expr):
1991 if not time_expr:
d631d5f9 1992 return
bf6427d2
YCH
1993
1994 mobj = re.match(r'^(?P<time_offset>\d+(?:\.\d+)?)s?$', time_expr)
1995 if mobj:
1996 return float(mobj.group('time_offset'))
1997
db2fe38b 1998 mobj = re.match(r'^(\d+):(\d\d):(\d\d(?:(?:\.|:)\d+)?)$', time_expr)
bf6427d2 1999 if mobj:
db2fe38b 2000 return 3600 * int(mobj.group(1)) + 60 * int(mobj.group(2)) + float(mobj.group(3).replace(':', '.'))
bf6427d2
YCH
2001
2002
c1c924ab
YCH
2003def srt_subtitles_timecode(seconds):
2004 return '%02d:%02d:%02d,%03d' % (seconds / 3600, (seconds % 3600) / 60, seconds % 60, (seconds % 1) * 1000)
bf6427d2
YCH
2005
2006
2007def dfxp2srt(dfxp_data):
4e335771
YCH
2008 _x = functools.partial(xpath_with_ns, ns_map={
2009 'ttml': 'http://www.w3.org/ns/ttml',
2010 'ttaf1': 'http://www.w3.org/2006/10/ttaf1',
2011 })
bf6427d2
YCH
2012
2013 def parse_node(node):
2014 str_or_empty = functools.partial(str_or_none, default='')
2015
2016 out = str_or_empty(node.text)
2017
2018 for child in node:
4e335771 2019 if child.tag in (_x('ttml:br'), _x('ttaf1:br'), 'br'):
bf6427d2 2020 out += '\n' + str_or_empty(child.tail)
4e335771 2021 elif child.tag in (_x('ttml:span'), _x('ttaf1:span'), 'span'):
bf6427d2
YCH
2022 out += str_or_empty(parse_node(child))
2023 else:
2024 out += str_or_empty(xml.etree.ElementTree.tostring(child))
2025
2026 return out
2027
36e6f62c 2028 dfxp = compat_etree_fromstring(dfxp_data.encode('utf-8'))
bf6427d2 2029 out = []
4e335771 2030 paras = dfxp.findall(_x('.//ttml:p')) or dfxp.findall(_x('.//ttaf1:p')) or dfxp.findall('.//p')
1b0427e6
YCH
2031
2032 if not paras:
2033 raise ValueError('Invalid dfxp/TTML subtitle')
bf6427d2
YCH
2034
2035 for para, index in zip(paras, itertools.count(1)):
d631d5f9 2036 begin_time = parse_dfxp_time_expr(para.attrib.get('begin'))
7dff0363 2037 end_time = parse_dfxp_time_expr(para.attrib.get('end'))
d631d5f9
YCH
2038 dur = parse_dfxp_time_expr(para.attrib.get('dur'))
2039 if begin_time is None:
2040 continue
7dff0363 2041 if not end_time:
d631d5f9
YCH
2042 if not dur:
2043 continue
2044 end_time = begin_time + dur
bf6427d2
YCH
2045 out.append('%d\n%s --> %s\n%s\n\n' % (
2046 index,
c1c924ab
YCH
2047 srt_subtitles_timecode(begin_time),
2048 srt_subtitles_timecode(end_time),
bf6427d2
YCH
2049 parse_node(para)))
2050
2051 return ''.join(out)
2052
2053
66e289ba
S
2054def cli_option(params, command_option, param):
2055 param = params.get(param)
2056 return [command_option, param] if param is not None else []
2057
2058
2059def cli_bool_option(params, command_option, param, true_value='true', false_value='false', separator=None):
2060 param = params.get(param)
2061 assert isinstance(param, bool)
2062 if separator:
2063 return [command_option + separator + (true_value if param else false_value)]
2064 return [command_option, true_value if param else false_value]
2065
2066
2067def cli_valueless_option(params, command_option, param, expected_value=True):
2068 param = params.get(param)
2069 return [command_option] if param == expected_value else []
2070
2071
2072def cli_configuration_args(params, param, default=[]):
2073 ex_args = params.get(param)
2074 if ex_args is None:
2075 return default
2076 assert isinstance(ex_args, list)
2077 return ex_args
2078
2079
39672624
YCH
2080class ISO639Utils(object):
2081 # See http://www.loc.gov/standards/iso639-2/ISO-639-2_utf-8.txt
2082 _lang_map = {
2083 'aa': 'aar',
2084 'ab': 'abk',
2085 'ae': 'ave',
2086 'af': 'afr',
2087 'ak': 'aka',
2088 'am': 'amh',
2089 'an': 'arg',
2090 'ar': 'ara',
2091 'as': 'asm',
2092 'av': 'ava',
2093 'ay': 'aym',
2094 'az': 'aze',
2095 'ba': 'bak',
2096 'be': 'bel',
2097 'bg': 'bul',
2098 'bh': 'bih',
2099 'bi': 'bis',
2100 'bm': 'bam',
2101 'bn': 'ben',
2102 'bo': 'bod',
2103 'br': 'bre',
2104 'bs': 'bos',
2105 'ca': 'cat',
2106 'ce': 'che',
2107 'ch': 'cha',
2108 'co': 'cos',
2109 'cr': 'cre',
2110 'cs': 'ces',
2111 'cu': 'chu',
2112 'cv': 'chv',
2113 'cy': 'cym',
2114 'da': 'dan',
2115 'de': 'deu',
2116 'dv': 'div',
2117 'dz': 'dzo',
2118 'ee': 'ewe',
2119 'el': 'ell',
2120 'en': 'eng',
2121 'eo': 'epo',
2122 'es': 'spa',
2123 'et': 'est',
2124 'eu': 'eus',
2125 'fa': 'fas',
2126 'ff': 'ful',
2127 'fi': 'fin',
2128 'fj': 'fij',
2129 'fo': 'fao',
2130 'fr': 'fra',
2131 'fy': 'fry',
2132 'ga': 'gle',
2133 'gd': 'gla',
2134 'gl': 'glg',
2135 'gn': 'grn',
2136 'gu': 'guj',
2137 'gv': 'glv',
2138 'ha': 'hau',
2139 'he': 'heb',
2140 'hi': 'hin',
2141 'ho': 'hmo',
2142 'hr': 'hrv',
2143 'ht': 'hat',
2144 'hu': 'hun',
2145 'hy': 'hye',
2146 'hz': 'her',
2147 'ia': 'ina',
2148 'id': 'ind',
2149 'ie': 'ile',
2150 'ig': 'ibo',
2151 'ii': 'iii',
2152 'ik': 'ipk',
2153 'io': 'ido',
2154 'is': 'isl',
2155 'it': 'ita',
2156 'iu': 'iku',
2157 'ja': 'jpn',
2158 'jv': 'jav',
2159 'ka': 'kat',
2160 'kg': 'kon',
2161 'ki': 'kik',
2162 'kj': 'kua',
2163 'kk': 'kaz',
2164 'kl': 'kal',
2165 'km': 'khm',
2166 'kn': 'kan',
2167 'ko': 'kor',
2168 'kr': 'kau',
2169 'ks': 'kas',
2170 'ku': 'kur',
2171 'kv': 'kom',
2172 'kw': 'cor',
2173 'ky': 'kir',
2174 'la': 'lat',
2175 'lb': 'ltz',
2176 'lg': 'lug',
2177 'li': 'lim',
2178 'ln': 'lin',
2179 'lo': 'lao',
2180 'lt': 'lit',
2181 'lu': 'lub',
2182 'lv': 'lav',
2183 'mg': 'mlg',
2184 'mh': 'mah',
2185 'mi': 'mri',
2186 'mk': 'mkd',
2187 'ml': 'mal',
2188 'mn': 'mon',
2189 'mr': 'mar',
2190 'ms': 'msa',
2191 'mt': 'mlt',
2192 'my': 'mya',
2193 'na': 'nau',
2194 'nb': 'nob',
2195 'nd': 'nde',
2196 'ne': 'nep',
2197 'ng': 'ndo',
2198 'nl': 'nld',
2199 'nn': 'nno',
2200 'no': 'nor',
2201 'nr': 'nbl',
2202 'nv': 'nav',
2203 'ny': 'nya',
2204 'oc': 'oci',
2205 'oj': 'oji',
2206 'om': 'orm',
2207 'or': 'ori',
2208 'os': 'oss',
2209 'pa': 'pan',
2210 'pi': 'pli',
2211 'pl': 'pol',
2212 'ps': 'pus',
2213 'pt': 'por',
2214 'qu': 'que',
2215 'rm': 'roh',
2216 'rn': 'run',
2217 'ro': 'ron',
2218 'ru': 'rus',
2219 'rw': 'kin',
2220 'sa': 'san',
2221 'sc': 'srd',
2222 'sd': 'snd',
2223 'se': 'sme',
2224 'sg': 'sag',
2225 'si': 'sin',
2226 'sk': 'slk',
2227 'sl': 'slv',
2228 'sm': 'smo',
2229 'sn': 'sna',
2230 'so': 'som',
2231 'sq': 'sqi',
2232 'sr': 'srp',
2233 'ss': 'ssw',
2234 'st': 'sot',
2235 'su': 'sun',
2236 'sv': 'swe',
2237 'sw': 'swa',
2238 'ta': 'tam',
2239 'te': 'tel',
2240 'tg': 'tgk',
2241 'th': 'tha',
2242 'ti': 'tir',
2243 'tk': 'tuk',
2244 'tl': 'tgl',
2245 'tn': 'tsn',
2246 'to': 'ton',
2247 'tr': 'tur',
2248 'ts': 'tso',
2249 'tt': 'tat',
2250 'tw': 'twi',
2251 'ty': 'tah',
2252 'ug': 'uig',
2253 'uk': 'ukr',
2254 'ur': 'urd',
2255 'uz': 'uzb',
2256 've': 'ven',
2257 'vi': 'vie',
2258 'vo': 'vol',
2259 'wa': 'wln',
2260 'wo': 'wol',
2261 'xh': 'xho',
2262 'yi': 'yid',
2263 'yo': 'yor',
2264 'za': 'zha',
2265 'zh': 'zho',
2266 'zu': 'zul',
2267 }
2268
2269 @classmethod
2270 def short2long(cls, code):
2271 """Convert language code from ISO 639-1 to ISO 639-2/T"""
2272 return cls._lang_map.get(code[:2])
2273
2274 @classmethod
2275 def long2short(cls, code):
2276 """Convert language code from ISO 639-2/T to ISO 639-1"""
2277 for short_name, long_name in cls._lang_map.items():
2278 if long_name == code:
2279 return short_name
2280
2281
4eb10f66
YCH
2282class ISO3166Utils(object):
2283 # From http://data.okfn.org/data/core/country-list
2284 _country_map = {
2285 'AF': 'Afghanistan',
2286 'AX': 'Åland Islands',
2287 'AL': 'Albania',
2288 'DZ': 'Algeria',
2289 'AS': 'American Samoa',
2290 'AD': 'Andorra',
2291 'AO': 'Angola',
2292 'AI': 'Anguilla',
2293 'AQ': 'Antarctica',
2294 'AG': 'Antigua and Barbuda',
2295 'AR': 'Argentina',
2296 'AM': 'Armenia',
2297 'AW': 'Aruba',
2298 'AU': 'Australia',
2299 'AT': 'Austria',
2300 'AZ': 'Azerbaijan',
2301 'BS': 'Bahamas',
2302 'BH': 'Bahrain',
2303 'BD': 'Bangladesh',
2304 'BB': 'Barbados',
2305 'BY': 'Belarus',
2306 'BE': 'Belgium',
2307 'BZ': 'Belize',
2308 'BJ': 'Benin',
2309 'BM': 'Bermuda',
2310 'BT': 'Bhutan',
2311 'BO': 'Bolivia, Plurinational State of',
2312 'BQ': 'Bonaire, Sint Eustatius and Saba',
2313 'BA': 'Bosnia and Herzegovina',
2314 'BW': 'Botswana',
2315 'BV': 'Bouvet Island',
2316 'BR': 'Brazil',
2317 'IO': 'British Indian Ocean Territory',
2318 'BN': 'Brunei Darussalam',
2319 'BG': 'Bulgaria',
2320 'BF': 'Burkina Faso',
2321 'BI': 'Burundi',
2322 'KH': 'Cambodia',
2323 'CM': 'Cameroon',
2324 'CA': 'Canada',
2325 'CV': 'Cape Verde',
2326 'KY': 'Cayman Islands',
2327 'CF': 'Central African Republic',
2328 'TD': 'Chad',
2329 'CL': 'Chile',
2330 'CN': 'China',
2331 'CX': 'Christmas Island',
2332 'CC': 'Cocos (Keeling) Islands',
2333 'CO': 'Colombia',
2334 'KM': 'Comoros',
2335 'CG': 'Congo',
2336 'CD': 'Congo, the Democratic Republic of the',
2337 'CK': 'Cook Islands',
2338 'CR': 'Costa Rica',
2339 'CI': 'Côte d\'Ivoire',
2340 'HR': 'Croatia',
2341 'CU': 'Cuba',
2342 'CW': 'Curaçao',
2343 'CY': 'Cyprus',
2344 'CZ': 'Czech Republic',
2345 'DK': 'Denmark',
2346 'DJ': 'Djibouti',
2347 'DM': 'Dominica',
2348 'DO': 'Dominican Republic',
2349 'EC': 'Ecuador',
2350 'EG': 'Egypt',
2351 'SV': 'El Salvador',
2352 'GQ': 'Equatorial Guinea',
2353 'ER': 'Eritrea',
2354 'EE': 'Estonia',
2355 'ET': 'Ethiopia',
2356 'FK': 'Falkland Islands (Malvinas)',
2357 'FO': 'Faroe Islands',
2358 'FJ': 'Fiji',
2359 'FI': 'Finland',
2360 'FR': 'France',
2361 'GF': 'French Guiana',
2362 'PF': 'French Polynesia',
2363 'TF': 'French Southern Territories',
2364 'GA': 'Gabon',
2365 'GM': 'Gambia',
2366 'GE': 'Georgia',
2367 'DE': 'Germany',
2368 'GH': 'Ghana',
2369 'GI': 'Gibraltar',
2370 'GR': 'Greece',
2371 'GL': 'Greenland',
2372 'GD': 'Grenada',
2373 'GP': 'Guadeloupe',
2374 'GU': 'Guam',
2375 'GT': 'Guatemala',
2376 'GG': 'Guernsey',
2377 'GN': 'Guinea',
2378 'GW': 'Guinea-Bissau',
2379 'GY': 'Guyana',
2380 'HT': 'Haiti',
2381 'HM': 'Heard Island and McDonald Islands',
2382 'VA': 'Holy See (Vatican City State)',
2383 'HN': 'Honduras',
2384 'HK': 'Hong Kong',
2385 'HU': 'Hungary',
2386 'IS': 'Iceland',
2387 'IN': 'India',
2388 'ID': 'Indonesia',
2389 'IR': 'Iran, Islamic Republic of',
2390 'IQ': 'Iraq',
2391 'IE': 'Ireland',
2392 'IM': 'Isle of Man',
2393 'IL': 'Israel',
2394 'IT': 'Italy',
2395 'JM': 'Jamaica',
2396 'JP': 'Japan',
2397 'JE': 'Jersey',
2398 'JO': 'Jordan',
2399 'KZ': 'Kazakhstan',
2400 'KE': 'Kenya',
2401 'KI': 'Kiribati',
2402 'KP': 'Korea, Democratic People\'s Republic of',
2403 'KR': 'Korea, Republic of',
2404 'KW': 'Kuwait',
2405 'KG': 'Kyrgyzstan',
2406 'LA': 'Lao People\'s Democratic Republic',
2407 'LV': 'Latvia',
2408 'LB': 'Lebanon',
2409 'LS': 'Lesotho',
2410 'LR': 'Liberia',
2411 'LY': 'Libya',
2412 'LI': 'Liechtenstein',
2413 'LT': 'Lithuania',
2414 'LU': 'Luxembourg',
2415 'MO': 'Macao',
2416 'MK': 'Macedonia, the Former Yugoslav Republic of',
2417 'MG': 'Madagascar',
2418 'MW': 'Malawi',
2419 'MY': 'Malaysia',
2420 'MV': 'Maldives',
2421 'ML': 'Mali',
2422 'MT': 'Malta',
2423 'MH': 'Marshall Islands',
2424 'MQ': 'Martinique',
2425 'MR': 'Mauritania',
2426 'MU': 'Mauritius',
2427 'YT': 'Mayotte',
2428 'MX': 'Mexico',
2429 'FM': 'Micronesia, Federated States of',
2430 'MD': 'Moldova, Republic of',
2431 'MC': 'Monaco',
2432 'MN': 'Mongolia',
2433 'ME': 'Montenegro',
2434 'MS': 'Montserrat',
2435 'MA': 'Morocco',
2436 'MZ': 'Mozambique',
2437 'MM': 'Myanmar',
2438 'NA': 'Namibia',
2439 'NR': 'Nauru',
2440 'NP': 'Nepal',
2441 'NL': 'Netherlands',
2442 'NC': 'New Caledonia',
2443 'NZ': 'New Zealand',
2444 'NI': 'Nicaragua',
2445 'NE': 'Niger',
2446 'NG': 'Nigeria',
2447 'NU': 'Niue',
2448 'NF': 'Norfolk Island',
2449 'MP': 'Northern Mariana Islands',
2450 'NO': 'Norway',
2451 'OM': 'Oman',
2452 'PK': 'Pakistan',
2453 'PW': 'Palau',
2454 'PS': 'Palestine, State of',
2455 'PA': 'Panama',
2456 'PG': 'Papua New Guinea',
2457 'PY': 'Paraguay',
2458 'PE': 'Peru',
2459 'PH': 'Philippines',
2460 'PN': 'Pitcairn',
2461 'PL': 'Poland',
2462 'PT': 'Portugal',
2463 'PR': 'Puerto Rico',
2464 'QA': 'Qatar',
2465 'RE': 'Réunion',
2466 'RO': 'Romania',
2467 'RU': 'Russian Federation',
2468 'RW': 'Rwanda',
2469 'BL': 'Saint Barthélemy',
2470 'SH': 'Saint Helena, Ascension and Tristan da Cunha',
2471 'KN': 'Saint Kitts and Nevis',
2472 'LC': 'Saint Lucia',
2473 'MF': 'Saint Martin (French part)',
2474 'PM': 'Saint Pierre and Miquelon',
2475 'VC': 'Saint Vincent and the Grenadines',
2476 'WS': 'Samoa',
2477 'SM': 'San Marino',
2478 'ST': 'Sao Tome and Principe',
2479 'SA': 'Saudi Arabia',
2480 'SN': 'Senegal',
2481 'RS': 'Serbia',
2482 'SC': 'Seychelles',
2483 'SL': 'Sierra Leone',
2484 'SG': 'Singapore',
2485 'SX': 'Sint Maarten (Dutch part)',
2486 'SK': 'Slovakia',
2487 'SI': 'Slovenia',
2488 'SB': 'Solomon Islands',
2489 'SO': 'Somalia',
2490 'ZA': 'South Africa',
2491 'GS': 'South Georgia and the South Sandwich Islands',
2492 'SS': 'South Sudan',
2493 'ES': 'Spain',
2494 'LK': 'Sri Lanka',
2495 'SD': 'Sudan',
2496 'SR': 'Suriname',
2497 'SJ': 'Svalbard and Jan Mayen',
2498 'SZ': 'Swaziland',
2499 'SE': 'Sweden',
2500 'CH': 'Switzerland',
2501 'SY': 'Syrian Arab Republic',
2502 'TW': 'Taiwan, Province of China',
2503 'TJ': 'Tajikistan',
2504 'TZ': 'Tanzania, United Republic of',
2505 'TH': 'Thailand',
2506 'TL': 'Timor-Leste',
2507 'TG': 'Togo',
2508 'TK': 'Tokelau',
2509 'TO': 'Tonga',
2510 'TT': 'Trinidad and Tobago',
2511 'TN': 'Tunisia',
2512 'TR': 'Turkey',
2513 'TM': 'Turkmenistan',
2514 'TC': 'Turks and Caicos Islands',
2515 'TV': 'Tuvalu',
2516 'UG': 'Uganda',
2517 'UA': 'Ukraine',
2518 'AE': 'United Arab Emirates',
2519 'GB': 'United Kingdom',
2520 'US': 'United States',
2521 'UM': 'United States Minor Outlying Islands',
2522 'UY': 'Uruguay',
2523 'UZ': 'Uzbekistan',
2524 'VU': 'Vanuatu',
2525 'VE': 'Venezuela, Bolivarian Republic of',
2526 'VN': 'Viet Nam',
2527 'VG': 'Virgin Islands, British',
2528 'VI': 'Virgin Islands, U.S.',
2529 'WF': 'Wallis and Futuna',
2530 'EH': 'Western Sahara',
2531 'YE': 'Yemen',
2532 'ZM': 'Zambia',
2533 'ZW': 'Zimbabwe',
2534 }
2535
2536 @classmethod
2537 def short2full(cls, code):
2538 """Convert an ISO 3166-2 country code to the corresponding full name"""
2539 return cls._country_map.get(code.upper())
2540
2541
91410c9b 2542class PerRequestProxyHandler(compat_urllib_request.ProxyHandler):
2461f79d
PH
2543 def __init__(self, proxies=None):
2544 # Set default handlers
2545 for type in ('http', 'https'):
2546 setattr(self, '%s_open' % type,
2547 lambda r, proxy='__noproxy__', type=type, meth=self.proxy_open:
2548 meth(r, proxy, type))
2549 return compat_urllib_request.ProxyHandler.__init__(self, proxies)
2550
91410c9b 2551 def proxy_open(self, req, proxy, type):
2461f79d 2552 req_proxy = req.headers.get('Ytdl-request-proxy')
91410c9b
PH
2553 if req_proxy is not None:
2554 proxy = req_proxy
2461f79d
PH
2555 del req.headers['Ytdl-request-proxy']
2556
2557 if proxy == '__noproxy__':
2558 return None # No Proxy
91410c9b
PH
2559 return compat_urllib_request.ProxyHandler.proxy_open(
2560 self, req, proxy, type)