~andrew-monkeysailor/python-snippets/ast-snippets

Viewing changes to AST/ASTcount.py

AST snippet

added added

removed removed

# [SNIPPET_NAME: Reading AST]

# [SNIPPET_CATEGORIES: AST]

# [SNIPPET_DESCRIPTION: Scans the AST to find import statements]

# [SNIPPET_AUTHOR: Andrew Lewis <andrew@monkeysailor.co.uk>]

# [SNIPPET_DOCS:]

# [SNIPPET_LICENSE: GPL]

import ast, _ast

class ASTcount():

""" Spit out a list of imports from a python file"""

def __init__(self, txt):

self.source = ast.parse(txt)

def imports(self):

# Get the names of import statements

# and return as a list

output = []

for each in self.source.body:

if type(each)== _ast.Import:

for modules in each.names:

output.append(modules.name)

return list(set(output))

if __name__ == "__main__":

# read in a python file

f = open("parser.py","r")

x = "".join(f.readlines())

f.close()

# create instance and print imports and variable names

counter = ASTcount(x)

print counter.imports()