pyapacheatlas_purview_service_principal_auth_and_entity_search.py

1import os
2from pyapacheatlas.auth import ServicePrincipalAuthentication
3from pyapacheatlas.core import AtlasClient
4
5if __name__ == "__main__":
6    # Authenticate using Service Principal
7    # Ensure these environment variables are set
8    auth = ServicePrincipalAuthentication(
9        tenant_id=os.environ.get("AZURE_TENANT_ID"),
10        client_id=os.environ.get("AZURE_CLIENT_ID"),
11        client_secret=os.environ.get("AZURE_CLIENT_SECRET")
12    )
13
14    # Initialize the Atlas Client
15    # For Azure Purview: https://<your-account-name>.purview.azure.com
16    # For Apache Atlas: http://localhost:21000/api/atlas/v2
17    client = AtlasClient(
18        endpoint_url=os.environ.get("ATLAS_ENDPOINT_URL"),
19        authentication=auth
20    )
21
22    # Perform a basic search
23    search_results = client.discovery.search_entities(query="*")
24
25    # Iterate and print the found entities
26    for entity in search_results:
27        print(f"Name: {entity.get('name')}, Type: {entity.get('typeName')}, Guid: {entity.get('guid')}")