o
    rZŽh*  ã                   @   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   s\   t |ƒdkr
tdƒ‚|| _|d | _|d | _t t |d ¡› dt |d ¡› ¡| _d S )Né   z'parens must contain exactly two stringsr   é   ú|)	ÚlenÚ
ValueErrorÚ_strictÚ_open_parenÚ_close_parenÚreÚcompileÚescapeÚ_paren_regexp)ÚselfÚparensÚstrict© r   úB/var/www/auris/lib/python3.10/site-packages/nltk/tokenize/sexpr.pyÚ__init__O   s   

 
ÿzSExprTokenizer.__init__c                 C   sø   g }d}d}| j  |¡D ]Q}| ¡ }|dkr&|||| ¡ …  ¡ 7 }| ¡ }|| jkr/|d7 }|| jkr]| jrC|dkrCtd| ¡  ƒ‚t	d|d ƒ}|dkr]| 
||| ¡ … ¡ | ¡ }q| jrk|dkrktd| ƒ‚|t|ƒk rz| 
||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ÚposÚ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>   s
   +S