Remove import imp (deprecated)

This commit is contained in:
allegroai 2021-01-18 11:32:41 +02:00
parent 661fac95a2
commit c8350a0438

View File

@ -5,19 +5,14 @@ from __future__ import print_function, division, absolute_import
import json import json
import os import os
import sys import sys
import six
import fnmatch import fnmatch
import importlib import importlib
import imp
import ast import ast
import doctest import doctest
import collections import collections
import functools import functools
from pathlib2 import Path from pathlib2 import Path
try:
from types import FileType # py2
except ImportError:
from io import IOBase as FileType # py3
from .utils import parse_git_config from .utils import parse_git_config
from .modules import ImportedModules from .modules import ImportedModules
@ -290,6 +285,9 @@ def is_std_or_local_lib(name):
str if local library str if local library
""" """
exist = True exist = True
if six.PY2:
import imp # noqa
from types import FileType # noqa
module_info = ('', '', '') module_info = ('', '', '')
try: try:
module_info = imp.find_module(name) module_info = imp.find_module(name)
@ -303,8 +301,16 @@ def is_std_or_local_lib(name):
exist = False exist = False
# Testcase: ResourceWarning # Testcase: ResourceWarning
if isinstance(module_info[0], FileType): if isinstance(module_info[0], FileType):
module_info[0].close() module_info[0].close() # noqa
mpath = module_info[1] mpath = module_info[1] # noqa
else:
module_info = None
try:
module_info = importlib.util.find_spec(name) # noqa
except ImportError:
exist = False
mpath = module_info.origin if module_info else None
if exist and mpath is not None: if exist and mpath is not None:
if ('site-packages' in mpath or if ('site-packages' in mpath or
'dist-packages' in mpath or 'dist-packages' in mpath or