pvw-cli 1.0.8__py3-none-any.whl → 1.0.10__py3-none-any.whl

purviewcli/client/_workflow.py

@@ -9,7 +9,7 @@ from .endpoints import ENDPOINTS, format_endpoint, get_api_version_params
 class Workflow(Endpoint):
     def __init__(self):
         Endpoint.__init__(self)
-        self.app = 'workflow'
+        self.app = 'datagovernance'  # Use datagovernance for workflow endpoints
 
     # ========== Workflow Management ==========
     
@@ -17,8 +17,8 @@ class Workflow(Endpoint):
     def workflowListWorkflows(self, args):
         """List all workflows"""
         self.method = 'GET'
-        self.endpoint = ENDPOINTS['workflow']['list_workflows']
-        self.params = get_api_version_params('workflow')
+        self.endpoint = '/datagovernance/dataaccess/workflows'
+        self.params = {}
 
     @decorator
     def workflowCreateWorkflow(self, args):

purviewcli/client/endpoint.py

@@ -37,6 +37,21 @@ def get_data(http_dict):
         client = SyncPurviewClient(config)
 
         # Make the request
+        # If debug enabled via PURVIEWCLI_DEBUG env var, print helpful diagnostics
+        debug = os.getenv("PURVIEWCLI_DEBUG")
+        if debug:
+            try:
+                base_info = {
+                    "app": http_dict.get("app"),
+                    "method": http_dict.get("method", "GET"),
+                    "endpoint": http_dict.get("endpoint", "/"),
+                    "params": http_dict.get("params"),
+                    "payload": http_dict.get("payload"),
+                }
+                print("[PURVIEWCLI DEBUG] Request:", json.dumps(base_info, default=str, indent=2))
+            except Exception:
+                print("[PURVIEWCLI DEBUG] Request: (could not serialize request info)")
+
         result = client.make_request(
             method=http_dict.get("method", "GET"),
             endpoint=http_dict.get("endpoint", "/"),
@@ -44,6 +59,12 @@ def get_data(http_dict):
             json=http_dict.get("payload"),
         )
 
+        if debug:
+            try:
+                print("[PURVIEWCLI DEBUG] Response:", json.dumps(result, default=str, indent=2))
+            except Exception:
+                print("[PURVIEWCLI DEBUG] Response: (could not serialize response)")
+
         # The synchronous client returns a wrapper dict like
         # {"status": "success", "data": <json>, "status_code": 200}
         # Normalize to return the raw JSON payload when available so

purviewcli/client/sync_client.py

@@ -60,21 +60,21 @@ class SyncPurviewClient:
                     "-o", "tsv"
                 ], capture_output=True, text=True, check=True)
                 atlas_url = result.stdout.strip()
-                # Extract account ID from URL like: https://c869cf92-11d8-4fbc-a7cf-6114d160dd71-api.purview-service.microsoft.com/catalog
+                
                 if atlas_url and "-api.purview-service.microsoft.com" in atlas_url:
                     account_id = atlas_url.split("://")[1].split("-api.purview-service.microsoft.com")[0]
                 else:
                     raise Exception(f"Could not extract account ID from Atlas URL: {atlas_url}")
             except Exception as e:
-                # Try to get tenant ID as fallback since it often matches the account ID
+                # For Unified Catalog, the account ID is typically the Azure Tenant ID
                 try:
                     tenant_result = subprocess.run([
                         "az", "account", "show", "--query", "tenantId", "-o", "tsv"
                     ], capture_output=True, text=True, check=True)
                     account_id = tenant_result.stdout.strip()
-                    print(f"Warning: Using tenant ID as account ID fallback: {account_id}")
+                    print(f"Info: Using Tenant ID as Purview Account ID for Unified Catalog: {account_id}")
                 except Exception:
-                    raise Exception(f"Could not determine Purview account ID. Please set PURVIEW_ACCOUNT_ID environment variable. Error: {e}")
+                    raise Exception(f"Could not determine Purview account ID. For Unified Catalog, this is typically your Azure Tenant ID. Please set PURVIEW_ACCOUNT_ID environment variable. Error: {e}")
         return account_id
 
     def _get_authentication_token(self, for_unified_catalog=False):
@@ -113,8 +113,13 @@ class SyncPurviewClient:
     def make_request(self, method: str, endpoint: str, **kwargs) -> Dict:
         """Make actual HTTP request to Microsoft Purview"""
         try:
-            # Determine if this is a Unified Catalog request
-            is_unified_catalog = endpoint.startswith('/datagovernance/catalog')
+            # Determine if this is a Unified Catalog / Data Map (Atlas) request
+            # Several endpoints use '/catalog' or '/datamap' prefixes (Atlas/DataMap APIs)
+            is_unified_catalog = (
+                endpoint.startswith('/datagovernance/catalog')
+                or endpoint.startswith('/catalog')
+                or endpoint.startswith('/datamap')
+            )
             
             # Get the appropriate authentication token and base URL
             if is_unified_catalog:
@@ -146,7 +151,7 @@ class SyncPurviewClient:
                 timeout=30,
             )
             # Handle the response
-            if response.status_code == 200:
+            if response.status_code in [200, 201]:
                 try:
                     data = response.json()
                     return {"status": "success", "data": data, "status_code": response.status_code}
@@ -172,7 +177,7 @@ class SyncPurviewClient:
                     timeout=30,
                 )
 
-                if response.status_code == 200:
+                if response.status_code in [200, 201]:
                     try:
                         data = response.json()
                         return {