pvw-cli 1.0.11__py3-none-any.whl → 1.0.14__py3-none-any.whl

purviewcli/cli/workflow.py

@@ -27,7 +27,7 @@ def list(ctx, output_json):
     try:
         if ctx.obj and ctx.obj.get("mock"):
             console.print("[yellow]🎭 Mock: workflow list command[/yellow]")
-            console.print("[green]✓ Mock workflow list completed successfully[/green]")
+            console.print("[green][OK] Mock workflow list completed successfully[/green]")
             return
 
         from purviewcli.client._workflow import Workflow
@@ -46,7 +46,7 @@ def list(ctx, output_json):
             workflows = []
 
         if not workflows:
-            console.print("[yellow]⚠ No workflows found[/yellow]")
+            console.print("[yellow][!] No workflows found[/yellow]")
             return
 
         # Output in JSON format if requested
@@ -81,7 +81,7 @@ def list(ctx, output_json):
         console.print(f"\n[dim]Total: {len(workflows)} workflow(s)[/dim]")
 
     except Exception as e:
-        console.print(f"[red]✗ Error executing workflow list: {str(e)}[/red]")
+        console.print(f"[red][X] Error executing workflow list: {str(e)}[/red]")
 
 
 @workflow.command()
@@ -100,7 +100,7 @@ def create(ctx, workflow_id, payload_file):
             console.print("[yellow]🎭 Mock: workflow create command[/yellow]")
             console.print(f"[dim]Workflow ID: {workflow_id}[/dim]")
             console.print(f"[dim]Payload File: {payload_file}[/dim]")
-            console.print("[green]✓ Mock workflow create completed successfully[/green]")
+            console.print("[green][OK] Mock workflow create completed successfully[/green]")
             return
 
         from purviewcli.client._workflow import Workflow
@@ -110,12 +110,12 @@ def create(ctx, workflow_id, payload_file):
         result = workflow_client.workflowCreateWorkflow(args)
 
         if result:
-            console.print("[green]✓ Workflow create completed successfully[/green]")
+            console.print("[green][OK] Workflow create completed successfully[/green]")
             console.print(json.dumps(result, indent=2))
         else:
-            console.print("[yellow]⚠ Workflow create completed with no result[/yellow]")
+            console.print("[yellow][!] Workflow create completed with no result[/yellow]")
     except Exception as e:
-        console.print(f"[red]✗ Error executing workflow create: {str(e)}[/red]")
+        console.print(f"[red][X] Error executing workflow create: {str(e)}[/red]")
 
 
 @workflow.command()
@@ -127,7 +127,7 @@ def get(ctx, workflow_id):
         if ctx.obj and ctx.obj.get("mock"):
             console.print("[yellow]🎭 Mock: workflow get command[/yellow]")
             console.print(f"[dim]Workflow ID: {workflow_id}[/dim]")
-            console.print("[green]✓ Mock workflow get completed successfully[/green]")
+            console.print("[green][OK] Mock workflow get completed successfully[/green]")
             return
 
         from purviewcli.client._workflow import Workflow
@@ -137,12 +137,12 @@ def get(ctx, workflow_id):
         result = workflow_client.workflowGetWorkflow(args)
 
         if result:
-            console.print("[green]✓ Workflow get completed successfully[/green]")
+            console.print("[green][OK] Workflow get completed successfully[/green]")
             console.print(json.dumps(result, indent=2))
         else:
-            console.print("[yellow]⚠ Workflow not found[/yellow]")
+            console.print("[yellow][!] Workflow not found[/yellow]")
     except Exception as e:
-        console.print(f"[red]✗ Error executing workflow get: {str(e)}[/red]")
+        console.print(f"[red][X] Error executing workflow get: {str(e)}[/red]")
 
 
 @workflow.command()
@@ -159,7 +159,7 @@ def execute(ctx, workflow_id, payload_file):
             console.print(f"[dim]Workflow ID: {workflow_id}[/dim]")
             if payload_file:
                 console.print(f"[dim]Payload File: {payload_file}[/dim]")
-            console.print("[green]✓ Mock workflow execute completed successfully[/green]")
+            console.print("[green][OK] Mock workflow execute completed successfully[/green]")
             return
 
         from purviewcli.client._workflow import Workflow
