B
    P?ð[7  ã               @   s4   d Z ddlZddlmZ G dd„ deƒZeƒ jZdS )aê  
S-Expression Tokenizer

``SExprTokenizer`` is used to find parenthesized expressions in a
string.  In particular, it divides a string into a sequence of
substrings that are either parenthesized expressions (including any
nested parenthesized expressions), or other whitespace-separated
tokens.

    >>> from nltk.tokenize import SExprTokenizer
    >>> SExprTokenizer().tokenize('(a b (c d)) e f (g)')
    ['(a b (c d))', 'e', 'f', '(g)']

By default, `SExprTokenizer` will raise a ``ValueError`` exception if
used to tokenize an expression with non-matching parentheses:

    >>> SExprTokenizer().tokenize('c) d) e (f (g')
    Traceback (most recent call last):
      ...
    ValueError: Un-matched close paren at char 1

The ``strict`` argument can be set to False to allow for
non-matching parentheses.  Any unmatched close parentheses will be
listed as their own s-expression; and the last partial sexpr with
unmatched open parentheses will be listed as its own sexpr:

    >>> SExprTokenizer(strict=False).tokenize('c) d) e (f (g')
    ['c', ')', 'd', ')', 'e', '(f (g']

The characters used for open and close parentheses may be customized
using the ``parens`` argument to the `SExprTokenizer` constructor:

    >>> SExprTokenizer(parens='{}').tokenize('{a b {c d}} e f {g}')
    ['{a b {c d}}', 'e', 'f', '{g}']

The s-expression tokenizer is also available as a function:

    >>> from nltk.tokenize import sexpr_tokenize
    >>> sexpr_tokenize('(a b (c d)) e f (g)')
    ['(a b (c d))', 'e', 'f', '(g)']

é    N)Ú
TokenizerIc               @   s"   e Zd ZdZd	dd„Zdd„ ZdS )
ÚSExprTokenizera\  
    A tokenizer that divides strings into s-expressions.
    An s-expresion can be either:

      - a parenthesized expression, including any nested parenthesized
        expressions, or
      - a sequence of non-whitespace non-parenthesis characters.

    For example, the string ``(a (b c)) d e (f)`` consists of four
    s-expressions: ``(a (b c))``, ``d``, ``e``, and ``(f)``.

    By default, the characters ``(`` and ``)`` are treated as open and
    close parentheses, but alternative strings may be specified.

    :param parens: A two-element sequence specifying the open and close parentheses
        that should be used to find sexprs.  This will typically be either a
        two-character string, or a list of two strings.
    :type parens: str or list
    :param strict: If true, then raise an exception when tokenizing an ill-formed sexpr.
    ú()Tc             C   sZ   t |ƒdkrtdƒ‚|| _|d | _|d | _t dt |d ¡t |d ¡f ¡| _d S )Né   z'parens must contain exactly two stringsr   é   z%s|%s)	ÚlenÚ
ValueErrorÚ_strictÚ_open_parenÚ_close_parenÚreÚcompileÚescapeÚ_paren_regexp)ÚselfZparensÚstrict© r   ú2lib/python3.7/site-packages/nltk/tokenize/sexpr.pyÚ__init__O   s    

zSExprTokenizer.__init__c             C   sü   g }d}d}x²| j  |¡D ]¢}| ¡ }|dkrN|||| ¡ …  ¡ 7 }| ¡ }|| jkr`|d7 }|| jkr| jrˆ|dkrˆtd| ¡  ƒ‚t	d|d ƒ}|dkr| 
||| ¡ … ¡ | ¡ }qW | jrÚ|dkrÚtd| ƒ‚|t|ƒk rø| 
||d… ¡ |S )aQ  
        Return a list of s-expressions extracted from *text*.
        For example:

            >>> SExprTokenizer().tokenize('(a b (c d)) e f (g)')
            ['(a b (c d))', 'e', 'f', '(g)']

        All parentheses are assumed to mark s-expressions.
        (No special processing is done to exclude parentheses that occur
        inside strings, or following backslash characters.)

        If the given expression contains non-matching parentheses,
        then the behavior of the tokenizer depends on the ``strict``
        parameter to the constructor.  If ``strict`` is ``True``, then
        raise a ``ValueError``.  If ``strict`` is ``False``, then any
        unmatched close parentheses will be listed as their own
        s-expression; and the last partial s-expression with unmatched open
        parentheses will be listed as its own s-expression:

            >>> SExprTokenizer(strict=False).tokenize('c) d) e (f (g')
            ['c', ')', 'd', ')', 'e', '(f (g']

        :param text: the string to be tokenized
        :type text: str or iter(str)
        :rtype: iter(str)
        r   r   z!Un-matched close paren at char %dz Un-matched open paren at char %dN)r   ÚfinditerÚgroupÚstartÚsplitr
   r   r	   r   ÚmaxÚappendÚendr   )r   ÚtextÚresultÚposZdepthÚmZparenr   r   r   ÚtokenizeY   s,    

zSExprTokenizer.tokenizeN)r   T)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r    r   r   r   r   r   9   s   

r   )r$   r   Znltk.tokenize.apir   r   r    Zsexpr_tokenizer   r   r   r   Ú<module>2   s   S