Python простой способ прочитать все операторы импорта из модуля Py - PullRequest
Новая
       24

Python простой способ прочитать все операторы импорта из модуля Py

4 голосов
/ 25 января 2012

Я пытаюсь создать вспомогательную функцию для чтения файла и макета всего импорта для модульного теста.Я должен прочитать файл против импорта, так как у меня нет этих вещей в пути Python.

Пример кода: 

#module.py
import com.stackoverflow.question
from com.stackoverflow.util import test_func
from com.stackoverflow.util import TestClass

#magic helper: what i want
magic = process('<path_to>/module.py')
for module in magic.modules_as_strings():
    #todo  would have to recuirsively add each path
    # so i would first create com, then com.stackoverflow, etc
    setattr(self, module, StubModules(module)
for obj in magic.sink:
    #these would be "from"  from x import Y
    #its basically just creating self.Y = object
    setattr(self, object)

Выше приведён код-макетЯ действительно ищу лучший способ просто токенизировать файл для "из / импорт операторов"

Это имеет смысл?Я знаю, что могу читать файл построчно, но я надеялся на более чёткий / лаконичный способ.

Дайте мне знать, если у вас есть какие-либо вопросы.

1 Ответ

11 голосов
/ 29 января 2012

Используя модуль AST , это довольно просто: 

import ast
from collections import namedtuple

Import = namedtuple("Import", ["module", "name", "alias"])

def get_imports(path):
    with open(path) as fh:        
       root = ast.parse(fh.read(), path)

    for node in ast.iter_child_nodes(root):
        if isinstance(node, ast.Import):
            module = []
        elif isinstance(node, ast.ImportFrom):  
            module = node.module.split('.')
        else:
            continue

        for n in node.names:
            yield Import(module, n.name.split('.'), n.asname)

Для такого модуля: 

from coco import bunny
from coco.bungy import carrot
from meta import teta
from rocket import spaceship as sp
import bingo
import com.stackoverflow
import motorbike as car
import module1, module2

s="a random variable"

def func():
    """And a function"""

Вывод: 

>>> for imp in get_imports("/path/to/file.py"): print imp
Import(module=['coco'], name=['bunny'], alias=None)
Import(module=['coco', 'bungy'], name=['carrot'], alias=None)
Import(module=['meta'], name=['teta'], alias=None)
Import(module=['rocket'], name=['spaceship'], alias='sp')
Import(module=[], name=['bingo'], alias=None)
Import(module=[], name=['com', 'stackoverflow'], alias=None)
Import(module=[], name=['motorbike'], alias='car')
Import(module=[], name=['module1'], alias=None)
Import(module=[], name=['module2'], alias=None)
...