@@ -171,12 +171,12 @@ def execute(ctx, workflow_id, payload_file):
         result = workflow_client.workflowExecuteWorkflow(args)
 
         if result:
-            console.print("[green]✓ Workflow execute completed successfully[/green]")
+            console.print("[green][OK] Workflow execute completed successfully[/green]")
             console.print(json.dumps(result, indent=2))
         else:
-            console.print("[yellow]⚠ Workflow execute completed with no result[/yellow]")
+            console.print("[yellow][!] Workflow execute completed with no result[/yellow]")
     except Exception as e:
-        console.print(f"[red]✗ Error executing workflow: {str(e)}[/red]")
+        console.print(f"[red][X] Error executing workflow: {str(e)}[/red]")
 
 
 @workflow.command()
@@ -188,7 +188,7 @@ def executions(ctx, workflow_id):
         if ctx.obj and ctx.obj.get("mock"):
             console.print("[yellow]🎭 Mock: workflow executions command[/yellow]")
             console.print(f"[dim]Workflow ID: {workflow_id}[/dim]")
-            console.print("[green]✓ Mock workflow executions completed successfully[/green]")
+            console.print("[green][OK] Mock workflow executions completed successfully[/green]")
             return
 
         from purviewcli.client._workflow import Workflow
@@ -198,12 +198,12 @@ def executions(ctx, workflow_id):
         result = workflow_client.workflowListWorkflowExecutions(args)
 
         if result:
-            console.print("[green]✓ Workflow executions list completed successfully[/green]")
+            console.print("[green][OK] Workflow executions list completed successfully[/green]")
             console.print(json.dumps(result, indent=2))
         else:
-            console.print("[yellow]⚠ No workflow executions found[/yellow]")
+            console.print("[yellow][!] No workflow executions found[/yellow]")
     except Exception as e:
-        console.print(f"[red]✗ Error listing workflow executions: {str(e)}[/red]")
+        console.print(f"[red][X] Error listing workflow executions: {str(e)}[/red]")
 
 
 # ========== Approval Commands ==========
@@ -222,7 +222,7 @@ def approvals(ctx, status, assigned_to):
                 console.print(f"[dim]Status Filter: {status}[/dim]")
             if assigned_to:
                 console.print(f"[dim]Assigned To: {assigned_to}[/dim]")
-            console.print("[green]✓ Mock workflow approvals completed successfully[/green]")
+            console.print("[green][OK] Mock workflow approvals completed successfully[/green]")
             return
 
         from purviewcli.client._workflow import Workflow
@@ -236,12 +236,12 @@ def approvals(ctx, status, assigned_to):
         result = workflow_client.workflowGetApprovalRequests(args)
 
         if result:
-            console.print("[green]✓ Approval requests list completed successfully[/green]")
+            console.print("[green][OK] Approval requests list completed successfully[/green]")
             console.print(json.dumps(result, indent=2))
         else:
-            console.print("[yellow]⚠ No approval requests found[/yellow]")
+            console.print("[yellow][!] No approval requests found[/yellow]")
     except Exception as e:
-        console.print(f"[red]✗ Error listing approval requests: {str(e)}[/red]")
+        console.print(f"[red][X] Error listing approval requests: {str(e)}[/red]")
 
 
 @workflow.command()
@@ -256,7 +256,7 @@ def approve(ctx, request_id, comments):
             console.print(f"[dim]Request ID: {request_id}[/dim]")
             if comments:
                 console.print(f"[dim]Comments: {comments}[/dim]")
-            console.print("[green]✓ Mock workflow approve completed successfully[/green]")
+            console.print("[green][OK] Mock workflow approve completed successfully[/green]")
             return
 
         from purviewcli.client._workflow import Workflow
@@ -268,12 +268,12 @@ def approve(ctx, request_id, comments):
         result = workflow_client.workflowApproveRequest(args)
 
         if result:
-            console.print("[green]✓ Request approved successfully[/green]")
+            console.print("[green][OK] Request approved successfully[/green]")
             console.print(json.dumps(result, indent=2))
         else:
