Tip
If you already have a Citus instance running, skip the steps to set up and clean up Citus.
-
Start a Citus instance:
docker run -d --rm --name citus_standalone -p 5432:5432 -e POSTGRES_PASSWORD=password citusdata/citus
-
Install the PostgreSQL ADBC driver:
dbc install postgresql
-
Customize the Python script
main.pyas needed- Change the connection arguments in
db_kwargs- Format
uriaccording to the connection URI format used by PostgreSQL, or keep it as is
- Format
- If you changed which database you're connecting to, also change the SQL SELECT statement in
cursor.execute()
- Change the connection arguments in
-
Run the Python script:
uv run main.py
Stop the Docker container running Citus:
docker stop citus_standalone