Simple proxy tor config with PySocks lib

proxy.py

# -*- coding: utf-8 -*-

import socket, socks


def create_connection(address):
    """
    Create connection with custom sock
    
    :param address: address for connection
    
    :return: connection sock
    :rtype: socks.socksocket
    """

    sock = socks.socksocket()
    sock.connect(address)
    return sock


def config(ip, port):
    """
    Config proxy
    
    :param ip: proxy ip
    :param port: proxy port
    
    :return: None
    :rtype: None
    """

    socks.setdefaultproxy(socks.PROXY_TYPE_SOCKS5, ip, port)


def patch():
    """
    Patch the socket module
    
    :return: None
    :rtype: None
    """

    config("127.0.0.1", 9050)

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