
    9Yf                     >    d dl Z d dlmZ d dlmZmZ  G d de      Zy)    N)api)report_errorsreport_warningsc                   ,    e Zd Zed        Zed        Zy)SearchClientc                     |j                  dd      }|j                  dd      }d|_        |j                  dd	      }|j                  dd
       |j	                  | j
                         y )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_parsers        W/var/www/highfloat_scraper/venv/lib/python3.12/site-packages/datadog/dogshell/search.pysetup_parserzSearchClient.setup_parser   st    &&x6F&G,,7,H $#..w=N.O!!'0Q!R!!szz!2    c                 p   |j                   t        _        t        j                  j	                  |j
                        }t        |       t        |       t        dk(  rut        |d   j                               D ]U  \  }}t        |      D ]B  \  }}|dk(  rt        d       t        |d|       't        dt        |      z  d|       D W y t        dk(  rt        t        j                  |             y t        |d   j                               D ]  \  }}|D ]  }t        |d|         y )	N)qprettyresultsr   
	 raw)timeoutr   _timeoutInfrastructurer	   r   r   r   formatlistitems	enumerateprintlenjsondumps)r   argsresfacetr"   idxresults          r   r   zSearchClient._query   s   ||  ''$**'5cX"&s9~';';'="> Ew#,W#5 EKCaxd%89#E
*:FCDEE u_$**S/""&s9~';';'="> 6w% 6FeV4566r   N)__name__
__module____qualname__classmethodr   r    r   r   r   r      s(    3 3 6 6r   r   )r0   datadogr   datadog.dogshell.commonr   r   objectr   r;   r   r   <module>r?      s   
   B66 6r   