a xdY=@s&ddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddZ ddZ ddZd d Zd d Zd dZddZddZddZd0ddZd1ddZddZddZd2ddZGd d!d!eZed"k Zd3d$d%Zd&d'Zd(d)Z d*d+Z!d,d-Z"d.d/Z#dS)4)print_functionNcCs.z t|tWSty(t|tYS0dSN) isinstance basestring NameErrorstrvaluer Q/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/lit/util.py is_strings  r cCsj|dur dSt|tur|St|tjr0|dkSt|rX|dvrHdS|dvrXdStd|dS)NFr)1trueonyesT)0falseoffnoz"{}" is not a valid boolean) typeboolrnumbersNumberr lower ValueErrorformatrr r r pythonize_bools    rcCs d|dS)Nz\br )wordr r r make_word_regex&srcCst|tr|S|dS)zReturn the parameter as type 'bytes', possibly encoding it. In Python2, the 'bytes' type is the same as 'str'. In Python3, they are distinct. utf-8)rbytesencodesr r r to_bytes*s r%cCsxt|tr|St|tr@z |dWSty>t|YS0z |dWStyrtdt|tfYn0dS)zReturn the parameter as type 'str', possibly encoding it. In Python2, the 'str' type is the same as 'bytes'. In Python3, the 'str' type is (essentially) Python2's 'unicode' type, and 'bytes' is distinct. r z not sure how to convert %s to %sN) rrr!decodeUnicodeDecodeErrorr"AttributeError TypeErrorr)br r r to_string;s       r+cCst|tr|dS|S)zReturn the parameter as type which supports unicode, possibly decoding it. In Python2, this is the unicode type. In Python3 it's the str type. r )rr!r&r#r r r to_unicodebs  r,cCsLzttd}Wnty0tp*d}Yn0tdkrHt|dS|S)zReturn the number of cores the current process can use, if supported. Otherwise, return the total number of cores (like `os.cpu_count()`). Default to 1 if undetermined. rWindows<)lenossched_getaffinityr( cpu_countplatformsystemmin)nr r r usable_core_countps   r8cCszttdkrhddlm}ddlm}m}tj|}| dd}t d|}|j |dsr||n t |Wn,tytd}|jtjkrYn0dS) Nr.r)windll) GetLastErrorWinError/\z\\?\%sr-)r4r5ctypesr9r:r;r1pathabspathreplacer,kernel32ZCreateDirectoryWmkdirOSErrorsysexc_infoerrnoEEXIST)r?r9r:r;ZNTPather r r rCs         rCcCs<|rtj|rdStj|}||kr0t|t|dS)zmkdir_p(path) - Make the "path" directory, if it does not exist; this will also make directories for any missing parent directories.N)r1r?existsdirnamemkdir_prC)r?parentr r r rLs  rLc#sv|durt}|durdh}t|D]Jtjtj|s&ds&|vs&tfdd|Dsjq&Vq&dS)aYields files in a directory. Filenames that are not excluded by rules below are yielded one at a time, as basenames (i.e., without dirname). Files starting with '.' are always skipped. If 'suffixes' is not None, then only filenames ending with one of its members will be yielded. These can be extensions, like '.exe', or strings, like 'Test'. (It is a lexicographic check; so an empty sequence will yield nothing, but a single empty string will yield all filenames.) If 'exclude_filenames' is not None, then none of the file basenames in it will be yielded. If specified, the containers for 'suffixes' and 'exclude_filenames' must support membership checking for strs. Args: dirname: a directory path. suffixes: (optional) a sequence of strings (set, list, etc.). exclude_filenames: (optional) a sequence of strings. Yields: Filenames as returned by os.listdir (generally, str). Nr.c3s|]}|VqdSr)endswith).0Zsfxfilenamer r z listdir_files..)setr1listdirr?isdirjoin startswithany)rKsuffixesZexclude_filenamesr rQr listdir_filessr\cCs|durtjdd}tj|rBtj|rBtjtj|S|sLtj}tj dkrltjdd d}ndg}| tj D]T}|D]J}tj |||}tj |rtj |stjtj|Sqq~dS)zwhich(command, [paths]) - Look up the given command in the paths string (or the PATH environment variable, if unspecified).NPATHr;PATHEXT)r1environgetr?isabsisfilenormcasenormpathdefpathpathsepsplitrXrJrWr@)commandpathspathextr?extpr r r whichs  rncCs*|D] }tjtj||sdSqdS)NFT)r1r?rJrX)dirtoolsZtoolr r r checkToolsPathsrqcCs(|tjD]}t||r |Sq dSr)rhr1rgrq)rprjr?r r r whichToolss  rrItemscCsL|jdddtdd|D}ttt|d}tdD]@}|d|}tt||}|dkrrqqB|dkrB|d8}qBd dt|D}|D]0\}} t t|| ||d} ||  |qd } d | d } t d |t | t |ddD]\}} t d| |fqt d|t | ttt|d}tdd|}|rf||d7}tttt |d}t dd|dddd| d|ddft | t tt|D]n\}}tt |t |}t| |}t d|||||||d|d|d| ||t ||t |f qt | dS)NcSs|dS)Nr-r )itemr r r rTz printHistogram..)keycSsg|] \}}|qSr r )rP_vr r r rTz"printHistogram.. )g@r-r-cSsg|] }tqSr )rU)rPir r r ry rT(-"z Slowest %s:iz %.2fs: %sz %s Times:rz[%s] :: [%s] :: [%s]Ranger|Z PercentageZCountz&[%*.*fs,%*.*fs) :: [%s%s] :: [%*d/%*d]* )sortmaxintmathceillog itertoolscycleranger6addprintreversedr0centerlist enumeratefloat)itemstitleZmaxValuepowerincZbarHNZhistonamerxbinZbarWhrr ZpDigitsZpfDigitsZcDigitsr}rowZpctwr r r printHistogramsP         rc@seZdZddZdS)ExecuteCommandTimeoutExceptioncCsTt|tsJt|tsJt|ts*Jt|ts8J||_||_||_||_dSr)rrrmsgouterrexitCode)selfrrrrr r r __init__-sz'ExecuteCommandTimeoutException.__init__N)__name__ __module__ __qualname__rr r r r r,srr.Fc s|durt|}|rtjntj}tj||tjtj||tdd}dgzV|dkrrfdd}t||}|j |d\} }  } W|dkr| n|dkr| 0t | } |rdnt | } drt d || | | d | tj krt| | | fS) aExecute command ``command`` (list of arguments or string) with. * working directory ``cwd`` (str), use None to use the current working directory * environment ``env`` (dict), use None for none * Input to the command ``input`` (str), use string to pass no input. * Max execution time ``timeout`` (int) seconds. Use 0 for no timeout. * ``redirect_stderr`` (bool), use True if redirect stderr to stdout Returns a tuple (out, err, exitCode) where * ``out`` (str) is the standard output of running the command * ``err`` (str) is the standard error of running the command * ``exitCode`` (int) is the exitCode of running the command If the timeout is hit an ``ExecuteCommandTimeoutException`` is raised. N)cwdstdinstdoutstderrenv close_fdsFrcsdd<tjdS)NTr)killProcessAndChildrenpidr Z hitTimeOutrmr r killProcessbsz#executeCommand..killProcess)inputrzReached timeout of {} seconds)rrrr)r% subprocessSTDOUTPIPEPopen kUseCloseFDs threadingTimerstart communicatewaitcancelr+rrsignalSIGINTKeyboardInterrupt) rirrrtimeoutredirect_stderrZerr_outZ timerObjectrrrrr rr executeCommand=sD     rcCsd|vpd|vS)zeWhether the given target triple is for macOS, e.g. x86_64-apple-darwin, arm64-apple-macos darwinZmacosr ) target_tripler r r isMacOSTriplesrcCst|jrz8tjgdtjtjd}|\}}|}|}WntyXd}Yn0|dkr|r| }| d|||j d<dS)N)xcrunz--show-sdk-path--sdkmacosxrrrzusing SDKROOT: %rZSDKROOT) rrrrrrstriprrDr&Znote environment)config lit_configcmdrrresZsdk_pathr r r usePlatformSdkOnDarwins       rcCsrt|jrnz8tjgdtjtjd}|\}}|}|}WntyXd}Yn0|dkrn|rn| SdS)N)rz--show-sdk-versionrrrrr) rrrrrrrrrDr&)rrrrrrr r r findPlatformSdkVersionOnMacOSs       rcCs8tdkrdSzddl}WdSty2YdS0dS)az Returns a tuple ( , ) where `` is True if `killProcessAndChildren()` is supported on the current host, returns False otherwise. `` is an empty string if `` is True, otherwise is contains a string describing why the function is not supported. AIX)TrrN)FzRequires the Python psutil module but it could not be found. Try installing it via pip or via your operating system's package manager.)r4r5psutil ImportError)rr r r !killProcessAndChildrenIsSupporteds  rc Cstdkr"tjd|ddnddl}zr||}z|jdd}Wntyd|j dd}Yn0|D]&}z | Wqj|j yYqj0qj| Wn|j yYn0dS)aPThis function kills a process with ``pid`` and all its running children (recursively). It is currently implemented using the psutil module on some platforms which provides a simple platform neutral implementation. TODO: Reimplement this without using psutil on all platforms so we can remove our dependency on it. rzkill -kill $(ps -o pid= -L{})T)shellrN) recursive) r4r5rcallrrProcesschildrenr(Z get_childrenkillZ NoSuchProcess)rrZ psutilProcZchildren_iteratorchildr r r rs"     r)NN)N)rs)NNNrF)$ __future__rrGrrrr1r4rrrErr rrr%r+r,r8rCrLr\rnrqrrr Exceptionrr5rrrrrrrr r r r sD ' )  . G