# hgdemandimport - global demand-loading of modules for Mercurial

#

# Copyright 2017 Facebook Inc.

#

# This software may be used and distributed according to the terms of the

# GNU General Public License version 2 or any later version.

'''demandimport - automatic demand-loading of modules'''

# This is in a separate package from mercurial because in Python 3,

# demand loading is per-package. Keeping demandimport in the mercurial package

# would disable demand loading for any modules in mercurial.

import os

import sys

from . import demandimportpy3 as demandimport

# Full module names which can't be lazy imported.

# Extensions can add to this set.

IGNORES = {

    '__future__',

    '_hashlib',

    # ImportError during pkg_resources/__init__.py:fixup_namespace_package

    '_imp',

    '_xmlplus',

    'fcntl',

    'nt',  # pathlib2 tests the existence of built-in 'nt' module

    'win32com.gen_py',

    'win32com.shell',  # 'appdirs' tries to import win32com.shell

    '_winreg',  # 2.7 mimetypes needs immediate ImportError

    'pythoncom',

    # imported by tarfile, not available under Windows

    'pwd',

    'grp',

    # imported by profile, itself imported by hotshot.stats,

    # not available under Windows

    'resource',

    # this trips up many extension authors

    'gtk',

    # setuptools' pkg_resources.py expects "from __main__ import x" to

    # raise ImportError if x not defined

    '__main__',

    '_ast',  # https://bugs.python.org/issue41631

    '_ssl',  # conditional imports in the stdlib, issue1964

    '_sre',  # issue4920

    'rfc822',

    'mimetools',

    'sqlalchemy.events',  # has import-time side effects (issue5085)

    'sqlalchemy.dialects',  # similar problems as above

    # setuptools 8 expects this module to explode early when not on windows

    'distutils.msvc9compiler',

    '__builtin__',

    'builtins',

    'urwid.command_map',  # for pudb

    'lzma',

    # setuptools uses this hack to inject it's own distutils at import time

    'setuptools',

    '_distutils_hack.override',

}

_pypy = '__pypy__' in sys.builtin_module_names

if _pypy:

    # _ctypes.pointer is shadowed by "from ... import pointer" (PyPy 5)

    IGNORES.add('_ctypes.pointer')

    # pure Python module on PyPy, must be loaded to raise ModuleNotFoundError

    # on non-Windows platforms

    IGNORES.add('msvcrt')

demandimport.init(IGNORES)

# Re-export.

isenabled = demandimport.isenabled

disable = demandimport.disable

deactivated = demandimport.deactivated

def enable():

    # chg pre-imports modules so do not enable demandimport for it

    if (

        'CHGINTERNALMARK' not in os.environ

        and os.environ.get('HGDEMANDIMPORT') != 'disable'

    ):

        demandimport.enable()