Using SQLAlchemy to connect to Microsoft Sql Server database using SOCKS 5 proxy with Python-TDS

pytds_sock5_with_sql_alchemy.py

import socks
import pytds
from sqlalchemy import create_engine
from sqlalchemy.pool import StaticPool

host = ...
port = 1433
user = ...
password = ...
socks_proxy_host = ...
socks_proxy_port = ...
socks_proxy_user = ...
socks_proxy_password = ...


def create_connection_socks5(host, port, socks_proxy_host, socks_proxy_port,
                                             socks_proxy_user, socks_proxy_password):
    s = socks.socksocket() # Same API as socket.socket in the standard lib

    s.set_proxy(
        socks.SOCKS5,
        socks_proxy_host,
        username=socks_proxy_user,
        password=socks_proxy_password,
        port=socks_proxy_port)

    # Can be treated identical to a regular socket object
    s.connect((host, port))

    return s


def create_pytds_connection_with_socks_proxy(host, port, user, password, socks_proxy_host, socks_proxy_port,
                                             socks_proxy_user, socks_proxy_password, database=""):
    return pytds.connect(host, database, user, password, port=port, sock=create_connection_socks5(
        host=host,
        port=port,
        socks_proxy_host=socks_proxy_host,
        socks_proxy_port=socks_proxy_port,
        socks_proxy_user=socks_proxy_user,
        socks_proxy_password=socks_proxy_password,
    ))


eng = create_engine(
    "mssql+pytds://{user}:{password}@{host}:{port}".format(
        host=host,
        port=port,
        user=user,
        password=password,
    ),
    # poolclass=StaticPool,
    creator=lambda: create_pytds_connection_with_socks_proxy(
        host=host,
        port=port,
        user=user,
        password=password,
        socks_proxy_host=socks_proxy_host,
        socks_proxy_port=socks_proxy_port,
        socks_proxy_user=socks_proxy_user,
        socks_proxy_password=socks_proxy_password,
    ))

print(eng.execute("select 1")).fetchone()