-            console.print("[yellow]⚠ Request approval completed with no result[/yellow]")
+            console.print("[yellow][!] Request approval completed with no result[/yellow]")
     except Exception as e:
-        console.print(f"[red]✗ Error approving request: {str(e)}[/red]")
+        console.print(f"[red][X] Error approving request: {str(e)}[/red]")
 
 
 @workflow.command()
@@ -288,7 +288,7 @@ def reject(ctx, request_id, comments):
             console.print(f"[dim]Request ID: {request_id}[/dim]")
             if comments:
                 console.print(f"[dim]Comments: {comments}[/dim]")
-            console.print("[green]✓ Mock workflow reject completed successfully[/green]")
+            console.print("[green][OK] Mock workflow reject completed successfully[/green]")
             return
 
         from purviewcli.client._workflow import Workflow
@@ -300,12 +300,12 @@ def reject(ctx, request_id, comments):
         result = workflow_client.workflowRejectRequest(args)
 
         if result:
-            console.print("[green]✓ Request rejected successfully[/green]")
+            console.print("[green][OK] Request rejected successfully[/green]")
             console.print(json.dumps(result, indent=2))
         else:
-            console.print("[yellow]⚠ Request rejection completed with no result[/yellow]")
+            console.print("[yellow][!] Request rejection completed with no result[/yellow]")
     except Exception as e:
-        console.print(f"[red]✗ Error rejecting request: {str(e)}[/red]")
+        console.print(f"[red][X] Error rejecting request: {str(e)}[/red]")
 
 
 # ========== Template Commands ==========
@@ -318,7 +318,7 @@ def templates(ctx):
     try:
         if ctx.obj and ctx.obj.get("mock"):
             console.print("[yellow]🎭 Mock: workflow templates command[/yellow]")
-            console.print("[green]✓ Mock workflow templates completed successfully[/green]")
+            console.print("[green][OK] Mock workflow templates completed successfully[/green]")
             return
 
         from purviewcli.client._workflow import Workflow
@@ -328,12 +328,12 @@ def templates(ctx):
         result = workflow_client.workflowListWorkflowTemplates(args)
 
         if result:
-            console.print("[green]✓ Workflow templates list completed successfully[/green]")
+            console.print("[green][OK] Workflow templates list completed successfully[/green]")
             console.print(json.dumps(result, indent=2))
         else:
-            console.print("[yellow]⚠ No workflow templates found[/yellow]")
+            console.print("[yellow][!] No workflow templates found[/yellow]")
     except Exception as e:
-        console.print(f"[red]✗ Error listing workflow templates: {str(e)}[/red]")
+        console.print(f"[red][X] Error listing workflow templates: {str(e)}[/red]")
 
 
 @workflow.command()
@@ -345,7 +345,7 @@ def template(ctx, template_id):
         if ctx.obj and ctx.obj.get("mock"):
             console.print("[yellow]🎭 Mock: workflow template command[/yellow]")
             console.print(f"[dim]Template ID: {template_id}[/dim]")
-            console.print("[green]✓ Mock workflow template completed successfully[/green]")
+            console.print("[green][OK] Mock workflow template completed successfully[/green]")
             return
 
         from purviewcli.client._workflow import Workflow
@@ -355,12 +355,12 @@ def template(ctx, template_id):
         result = workflow_client.workflowGetWorkflowTemplate(args)
 
         if result:
-            console.print("[green]✓ Workflow template get completed successfully[/green]")
+            console.print("[green][OK] Workflow template get completed successfully[/green]")
             console.print(json.dumps(result, indent=2))
         else:
-            console.print("[yellow]⚠ Workflow template not found[/yellow]")
+            console.print("[yellow][!] Workflow template not found[/yellow]")
     except Exception as e:
-        console.print(f"[red]✗ Error getting workflow template: {str(e)}[/red]")
+        console.print(f"[red][X] Error getting workflow template: {str(e)}[/red]")
 
 
 # ========== Validation Commands ==========
@@ -380,7 +380,7 @@ def validate(ctx, payload_file):
         if ctx.obj and ctx.obj.get("mock"):
             console.print("[yellow]🎭 Mock: workflow validate command[/yellow]")
             console.print(f"[dim]Payload File: {payload_file}[/dim]")
