Merge pull request #15 from returntocorp/python_attributes

* lang_python/parsing/parser_python.mly: adding attribute in AST
semgrep · Feb 3, 2020 · 22eea45 · 22eea45
2 parents 4d20941 + 62fbeb4
commit 22eea45
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/lang_python/parsing/parser_python.mly b/lang_python/parsing/parser_python.mly
@@ -27,6 +27,7 @@
  * old src: 
  *  - http://inst.eecs.berkeley.edu/~cs164/sp10/python-grammar.html
  *)
+open Common
 open Ast_python
 
 (* intermediate helper type *)
@@ -489,11 +490,22 @@ compound_stmt:
      *)*/
   | async_stmt  { $1 }
 
-/*(* TODO: add attributes *)*/
 decorated:
-  | decorators classdef { $2 }
-  | decorators funcdef { $2 }
-  | decorators async_funcdef { $2 }
+  | decorators classdef { 
+     match $2 with 
+     | ClassDef (a, b, c, d) -> ClassDef (a, b, c, $1 @ d)
+     | _ -> raise Impossible
+   }
+  | decorators funcdef { 
+     match $2 with 
+     | FunctionDef (a, b, c, d, e) -> FunctionDef (a, b, c, d, $1 @ e)
+     | _ -> raise Impossible
+   }
+  | decorators async_funcdef {
+     match $2 with 
+     | FunctionDef (a, b, c, d, e) -> FunctionDef (a, b, c, d, $1 @ e)
+     | _ -> raise Impossible
+  }
 
 /*(* this is always preceded by a COLON *)*/
 suite: