
    ∋d              	      x   d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZ  ej	        d e
d                     ej        dd	
           ej        dedd           ej        dedd           ej                    dd                                                            Zedk    r e             dS dS )    )annotationsN)run_specspecT)ignore_unknown_options)namecontext_settingsargs)nargsz--spec )typedefaulthelpz--spec-fileliststr	spec_filereturnNonec                    |r|s|s%|s#t          d           t          j        d           i |rNt          |          5 }                    t          j        |                     ddd           n# 1 swxY w Y   |r'                    t          j        |                     dv rd         	                    dd          i fd}t          j         |                       dS )	z:Launch a Dask process defined by a JSON/YAML specificationz2Must specify exactly one of --spec and --spec-file   Nclsoptsr   r   c                   K   t          gR   d {V } 	 t          j        d |                                 D               d {V  d S # t          $ r4 t          j        d |                                 D               d {V  Y d S w xY w)Nc              3  >   K   | ]}|                                 V  d S N)finished.0ws     9lib/python3.11/site-packages/distributed/cli/dask_spec.py	<genexpr>z$main.<locals>.run.<locals>.<genexpr>&   s*      "J"JA1::<<"J"J"J"J"J"J    c              3  >   K   | ]}|                                 V  d S r   )closer   s     r    r!   z$main.<locals>.run.<locals>.<genexpr>(   s*      "G"G17799"G"G"G"G"G"Gr"   )r   asynciogathervaluesKeyboardInterrupt)servers_specr	   s    r    runzmain.<locals>.run#   s       ..........	I."J"J9I9I"J"J"JKKKKKKKKKK  	I 	I 	I."G"Ggnn6F6F"G"G"GHHHHHHHHHHH	Is   0A
 
:BB)printsysexitopenupdateyaml	safe_loadjsonloadsgetr%   r+   )r	   r   r   fr+   r*   s   `    @r    mainr7      sU     	  i BCCCE ,)__ 	,LL**+++	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	,  'TZ%%&&&~~v""61--u5I I I I I I Ks   (A55A9<A9__main__)r	   r   r   r   r   r   r   r   )
__future__r   r%   r3   r-   clickr1   distributed.deploy.specr   commanddictargumentoptionr   version_optionr7   __name__ r"   r    <module>rC      s,   " " " " " "   



   , , , , , , FTT-N-N-NOOOb!!!hS"2666m#t"===    >= 76 "! PO
6 zDFFFFF r"   