a h @sddlZddlZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z m Z mZmZdgZeeeefdddZeeed d d Zeeeeeefd d dZeedddZeeeeeffddddZGdddZdS)N) timedelta)EventThread)AnyCallableOptionalUnionparse_rendezvous_endpoint) config_strreturncCsi}|}|s|S|d}|D]Z}|dd^}}|}|sJtd|r\|d}nd}|sttd|d|||<q"|S) zExtract key-value pairs from a rendezvous configuration string. Args: config_str: A string in format =,...,=. ,=zZThe rendezvous configuration string must be in format =,...,=.rNz%The rendezvous configuration option 'z' must have a value specified.)stripsplit ValueError)r config key_valueskvkeyvaluesvaluerX/var/www/auris/lib/python3.9/site-packages/torch/distributed/elastic/rendezvous/utils.py_parse_rendezvous_configs*   r)port_strr cCs|rtd|rt|SdS)z1Try to extract the port number from ``port_str``.z ^[0-9]{1,5}$N)rematchint)rrrr_try_parse_port<sr)endpoint default_portr cCs|dur|}|sd|fS|ddkrF|ddkrF|ggR^}}n|dd^}}t|dkr|ddkr|ddkr|dd}t|dkrt|d}|dus|d krtd |d n|}td |std |d||fS)a4Extract the hostname and the port number from a rendezvous endpoint. Args: endpoint: A string in format [:]. default_port: The port number to use if the endpoint does not include one. Returns: A tuple of hostname and port number. N localhostr[]:riz,The port number of the rendezvous endpoint 'z)' must be an integer between 0 and 65536.z ^[\w\.:-]+$z)The hostname of the rendezvous endpoint 'zN' must be a dot-separated list of labels, an IPv4 address, or an IPv6 address.)rrsplitlenrrrr)r r!hostrestportrrrr Cs*$      )r)r c Cs"|dkr dSzt|}Wnty0d}Yn0|r@|jr@dSztj|dtjtjd}Wn.ttjfy}zg}WYd}~n d}~00dd|D}t }||krdStj|dtjtjd}|D]V}|dr|d|krdS|r|dd t |krdS|dd |vrdSqd S) a<Indicate whether ``host`` matches the hostname of this machine. This function compares ``host`` to the hostname as well as to the IP addresses of this machine. Note that it may return a false negative if this machine has CNAME records beyond its FQDN or IP addresses assigned to secondary NICs. r"TN)protoflagscSsg|]}|ddqS)rr).0Zhost_addr_inforrr z-_matches_machine_hostname..r.rF) ipaddress ip_addressr is_loopbacksocket getaddrinfo IPPROTO_TCP AI_CANONNAMEgaierror gethostnamestr)r)addrZhost_addr_list_Z host_ip_listZ this_hostZ addr_listZ addr_inforrr_matches_machine_hostnamets:     r?)secondsr cCs*t|trtj|}|dkr&t|dS)zSuspend the current thread for ``seconds``. Args: seconds: Either the delay, in seconds, or a tuple of a lower and an upper bound within which a random delay will be picked. g{Gz?N) isinstancetuplerandomuniformtimesleep)r@rrr_delays  rGc@seZdZUdZGdddZeeed<eeed<ee j ed<eed<e e de e d d d d Zeeed ddZed dddZd d ddZd d ddZed d ddZeddZd S)_PeriodicTimerzRepresent a timer that periodically runs a specified function. Args: interval: The interval, in seconds, between each run. function: The function to run. c@sJeZdZUeed<eded<eedfed<ee efed<e ed<dS) z_PeriodicTimer._Contextinterval.Nfunction.argskwargs stop_eventN) __name__ __module__ __qualname__float__annotations__rrBrdictr<rrrrr_Contexts  rU_name_thread _finalizer_ctxrJN)rIrKrLrMr cOsVd|_||_||j_||j_|p*d|j_|p6i|j_t|j_ d|_ d|_ dS)Nr) rVrUrY total_secondsrIrKrLrMrrNrWrX)selfrIrKrLrMrrr__init__s     z_PeriodicTimer.__init__)r cCs|jS)zGet the name of the timer.)rVr[rrrnamesz_PeriodicTimer.name)r^r cCs|jrtd||_dS)zSet the name of the timer. The specified name will be assigned to the background thread and serves for debugging and troubleshooting purposes. The timer has already started.N)rW RuntimeErrorrV)r[r^rrrset_namesz_PeriodicTimer.set_namecCs\|jrtdt|j|jpd|jfdd|_t||j|j|jj |_ d|j _ |j dS)zStart the timer.r_Z PeriodicTimerT)targetr^rLdaemonFN) rWr`r_runrVrYweakreffinalize _stop_threadrNrXatexitstartr]rrrris z_PeriodicTimer.startcCs|jr|dS)z'Stop the timer at the next opportunity.N)rXr]rrrcancelsz_PeriodicTimer.cancelcCs(|j|js$|j|ji|jqdSN)rNwaitrIrKrLrM)ctxrrrrdsz_PeriodicTimer._runcCs||dSrk)setjoin)threadrNrrrrgsz_PeriodicTimer._stop_thread)rOrPrQ__doc__rUrr<rSrrerfrrrr\propertyr^rarirj staticmethodrdrgrrrrrHs*      rH)r3rCrr6rEredatetimer threadingrrtypingrrrr__all__r<rTrrrrBr boolr?rRrGrHrrrrs" & 14