Lg jddlmZddlZddlZddlmZddlmZddlZddl Z ddl Z ddl m Z ddl m Z ddgd d Zeje ej e d Zej%d ZejdZej%dZejdZej%dej.ddd%dZej%dej.dddZej%dej.ddej.ddej4dddej6ddZ d% d&d Zej%d!d"Zd#Zd$Z y)') annotationsN)reduce)Any) __version__)importlib_metadataz-hz--helpX)help_option_namesmax_content_width)context_settingscy)zDask command line interface.Nr (lib/python3.12/site-packages/dask/cli.pyclirs rc.ddl}|jdy)zBOpen Dask documentation (https://docs.dask.org/) in a web browser.rNzhttps://docs.dask.org) webbrowseropen)rs rdocsrsOO+,rcy)z)Information about your dask installation.Nr r rrinfor$ rcddlm}|y)z(Print versions of Dask related projects.r show_versionsN) dask.utilsrrs rversionsr*s )Orcy)zDask config settingsNr r rrconfigr2rrget)namekeyT)requiredc td|jdtjj}tttfr)t jtj|y |t jd y t j|y #tt f$r=t jt jd|ddtdYwxYw) z&Print config key, or the whole config.c ||SNr )dks rzconfig_get..=s1Q4r.zSection not found: red)fgT)errNNone)rsplitdaskrKeyError TypeErrorclickechostyleexit isinstancelistdictecho_via_pageryamldump)r!datas r config_getr>8s'39K9KL $t % TYYt_-  6 4 i  5;;!4SE:uE4P Qs5B$$A C0/C0findc ttjj|}|rt j d|dt tt|d}|D]}ttjj|g}tjj||}d|t t|z z}t j |d|d|d|d y t j d |d t j d jtttjjy )z9Find a Dask config key by searching config location pathszFound [z] in the following files:c*tt|Sr%)lenstr)vs rr(zconfig_find..Qs 3s1v;r)r!r z [=]zUnable to find [z ] in any of the following paths: N)r8r0rpaths_containing_keyr3r4rBrCmaxnext collect_yamlrjoinmappaths)r!rPmax_lenpathrvaluespacings r config_findrUJs 11#6 7E  WSE!:;<c#e)>?@AD$++22D6:;FKKOOCO7EWs3t9~56G JJ$q C5%: ;   %cU*JKL 499Sdkk&7&789:rsetrSz--fileFzFile to use, defaulting to the first config file containing the key, 'DASK_CONFIG' environment variable location or finally, '~/.config/dask/dask.yaml')defaultr"typehelpctjj|}t|||\}}t j d|d|d|y)z$Set a Dask config key to a new valuez Updated [z] to [z], config saved to N)r0rinterpret_value save_configr3r4)r!rSfile_rRs r config_setr_\sJ KK ' ' .E#ud+GAt JJ3%veW,?vFGrc<|Wttjj|t j tjj dz }|jjdd|jr(tj|jxsi}ni}tjj||i| |jtj|||fS#t $r}t#d|d|d}~wwxYw)z] Save new config values to dask config file, return new config and path to file. Nz dask.yamlT)exist_okparentsrEz. For some reason we couldn't write config to z. Perhaps you don't have permissions here? You can change the directory where Dask writes config files to somewhere else using the DASK_CONFIG environment variable. For example, you could set the following: export DASK_CONFIG=~/.dask )rLr0rrJpathlibPathPATHparentmkdirexistsr; safe_load read_textrV write_textr<OSError RuntimeError)r!rS config_fileres rr\r\qs KK , ,S 1 LL)) *[ 8 dD9 5 5 78>BKKOOS%LO0tyy01" ; ! --8M :    s$C== DDDr8ctjtjtj j y)zPrint the whole config.N)r3r:r;r<r0rr rr config_listrqs% 4;;#5#567rc |j}t |t jt jfs#tjdt|dy|j|jvr#tjd|jd|j|y#t$r0}tjd|jd|dYd}~yd}~wwxYw)aAdd `entry_point` command to `interface`. Parameters ---------- interface : click.Command or click.Group The click interface to augment with `entry_point`. entry_point : importlib.metadata.EntryPoint The entry point which loads to a ``click.Command`` or ``click.Group`` instance to be added as a sub-command or sub-group in `interface`. z)While registering the command with name 'z', an exception occurred; r)NzRentry points in 'dask_cli' must be instances of click.Command or click.Group, not zC', an existing command or group; the original has been overwritten.) load Exceptionwarningswarnr r7r3CommandGrouprXcommands add_command) interface entry_pointcommandros r_register_command_epr~s""$ g u{{; <  115gq B  ||y))) 7 ~FL L '"'  7 8H8H7IJ##$#Q (   sB,, C%5&C  C%cntjdD]}tt|ty)z$Run the dask command line interface.dask_cli)groupN)r entry_pointsr~r)eps rrun_clirs+ !--J?S"%@Err%)r!rCrSrrnzpathlib.Path | Nonereturnztuple[dict, pathlib.Path])! __future__rrcru functoolsrtypingrr3r;r0rdask._compatibilityrCONTEXT_SETTINGSrversion_optionrr}rrrrargumentr>rUoptionrdr_r\rqr~rr rrrs " 2)  ./k" #0 --      U% &  V% ;& ; U%$'     % H (&H(,* * *%* *ZV88 "#J r