Merge lp:~andrew-monkeysailor/python-snippets/ast-snippets into lp:~jonobacon/python-snippets/trunk

[Monkeysailor]

Adds AST folder for AST module examples, includes an example based on finding import statements in a python file.

[Jono Bacon (jonobacon)]

Thanks for the snippet, pal!

I am getting:

Traceback (most recent call last):
  File "/usr/share/python-snippets/AST/ASTcount.py", line 28, in <module>
    f = open("parser.py","r")
IOError: [Errno 2] No such file or directory: 'parser.py'

could you take a look?

review: Needs Fixing

[lp:~andrew-monkeysailor/python-snippets/ast-snippets]

83. By user <user@user-PC> on 2010-04-10

corrected rudimentary execution error.

[Monkeysailor (andrew-monkeysailor)]

should be fixed now.

[Jono Bacon (jonobacon)]

Thanks for the re-submission. Now I am getting:

Traceback (most recent call last):
  File "/usr/share/python-snippets/AST/ASTcount.py", line 33, in <module>
    counter = ASTcount(x)
  File "/usr/share/python-snippets/AST/ASTcount.py", line 13, in __init__
    self.source = ast.parse(txt)
  File "/usr/lib/python2.6/ast.py", line 37, in parse
    return compile(expr, filename, mode, PyCF_ONLY_AST)
  File "<unknown>", line 7

    ^
SyntaxError: invalid syntax

[Jono Bacon (jonobacon)]

review: Needs Fixing

[Monkeysailor (andrew-monkeysailor)]

On 24 April 2010 19:28, Jono Bacon <email address hidden> wrote:
> Review: Needs Fixing
>
> --
> https://code.edge.launchpad.net/~andrew-monkeysailor/python-snippets/ast-snippets/+merge/23144
> You are the owner of lp:~andrew-monkeysailor/python-snippets/ast-snippets.
>

Weird... it's working for me.

Unmerged revisions

83. By user <user@user-PC> on 2010-04-10

corrected rudimentary execution error.

82. By user <user@user-PC> on 2010-04-05

AST snippet

=== added directory 'AST'
=== added file 'AST/ASTcount.py'
--- AST/ASTcount.py	1970-01-01 00:00:00 +0000
+++ AST/ASTcount.py	2010-04-10 20:43:22 +0000
@@ -0,0 +1,34 @@
+# [SNIPPET_NAME: Reading AST]
+# [SNIPPET_CATEGORIES: AST]
+# [SNIPPET_DESCRIPTION: Scans the AST to find import statements]
+# [SNIPPET_AUTHOR: Andrew Lewis <[e-mail]>]
+# [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("ASTcount.py","r")
+    x = "".join(f.readlines())
+    f.close()
+
+    # create instance and print imports and variable names
+    counter = ASTcount(x)
+    print counter.imports()

=== modified file 'CATEGORIES'
--- CATEGORIES	2010-04-10 20:36:15 +0000
+++ CATEGORIES	2010-04-10 20:43:22 +0000
@@ -13,6 +13,7 @@
 When adding a snippet please try to use one of the following categories if suitable:
 
     Application Indicator   Application indicator examples.
+    AST                         Abstract Syntax Tree examples
     bzrlib                  Bazaar source control system Python module.
     Cairo		            Cairo drawing examples
     Clutter                 Clutter toolkit examples.