monkey patch socket module

monkey.py

import os

proxy = os.getenv('socks5_proxy')
if proxy:
    host, port = proxy.split(':')
    import socks # pip install pysocks
    import socket

    def create_connection(address, timeout=None, source_address=None):
        sock = socks.socksocket()
        sock.connect(address)
        return sock

    socks.setdefaultproxy(socks.PROXY_TYPE_SOCKS5, host, int(port, 10))

    socket.socket = socks.socksocket
    socket.create_connection = create_connection