U
    	ڲg\d                     @   s   U d dl mZmZmZmZmZ d dlZd dlmZm	Z	m
Z
mZmZ ddlmZ eefZeedf ed< edd	 d
D ZG dd dZG dd dZee ee dS )    )MutableMappingMappingMutableSequenceIteratorIterableN)TupleAnyDictSetList   )replaced_by_pep8.str_typec                 c   s   | ]
}|V  qd S N ).0_r   r   5/tmp/pip-unpacked-wheel-8n0dx3ox/pyparsing/results.py	<genexpr>   s     r   r   c                   @   sL   e Zd ZU edef ed< dgZdedddZdd Zdd	 Z	d
d Z
dS )_ParseResultsWithOffsetParseResultstup)p1p2c                 C   s   ||f| _ d S r   r   )selfr   r   r   r   r   __init__   s    z _ParseResultsWithOffset.__init__c                 C   s
   | j | S r   r   r   ir   r   r   __getitem__   s    z#_ParseResultsWithOffset.__getitem__c                 C   s   | j S r   r   r   r   r   r   __getstate__   s    z$_ParseResultsWithOffset.__getstate__c                 G   s   |d | _ d S Nr   r   )r   argsr   r   r   __setstate__    s    z$_ParseResultsWithOffset.__setstate__N)__name__
__module____qualname__r   int__annotations__	__slots__r   r   r!   r$   r   r   r   r   r      s   
r   c                   @   sR  e Zd ZU dZdg dfZeedf ed< eed< d ed< e	e ed< e
ed	< ee ed
< eeef ed< dZG dd deZdbddZddddefddZdd ZefddZdd Ze
dddZedddZe
ddd Zedd!d"Zedd#d$Zd%d& Zd'd( Zd)d* Ze
dd+d,Zd-d. Z dcd/d0Z!d1d2 Z"d3d4 Z#d5d6 Z$d7d8 Z%d9d: Z&d d d;d<d=Z'd d d;d>d?Z(d dd@dAZ)eddBdCZ*eddDdEZ+dddGdHZ,eddIdJZ-e.ddKdLZ/d ddMdNZ0d ddOdPZ1eddQdRZ2deeddTdUZ3dVdW Z4dXdY Z5dZd[ Z6d\d] Z7d^d_ Z8e9dfd dd`daZ:e-Z;e/Z<e2Z=dS )gr   a{  Structured parse results, to provide multiple means of access to
    the parsed data:

    - as a list (``len(results)``)
    - by list index (``results[0], results[1]``, etc.)
    - by attribute (``results.<results_name>`` - see :class:`ParserElement.set_results_name`)

    Example::

        integer = Word(nums)
        date_str = (integer.set_results_name("year") + '/'
                    + integer.set_results_name("month") + '/'
                    + integer.set_results_name("day"))
        # equivalent form:
        # date_str = (integer("year") + '/'
        #             + integer("month") + '/'
        #             + integer("day"))

        # parse_string returns a ParseResults object
        result = date_str.parse_string("1999/12/31")

        def test(s, fn=repr):
            print(f"{s} -> {fn(eval(s))}")
        test("list(result)")
        test("result[0]")
        test("result['month']")
        test("result.day")
        test("'month' in result")
        test("'minutes' in result")
        test("result.dump()", str)

    prints::

        list(result) -> ['1999', '/', '12', '/', '31']
        result[0] -> '1999'
        result['month'] -> '12'
        result.day -> '31'
        'month' in result -> True
        'minutes' in result -> False
        result.dump() -> ['1999', '/', '12', '/', '31']
        - day: '31'
        - month: '12'
        - year: '1999'
    Nr   ._null_values_name_parent
