pyapacheatlas

import os from azure.identity import DefaultAzureCredential from pyapacheatlas.auth import PurviewClient from pyapacheatlas.core.atlas_entities import AtlasEntity # Replace with your Purview account name purview_account_name = os.environ.get("PURVIEW_NAME", "YOUR_PURVIEW_ACCOUNT_NAME") try: # Authenticate using DefaultAzureCredential (ensure 'azure-identity' is installed) credential = DefaultAzureCredential() client = PurviewClient( account_name=purview_account_name, authentication=credential ) # Example: Define a simple Atlas entity (e.g., a table) table_entity = AtlasEntity( name="my_example_table", typeName="azure_sql_table", qualifiedName=f"mssql://{purview_account_name}/server/database/my_example_table", guid="-100" # Use negative GUIDs for temporary entities in bulk uploads ) # Example: Upload the entity to Purview # For real use, ensure 'YOUR_PURVIEW_ACCOUNT_NAME' is replaced or PURVIEW_NAME env var is set # response = client.upload_entities([table_entity]) # print(f"Successfully uploaded entity: {response.json()}") # Example: Search for entities (using the recommended discover method) search_results = client.discover.search_entities(query="my_example_table") print(f"Search results (first 5): {search_results.json().get('value', [])[:5]}") except Exception as e: print(f"An error occurred: {e}") print("Ensure 'PURVIEW_NAME' environment variable is set or replace 'YOUR_PURVIEW_ACCOUNT_NAME'.") print("Also ensure your Azure AD account has permissions to the Purview account.")