U
    luf                      @  s  U d Z ddlmZ ddlZddlZddlZddlZddlmZmZ ddl	m
Z
mZmZmZ ddlmZ g Zded< g Zded	< d
aded< dddddZdddddZdddddZdddddZG dd deZdddd d!Zed"ejZdd#d$d%Zdd#d&d'Zed(ejZ dd#d)d*Z!dd#d+d,Z"dd#d-d.Z#dd#d/d0Z$edBd2dd3d4d5Z%ed6d#d7d8Z&ed6d#d9d:Z'edd#d;d<Z(edd2d=d>d?Z)edd2d=d@dAZ*dS )Cz-Utilities for identifying local IP addresses.    )annotationsN)PIPEPopen)AnyCallableIterableSequence)warnlist	LOCAL_IPS
PUBLIC_IPS str	LOCALHOSTr   )elemsreturnc                 C  s4   t  }g }| D ] }||kr|| || q|S )zuniq_stable(elems) -> list

    Return from an iterable, a list of all the unique elements in the input,
    maintaining the order in which they first appear.
    )setappendadd)r   seenvaluex r   B/tmp/pip-unpacked-wheel-naub1w99/jupyter_client/localinterfaces.py_uniq_stable   s    
r   zstr | Sequence[str])cmdr   c                 C  sp   d}t jdkr&t }| jtjO  _t| tt|d}| \}}|j	rdd
| |dd}t||ddS )z4Get output of a command, raising IOError if it failsNnt)stdoutstderrstartupinfozFailed to run {}: {}utf8replace)osname
subprocessZSTARTUPINFOZdwFlagsZSTARTF_USESHOWWINDOWr   r   communicate
returncodeformatdecodeOSError)r   r   pr   r   msgr   r   r   _get_output#   s    
r,   r   )fr   c                   s   d _ ddd fdd}|S )z%decorator to only run a function onceFr   )kwargsr   c                    s    j r
d S  f | }d _ |S )NTcalled)r.   retr-   r   r   wrapped5   s
    
z_only_once.<locals>.wrappedr/   )r-   r3   r   r2   r   
_only_once1   s    r4   c                   s   dddd fdd}|S )z2decorator to ensure load_ips has been run before fr   )argsr.   r   c                    s   t    | |S )N)	_load_ips)r5   r.   r2   r   r   
ips_loadedB   s    z!_requires_ips.<locals>.ips_loadedr   )r-   r7   r   r2   r   _requires_ips?   s    r8   c                   @  s   e Zd ZdS )NoIPAddressesN)__name__
__module____qualname__r   r   r   r   r9   J   s   r9   zSequence[str] | NoneNone)addrsr   c                 C  s   | st g }g }| D ],}|| |ds8|| qts|aqtrNtdkr^da|dt |ddg t|tdd< t|tdd< dS )z7populate local and public IPs from flat list of all IPs127.	127.0.0.1r   0.0.0.0r   N)	r9   r   
startswithr   insertextendr   r   r   )r>   
public_ips	local_ipsipr   r   r   _populate_from_listN   s     

rH   zinet\b.*?(\d+\.\d+\.\d+\.\d+))r   c                  C  sn   zt d} W n tk
r(   t d} Y nX |  }g }|D ]&}t| }|r:||d q:t| dS )z0load ip addresses from `ifconfig` output (posix)Zifconfigz/sbin/ifconfig   N)	r,   r)   
splitlines_ifconfig_ipv4_patmatchstripr   grouprH   outlinesr>   linemr   r   r   _load_ips_ifconfigk   s    rT   c                  C  sn   t ddddg} |  }g }|D ]@}|  }t|dkr |d dkr ||d dd  q t| d	S )
z/load ip addresses from `ip addr` output (Linux)rG   z-fZinetaddr   r   rI   /N)r,   rJ   lowersplitlenr   rH   )rP   rQ   r>   rR   blocksr   r   r   _load_ips_ip}   s    r\   zipv4.*?(\d+\.\d+\.\d+\.\d+)$c                  C  sL   t d} |  }g }|D ]&}t| }|r||d qt| dS )z2load ip addresses from `ipconfig` output (Windows)ZipconfigrI   N)r,   rJ   _ipconfig_ipv4_patrL   rM   r   rN   rH   rO   r   r   r   _load_ips_ipconfig   s    r^   c                  C  s   ddl } g }g }|  D ]d}| || jg }|D ]F}|d}|sHq4|dsh|dsh|| ntsp|a|| q4qtsda|dt |	ddg t
|tdd< t
|tdd< dS )	z load ip addresses with netifacesr   NrU   lor?   r@   rA   r   )	netifacesZ
interfacesZifaddressesgetAF_INETrB   r   r   rC   rD   r   r   r   )r`   rF   rE   ZifaceZipv4sentryrU   r   r   r   _load_ips_netifaces   s(    
rd   c               
   C  s   zt dd tdd< W n" tk
r<   dgtdd< Y nX zvz\t  } t | d tdd< | dst	dd tD rt t  d d tdd< W n tk
r   Y nX W 5 tttdd< tt X tdd	g tttdd< td
 a
dS )zJload ip addresses with socket.gethostbyname_ex

    This can be slow.
    	localhostrV   Nr@   z.localc                 s  s   | ]}| d V  qdS )Z127N)rB   ).0rG   r   r   r   	<genexpr>   s     z*_load_ips_gethostbyname.<locals>.<genexpr>rA   r   r   )socketgethostbyname_exr   r)   r   r   rD   gethostnameendswithallr   )hostnamer   r   r   _load_ips_gethostbyname   s     "
rn   c                   C  s&   da t ddgtdd< g tdd< dS )z&Fallback in case of unexpected failurer@   rA   r   N)r   r   r   r   r   r   r   _load_ips_dumb   s    ro   Tbool)suppress_exceptionsr   c              
   C  s   zz
t  W W S  tk
r    Y nX tjdkrRz
t W W S  ttfk
rN   Y qX nHz
t W W S  ttfk
rt   Y nX z
t W W S  ttfk
r   Y nX t	 W S  t
k
r } z| s td| dd W 5 d}~X Y nX t  dS )aG  load the IPs that point to this machine

    This function will only ever be called once.

    It will use netifaces to do it quickly if available.
    Then it will fallback on parsing the output of ifconfig / ip addr / ipconfig, as appropriate.
    Finally, it will fallback on socket.gethostbyname_ex, which can be slow.
    r   z9Unexpected error discovering local network interfaces: %srV   )
stacklevelN)rd   ImportErrorr"   r#   r^   r)   r9   r\   rT   rn   	Exceptionr	   ro   )rq   er   r   r   r6      s0    




"r6   z	list[str]c                   C  s   t S )z2return the IP addresses that point to this machiner   r   r   r   r   rF     s    rF   c                   C  s   t S )zKreturn the IP addresses for this machine that are visible to other machinesr   r   r   r   r   rE     s    rE   c                   C  s   t S )z1return ip for localhost (almost always 127.0.0.1))r   r   r   r   r   re     s    re   )rG   r   c                 C  s   | t kS )z does `ip` point to this machine?rv   rG   r   r   r   is_local_ip   s    ry   c                 C  s   | t kS )z#is `ip` a publicly visible address?rw   rx   r   r   r   is_public_ip&  s    rz   )T)+__doc__
__future__r   r"   rerh   r$   r   r   typingr   r   r   r   warningsr	   r   __annotations__r   r   r   r,   r4   r8   rt   r9   rH   compile
IGNORECASErK   rT   r\   r]   r^   rd   rn   ro   r6   rF   rE   re   ry   rz   r   r   r   r   <module>   sJ    -