_all_names_modal_toklist_tokdict)r,   r-   r.   r/   r0   r1   c                   @   s   e Zd ZdZdddZdS )zParseResults.Lista  
        Simple wrapper class to distinguish parsed list results that should be preserved
        as actual Python lists, instead of being converted to :class:`ParseResults`::

            LBRACK, RBRACK = map(pp.Suppress, "[]")
            element = pp.Forward()
            item = ppc.integer
            element_list = LBRACK + pp.DelimitedList(element) + RBRACK

            # add parse actions to convert from ParseResults to actual Python collection types
            def as_python_list(t):
                return pp.ParseResults.List(t.as_list())
            element_list.add_parse_action(as_python_list)

            element <<= item | element_list

            element.run_tests('''
                100
                [2,3,4]
                [[2, 1],3,4]
                [(2, 1),3,4]
                (2,3,4)
                ''', post_parse=lambda s, r: (r[0], type(r[0])))

        prints::

            100
            (100, <class 'int'>)

            [2,3,4]
            ([2, 3, 4], <class 'list'>)

            [[2, 1],3,4]
            ([[2, 1], 3, 4], <class 'list'>)

        (Used internally by :class:`Group` when `aslist=True`.)
        Nc                 C   s:   |d krg }t |ts0t| j dt|j t| S )Nz* may only be constructed with a list, not )
isinstancelist	TypeErrorr%   type__new__)clsZ	containedr   r   r   r6      s    
zParseResults.List.__new__)N)r%   r&   r'   __doc__r6   r   r   r   r   r   d   s   &r   c                 K   s   t |tr|S t| }d |_d |_t |_|d kr<g |_n<t |t	t
frpt |tjrd|d d  gnt	||_n|g|_t |_|S r   )r2   r   objectr6   r,   r-   setr.   r0   r3   _generator_typer   dictr1   )r7   toklistnamekwargsr   r   r   r   r6      s     


zParseResults.__new__Tc              
   C   s   |  || _ |d ks|dkrd S ||tr0t|}|s<|h| _|| _|| jkrPd S ||ttfrd|g}|r||trt	t|j
d| |< nt	t|d d| |< || | _d S z|d | |< W n2 tttfk
r   || k	r|| |< n|| _Y nX d S )N r   )r/   r(   strr.   r,   r+   r   r5   r   r   r0   KeyErrorr4   
IndexError)r   r=   r>   asListmodalr2   r   r   r   r      s2    




zParseResults.__init__c                 C   sL   t |ttfr| j| S || jkr4| j| d d S tdd | j| D S )Nr   c                 S   s   g | ]}|d  qS )r   r   r   vr   r   r   
<listcomp>   s     z,ParseResults.__getitem__.<locals>.<listcomp>)r2   r(   slicer0   r.   r1   r   r   r   r   r   r      s
    

zParseResults.__getitem__c                 C   s   ||t r0| j|t |g | j|< |d }nD||ttfrN|| j|< |}n&| j|t t |dg | j|< |}||tr| |_d S r"   )	r   r1   getr3   r(   rJ   r0   r   r-   )r   krH   r2   subr   r   r   __setitem__   s    



zParseResults.__setitem__c           	      C   s   t |ttfs| j|= d S t| j}| j|= t |trT|dk rF||7 }t||d }tt|| }|	  | j
 D ]:}|D ]0}t|D ]"\}\}}t||||k ||< qqqxd S )Nr   r   )r2   r(   rJ   r1   lenr0   r3   rangeindicesreversevalues	enumerater   )	r   r   mylenremovedoccurrencesjrL   valuepositionr   r   r   __delitem__   s$    

 
zParseResults.__delitem__)returnc                 C   s
   || j kS r   r1   )r   rL   r   r   r   __contains__  s    zParseResults.__contains__c                 C   s
   t | jS r   )rO   r0   r    r   r   r   __len__  s    zParseResults.__len__c                 C   s   | j p
| j  S r   )r0   r1   r    r   r   r   __bool__
  s    zParseResults.__bool__c                 C   s
   t | jS r   iterr0   r    r   r   r   __iter__  s    zParseResults.__iter__c                 C   s   t | jd d d S )NrF   ra   r    r   r   r   __reversed__  s    zParseResults.__reversed__c                 C   s
   t | jS r   )rb   r1   r    r   r   r   keys  s    zParseResults.keysc                    s    fdd   D S )Nc                 3   s   | ]} | V  qd S r   r   r   rL   r    r   r   r     s     z&ParseResults.values.<locals>.<genexpr>re   r    r   r    r   rS     s    zParseResults.valuesc                    s    fdd   D S )Nc                 3   s   | ]}| | fV  qd S r   r   rf   r    r   r   r     s     z%ParseResults.items.<locals>.<genexpr>rg   r    r   r    r   items  s    zParseResults.itemsc                 C   s
   | j   S )z
        Since ``keys()`` returns an iterator, this method is helpful in bypassing
        code that looks for the existence of any defined results names.r]   r    r   r   r   haskeys  s    zParseResults.haskeysc                 O   s   |s
