inishchith · January 13, 2026 19:44
diff --git a/set_oauth_cookie_by_guids.py b/set_oauth_cookie_by_guids.py
 """
 Script to set connectionInsightsViaOAuthCookie to True for ALL connections.

 Requires environment variables:
  - ATLAN_API_KEY: Your Atlan API key
  - ATLAN_BASE_URL: Your Atlan instance URL (e.g., https://tenant.atlan.com)
 """

 from pyatlan.client.atlan import AtlanClient
 from pyatlan.model.assets import Connection
 from pyatlan.model.fluent_search import FluentSearch


 def main():
    client = AtlanClient()

    print("Searching for all connections...")

    results = (
        FluentSearch()
        .select()
        .where(FluentSearch.asset_type(Connection))
        .include_on_results(Connection.CONNECTION_INSIGHTS_VIA_O_AUTH_COOKIE)
        .include_on_results(Connection.NAME)
        .execute(client=client)
    )

    updated_count = 0
    failed_count = 0
    skipped_count = 0

    for connection in results:
        if connection.connection_insights_via_o_auth_cookie is True:
            print(f"  Skipping '{connection.name}' (already enabled)")
            skipped_count += 1
            continue

        try:
            print(f"  Updating '{connection.name}' ({connection.guid})...")

            to_update = Connection.updater(
                qualified_name=connection.qualified_name,
                name=connection.name
            )
            to_update.connection_insights_via_o_auth_cookie = True

            client.asset.save(to_update)
            updated_count += 1
            print(f"    Success")

        except Exception as e:
            failed_count += 1
            print(f"    Failed: {e}")

    print(f"\nSummary:")
    print(f"  Updated: {updated_count}")
    print(f"  Skipped (already enabled): {skipped_count}")
    print(f"  Failed: {failed_count}")


 if __name__ == "__main__":
    main()
	"""
	Script to set connectionInsightsViaOAuthCookie to True for ALL connections.

	Requires environment variables:
	- ATLAN_API_KEY: Your Atlan API key
	- ATLAN_BASE_URL: Your Atlan instance URL (e.g., https://tenant.atlan.com)
	"""

	from pyatlan.client.atlan import AtlanClient
	from pyatlan.model.assets import Connection
	from pyatlan.model.fluent_search import FluentSearch


	def main():
	client = AtlanClient()

	print("Searching for all connections...")

	results = (
	FluentSearch()
	.select()
	.where(FluentSearch.asset_type(Connection))
	.include_on_results(Connection.CONNECTION_INSIGHTS_VIA_O_AUTH_COOKIE)
	.include_on_results(Connection.NAME)
	.execute(client=client)
	)

	updated_count = 0
	failed_count = 0
	skipped_count = 0

	for connection in results:
	if connection.connection_insights_via_o_auth_cookie is True:
	print(f" Skipping '{connection.name}' (already enabled)")
	skipped_count += 1
	continue

	try:
	print(f" Updating '{connection.name}' ({connection.guid})...")

	to_update = Connection.updater(
	qualified_name=connection.qualified_name,
	name=connection.name
	)
	to_update.connection_insights_via_o_auth_cookie = True

	client.asset.save(to_update)
	updated_count += 1
	print(f" Success")

	except Exception as e:
	failed_count += 1
	print(f" Failed: {e}")

	print(f"\nSummary:")
	print(f" Updated: {updated_count}")
	print(f" Skipped (already enabled): {skipped_count}")
	print(f" Failed: {failed_count}")


	if __name__ == "__main__":
	main()
No results found