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