B ]t\@slddlZddlZddlZejZejZejjZejj Z da da ddZ d ddZd ddZejd d d ZdS)NFcsfdd}|S)z Wraps ``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|dtjrF|djtjtjfkr0||S|ddd}d}n@sz;check_internet_off..new_function..z[An attempt was made to connect to the internet by a test that was not marked `remote_data`.) isinstancesocketZfamilyZAF_INETZAF_INET6tuplelenZ gethostnameZgetfqdnanyIOError)argskwargsZaddr_argZ valid_hostsZhostnameZfqdnZnew_addr)original_function)rr new_function#s(     " z(check_internet_off..new_functionr)rrr)rr check_internet_offs "rcCsptrdSdad}|rtdtjatji}tj|}tj|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]. NTzInternet access disabled) INTERNET_OFFprinturllibrequestZ build_opener _orig_openerZ ProxyHandlerinstall_openerrsocket_create_connectionr create_connection socket_bindbindsocket_connectconnect)verboseZ__tracebackhide__Zno_proxy_handleropenerrrr turn_off_internetHs        r$cCs>tsdSda|rtdtjttt_t tj_ t tj_ tS)zL Restore internet access. Not used, but kept in case it is needed. NFzInternet access enabled) rrrrrrrr rrrr r!)r"rrr turn_on_internetks r%c cs.t}t|dz dVWd|s(t|dXdS)a7Context manager to temporarily disable 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`. )r"N)rr$r%)r"Zalready_disabledrrr no_internets   r&)F)F)F) contextlibr Zurllib.requestrZsocket_originalrrrrr!r rrrr$r%contextmanagerr&rrrr s, #