dg}|  D ],\}}|dkr0|d |f}qtd|qt|d tsft|dksf|d | kr|d }| | }| |= |S |d }|S dS )a  
        Removes and returns item at specified index (default= ``last``).
        Supports both ``list`` and ``dict`` semantics for ``pop()``. If
        passed no argument or an integer argument, it will use ``list``
        semantics and pop tokens from the list of parsed tokens. If passed
        a non-integer argument (most likely a string), it will use ``dict``
        semantics and pop the corresponding value from any defined results
        names. A second default return value argument is supported, just as in
        ``dict.pop()``.

        Example::

            numlist = Word(nums)[...]
            print(numlist.parse_string("0 123 321")) # -> ['0', '123', '321']

            def remove_first(tokens):
                tokens.pop(0)
            numlist.add_parse_action(remove_first)
            print(numlist.parse_string("0 123 321")) # -> ['123', '321']

            label = Word(alphas)
            patt = label("LABEL") + Word(nums)[1, ...]
            print(patt.parse_string("AAB 123 321").dump())

            # Use pop() in a parse action to remove named result (note that corresponding value is not
            # removed from list form of results)
            def remove_LABEL(tokens):
                tokens.pop("LABEL")
                return tokens
            patt.add_parse_action(remove_LABEL)
            print(patt.parse_string("AAB 123 321").dump())

        prints::

            ['AAB', '123', '321']
            - LABEL: 'AAB'

            ['AAB', '123', '321']
        rF   defaultr   z)pop() got an unexpected keyword argument r   N)rh   r4   r2   r(   rO   )r   r#   r?   rL   rH   indexretdefaultvaluer   r   r   pop"  s    (&zParseResults.popc                 C   s   || kr| | S |S dS )a^  
        Returns named result matching the given key, or if there is no
        such name, then returns the given ``default_value`` or ``None`` if no
        ``default_value`` is specified.

        Similar to ``dict.get()``.

        Example::

            integer = Word(nums)
            date_str = integer("year") + '/' + integer("month") + '/' + integer("day")

            result = date_str.parse_string("1999/12/31")
            print(result.get("year")) # -> '1999'
            print(result.get("hour", "not specified")) # -> 'not specified'
            print(result.get("hour")) # -> None
        Nr   )r   keydefault_valuer   r   r   rK   Z  s    zParseResults.getc                 C   sN   | j || | j D ]0}t|D ]"\}\}}t||||k ||< q$qdS )a;  
        Inserts new element at location index in the list of parsed tokens.

        Similar to ``list.insert()``.

        Example::

            numlist = Word(nums)[...]
            print(numlist.parse_string("0 123 321")) # -> ['0', '123', '321']

            # use a parse action to insert the parse location in the front of the parsed results
            def insert_locn(locn, tokens):
                tokens.insert(0, locn)
            numlist.add_parse_action(insert_locn)
            print(numlist.parse_string("0 123 321")) # -> [0, '0', '123', '321']
        N)r0   insertr1   rS   rT   r   )r   rk   Z
