o @a@sdZddlmZmZmZmZmZddlmZddl m Z dZ Gddde Z Gdd d eZGd d d eZGd d d eZGdddeZGdddeZdS)z9 Module to looks for a specified pattern in a given AST. )AST iter_fields NodeVisitorDictSet) permutations)isnan c@eZdZdZdS)DamnTooLongPatternz@ Exception for long dict/set comparison to reduce compile time. N__name__ __module__ __qualname____doc__rr>/usr/lib/python3/dist-packages/pythran/analyses/ast_matcher.pyr r cs"eZdZdZdfdd ZZS) Placeholderz= Class to save information from ast while check for pattern. Ncs||_||_tt|dS)z1 Placehorder are identified using an identifier. N)idtypesuperr__init__)selfZ identifierr __class__rrrszPlaceholder.__init__Nr rrrr __classcell__rrrrrsrc@r )AST_anyz< Class to specify we don't care about a field value in ast. Nr rrrrrrrcs eZdZdZfddZZS)AST_orz Class to specify multiple possibles value for a given field in ast. Attributes ---------- args: [ast field value] List of possible value for a field of an ast. cs||_tt|dS)z) Initialiser to keep track of arguments. N)argsrr r)rr!rrrr*szAST_or.__init__rrrrrr s r c@sheZdZdZddZddZddZedd Zd d Z d d Z ddZ ddZ eddZ ddZdS)Checka Checker for ast <-> pattern. NodeVisitor is needed for specific behavior checker. Attributes ---------- node : AST node we want to compare with pattern placeholders : [AST] list of placeholder value for later comparison or replacement. cCs||_||_dS)z Initialize attributes. N)node placeholders)rr#r$rrrr?s zCheck.__init__cs0t|t|kr dStfddt||DS)z" Check if list of node are equal. Fc3s&|]\}}t|j|VqdSr)r"r$visit).0Znode_eltZ pattern_eltrrr Hz#Check.check_list..)lenallzip)rZ node_listZ pattern_listrr'r check_listDs  zCheck.check_listcCsZ|j|jvrt|j|j|j|jsdS|jdur$t|j|js$dS|j|j|j<dS)z Save matching node or compare it with the existing one. FIXME : What if the new placeholder is a better choice? FNT)rr$r"r#r%r isinstancerpatternrrrvisit_PlaceholderKs  zCheck.visit_PlaceholdercCsdS)z Every node match with it. Tr)_rrr visit_AST_any\szCheck.visit_AST_anycstfdd|jDS)z; Match if any of the or content match with the other node. c3s|] }j|VqdSr) field_matchr#)r&Zvalue_orr'rrr(csz%Check.visit_AST_or..)anyr!r/rr'r visit_AST_oras zCheck.visit_AST_orcsBtjtsdSt|jtkrtdtfddt|jDS)z Set have unordered values. FzPattern for Set is too longc3s |] }jj|VqdSr)r-r#elts)r&Z pattern_eltsr'rrr(lsz"Check.visit_Set..) r.r#rr*r7MAX_UNORDERED_LENGTHr r5rr/rr'r visit_Setfs  zCheck.visit_Setcst|jtsdStjtkrtdttt|jjD]-}t |D]\}}| |jj|j|s6nq#fdd|D}| |jj |SqdS)z' Dict can match with unordered values. FzPattern for Dict is too longcsg|]}j|qSr)values)r&ir0rr {sz$Check.visit_Dict..) r.r#rr*keysr8r rrange enumerater4r-r:)rr0Z permutationr;valueZpattern_valuesrr<r visit_Dictos zCheck.visit_DictcCs>t|tr |||St|trt||j|St||S)z Check if two fields match. Field match if: - If it is a list, all values have to match. - If if is a node, recursively check it. - Otherwise, check values are equal. )r.listr-rr"r$r% strict_eq)rZ node_fieldZ pattern_fieldrrrr4s   zCheck.field_matchcCs4||krdSz t|ot|WStyYdSw)NTF)r TypeError)Zf0f1rrrrDs zCheck.strict_eqcs2ttjs dStfddtjDS)z Check if the pattern match with the checked node. a node match if: - type match - all field match Fc3s&|]\}}|t|VqdSr)r4getattr)r&ZfieldrAr0rrrr(r)z&Check.generic_visit..)r.rr#r+rr/rrHr generic_visits zCheck.generic_visitN)r rrrrr-r1 staticmethodr3r6r9rBr4rDrIrrrrr"0s     r"cs0eZdZdZfddZddZddZZS) ASTMatchera Visitor to gather node matching with a given pattern. Examples -------- >>> import gast as ast >>> code = "[(i, j) for i in range(a) for j in range(b)]" >>> pattern = ast.Call(func=ast.Name('range', ctx=ast.Load(), ... annotation=None, ... type_comment=None), ... args=AST_any(), keywords=[]) >>> len(ASTMatcher(pattern).search(ast.parse(code))) 2 >>> code = "[(i, j) for i in range(a) for j in range(b)]" >>> pattern = ast.Call(func=ast.Name(id=AST_or('range', 'range'), ... ctx=ast.Load(), ... annotation=None, ... type_comment=None), ... args=AST_any(), keywords=[]) >>> len(ASTMatcher(pattern).search(ast.parse(code))) 2 >>> code = "{1:2, 3:4}" >>> pattern = ast.Dict(keys=[ast.Constant(3, None), ast.Constant(1, None)], ... values=[ast.Constant(4, None), ... ast.Constant(2, None)]) >>> len(ASTMatcher(pattern).search(ast.parse(code))) 1 >>> code = "{1, 2, 3}" >>> pattern = ast.Set(elts=[ast.Constant(3, None), ... ast.Constant(2, None), ... ast.Constant(1, None)]) >>> len(ASTMatcher(pattern).search(ast.parse(code))) 1 cs ||_t|_tt|dS)z? Basic initialiser saving pattern and initialising result set. N)r0setresultrrKrr/rrrrszASTMatcher.__init__cCs.t|t|jr|j|||dS)z Visitor looking for matching between current node and pattern. If it match, save it but whatever happen, keep going. N)r"dictr%r0rMaddrIrr#rrrr%s zASTMatcher.visitcCs|||jS)z9 Facility to get values of the matcher for a given node. )r%rMrPrrrsearchs zASTMatcher.search)r rrrrr%rQrrrrrrKs  # rKN)rZgastrrrrr itertoolsrZmathrr8 Exceptionr rrr r"rKrrrrs   v