yt_dlp/compat/__init__.py

   1 import os
   2 import sys
   3 import xml.etree.ElementTree as etree
   4
   5 from .compat_utils import passthrough_module
   6
   7 passthrough_module(__name__, '._deprecated')
   8 del passthrough_module
   9
  10
  11 # HTMLParseError has been deprecated in Python 3.3 and removed in
  12 # Python 3.5. Introducing dummy exception for Python >3.5 for compatible
  13 # and uniform cross-version exception handling
  14 class compat_HTMLParseError(ValueError):
  15     pass
  16
  17
  18 class _TreeBuilder(etree.TreeBuilder):
  19     def doctype(self, name, pubid, system):
  20         pass
  21
  22
  23 def compat_etree_fromstring(text):
  24     return etree.XML(text, parser=etree.XMLParser(target=_TreeBuilder()))
  25
  26
  27 compat_os_name = os._name if os.name == 'java' else os.name
  28
  29
  30 def compat_shlex_quote(s):
  31     from ..utils import shell_quote
  32     return shell_quote(s)
  33
  34
  35 def compat_ord(c):
  36     return c if isinstance(c, int) else ord(c)
  37
  38
  39 if compat_os_name == 'nt' and sys.version_info < (3, 8):
  40     # os.path.realpath on Windows does not follow symbolic links
  41     # prior to Python 3.8 (see https://bugs.python.org/issue9949)
  42     def compat_realpath(path):
  43         while os.path.islink(path):
  44             path = os.path.abspath(os.readlink(path))
  45         return os.path.realpath(path)
  46 else:
  47     compat_realpath = os.path.realpath
  48
  49
  50 # Python 3.8+ does not honor %HOME% on windows, but this breaks compatibility with youtube-dl
  51 # See https://github.com/yt-dlp/yt-dlp/issues/792
  52 # https://docs.python.org/3/library/os.path.html#os.path.expanduser
  53 if compat_os_name in ('nt', 'ce'):
  54     def compat_expanduser(path):
  55         HOME = os.environ.get('HOME')
  56         if not HOME:
  57             return os.path.expanduser(path)
  58         elif not path.startswith('~'):
  59             return path
  60         i = path.replace('\\', '/', 1).find('/')  # ~user
  61         if i < 0:
  62             i = len(path)
  63         userhome = os.path.join(os.path.dirname(HOME), path[1:i]) if i > 1 else HOME
  64         return userhome + path[i:]
  65 else:
  66     compat_expanduser = os.path.expanduser
  67
  68
  69 def urllib_req_to_req(urllib_request):
  70     """Convert urllib Request to a networking Request"""
  71     from ..networking import Request
  72     from ..utils.networking import HTTPHeaderDict
  73     return Request(
  74         urllib_request.get_full_url(), data=urllib_request.data, method=urllib_request.get_method(),
  75         headers=HTTPHeaderDict(urllib_request.headers, urllib_request.unredirected_hdrs),
  76         extensions={'timeout': urllib_request.timeout} if hasattr(urllib_request, 'timeout') else None)