ins_stringrW   rL   rY   rZ   r   r   r   rq   q  s     
zParseResults.insertc                 C   s   | j | dS )a  
        Add single element to end of ``ParseResults`` list of elements.

        Example::

            numlist = Word(nums)[...]
            print(numlist.parse_string("0 123 321")) # -> ['0', '123', '321']

            # use a parse action to compute the sum of the parsed integers, and add it to the end
            def append_sum(tokens):
                tokens.append(sum(map(int, tokens)))
            numlist.add_parse_action(append_sum)
            print(numlist.parse_string("0 123 321")) # -> ['0', '123', '321', 444]
        N)r0   append)r   itemr   r   r   rr     s    zParseResults.appendc                 C   s&   t |tr| | n| j| dS )a  
        Add sequence of elements to end of ``ParseResults`` list of elements.

        Example::

            patt = Word(alphas)[1, ...]

            # use a parse action to append the reverse of the matched strings, to make a palindrome
            def make_palindrome(tokens):
                tokens.extend(reversed([t[::-1] for t in tokens]))
                return ''.join(tokens)
            patt.add_parse_action(make_palindrome)
            print(patt.parse_string("lskdj sdlkjf lksd")) # -> 'lskdjsdlkjflksddsklfjkldsjdksl'
        N)r2   r   __iadd__r0   extend)r   itemseqr   r   r   ru     s    
zParseResults.extendc                 C   s   | j dd= | j  dS )z7
        Clear all elements and results names.
        N)r0   r1   clearr    r   r   r   rw     s    zParseResults.clearc                 C   s8   z
| | W S  t k
r2   |dr,t|Y dS X d S )N__r@   )rB   
startswithAttributeError)r   r>   r   r   r   __getattr__  s    

zParseResults.__getattr__)otherr\   c                 C   s   |   }||7 }|S r   )copy)r   r|   rl   r   r   r   __add__  s    zParseResults.__add__c                    s   |s| S |j rnt| jfdd |j  } fdd|D }|D ](\}}|| |< t|d trD| |d _qD|  j|j7  _|  j|jO  _| S )Nc                    s   | dk r S |   S r"   r   )a)offsetr   r   <lambda>      z'ParseResults.__iadd__.<locals>.<lambda>c              	      s4   g | ],\}}|D ]}|t |d   |d fqqS )r   r   )r   )r   rL   vlistrH   )	addoffsetr   r   rI     s    z)ParseResults.__iadd__.<locals>.<listcomp>r   )r1   rO   r0   rh   r2   r   r-   r.   )r   r|   
otheritemsotherdictitemsrL   rH   r   )r   r   r   rt     s     


zParseResults.__iadd__c                 C   s&   t |tr|dkr|  S ||  S d S r"   )r2   r(   r}   )r   r|   r   r   r   __radd__  s    zParseResults.__radd__c                 C   s"   t | j d| jd|   dS )N(, ))r5   r%   r0   as_dictr    r   r   r   __repr__  s    zParseResults.__repr__c                 C   s   dd dd | jD  d S )N[r   c                 S   s&   g | ]}t |trt|nt|qS r   )r2   r   rA   repr)r   r   r   r   r   rI     s   z(ParseResults.__str__.<locals>.<listcomp>])joinr0   r    r   r   r   __str__  s    zParseResults.__str__r@   c                 C   sL   g }| j D ]<}|r |r || t|tr8|| 7 }q
|t| q
|S r   )r0   rr   r2   r   _asStringListrA   )r   sepoutrs   r   r   r   r     s    


zParseResults._asStringListc                 C   s   dd | j D S )ax  
        Returns the parse results as a nested list of matching tokens, all converted to strings.

        Example::

            patt = Word(alphas)[1, ...]
            result = patt.parse_string("sldkj lsdkj sldkj")
            # even though the result prints in string-like form, it is actually a pyparsing ParseResults
            print(type(result), result) # -> <class 'pyparsing.ParseResults'> ['sldkj', 'lsdkj', 'sldkj']

            # Use as_list() to create an actual list
            result_list = result.as_list()
            print(type(result_list), result_list) # -> <class 'list'> ['sldkj', 'lsdkj', 'sldkj']
        c                 S   s"   g | ]}t |tr| n|qS r   )r2   r   as_list)r   resr   r   r   rI   
  s   z(ParseResults.as_list.<locals>.<listcomp>)r0   r    r   r   r   r     s    zParseResults.as_listc                    s&    fdd t  fdd|  D S )a  
        Returns the named parse results as a nested dictionary.

        Example::

            integer = Word(nums)
            date_str = integer("year") + '/' + integer("month") + '/' + integer("day")

            result = date_str.parse_string('12/31/1999')
            print(type(result), repr(result)) # -> <class 'pyparsing.ParseResults'> (['12', '/', '31', '/', '1999'], {'day': [('1999', 4)], 'year': [('12', 0)], 'month': [('31', 2)]})

            result_dict = result.as_dict()
            print(type(result_dict), repr(result_dict)) # -> <class 'dict'> {'day': '1999', 'year': '12', 'month': '31'}

            # even though a ParseResults supports dict-like access, sometime you just need to have a dict
            import json
            print(json.dumps(result)) # -> Exception: TypeError: ... is not JSON serializable
            print(json.dumps(result.as_dict())) # -> {"month": "31", "day": "1999", "year": "12"}
        c                    s4   t | tr,|  r|  S  fdd| D S | S d S )Nc                    s   g | ]} |qS r   r   rG   to_itemr   r   rI   &  s     z9ParseResults.as_dict.<locals>.to_item.<locals>.<listcomp>)r2   r   ri   r   objr   r   r   r   $  s    
