OwlCyberSecurity - MANAGER
Edit File: btm_matcher.cpython-35.opt-2.pyc
���]� � @ s� d Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z Gd d � d e � Z Gd d � d e � Z i a d d � Z d S) z+George Boutsioukis <gboutsioukis@gmail.com>� N)� defaultdict� )�pytree)� reduce_treec @ s( e Z d Z e j � Z d d � Z d S)�BMNodec C s1 i | _ g | _ t t j � | _ d | _ d S)N� )�transition_table�fixers�nextr �count�id�content)�self� r �0/opt/alt/python35/lib64/python3.5/btm_matcher.py�__init__ s zBMNode.__init__N)�__name__� __module__� __qualname__� itertoolsr r r r r r r s r c @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d � Z d S) � BottomMatcherc C sF t � | _ t � | _ | j g | _ g | _ t j d � | _ d S)NZRefactoringTool) �set�matchr �rootZnodesr �logging� getLogger�logger)r r r r r s zBottomMatcher.__init__c C sh | j j | � t | j � } | j � } | j | d | j �} x | D] } | j j | � qJ Wd S)N�start)r �appendr Z pattern_treeZget_linear_subpattern�addr )r �fixerZtreeZlinear� match_nodesZ match_noder r r � add_fixer% s zBottomMatcher.add_fixerc C s | s | g St | d t � r� g } x[ | d D]O } | j | d | �} x1 | D]) } | j | j | d d � | � � qS Wq1 W| S| d | j k r� t � } | | j | d <n | j | d } | d d � r� | j | d d � d | �} n | g } | Sd S)Nr r r )� isinstance�tupler �extendr r )r �patternr r! Z alternativeZ end_nodes�endZ next_noder r r r 1 s" + " zBottomMatcher.addc C s� | j } t t � } xr| D]j} | } x[| r�d | _ x< | j D]1 } t | t j � rD | j d k rD d | _ PqD W| j d k r� | j } n | j } | | j k r� | j | } x� | j D]- } | | k r� g | | <| | j | � q� Wn� | j } | j d k r | j j r P| | j k ry| j | } x> | j D]3 } | | j � k rdg | | <| | j | � qBW| j } q+ Wq W| S)NT�;Fr )r r �listZ was_checkedZchildrenr# r ZLeaf�value�typer r r �parent�keys) r ZleavesZcurrent_ac_node�resultsZleafZcurrent_ast_nodeZchildZ node_tokenr r r r �runS s> ! zBottomMatcher.runc s7 t d � � f d d � � � | j � t d � d S)Nz digraph g{c s} xv | j j � D]e } | j | } t d | j | j t | � t | j � f � | d k rk t | j � � | � q Wd S)Nz%d -> %d [label=%s] //%sr )r r- �printr � type_repr�strr r )ZnodeZ subnode_keyZsubnode)� print_noder r r3 � s ) z*BottomMatcher.print_ac.<locals>.print_node�})r0 r )r r )r3 r �print_ac� s zBottomMatcher.print_acN)r r r r r" r r/ r5 r r r r r s " =r c C sb t sR d d l m } x9 | j j � D]( \ } } t | � t k r& | t | <q& Wt j | | � S)Nr )�python_symbols)� _type_reprsZpygramr6 �__dict__�itemsr+ �int� setdefault)Ztype_numr6 �name�valr r r r1 � s r1 )� __author__r r � collectionsr r r Z btm_utilsr �objectr r r7 r1 r r r r �<module> s �