inishchith · January 13, 2026 19:44
diff --git a/set_oauth_cookie_all_connections.py b/set_oauth_cookie_all_connections.py
 """
 Script to set connectionInsightsViaOAuthCookie to True for specific connection GUIDs.

 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

 # Add your connection GUIDs here
 CONNECTION_GUIDS = [
    "guid-1-here",
    "guid-2-here",
    # Add more GUIDs as needed
 ]


 def main():
    client = AtlanClient()

    print(f"Processing {len(CONNECTION_GUIDS)} connection(s)...")

    updated_count = 0
    failed_count = 0
    not_found_count = 0

    for guid in CONNECTION_GUIDS:
        try:
            print(f"  Fetching connection {guid}...")

            connection = client.asset.get_by_guid(
                guid=guid,
                asset_type=Connection,
            )

            if connection is None:
                print(f"    Not found")
                not_found_count += 1
                continue

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

            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"  Not found: {not_found_count}")
    print(f"  Failed: {failed_count}")


 if __name__ == "__main__":
    main()
	"""
	Script to set connectionInsightsViaOAuthCookie to True for specific connection GUIDs.

	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

	# Add your connection GUIDs here
	CONNECTION_GUIDS = [
	"guid-1-here",
	"guid-2-here",
	# Add more GUIDs as needed
	]


	def main():
	client = AtlanClient()

	print(f"Processing {len(CONNECTION_GUIDS)} connection(s)...")

	updated_count = 0
	failed_count = 0
	not_found_count = 0

	for guid in CONNECTION_GUIDS:
	try:
	print(f" Fetching connection {guid}...")

	connection = client.asset.get_by_guid(
	guid=guid,
	asset_type=Connection,
	)

	if connection is None:
	print(f" Not found")
	not_found_count += 1
	continue

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

	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" Not found: {not_found_count}")
	print(f" Failed: {failed_count}")


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