Отмена импорта модуля *

Я думаю, что решения PurityLake и Martijn Pieters с вспомогательным ручным управлением, вероятно, являются лучшим способом. Но невозможно сделать это программно.

Во-первых, вам нужно получить список всех имен, существующих в словаре модуля, которые могут использоваться в коде. Я предполагаю, что ваш код напрямую не вызывает какие-либо функции dunder и т. д.

Затем вам нужно перебрать их, используя inspect.getmodule(), чтобы узнать, в каком модуле был изначально определен каждый объект. И я предполагаю, что вы не используете ничего, что было дважды from foo import *-ed. Составьте список всех имен, которые были определены в модулях numpy и scipy.

Теперь вы можете взять этот вывод и просто заменить каждый foo на numpy.foo.

Итак, собирая это, что-то вроде этого:

for modname in sys.argv[1:]:
    with open(modname + '.py') as srcfile:
        src = srcfile.read()
    src = src.replace('from numpy import *', 'import numpy')
    src = src.replace('from scipy import *', 'import scipy')
    mod = __import__(modname)
    for name in dir(mod):
        original_mod = inspect.getmodule(getattr(mod, name))
        if original_mod.__name__ == 'numpy':
            src = src.replace(name, 'numpy.'+name)
        elif original_mod.__name__ == 'scipy':
            src = src.replace(name, 'scipy.'+name)
    with open(modname + '.tmp') as dstfile:
        dstfile.write(src)
    os.rename(modname + '.py', modname + '.bak')
    os.rename(modname + '.tmp', modname + '.py')

Если одно из предположений неверно, изменить код несложно. Кроме того, вы можете использовать tempfile.NamedTemporaryFile и другие улучшения, чтобы случайно не перезаписать временные файлы. (Я просто не хотел иметь дело с головной болью написания чего-то кросс-платформенного; если вы не работаете в Windows, это легко.) И, очевидно, добавить некоторую обработку ошибок и, возможно, некоторые отчеты.

да. Удалите импорт и запустите линтер на модуле.

Я рекомендую использовать flake8, хотя это также может создать много шума из-за ошибок стиля.

Просто удалить импорт и попытаться запустить код, вероятно, будет недостаточно, так как многие ошибки имен не будут возникать до тех пор, пока вы не запустите только правильную строку кода с правильным вводом. Вместо этого линтер проанализирует код путем синтаксического анализа и обнаружит потенциальные NameError без необходимости запуска кода.

Все это предполагает, что не существует надежных модульных тестов или что тесты не обеспечивают достаточного покрытия.

В этом случае, когда есть несколько from module import * строк, это становится немного более болезненным, поскольку вам нужно выяснить для каждого отсутствующего имени, какой модуль предоставил это имя. Это потребует ручной работы, но вы можете просто импортировать модуль в интерпретатор Python и проверить, определено ли отсутствующее имя в этом модуле:

>>> import scipy, numpy
>>> 'loadtxt' in dir(numpy)
True

Вам нужно принять во внимание, что в этом конкретном случае между модулями numpy и scipy есть перекрытие; для любого имени, определенного в обоих модулях, выигрывает модуль, импортированный последним.

Обратите внимание, что оставление любой строки from module import * на месте означает, что линтер не сможет определить, какие имена могут вызывать ошибки NameError!

Для этого я сделал небольшую утилиту, которую я называю «dedazzler». Он найдет строки «из импорта модуля *», а затем расширит «каталог» целевых модулей, заменив строки.

После его запуска вам все равно нужно запустить линтер. Вот особенно интересная часть кода:

import re

star_match = re.compile('from\s(?P<module>[\.\w]+)\simport\s[*]')
now = str(time.time())
error = lambda x: sys.stderr.write(x + '\n')

def replace_imports(lines):
    """
    Iterates through lines in a Python file, looks for 'from module import *'
    statements, and attempts to fix them.
    """
    for line_num, line in enumerate(lines):
        match = star_match.search(line)
        if match:
            newline = import_generator(match.groupdict()['module'])
            if newline:
                lines[line_num] = newline
    return lines

def import_generator(modulename):
    try:
        prop_depth = modulename.split('.')[1:]
        namespace = __import__(modulename)
        for prop in prop_depth:
            namespace = getattr(namespace, prop)
    except ImportError:
        error("Couldn't import module '%s'!" % modulename)
        return
    directory = [ name for name in dir(namespace) if not name.startswith('_') ]
    return "from %s import %s\n"% (modulename, ', '.join(directory))

Я поддерживаю это в более полезной автономной форме утилиты здесь:

https://github.com/USGM/dedazzler/

хорошо, вот что я думаю, что вы могли бы сделать, сломать программу. удалите импорт и обратите внимание на допущенные ошибки. Затем импортируйте только те модули, которые вам нужны, это может занять некоторое время, но это единственный известный мне способ сделать это, я буду приятно удивлен, если кто-то знает инструмент, который поможет

РЕДАКТИРОВАТЬ: ах да, линтер, я об этом не подумал.