extract_imports.py

import re

def get_imports(script):
    res = []
    patt = re.compile('^(?:import|from).*$', re.MULTILINE)
    raw_imports = re.findall(patt, script)
    if raw_imports == []:
        return None
    for imp in raw_imports:
        if '#' in imp:
            imp = imp.split('#')[0]
        if 'as' in imp:
            imp = imp.split(' as ')[0]
        if 'from ' in imp:
            if '*' not in imp.split('import')[-1]:
                root_module = imp.split(' ')[1] + '.'
            else:
                imp = imp.split('import')[0].replace('from ', 'import ')
                root_module = ''
        else:
            root_module = ''
        for module in imp.split('import ')[-1].split(','):
            neat = root_module + module.strip(' ,')
            res.append(neat.strip())
    return res