Issue with "production" code

[mateusoliveira43]

Issue with "production" code

[mateusoliveira43]

first ideas

import ast
from collections import namedtuple

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


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

    for node in ast.walk(root):
        if isinstance(node, ast.Import):
            module = []
        elif isinstance(node, ast.ImportFrom):
            if node.module:
                module = node.module.split('.')
            else:
                module = []
            for _ in range(node.level):
                module.insert(0, ".")
        else:
            continue

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

    return imports