import psycopg

with psycopg.connect("dbname=pgreplay_db user=pgreplay_user password=pgreplay_pass", autocommit=True) as conn:
    with conn.cursor() as cur:
        with conn.transaction():
            cur.execute("CREATE TABLE IF NOT EXISTS foo (a int);")
            for _ in range(6):
                cur.execute(f"INSERT INTO foo VALUES (1);")
            raise psycopg.Rollback