o Uj @sdZddlZddlZddlmZdeeeeefeeeefBeeefBdBdeedBfddZ dd eed edBddfd d Z deeefeeeeefBdeej efeej eeefBfd dZ d eedeeefeeeeefBddfddZdS)z'Utility functions for aiohappyeyeballs.N) AddrInfoTypeaddrreturncCs|durdS|d}|d}d|v}|r6d}d}t|}|dkr$|d}|dkr,|d}||||f}tj}n||f}tj}|tjtjd|fgS) z7Convert an address tuple to a list of addr_info tuples.Nrr:)lensocketAF_INET6AF_INET SOCK_STREAM IPPROTO_TCP)rhostportis_ipv6flowinfoscopeidaddr_lenfamilyrO/opt/telega-widget/.venv/lib/python3.10/site-packages/aiohappyeyeballs/utils.pyaddr_to_addr_infos s$ r addr_infos interleavecCs`|durd}i}g}|D]}|d}||d}||kr!|||d||<q ||dd<dS)z Pop addr_info from the list of addr_infos by family up to interleave times. The interleave parameter is used to know how many addr_infos for each family should be popped of the top of the list. Nrr)getappend)rrseenkept addr_inforcountrrrpop_addr_infos_interleave"s   r#cCst|dg|ddRS)z+Convert an address tuple to an IPv4Address.rrN) ipaddress ip_addressrrrr_addr_tuple_to_ip_address8sr'cslfdd|D}t|t|krtfdd|D}t|t|kr.tdd||dd<dS)z Remove an address from the list of addr_infos. The addr value is typically the return value of sock.getpeername(). csg|] }|dkr|qSr.0air&rr Isz%remove_addr_infos..cs g|] }t|dkr|qSr()r'r*) match_addrrrr-MszAddress z not found in addr_infosN)r r' ValueError)rrr r)rr.rremove_addr_infos?s  r0)N)__doc__r$r typesrtuplestrintlistrr# IPv4Address IPv6Addressr'r0rrrrs: ,