"z%ParseResults.as_dict.<locals>.to_itemc                 3   s   | ]\}}| |fV  qd S r   r   r   rL   rH   r   r   r   r   *  s     z'ParseResults.as_dict.<locals>.<genexpr>)r<   rh   r    r   r   r   r     s    zParseResults.as_dictc                 C   s:   t | j}| j |_| j|_| j| jO  _| j|_|S )a  
        Returns a new shallow copy of a :class:`ParseResults` object. `ParseResults`
        items contained within the source are shared with the copy. Use
        :class:`ParseResults.deepcopy()` to create a copy with its own separate
        content values.
        )r   r0   r1   r}   r-   r.   r,   )r   rl   r   r   r   r}   ,  s    
zParseResults.copyc                 C   s   |   }t| jD ]\}}t|tr4| |j|< qt|ttfrDqt|trt	|  |j|< }|
 D ]"\}}t|tr| n|||< qjqt|trt	|dd |D |j|< q|S )zL
        Returns a new deep copy of a :class:`ParseResults` object.
        c                 s   s$   | ]}t |tr| n|V  qd S r   )r2   r   deepcopyrG   r   r   r   r   J  s    z(ParseResults.deepcopy.<locals>.<genexpr>)r}   rT   r0   r2   r   r   rA   bytesr   r5   rh   r   )r   rl   r   r   destrL   rH   r   r   r   r   :  s    


zParseResults.deepcopyc                    s    j r j S  jr: j}|j }t fdd|D dS t dkrt jdkrtt j d d dkrtt j S dS dS )a  
        Returns the results name for this token expression. Useful when several
        different expressions might match at a particular location.

        Example::

            integer = Word(nums)
            ssn_expr = Regex(r"\d\d\d-\d\d-\d\d\d\d")
            house_number_expr = Suppress('#') + Word(nums, alphanums)
            user_data = (Group(house_number_expr)("house_number")
                        | Group(ssn_expr)("ssn")
                        | Group(integer)("age"))
            user_info = user_data[1, ...]

            result = user_info.parse_string("22 111-22-3333 #221B")
            for item in result:
                print(item.get_name(), ':', item[0])

        prints::

            age : 22
            ssn : 111-22-3333
            house_number : 221B
        c                 3   s,   | ]$\}}|D ]\}}| kr|V  qqd S r   r   )r   rL   r   rH   locr    r   r   r   n  s
    z(ParseResults.get_name.<locals>.<genexpr>Nr   r   )r   rF   )	r,   r-   r1   rh   nextrO   rb   rS   re   )r   parZparent_tokdict_itemsr   r    r   get_nameO  s&    



zParseResults.get_namer   c                 C   s  g }d}| |r |t|   nd |s4d|S |  rtdd |  D }|D ]~\}}	|rl| | | | d|  d| d t|	ts| t	|	 qV|	s| t|	 qV| |	j
||||d d	 qVtd
d | D sd|S | }	d}
d}t|	D ]\}}t|trl|j
||||d d	}| | | |
|  d| d| | |
|d   | 
 n:| | | |
