o
    Zh                     @   s   d Z ddlZddlmZ ddlmZ ddlmZ ddlm	Z	 e
 ZG dd deZd	d
 Zdd Zdd Zdd Zdd Zdd ZdddZdS )ap  
This module is here for string completions. This means mostly stuff where
strings are returned, like `foo = dict(bar=3); foo["ba` would complete to
`"bar"]`.

It however does the same for numbers. The difference between string completions
and other completions is mostly that this module doesn't return defined
names in a module, but pretty much an arbitrary string.
    N)AbstractArbitraryName)infer_call_of_leaf)
Completion)cut_value_at_positionc                   @   s   e Zd ZdZdZdS )
StringNamestringFN)__name__
__module____qualname__Zapi_typeZis_value_name r   r   G/var/www/html/lang_env/lib/python3.10/site-packages/jedi/api/strings.pyr      s    r   c                 C   s   |}|dkr
|  }d}|rt|||dd}|dkrU|d u r'||ur't||}| |}|   }	}
|	dv r9|	j}	|	jdv rUt||
}tt| j	||d u rNdn|||dS g S )N[ T)invert_result))]})Zatomtrailername)fuzzy)
Zget_previous_leafget_quote_endingr   Zcreate_contextparenttyper   list_completions_for_dictsinference_state)Zmodule_context
code_linesleafpositionr   r   Zbracket_leafcut_end_quotecontextZbefore_nodeZbefore_bracket_leafvaluesr   r   r   complete_dict   s.   



r"   c                 c   sh    t t|dd dD ]&}t||}||r1t| |d t| p!d  }t| |d t||dV  qd S )Nc                 S   s   t | S )N)repr)xr   r   r   <lambda>8   s    z(_completions_for_dicts.<locals>.<lambda>)key)stackZlike_name_lengthZis_fuzzy)sorted_get_python_keys_create_repr_string
startswithr   lenr   )r   dictsliteral_stringr   r   dict_keyZdict_key_strr   r   r   r   r   7   s   

r   c                 C   sf   t |ttfr	| st|S t|}t| \}}|d u r|S ||d kr'|| S || |dd  | S )Nr      )
isinstancestrbytesr#   _get_string_prefix_and_quote)r.   r/   rprefixquoter   r   r   r*   E   s   r*   c                 c   sB    | D ]}|j dkr| D ]}|jtd}|tur|V  qqd S )Ndict)default)Z
array_typeZget_key_valuesZget_safe_value	_sentinel)r-   dctr&   r/   r   r   r   r)   R   s   
r)   c                 C   s,   t d| }|d u rdS |d|dfS )Nz(\w*)("""|\'{3}|"|\'))NNr0      )rematchgroup)r   r?   r   r   r   r5   [   s   r5   c                 C   s0   | |d d  |d |d t |  }||kS )Nr   r0   )r,   )r   r8   r   r   r   r   r   _matches_quote_at_positionb   s   (rA   Fc                 C   s0   t | \}}|d u rdS t||||krdS |S )Nr   )r5   rA   )r   r   r   r   _r8   r   r   r   r   g   s   r   )F)__doc__r>   Zjedi.inference.namesr   Zjedi.inference.helpersr   Zjedi.api.classesr   Zjedi.parser_utilsr   objectr;   r   r"   r   r*   r)   r5   rA   r   r   r   r   r   <module>   s    		