Access list of active server connections to CrateDB when using SQLAlchemy

cratedb-sqlalchemy-active-servers.py

# Connect using SQLAlchemy Core.
import sqlalchemy as sa
from pprint import pp

dburi = "crate://localhost:4200"
query = "SELECT country, mountain, coordinates, height FROM sys.summits ORDER BY country;"


def print_active_servers(connection: sa.engine.Connection):
    print("Active Servers:", connection.connection.dbapi_connection.client.active_servers)


if __name__ == "__main__":
    engine = sa.create_engine(dburi, echo=True)
    with engine.connect() as connection:
        print_active_servers(connection)
        with connection.execute(sa.text(query)) as result:
            pp(result.mappings().fetchall())
        print_active_servers(connection)