|  d| d| | |
|d   | 
 qd|S )aM  
        Diagnostic method for listing out the contents of
        a :class:`ParseResults`. Accepts an optional ``indent`` argument so
        that this string can be embedded in a nested display of other data.

        Example::

            integer = Word(nums)
            date_str = integer("year") + '/' + integer("month") + '/' + integer("day")

            result = date_str.parse_string('1999/12/31')
            print(result.dump())

        prints::

            ['1999', '/', '12', '/', '31']
            - day: '31'
            - month: '12'
            - year: '1999'
        
r@   c                 s   s   | ]\}}t ||fV  qd S r   )rA   r   r   r   r   r     s     z$ParseResults.dump.<locals>.<genexpr>z  z- z: r   )indentfullinclude_list_depthc                 s   s   | ]}t |tV  qd S r   )r2   r   )r   vvr   r   r   r     s     r   z]:)rr   rA   r   r   ri   sortedrh   r2   r   r   dumpanyrT   )r   r   r   r   r   r   NLrh   rL   rH   incrnlr   r   Zvv_dumpr   r   r   r     sZ    



22zParseResults.dumpc                 O   s   t j |  f|| dS )a$  
        Pretty-printer for parsed results as a list, using the
        `pprint <https://docs.python.org/3/library/pprint.html>`_ module.
        Accepts additional positional or keyword args as defined for
        `pprint.pprint <https://docs.python.org/3/library/pprint.html#pprint.pprint>`_ .

        Example::

            ident = Word(alphas, alphanums)
            num = Word(nums)
            func = Forward()
            term = ident | num | Group('(' + func + ')')
            func <<= ident + Group(Optional(DelimitedList(term)))
            result = func.parse_string("fna a,b,(fnb c,d,200),100")
            result.pprint(width=40)

        prints::

            ['fna',
             ['a',
              'b',
              ['(', 'fnb', ['c', 'd', '200'], ')'],
              '100']]
        N)pprintr   )r   r#   r?   r   r   r   r     s    zParseResults.pprintc                 C   s   | j | j d | j| jffS r   )r0   r1   r}   r.   r,   r    r   r   r   r!     s    zParseResults.__getstate__c                 C   s*   |\| _ \| _}}| _t|| _d | _d S r   )r0   r1   r,   r:   r.   r-   )r   stater   inAccumNamesr   r   r   r$     s    
zParseResults.__setstate__c                 C   s   | j | jfS r   )r0   r,   r    r   r   r   __getnewargs__  s    zParseResults.__getnewargs__c                 C   s   t t| t|   S r   )dirr5   r3   re   r    r   r   r   __dir__  s    zParseResults.__dir__c                 C   sr   dd }| g }|  D ]>\}}t|tr>|| j||d7 }q|| |g|||d7 }q|dk	rn| |g|d}|S )z
        Helper classmethod to construct a ``ParseResults`` from a ``dict``, preserving the
        name-value relations as results names. If an optional ``name`` argument is
        given, a nested ``ParseResults`` will be returned.
        c                 S   s4   zt |  W n tk
r"   Y dS X t| t S d S )NF)rb   	Exceptionr2   r   r   r   r   r   is_iterable  s
    z+ParseResults.from_dict.<locals>.is_iterable)r>   )r>   rD   N)rh   r2   r   	from_dict)r7   r|   r>   r   rl   rL   rH   r   r   r   r     s    	
zParseResults.from_dict)NN)N)r@   )r@   TTr   )N)>r%   r&   r'   r8   r+   r   r   r)   rA   r
   boolr   r	   r*   r3   r6   r2   r   r   rN   r[   r^   r(   r_   r`   r   rc   rd   re   rS   rh   ri   rn   rK   rq   rr   ru   rw   r{   r~   rt   r   r   r   r   r   r<   r   r}   r   r   r   r   r!   r$   r   r   classmethodr   rD   asDictgetNamer   r   r   r   r   $   sv   
-	2
    
'	8

0Jr   )collections.abcr   r   r   r   r   r   typingr   r   r	   r
   r   utilr   rA   r   r   r5   r)   r;   r   r   registerr   r   r   r   <module>   s            