-            console.print("[green]✓ Mock workflow validate completed successfully[/green]")
+            console.print("[green][OK] Mock workflow validate completed successfully[/green]")
             return
 
         from purviewcli.client._workflow import Workflow
@@ -390,12 +390,12 @@ def validate(ctx, payload_file):
         result = workflow_client.workflowValidateWorkflow(args)
 
         if result:
-            console.print("[green]✓ Workflow validation completed successfully[/green]")
+            console.print("[green][OK] Workflow validation completed successfully[/green]")
             console.print(json.dumps(result, indent=2))
         else:
-            console.print("[yellow]⚠ Workflow validation completed with no result[/yellow]")
+            console.print("[yellow][!] Workflow validation completed with no result[/yellow]")
     except Exception as e:
-        console.print(f"[red]✗ Error validating workflow: {str(e)}[/red]")
+        console.print(f"[red][X] Error validating workflow: {str(e)}[/red]")
 
 
 if __name__ == "__main__":

purviewcli/client/_unified_catalog.py

@@ -39,12 +39,22 @@ class UnifiedCatalogClient(Endpoint):
         """Create a new governance domain."""
         self.method = "POST"
         self.endpoint = "/datagovernance/catalog/businessdomains"
-        self.payload = get_json(args, "--payloadFile") or {
-            "name": args.get("--name", [""])[0],
-            "description": args.get("--description", [""])[0],
-            "type": args.get("--type", ["FunctionalUnit"])[0],
-            "status": args.get("--status", ["Draft"])[0],
-        }
+        # Allow payload file to fully control creation; otherwise build payload from flags
+        payload = get_json(args, "--payloadFile")
+        if not payload:
+            payload = {
+                "name": args.get("--name", [""])[0],
+                "description": args.get("--description", [""])[0],
+                "type": args.get("--type", ["FunctionalUnit"])[0],
+                "status": args.get("--status", ["Draft"])[0],
+            }
+            # Support parent domain ID passed via CLI as --parent-domain-id
+            parent_id = args.get("--parent-domain-id", [""])[0]
+            if parent_id:
+                payload["parentId"] = parent_id
+
+        # If payload file contains parentId or parentDomainId, keep it as-is
+        self.payload = payload
 
     @decorator
     def update_governance_domain(self, args):
@@ -411,57 +421,102 @@ class UnifiedCatalogClient(Endpoint):
 
         self.payload = payload
 
-    @decorator
     def update_term(self, args):
-        """Update an existing Unified Catalog term."""
+        """Update an existing Unified Catalog term (supports partial updates)."""
+        from purviewcli.client.endpoint import get_data
+        
         term_id = args.get("--term-id", [""])[0]
-        self.method = "PUT"
-        self.endpoint = f"/datagovernance/catalog/terms/{term_id}"
-
-        # Build payload with all fields (UC API requires full object)
-        domain_id = args.get("--governance-domain-id", [""])[0]
-        name = args.get("--name", [""])[0]
-        description = args.get("--description", [""])[0]
-        status = args.get("--status", ["Draft"])[0]
         
-        # Get owner IDs if provided
-        owner_ids = args.get("--owner-id", [])
-        owners = []
-        if owner_ids:
-            for owner_id in owner_ids:
-                owners.append({"id": owner_id})
+        # First, fetch the existing term to get current values
+        fetch_client = UnifiedCatalogClient()
+        existing_term = fetch_client.get_term_by_id({"--term-id": [term_id]})
         
-        # Get acronyms if provided
-        acronyms = args.get("--acronym", [])
+        if not existing_term or (isinstance(existing_term, dict) and existing_term.get("error")):
+            return {"error": f"Could not fetch existing term {term_id}"}
         
