o Zh @sddlZddlZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z m Z mZmZdgZdedeeeffdd Zd edeefd d Zd eededeeeffddZdedefddZdeeeeeffddfddZGdddZdS)N) timedelta)EventThread)AnyCallableOptionalUnionparse_rendezvous_endpoint config_strreturncCsi}|}|s |S|d}|D]-}|dd^}}|}|s%td|r.|d}nd}|s:td|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_valueskvkeyvaluesvaluerY/var/www/auris/lib/python3.10/site-packages/torch/distributed/elastic/rendezvous/utils.py_parse_rendezvous_configs*   rport_strcCs|r td|r t|SdS)z1Try to extract the port number from ``port_str``.z ^[0-9]{1,5}$N)rematchint)rrrr_try_parse_port<srendpoint default_portcCs|dur|}|sd|fS|ddkr#|ddkr#|ggR^}}n|dd^}}t|dkrC|ddkrC|ddkrC|dd}t|dkr`t|d}|dusW|d kr_td |d n|}td |sptd |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)c Cs|dkrdSzt|}Wn tyd}Ynw|r |jr dSz tj|dtjtjd}WnttjfyD}zg}WYd}~nd}~wwdd|D}t }||krVdStj|dtjtjd}|D])}|drr|d|krrdS|r|dd t |krdS|dd |vrdSqcd 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 sz-_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_list addr_inforrr_matches_machine_hostnametsB     r?secondscCs.t|tr tj|}|dkrt|dSdS)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<de d e d d e d e d df ddZed eefddZded dfddZdddZdddZedddZeddZdS)_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_ctxrIrKrJrLrMr NcOsVd|_||_||j_||j_|pd|j_|pi|j_t|j_ d|_ d|_ dS)Nr) rVrUrY total_secondsrIrKrLrMrrNrWrX)selfrIrKrLrMrrr__init__s      z_PeriodicTimer.__init__cCs|jS)zGet the name of the timer.)rVr[rrrnamesz_PeriodicTimer.namer^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_names z_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 |dSdS)z'Stop the timer at the next opportunity.N)rXr]rrrcancels z_PeriodicTimer.cancelcCs8|j|js|j|ji|j|j|jrdSdSN)rNwaitrIrKrLrM)ctxrrrrdsz_PeriodicTimer._runcCs||dSrk)setjoin)threadrNrrrrgs z_PeriodicTimer._stop_thread)r N)rOrPrQ__doc__rUrr;rSrrerfrrrr\propertyr^rarirj staticmethodrdrgrrrrrHs6      rH)r2rCrr5rEredatetimer threadingrrtypingrrrr__all__r;rTrrrrBr boolr?rRrGrHrrrrs, &  1"4