o i/@sddlZddlZddlmZejdedddejZejZejj Z ejj Z da dadd Zdd d Zdd d ZejdddZdS)N)cbookz3.2modulezpytest-remotedata)nameobj_type alternativeFcsfdd}|S)z Wrap ``original_function``, which in most cases is assumed to be a `socket.socket` method, to raise an `IOError` for any operations on non-local AF_INET sockets. cst|dtjr%|djtjtjfvr|i|S|ddd}d}n t|dtr4t|ddks;|i|S|ddd}d}t}t}||fvrnd||df}|d||f||dd}t fdd|Dr|i|St d ) Nr) localhost 127.0.0.1z::1)rr rc3s|]}|vVqdS)N).0hhostr ^/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/matplotlib/testing/disable_internet.py Csz;check_internet_off..new_function..z[An attempt was made to connect to the internet by a test that was not marked `remote_data`.) isinstancesocketfamilyAF_INETAF_INET6tuplelen gethostnamegetfqdnanyIOError)argskwargsaddr_arg valid_hostshostnamefqdnnew_addroriginal_functionrr new_function&s(   "z(check_internet_off..new_functionr )r%r&r r$rcheck_internet_offs "r'cCsxddl}trdSdad}|rtd|ja|ji}|j|}|j|tt t _ tt t j _ ttt j _t S)a1 Disable internet access via python by preventing connections from being created using the socket module. Presumably this could be worked around by using some other means of accessing the internet, but all default python modules (urllib, requests, etc.) use socket [citation needed]. rNTzInternet access disabled)urllib.request INTERNET_OFFprintrequest build_opener _orig_opener ProxyHandlerinstall_openerr'socket_create_connectionrcreate_connection socket_bindbindsocket_connectconnect)verboseurllib__tracebackhide__no_proxy_handleropenerr r rturn_off_internetKs       r;cCsFddl}tsdSda|rtd|jttt_t tj_ t tj_ tS)zL Restore internet access. Not used, but kept in case it is needed. rNFzInternet access enabled) r(r)r*r+r/r-r0rr1r2r3r4r5)r6r7r r rturn_on_internetos r<c csBt}t|dzdVW|st|ddSdS|s t|dww)a* Temporarily disables internet access (if not already disabled). If it was already disabled before entering the context manager (i.e. `turn_off_internet` was called previously) then this is a no-op and leaves internet access disabled until a manual call to `turn_on_internet`. )r6N)r)r;r<)r6already_disabledr r r no_internets  r>)F) contextlibr matplotlibrwarn_deprecated__name__socket_originalr1r0r3r2r5r4r)r-r'r;r<contextmanagerr>r r r rs"   , $