unitysvc-services 0.2.1__py3-none-any.whl → 0.2.3__py3-none-any.whl

unitysvc_services/publisher.py

@@ -19,7 +19,7 @@ from .validator import DataValidator
 class ServiceDataPublisher:
     """Publishes service data to UnitySVC backend endpoints."""
 
-    def __init__(self):
+    def __init__(self) -> None:
         self.base_url = os.environ.get("UNITYSVC_BASE_URL")
         if not self.base_url:
             raise ValueError("UNITYSVC_BASE_URL environment variable not set")
@@ -139,7 +139,7 @@ class ServiceDataPublisher:
 
         # Post to the endpoint
         response = self.client.post(
-            f"{self.base_url}/publish/service_offering",
+            f"{self.base_url}/publish/offering",
             json=data_with_content,
         )
         response.raise_for_status()
@@ -250,7 +250,7 @@ class ServiceDataPublisher:
 
         # Post to the endpoint
         response = self.client.post(
-            f"{self.base_url}/publish/service_listing",
+            f"{self.base_url}/publish/listing",
             json=data_with_content,
         )
         response.raise_for_status()

unitysvc_services/query.py

@@ -16,7 +16,7 @@ console = Console()
 class ServiceDataQuery:
     """Query service data from UnitySVC backend endpoints."""
 
-    def __init__(self):
+    def __init__(self) -> None:
         """Initialize query client from environment variables.
 
         Raises:
@@ -41,14 +41,14 @@ class ServiceDataQuery:
 
     def list_service_offerings(self) -> list[dict[str, Any]]:
         """List all service offerings from the backend."""
-        response = self.client.get(f"{self.base_url}/publish/service_offerings")
+        response = self.client.get(f"{self.base_url}/publish/offerings")
         response.raise_for_status()
         result = response.json()
         return result.get("data", result) if isinstance(result, dict) else result
 
     def list_service_listings(self) -> list[dict[str, Any]]:
         """List all service listings from the backend."""
-        response = self.client.get(f"{self.base_url}/publish/services")
+        response = self.client.get(f"{self.base_url}/publish/listings")
         response.raise_for_status()
         result = response.json()
         return result.get("data", result) if isinstance(result, dict) else result
@@ -69,7 +69,7 @@ class ServiceDataQuery:
 
     def list_access_interfaces(self) -> dict[str, Any]:
         """List all access interfaces from the backend (private endpoint)."""
-        response = self.client.get(f"{self.base_url}/private/access_interfaces")
+        response = self.client.get(f"{self.base_url}/private/interfaces")
         response.raise_for_status()
         return response.json()
 
@@ -100,37 +100,136 @@ def query_sellers(
         "-f",
         help="Output format: table, json",
     ),
+    fields: str = typer.Option(
+        "id,name,display_name,seller_type",
+        "--fields",
+        help=(
+            "Comma-separated list of fields to display. Available fields: "
+            "id, name, display_name, seller_type, contact_email, "
+            "secondary_contact_email, homepage, description, "
+            "business_registration, tax_id, account_manager_id, "
+            "created_at, updated_at, status"
+        ),
+    ),
 ):
-    """Query all sellers from the backend."""
+    """Query all sellers from the backend.
+
+    Examples:
+        # Use default fields
+        unitysvc_services query sellers
+
+        # Show only specific fields
+        unitysvc_services query sellers --fields id,name,contact_email
+
+        # Show all available fields
+        unitysvc_services query sellers --fields \\
+            id,name,display_name,seller_type,contact_email,homepage,created_at,updated_at
+    """
+    # Parse fields list
+    field_list = [f.strip() for f in fields.split(",")]
+
+    # Define allowed fields from SellerPublic model
+    allowed_fields = {
+        "id",
+        "name",
+        "display_name",
+        "seller_type",
+        "contact_email",
+        "secondary_contact_email",
+        "homepage",
+        "description",
+        "business_registration",
+        "tax_id",
+        "account_manager_id",
+        "created_at",
+        "updated_at",
+        "status",
+    }
+
+    # Validate fields
+    invalid_fields = [f for f in field_list if f not in allowed_fields]
+    if invalid_fields:
+        console.print(
+            f"[red]✗[/red] Invalid field(s): {', '.join(invalid_fields)}",
+            style="bold red",
+        )
+        console.print(f"[yellow]Allowed fields:[/yellow] {', '.join(sorted(allowed_fields))}")
+        raise typer.Exit(code=1)
+
     try:
         with ServiceDataQuery() as query:
             sellers = query.list_sellers()
 
             if format == "json":
-                console.print(json.dumps(sellers, indent=2))
+                # For JSON, filter fields if not all are requested
+                if set(field_list) != allowed_fields:
+                    filtered_sellers = [{k: v for k, v in seller.items() if k in field_list} for seller in sellers]
+                    console.print(json.dumps(filtered_sellers, indent=2))
+                else:
+                    console.print(json.dumps(sellers, indent=2))
             else:
                 if not sellers:
                     console.print("[yellow]No sellers found.[/yellow]")
                 else:
                     table = Table(title="Sellers")
-                    table.add_column("ID", style="cyan")
-                    table.add_column("Name", style="green")
-                    table.add_column("Display Name", style="blue")
-                    table.add_column("Type", style="magenta")
-                    table.add_column("Contact Email", style="yellow")
-                    table.add_column("Active", style="white")
 
+                    # Define column styles
+                    field_styles = {
+                        "id": "cyan",
+                        "name": "green",
+                        "display_name": "blue",
+                        "seller_type": "magenta",
+                        "contact_email": "yellow",
+                        "secondary_contact_email": "yellow",
+                        "homepage": "blue",
+                        "description": "white",
+                        "business_registration": "white",
+                        "tax_id": "white",
+                        "account_manager_id": "cyan",
+                        "created_at": "white",
+                        "updated_at": "white",
+                        "status": "green",
+                    }
+
+                    # Define column headers
+                    field_headers = {
+                        "id": "ID",
+                        "name": "Name",
+                        "display_name": "Display Name",
+                        "seller_type": "Type",
+                        "contact_email": "Contact Email",
+                        "secondary_contact_email": "Secondary Email",
+                        "homepage": "Homepage",
+                        "description": "Description",
+                        "business_registration": "Business Reg",
+                        "tax_id": "Tax ID",
+                        "account_manager_id": "Account Manager ID",
+                        "created_at": "Created At",
+                        "updated_at": "Updated At",
+                        "status": "Status",
+                    }
+
+                    # Add columns based on requested fields
+                    for field in field_list:
+                        header = field_headers.get(field, field.title())
+                        style = field_styles.get(field, "white")
+                        table.add_column(header, style=style)
+
+                    # Add rows
                     for seller in sellers:
-                        table.add_row(
-                            str(seller.get("id", "N/A")),
-                            seller.get("name", "N/A"),
-                            seller.get("display_name", "N/A"),
-                            seller.get("seller_type", "N/A"),
-                            seller.get("contact_email", "N/A"),
-                            "✓" if seller.get("is_active") else "✗",
-                        )
+                        row = []
+                        for field in field_list:
+                            value = seller.get(field)
+                            if value is None:
+                                row.append("N/A")
+                            elif isinstance(value, dict | list):
+                                row.append(str(value)[:50])  # Truncate complex types
+                            else:
+                                row.append(str(value))
+                        table.add_row(*row)
 
                     console.print(table)
+                    console.print(f"\n[green]Total:[/green] {len(sellers)} seller(s)")
     except ValueError as e:
         console.print(f"[red]✗[/red] {e}", style="bold red")
         raise typer.Exit(code=1)
@@ -147,33 +246,124 @@ def query_providers(
         "-f",
         help="Output format: table, json",
     ),
+    fields: str = typer.Option(
+        "id,name,display_name,status",
+        "--fields",
+        help=(
+            "Comma-separated list of fields to display. Available fields: "
+            "id, name, display_name, contact_email, secondary_contact_email, "
+            "homepage, description, status, created_at, updated_at"
+        ),
+    ),
 ):
-    """Query all providers from the backend."""
+    """Query all providers from the backend.
+
+    Examples:
+        # Use default fields
+        unitysvc_services query providers
+
+        # Show only specific fields
+        unitysvc_services query providers --fields id,name,contact_email
+
+        # Show all available fields
+        unitysvc_services query providers --fields \\
+            id,name,display_name,contact_email,homepage,status,created_at,updated_at
+    """
+    # Parse fields list
+    field_list = [f.strip() for f in fields.split(",")]
+
+    # Define allowed fields from ProviderPublic model
+    allowed_fields = {
+        "id",
+        "name",
+        "display_name",
+        "contact_email",
+        "secondary_contact_email",
+        "homepage",
+        "description",
+        "status",
+        "created_at",
+        "updated_at",
+    }
+
+    # Validate fields
+    invalid_fields = [f for f in field_list if f not in allowed_fields]
+    if invalid_fields:
+        console.print(
+            f"[red]✗[/red] Invalid field(s): {', '.join(invalid_fields)}",
+            style="bold red",
+        )
+        console.print(f"[yellow]Allowed fields:[/yellow] {', '.join(sorted(allowed_fields))}")
+        raise typer.Exit(code=1)
+
     try:
         with ServiceDataQuery() as query:
             providers = query.list_providers()
 
             if format == "json":
-                console.print(json.dumps(providers, indent=2))
+                # For JSON, filter fields if not all are requested
+                if set(field_list) != allowed_fields:
+                    filtered_providers = [
+                        {k: v for k, v in provider.items() if k in field_list} for provider in providers
+                    ]
+                    console.print(json.dumps(filtered_providers, indent=2))
+                else:
+                    console.print(json.dumps(providers, indent=2))
             else:
                 if not providers:
                     console.print("[yellow]No providers found.[/yellow]")
                 else:
                     table = Table(title="Providers")
-                    table.add_column("ID", style="cyan")
-                    table.add_column("Name", style="green")
-                    table.add_column("Display Name", style="blue")
-                    table.add_column("Time Created", style="magenta")
 
+                    # Define column styles
+                    field_styles = {
+                        "id": "cyan",
+                        "name": "green",
+                        "display_name": "blue",
+                        "contact_email": "yellow",
+                        "secondary_contact_email": "yellow",
+                        "homepage": "blue",
+                        "description": "white",
+                        "status": "green",
+                        "created_at": "magenta",
+                        "updated_at": "magenta",
+                    }
+
+                    # Define column headers
+                    field_headers = {
+                        "id": "ID",
+                        "name": "Name",
+                        "display_name": "Display Name",
+                        "contact_email": "Contact Email",
+                        "secondary_contact_email": "Secondary Email",
+                        "homepage": "Homepage",
+                        "description": "Description",
+                        "status": "Status",
+                        "created_at": "Created At",
+                        "updated_at": "Updated At",
+                    }
+
+                    # Add columns based on requested fields
+                    for field in field_list:
+                        header = field_headers.get(field, field.title())
+                        style = field_styles.get(field, "white")
+                        table.add_column(header, style=style)
+
+                    # Add rows
                     for provider in providers:
-                        table.add_row(
-                            str(provider.get("id", "N/A")),
-                            provider.get("name", "N/A"),
-                            provider.get("display_name", "N/A"),
-                            str(provider.get("time_created", "N/A")),
-                        )
+                        row = []
+                        for field in field_list:
+                            value = provider.get(field)
+                            if value is None:
+                                row.append("N/A")
+                            elif isinstance(value, dict | list):
+                                row.append(str(value)[:50])  # Truncate complex types
+                            else:
+                                row.append(str(value))
+                        table.add_row(*row)
 
                     console.print(table)
+                    console.print(f"\n[green]Total:[/green] {len(providers)} provider(s)")
     except ValueError as e:
         console.print(f"[red]✗[/red] {e}", style="bold red")
         raise typer.Exit(code=1)
@@ -190,35 +380,91 @@ def query_offerings(
         "-f",
         help="Output format: table, json",
     ),
+    fields: str = typer.Option(
+        "id,service_name,service_type,provider_name,status",
+        "--fields",
+        help=(
+            "Comma-separated list of fields to display. Available fields: "
+            "id, definition_id, provider_id, status, price, service_name, "
+            "service_type, provider_name"
+        ),
+    ),
 ):
-    """Query all service offerings from UnitySVC backend."""
+    """Query all service offerings from UnitySVC backend.
+
+    Examples:
+        # Use default fields
+        unitysvc_services query offerings
+
+        # Show only specific fields
+        unitysvc_services query offerings --fields id,service_name,status
+
+        # Show all available fields
+        unitysvc_services query offerings --fields \\
+            id,service_name,service_type,provider_name,status,price,definition_id,provider_id
+    """
+    # Parse fields list
+    field_list = [f.strip() for f in fields.split(",")]
+
+    # Define allowed fields from ServiceOfferingPublic model
+    allowed_fields = {
+        "id",
+        "definition_id",
+        "provider_id",
+        "status",
+        "price",
+        "service_name",
+        "service_type",
+        "provider_name",
+    }
+
+    # Validate fields
+    invalid_fields = [f for f in field_list if f not in allowed_fields]
+    if invalid_fields:
+        console.print(
+            f"[red]Error:[/red] Invalid field(s): {', '.join(invalid_fields)}",
+            style="bold red",
+        )
+        console.print(f"[yellow]Available fields:[/yellow] {', '.join(sorted(allowed_fields))}")
+        raise typer.Exit(code=1)
+
     try:
         with ServiceDataQuery() as query:
             offerings = query.list_service_offerings()
 
             if format == "json":
-                console.print(json.dumps(offerings, indent=2))
+                # For JSON, filter fields if not all are requested
+                if set(field_list) != allowed_fields:
+                    filtered_offerings = [
+                        {k: v for k, v in offering.items() if k in field_list} for offering in offerings
+                    ]
+                    console.print(json.dumps(filtered_offerings, indent=2))
+                else:
+                    console.print(json.dumps(offerings, indent=2))
             else:
                 if not offerings:
                     console.print("[yellow]No service offerings found.[/yellow]")
                 else:
-                    table = Table(title="Service Offerings", show_lines=True)
-                    table.add_column("ID", style="cyan")
-                    table.add_column("Name", style="green")
-                    table.add_column("Display Name", style="blue")
-                    table.add_column("Type", style="magenta")
-                    table.add_column("Status", style="yellow")
-                    table.add_column("Version")
+                    table = Table(title="Service Offerings")
+
+                    # Add columns dynamically based on selected fields
+                    for field in field_list:
+                        # Capitalize and format field names for display
+                        column_name = field.replace("_", " ").title()
+                        table.add_column(column_name)
 
+                    # Add rows
                     for offering in offerings:
-                        table.add_row(
-                            str(offering.get("id", "N/A")),
-                            offering.get("name", "N/A"),
-                            offering.get("display_name", "N/A"),
-                            offering.get("service_type", "N/A"),
-                            offering.get("upstream_status", "N/A"),
-                            offering.get("version", "N/A"),
-                        )
+                        row = []
+                        for field in field_list:
+                            value = offering.get(field)
+                            if value is None:
+                                row.append("N/A")
+                            elif isinstance(value, dict | list):
+                                row.append(str(value)[:50])  # Truncate complex types
+                            else:
+                                row.append(str(value))
+                        table.add_row(*row)
 
                     console.print(table)
                     console.print(f"\n[green]Total:[/green] {len(offerings)} service offering(s)")
@@ -238,34 +484,96 @@ def query_listings(
         "-f",
         help="Output format: table, json",
     ),
+    fields: str = typer.Option(
+        "id,service_name,service_type,seller_name,listing_type,status",
+        "--fields",
+        help=(
+            "Comma-separated list of fields to display. Available fields: "
+            "id, offering_id, seller_id, status, created_at, updated_at, "
+            "parameters_schema, parameters_ui_schema, tags, service_name, "
+            "service_type, provider_name, seller_name, listing_type"
+        ),
+    ),
 ):
-    """Query all service listings from UnitySVC backend."""
+    """Query all service listings from UnitySVC backend.
+
+    Examples:
+        # Use default fields
+        unitysvc_services query listings
+
+        # Show only specific fields
+        unitysvc_services query listings --fields id,service_name,status
+
+        # Show all available fields
+        unitysvc_services query listings --fields \\
+            id,service_name,service_type,seller_name,listing_type,status,provider_name
+    """
+    # Parse fields list
+    field_list = [f.strip() for f in fields.split(",")]
+
+    # Define allowed fields from ServiceListingPublic model
+    allowed_fields = {
+        "id",
+        "offering_id",
+        "seller_id",
+        "status",
+        "created_at",
+        "updated_at",
+        "parameters_schema",
+        "parameters_ui_schema",
+        "tags",
+        "service_name",
+        "service_type",
+        "provider_name",
+        "seller_name",
+        "listing_type",
+    }
+
+    # Validate fields
+    invalid_fields = [f for f in field_list if f not in allowed_fields]
+    if invalid_fields:
+        console.print(
+            f"[red]Error:[/red] Invalid field(s): {', '.join(invalid_fields)}",
+            style="bold red",
+        )
+        console.print(f"[yellow]Available fields:[/yellow] {', '.join(sorted(allowed_fields))}")
+        raise typer.Exit(code=1)
+
     try:
         with ServiceDataQuery() as query:
             listings = query.list_service_listings()
 
             if format == "json":
-                console.print(json.dumps(listings, indent=2))
+                # For JSON, filter fields if not all are requested
+                if set(field_list) != allowed_fields:
+                    filtered_listings = [{k: v for k, v in listing.items() if k in field_list} for listing in listings]
+                    console.print(json.dumps(filtered_listings, indent=2))
+                else:
+                    console.print(json.dumps(listings, indent=2))
             else:
                 if not listings:
                     console.print("[yellow]No service listings found.[/yellow]")
                 else:
-                    table = Table(title="Service Listings", show_lines=True)
-                    table.add_column("ID", style="cyan")
-                    table.add_column("Service ID", style="blue")
-                    table.add_column("Seller", style="green")
-                    table.add_column("Status", style="yellow")
-                    table.add_column("Interfaces")
+                    table = Table(title="Service Listings")
 
+                    # Add columns dynamically based on selected fields
+                    for field in field_list:
+                        # Capitalize and format field names for display
+                        column_name = field.replace("_", " ").title()
+                        table.add_column(column_name)
+
+                    # Add rows
                     for listing in listings:
-                        interfaces_count = len(listing.get("user_access_interfaces", []))
-                        table.add_row(
-                            str(listing.get("id", "N/A")),
-                            str(listing.get("service_id", "N/A")),
-                            listing.get("seller_name", "N/A"),
-                            listing.get("listing_status", "N/A"),
-                            str(interfaces_count),
-                        )
+                        row = []
+                        for field in field_list:
+                            value = listing.get(field)
+                            if value is None:
+                                row.append("N/A")
+                            elif isinstance(value, dict | list):
+                                row.append(str(value)[:50])  # Truncate complex types
+                            else:
+                                row.append(str(value))
+                        table.add_row(*row)
 
                     console.print(table)
                     console.print(f"\n[green]Total:[/green] {len(listings)} service listing(s)")
@@ -285,36 +593,101 @@ def query_interfaces(
         "-f",
         help="Output format: table, json",
     ),
+    fields: str = typer.Option(
+        "id,name,context_type,access_method,is_active",
+        "--fields",
+        help=(
+            "Comma-separated list of fields to display. Available fields: "
+            "id, entity_id, context_type, access_method, api_endpoint, name, "
+            "description, request_transformer, rate_limits, constraint, "
+            "is_active, is_primary, sort_order, created_at, updated_at"
+        ),
+    ),
 ):
-    """Query all access interfaces from UnitySVC backend (private endpoint)."""
+    """Query all access interfaces from UnitySVC backend (private endpoint).
+
+    Examples:
+        # Use default fields
+        unitysvc_services query interfaces
+
+        # Show only specific fields
+        unitysvc_services query interfaces --fields id,name,access_method
+
+        # Show all available fields
+        unitysvc_services query interfaces --fields \\
+            id,name,context_type,access_method,entity_id,is_active,is_primary
+    """
+    # Parse fields list
+    field_list = [f.strip() for f in fields.split(",")]
+
+    # Define allowed fields from AccessInterfacePublic model
+    allowed_fields = {
+        "id",
+        "entity_id",
+        "context_type",
+        "access_method",
+        "api_endpoint",
+        "name",
+        "description",
+        "request_transformer",
+        "rate_limits",
+        "constraint",
+        "is_active",
+        "is_primary",
+        "sort_order",
+        "created_at",
+        "updated_at",
+    }
+
+    # Validate fields
+    invalid_fields = [f for f in field_list if f not in allowed_fields]
+    if invalid_fields:
+        console.print(
+            f"[red]Error:[/red] Invalid field(s): {', '.join(invalid_fields)}",
+            style="bold red",
+        )
+        console.print(f"[yellow]Available fields:[/yellow] {', '.join(sorted(allowed_fields))}")
+        raise typer.Exit(code=1)
+
     try:
         with ServiceDataQuery() as query:
             data = query.list_access_interfaces()
 
             if format == "json":
-                console.print(json.dumps(data, indent=2))
+                # For JSON, filter fields if not all are requested
+                interfaces = data.get("data", [])
+                if set(field_list) != allowed_fields:
+                    filtered_interfaces = [
+                        {k: v for k, v in interface.items() if k in field_list} for interface in interfaces
+                    ]
+                    console.print(json.dumps({"data": filtered_interfaces, "count": data.get("count", 0)}, indent=2))
+                else:
+                    console.print(json.dumps(data, indent=2))
             else:
                 interfaces = data.get("data", [])
                 if not interfaces:
                     console.print("[yellow]No access interfaces found.[/yellow]")
                 else:
-                    table = Table(title="Access Interfaces", show_lines=True)
-                    table.add_column("ID", style="cyan")
-                    table.add_column("Name", style="green")
-                    table.add_column("Context", style="blue")
-                    table.add_column("Entity ID", style="yellow")
-                    table.add_column("Method", style="magenta")
-                    table.add_column("Active", style="green")
+                    table = Table(title="Access Interfaces")
+
+                    # Add columns dynamically based on selected fields
+                    for field in field_list:
+                        # Capitalize and format field names for display
+                        column_name = field.replace("_", " ").title()
+                        table.add_column(column_name)
 
+                    # Add rows
                     for interface in interfaces:
-                        table.add_row(
-                            str(interface.get("id", "N/A"))[:8] + "...",
-                            interface.get("name", "N/A"),
-                            interface.get("context_type", "N/A"),
-                            str(interface.get("entity_id", "N/A"))[:8] + "...",
-                            interface.get("access_method", "N/A"),
-                            "✓" if interface.get("is_active") else "✗",
-                        )
+                        row = []
+                        for field in field_list:
+                            value = interface.get(field)
+                            if value is None:
+                                row.append("N/A")
+                            elif isinstance(value, dict | list):
+                                row.append(str(value)[:50])  # Truncate complex types
+                            else:
+                                row.append(str(value))
+                        table.add_row(*row)
 
                     console.print(table)
                     console.print(f"\n[green]Total:[/green] {data.get('count', 0)} access interface(s)")
@@ -334,36 +707,102 @@ def query_documents(
         "-f",
         help="Output format: table, json",
     ),
+    fields: str = typer.Option(
+        "id,title,category,mime_type,context_type,is_public",
+        "--fields",
+        help=(
+            "Comma-separated list of fields to display. Available fields: "
+            "id, entity_id, context_type, title, description, mime_type, "
+            "version, category, meta, sort_order, is_active, is_public, "
+            "external_url, object_key, filename, filesize, created_at, updated_at"
+        ),
+    ),
 ):
-    """Query all documents from UnitySVC backend (private endpoint)."""
+    """Query all documents from UnitySVC backend (private endpoint).
+
+    Examples:
+        # Use default fields
+        unitysvc_services query documents
+
+        # Show only specific fields
+        unitysvc_services query documents --fields id,title,category
+
+        # Show all available fields
+        unitysvc_services query documents --fields \\
+            id,title,category,mime_type,context_type,is_public,filename,filesize
+    """
+    # Parse fields list
+    field_list = [f.strip() for f in fields.split(",")]
+
+    # Define allowed fields from DocumentPublic model
+    allowed_fields = {
+        "id",
+        "entity_id",
+        "context_type",
+        "title",
+        "description",
+        "mime_type",
+        "version",
+        "category",
+        "meta",
+        "sort_order",
+        "is_active",
+        "is_public",
+        "external_url",
+        "object_key",
+        "filename",
+        "filesize",
+        "created_at",
+        "updated_at",
+    }
+
+    # Validate fields
+    invalid_fields = [f for f in field_list if f not in allowed_fields]
+    if invalid_fields:
+        console.print(
+            f"[red]Error:[/red] Invalid field(s): {', '.join(invalid_fields)}",
+            style="bold red",
+        )
+        console.print(f"[yellow]Available fields:[/yellow] {', '.join(sorted(allowed_fields))}")
+        raise typer.Exit(code=1)
+
     try:
         with ServiceDataQuery() as query:
             data = query.list_documents()
 
             if format == "json":
-                console.print(json.dumps(data, indent=2))
+                # For JSON, filter fields if not all are requested
+                documents = data.get("data", [])
+                if set(field_list) != allowed_fields:
+                    filtered_documents = [{k: v for k, v in doc.items() if k in field_list} for doc in documents]
+                    console.print(json.dumps({"data": filtered_documents, "count": data.get("count", 0)}, indent=2))
+                else:
+                    console.print(json.dumps(data, indent=2))
             else:
                 documents = data.get("data", [])
                 if not documents:
                     console.print("[yellow]No documents found.[/yellow]")
                 else:
-                    table = Table(title="Documents", show_lines=True)
-                    table.add_column("ID", style="cyan")
-                    table.add_column("Title", style="green")
-                    table.add_column("Category", style="blue")
-                    table.add_column("MIME Type", style="yellow")
-                    table.add_column("Context", style="magenta")
-                    table.add_column("Public", style="red")
+                    table = Table(title="Documents")
+
+                    # Add columns dynamically based on selected fields
+                    for field in field_list:
+                        # Capitalize and format field names for display
+                        column_name = field.replace("_", " ").title()
+                        table.add_column(column_name)
 
+                    # Add rows
                     for doc in documents:
-                        table.add_row(
-                            str(doc.get("id", "N/A"))[:8] + "...",
-                            doc.get("title", "N/A")[:40],
-                            doc.get("category", "N/A"),
-                            doc.get("mime_type", "N/A"),
-                            doc.get("context_type", "N/A"),
-                            "✓" if doc.get("is_public") else "✗",
-                        )
+                        row = []
+                        for field in field_list:
+                            value = doc.get(field)
+                            if value is None:
+                                row.append("N/A")
+                            elif isinstance(value, dict | list):
+                                row.append(str(value)[:50])  # Truncate complex types
+                            else:
+                                row.append(str(value))
+                        table.add_row(*row)
 
                     console.print(table)
                     console.print(f"\n[green]Total:[/green] {data.get('count', 0)} document(s)")

{unitysvc_services-0.2.1.dist-info → unitysvc_services-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unitysvc-services
-Version: 0.2.1
+Version: 0.2.3
 Summary: SDK for digital service providers on UnitySVC
 Author-email: Bo Peng <bo.peng@unitysvc.com>
 Maintainer-email: Bo Peng <bo.peng@unitysvc.com>

{unitysvc_services-0.2.1.dist-info → unitysvc_services-0.2.3.dist-info}/RECORD

@@ -3,8 +3,9 @@ unitysvc_services/cli.py,sha256=OK0IZyAckxP15jRWU_W49hl3t7XcNRtd8BoDMyRKqNM,682
 unitysvc_services/format_data.py,sha256=Jl9Vj3fRX852fHSUa5DzO-oiFQwuQHC3WMCDNIlo1Lc,5460
 unitysvc_services/list.py,sha256=QDp9BByaoeFeJxXJN9RQ-jU99mH9Guq9ampfXCbpZmI,7033
 unitysvc_services/populate.py,sha256=zkcjIy8BWuQSO7JwiRNHKgGoxQvc3ujluUQdYixdBvY,6626
-unitysvc_services/publisher.py,sha256=srOwISXimWTbtIS9g6egLm7JHPl7wCvjoDlY17tQaeU,35375
-unitysvc_services/query.py,sha256=lxNxsgjuxJ0UEouYMorjG4MpklKi7UCYPKgdJH9MFTA,14553
+unitysvc_services/publisher.py,sha256=qEkui7T7DQr_CfXzDvWMuTJrPpKbcldfwPi6Ocv-ILA,35367
+unitysvc_services/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+unitysvc_services/query.py,sha256=p-fG1iosAuzdHYnaYmEe2zdHQVZzWk2LhWmu5B4sAdA,30203
 unitysvc_services/scaffold.py,sha256=Y73IX8vskImxSvxDgR0mvEFuAMYnBKfttn3bjcz3jmQ,40331
 unitysvc_services/update.py,sha256=K9swocTUnqqiSgARo6GmuzTzUySSpyqqPPW4xF7ZU-g,9659
 unitysvc_services/utils.py,sha256=GN0gkVTU8fOx2G0EbqnWmx8w9eFsoPfRprPjwCyPYkE,11371
@@ -15,9 +16,9 @@ unitysvc_services/models/listing_v1.py,sha256=CC_GXoN3NHJFzEQ3cBHDQpdUaBNpvLdHAR
 unitysvc_services/models/provider_v1.py,sha256=cYK5kDDmzQEnLvUC2C8dKz-ZXci7hVn3fjNrJkaSr10,2050
 unitysvc_services/models/seller_v1.py,sha256=SU4rqYAh9hE4EeUrEkqaVrLwusenV7MotPF77VcsRKo,3263
 unitysvc_services/models/service_v1.py,sha256=u16zqM3khrJoTw_v0d45tMcKXjko5k_v3w8xwUtZ6nM,2720
-unitysvc_services-0.2.1.dist-info/licenses/LICENSE,sha256=_p8V6A8OMPu2HIztn3O01v0-urZFwk0Dd3Yk_PTIlL8,1065
-unitysvc_services-0.2.1.dist-info/METADATA,sha256=KmRHbFRzKhpl0FGM8CCXE8nrAHMp5KJ0fdZBUvQHQgw,6515
-unitysvc_services-0.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-unitysvc_services-0.2.1.dist-info/entry_points.txt,sha256=-vodnbPmo7QQmFu8jdG6sCyGRVM727w9Nhwp4Vwau_k,64
-unitysvc_services-0.2.1.dist-info/top_level.txt,sha256=GIotQj-Ro2ruR7eupM1r58PWqIHTAq647ORL7E2kneo,18
-unitysvc_services-0.2.1.dist-info/RECORD,,
+unitysvc_services-0.2.3.dist-info/licenses/LICENSE,sha256=_p8V6A8OMPu2HIztn3O01v0-urZFwk0Dd3Yk_PTIlL8,1065
+unitysvc_services-0.2.3.dist-info/METADATA,sha256=MqWBQxgCpX6dpqmS1K95RXr-2yhvS8YyHDqfE0mZiYU,6515
+unitysvc_services-0.2.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+unitysvc_services-0.2.3.dist-info/entry_points.txt,sha256=-vodnbPmo7QQmFu8jdG6sCyGRVM727w9Nhwp4Vwau_k,64
+unitysvc_services-0.2.3.dist-info/top_level.txt,sha256=GIotQj-Ro2ruR7eupM1r58PWqIHTAq647ORL7E2kneo,18
+unitysvc_services-0.2.3.dist-info/RECORD,,