CockroachDB uses the PostgreSQL protocol so you can reuse any Postgres drivers and tooling to connect.

They have an entire section in their documentation for Python and SQLAlchemy: https://www.cockroachlabs.com/docs/stable/build-a-python-app...

sqlalchemy has a cockroach dialect.