EVOLUTION-MANAGER
Edit File: __init__.cpython-36.opt-1.pyc
3 �f � @ s& d dl Z d dlmZ G dd� d�ZdS )� N)�Lockc @ sd e Zd ZdZg Zg ZdZdZdZdd� Z dd� Z dd � Z d d � Z d d � Z dd� Zdd� Zdd� ZdS )�SoSMapz�Standardized way to store items with their obfuscated counterparts. Each type of sanitization that SoSCleaner supports should have a corresponding SoSMap() object, to allow for easy retrieval of obfuscated items. TFc C s i | _ t� | _g | _t� | _d S )N)�dataset�set� _regexes_made�compiled_regexesr �lock)�self� r � /usr/lib/python3.6/__init__.py�__init__ s zSoSMap.__init__c C s\ | s0|| j ks0|| jj� ks0| jr4t|�dkr4dS x"| jD ]}tj||tj�r<dS q<W dS )z�Some items need to be completely ignored, for example link-local or loopback addresses should not be obfuscated � TF) � skip_keysr �values�ignore_short_items�len�ignore_matches�re�match�I)r �item�skipr r r � ignore_item% s zSoSMap.ignore_itemc C sJ | j |�r|S | j�, | j|�| j|< | jr6| j|� | j| S Q R X dS )z�Add a particular item to the map, generating an obfuscated pair for it. Positional arguments: :param item: The plaintext object to obfuscate N)r r � sanitize_itemr �compile_regexes�add_regex_item)r r r r r �add1 s z SoSMap.addc C sR | j |�rdS || jkrN| jj|� | jj|| j|�f� | jjdd� dd� dS )z�Add an item to the regexes dict and then re-sort the list that the parsers will use during parse_line() :param item: The unobfuscated item to generate a regex for :type item: ``str`` Nc S s t | d �S )Nr )r )�xr r r �<lambda>V s z'SoSMap.add_regex_item.<locals>.<lambda>T)�key�reverse)r r r r �append�get_regex_result�sort)r r r r r r A s zSoSMap.add_regex_itemc C s2 | j rdtj|�� d�}n tj|�}tj|tj�S )a9 Generate the object/value that is used by the parser when iterating over pre-generated regexes during parse_line(). For most parsers this will simply be a ``re.Pattern()`` object, but for more complex parsers this can be overridden to provide a different object, e.g. a tuple, for that parer's specific iteration needs. :param item: The unobfuscated string to generate the regex for :type item: ``str`` :returns: A compiled regex pattern for the item :rtype: ``re.Pattern`` z (?=\b|_|-))�match_full_words_onlyr �escape�compiler )r r r r r r" X s zSoSMap.get_regex_resultc C s |S )z�Perform the obfuscation relevant to the item being added to the map. This should be overridden by each type of map that subclasses SoSMap Positional arguments: :param item: The plaintext object to obfuscate r )r r r r r r k s zSoSMap.sanitize_itemc C s, | j |�r|S || jkr"| j|�S | j| S )z�Retrieve an item's obfuscated counterpart from the map. If the item does not yet exist in the map, add it by generating one on the fly )r r r )r r r r r �getv s z SoSMap.getc C s | j j|� dS )a) Update the map using information from a previous run to ensure that we have consistent obfuscation between reports Positional arguments: :param map_dict: A dict of mappings with the form of {clean_entry: 'obfuscated_entry'} N)r �update)r Zmap_dictr r r � conf_update� s zSoSMap.conf_updateN)�__name__� __module__� __qualname__�__doc__r r r r r$ r r r r r" r r' r) r r r r r s r )r Z threadingr r r r r r �<module> s