-        # Get resources if provided
-        resources = []
+        # Start with existing term data
+        payload = {
+            "id": term_id,
+            "name": existing_term.get("name", ""),
+            "description": existing_term.get("description", ""),
+            "domain": existing_term.get("domain", ""),
+            "status": existing_term.get("status", "Draft"),
+        }
+        
+        # Update with provided values (only if explicitly provided)
+        if args.get("--name"):
+            payload["name"] = args["--name"][0]
+        if "--description" in args:  # Allow empty string
+            payload["description"] = args.get("--description", [""])[0]
+        if args.get("--governance-domain-id"):
+            payload["domain"] = args["--governance-domain-id"][0]
+        if args.get("--status"):
+            payload["status"] = args["--status"][0]
+        
+        # Handle owners - replace or add to existing
+        contacts = existing_term.get("contacts") or {}
+        existing_owners = contacts.get("owner", []) if isinstance(contacts, dict) else []
+        if args.get("--owner-id"):
+            # Replace owners
+            owners = [{"id": oid} for oid in args["--owner-id"]]
+            payload["contacts"] = {"owner": owners}
+        elif args.get("--add-owner-id"):
+            # Add to existing owners
+            existing_owner_ids = set()
+            if isinstance(existing_owners, list):
+                for o in existing_owners:
+                    if isinstance(o, dict) and o.get("id"):
+                        existing_owner_ids.add(o.get("id"))
+            new_owner_ids = args["--add-owner-id"]
+            combined_owner_ids = existing_owner_ids.union(set(new_owner_ids))
+            owners = [{"id": oid} for oid in combined_owner_ids]
+            payload["contacts"] = {"owner": owners}
+        elif existing_owners:
+            # Keep existing owners
+            payload["contacts"] = {"owner": existing_owners}
+        
+        # Handle acronyms - replace or add to existing
+        existing_acronyms = existing_term.get("acronyms", []) or []
+        if args.get("--acronym"):
+            # Replace acronyms
+            payload["acronyms"] = list(args["--acronym"])
+        elif args.get("--add-acronym"):
+            # Add to existing acronyms
+            combined_acronyms = list(set(existing_acronyms + list(args["--add-acronym"])))
+            payload["acronyms"] = combined_acronyms
+        elif existing_acronyms:
+            # Keep existing acronyms
+            payload["acronyms"] = existing_acronyms
+        
+        # Handle resources - replace with new ones if provided
+        existing_resources = existing_term.get("resources", []) or []
         resource_names = args.get("--resource-name", [])
         resource_urls = args.get("--resource-url", [])
         if resource_names and resource_urls:
+            # Replace resources
+            resources = []
             for i in range(min(len(resource_names), len(resource_urls))):
                 resources.append({
                     "name": resource_names[i],
                     "url": resource_urls[i]
                 })
-        
-        payload = {
-            "id": term_id,
-            "name": name,
-            "description": description,
-            "domain": domain_id,
-            "status": status,
+            payload["resources"] = resources
+        elif existing_resources:
+            # Keep existing resources
+            payload["resources"] = existing_resources
+
+        # Now make the actual PUT request
+        http_dict = {
+            "app": "datagovernance",
+            "method": "PUT",
+            "endpoint": f"/datagovernance/catalog/terms/{term_id}",
+            "params": {},
+            "payload": payload,
+            "files": None,
+            "headers": {},
         }
         
-        # Add optional fields
-        if owners:
-            payload["contacts"] = {"owner": owners}
-        if acronyms:
-            payload["acronyms"] = acronyms
-        if resources:
-            payload["resources"] = resources
-
-        self.payload = payload
+        return get_data(http_dict)
 
     @decorator
     def delete_term(self, args):

purviewcli/client/api_client.py

@@ -5,7 +5,10 @@ Supports the latest Microsoft Purview REST API specifications with comprehensive
 
 import json
 import asyncio
-import aiohttp
+try:
+    import aiohttp
+except Exception:
+    aiohttp = None
 import pandas as pd
 from typing import Dict, List, Optional, Union, Any
 from dataclasses import dataclass
@@ -73,6 +76,11 @@ class PurviewClient:
 
     async def _initialize_session(self):
         """Initialize HTTP session and authentication"""
+        if aiohttp is None:
+            raise RuntimeError(
+                "The 'aiohttp' package is required for Purview async operations. "
+                "Install it in your environment (e.g. '.venv\\Scripts\\pip.exe install aiohttp' or 'pip install aiohttp')."
+            )
         self._credential = DefaultAzureCredential()
 
         try: