o
    i                     @   s8   d dl Z d dlmZ d dlmZmZ G dd deZdS )    N)api)report_errorsreport_warningsc                   @   s$   e Zd Zedd Zedd ZdS )SearchClientc                 C   sP   |j ddd}|jddd}d|_|j dd	d}|jdd
d |j| jd d S )Nsearchzsearch datadog)helpVerbsverb)titledestTqueryzSearch datadog.zoptionally faceted search query)func)
add_parseradd_subparsersrequiredadd_argumentset_defaults_query)cls
subparsersparserverb_parsersquery_parser r   K/home/ubuntu/.local/lib/python3.10/site-packages/datadog/dogshell/search.pysetup_parser   s   zSearchClient.setup_parserc                 C   s   |j t_tjj|jd}t| t| tdkrMt	|d 
 D ]*\}}t|D ]!\}}|dkr=td td||f  q(tddt| |f  q(q d S tdkrZtt| d S t	|d 
 D ]\}}|D ]
}td||f  qhqbd S )	N)qprettyresultsr   
z%s	%s raw)timeoutr   _timeoutInfrastructurer   r   r   r   formatlistitems	enumerateprintlenjsondumps)r   argsresfacetr   idxresultr   r   r   r      s(   zSearchClient._queryN)__name__
__module____qualname__classmethodr   r   r   r   r   r   r      s
    
	r   )r+   datadogr   datadog.dogshell.commonr   r   objectr   r   r   r   r   <module>   s   