unitysvc-services 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

unitysvc_services/format_data.py

@@ -1,6 +1,5 @@
 """Format command - format data files."""
 
-import os
 from pathlib import Path
 
 import typer
@@ -14,7 +13,7 @@ console = Console()
 def format_data(
     data_dir: Path | None = typer.Argument(
         None,
-        help="Directory containing data files to format (default: ./data or UNITYSVC_DATA_DIR env var)",
+        help="Directory containing data files to format (default: current directory)",
     ),
     check_only: bool = typer.Option(
         False,
@@ -35,11 +34,7 @@ def format_data(
 
     # Set data directory
     if data_dir is None:
-        data_dir_str = os.getenv("UNITYSVC_DATA_DIR")
-        if data_dir_str:
-            data_dir = Path(data_dir_str)
-        else:
-            data_dir = Path.cwd() / "data"
+        data_dir = Path.cwd()
 
     if not data_dir.is_absolute():
         data_dir = Path.cwd() / data_dir

unitysvc_services/list.py

@@ -1,6 +1,5 @@
 """List command group - list local data files."""
 
-import os
 from pathlib import Path
 
 import typer
@@ -21,25 +20,19 @@ console = Console()
 def list_providers(
     data_dir: Path | None = typer.Argument(
         None,
-        help="Directory containing provider files (default: ./data or UNITYSVC_DATA_DIR env var)",
+        help="Directory containing provider files (default: current directory)",
     ),
 ):
     """List all provider files found in the data directory."""
     # Set data directory
     if data_dir is None:
-        data_dir_str = os.getenv("UNITYSVC_DATA_DIR")
-        if data_dir_str:
-            data_dir = Path(data_dir_str)
-        else:
-            data_dir = Path.cwd() / "data"
+        data_dir = Path.cwd()
 
     if not data_dir.is_absolute():
         data_dir = Path.cwd() / data_dir
 
     if not data_dir.exists():
-        console.print(
-            f"[red]✗[/red] Data directory not found: {data_dir}", style="bold red"
-        )
+        console.print(f"[red]✗[/red] Data directory not found: {data_dir}", style="bold red")
         raise typer.Exit(code=1)
 
     console.print(f"[blue]Searching for providers in:[/blue] {data_dir}\n")
@@ -72,25 +65,19 @@ def list_providers(
 def list_sellers(
     data_dir: Path | None = typer.Argument(
         None,
-        help="Directory containing seller files (default: ./data or UNITYSVC_DATA_DIR env var)",
+        help="Directory containing seller files (default: current directory)",
     ),
 ):
     """List all seller files found in the data directory."""
     # Set data directory
     if data_dir is None:
-        data_dir_str = os.getenv("UNITYSVC_DATA_DIR")
-        if data_dir_str:
-            data_dir = Path(data_dir_str)
-        else:
-            data_dir = Path.cwd() / "data"
+        data_dir = Path.cwd()
 
     if not data_dir.is_absolute():
         data_dir = Path.cwd() / data_dir
 
     if not data_dir.exists():
-        console.print(
-            f"[red]✗[/red] Data directory not found: {data_dir}", style="bold red"
-        )
+        console.print(f"[red]✗[/red] Data directory not found: {data_dir}", style="bold red")
         raise typer.Exit(code=1)
 
     console.print(f"[blue]Searching for sellers in:[/blue] {data_dir}\n")
@@ -123,25 +110,19 @@ def list_sellers(
 def list_offerings(
     data_dir: Path | None = typer.Argument(
         None,
-        help="Directory containing service files (default: ./data or UNITYSVC_DATA_DIR env var)",
+        help="Directory containing service files (default: current directory)",
     ),
 ):
     """List all service offering files found in the data directory."""
     # Set data directory
     if data_dir is None:
-        data_dir_str = os.getenv("UNITYSVC_DATA_DIR")
-        if data_dir_str:
-            data_dir = Path(data_dir_str)
-        else:
-            data_dir = Path.cwd() / "data"
+        data_dir = Path.cwd()
 
     if not data_dir.is_absolute():
         data_dir = Path.cwd() / data_dir
 
     if not data_dir.exists():
-        console.print(
-            f"[red]✗[/red] Data directory not found: {data_dir}", style="bold red"
-        )
+        console.print(f"[red]✗[/red] Data directory not found: {data_dir}", style="bold red")
         raise typer.Exit(code=1)
 
     console.print(f"[blue]Searching for service offerings in:[/blue] {data_dir}\n")
@@ -172,34 +153,26 @@ def list_offerings(
         )
 
     console.print(table)
-    console.print(
-        f"\n[green]Total:[/green] {len(service_files)} service offering file(s)"
-    )
+    console.print(f"\n[green]Total:[/green] {len(service_files)} service offering file(s)")
 
 
 @app.command("listings")
 def list_listings(
     data_dir: Path | None = typer.Argument(
         None,
-        help="Directory containing listing files (default: ./data or UNITYSVC_DATA_DIR env var)",
+        help="Directory containing listing files (default: current directory)",
     ),
 ):
     """List all service listing files found in the data directory."""
     # Set data directory
     if data_dir is None:
-        data_dir_str = os.getenv("UNITYSVC_DATA_DIR")
-        if data_dir_str:
-            data_dir = Path(data_dir_str)
-        else:
-            data_dir = Path.cwd() / "data"
+        data_dir = Path.cwd()
 
     if not data_dir.is_absolute():
         data_dir = Path.cwd() / data_dir
 
     if not data_dir.exists():
-        console.print(
-            f"[red]✗[/red] Data directory not found: {data_dir}", style="bold red"
-        )
+        console.print(f"[red]✗[/red] Data directory not found: {data_dir}", style="bold red")
         raise typer.Exit(code=1)
 
     console.print(f"[blue]Searching for service listings in:[/blue] {data_dir}\n")
@@ -240,6 +213,4 @@ def list_listings(
         )
 
     console.print(table)
-    console.print(
-        f"\n[green]Total:[/green] {len(listing_files)} service listing file(s)"
-    )
+    console.print(f"\n[green]Total:[/green] {len(listing_files)} service listing file(s)")

unitysvc_services/models/base.py

@@ -206,6 +206,22 @@ class UpstreamStatusEnum(StrEnum):
     deprecated = "deprecated"
 
 
+class ProviderStatusEnum(StrEnum):
+    """Provider status enum."""
+
+    active = "active"
+    disabled = "disabled"
+    incomplete = "incomplete"  # Provider information is incomplete
+
+
+class SellerStatusEnum(StrEnum):
+    """Seller status enum."""
+
+    active = "active"
+    disabled = "disabled"
+    incomplete = "incomplete"  # Seller information is incomplete
+
+
 class Document(BaseModel):
     model_config = ConfigDict(extra="forbid")
 

unitysvc_services/models/listing_v1.py

@@ -30,9 +30,7 @@ class ListingV1(BaseModel):
         ),
     )
 
-    seller_name: str = Field(
-        description="Name of the seller offering this service listing"
-    )
+    seller_name: str | None = Field(default=None, description="Name of the seller offering this service listing")
 
     # unique name for each provider, usually following upstream naming convention
     # status of the service, public, deprecated etc

unitysvc_services/models/provider_v1.py

@@ -3,7 +3,7 @@ from typing import Any
 
 from pydantic import BaseModel, ConfigDict, EmailStr, Field, HttpUrl
 
-from unitysvc_services.models.base import AccessInterface, Document
+from unitysvc_services.models.base import AccessInterface, Document, ProviderStatusEnum
 
 
 class ProviderV1(BaseModel):
@@ -51,3 +51,9 @@ class ProviderV1(BaseModel):
     homepage: HttpUrl
     contact_email: EmailStr
     secondary_contact_email: EmailStr | None = None
+
+    # Status field to track provider state
+    status: ProviderStatusEnum = Field(
+        default=ProviderStatusEnum.active,
+        description="Provider status: active, disabled, or incomplete",
+    )

unitysvc_services/models/seller_v1.py

@@ -2,7 +2,7 @@ from datetime import datetime
 
 from pydantic import BaseModel, ConfigDict, EmailStr, Field, HttpUrl
 
-from unitysvc_services.models.base import Document, SellerTypeEnum
+from unitysvc_services.models.base import Document, SellerStatusEnum, SellerTypeEnum
 
 
 class SellerV1(BaseModel):
@@ -98,10 +98,10 @@ class SellerV1(BaseModel):
     # fields for business operation purposes
     #
 
-    # Status flags - these would typically be set by the backend
-    is_active: bool = Field(
-        default=True,
-        description="Whether the seller is active on the marketplace",
+    # Status field to track seller state
+    status: SellerStatusEnum = Field(
+        default=SellerStatusEnum.active,
+        description="Seller status: active, disabled, or incomplete",
     )
 
     is_verified: bool = Field(

unitysvc_services/populate.py

@@ -19,7 +19,7 @@ console = Console()
 def populate(
     data_dir: Path | None = typer.Argument(
         None,
-        help="Directory containing provider data files (default: ./data or UNITYSVC_DATA_DIR env var)",
+        help="Directory containing provider data files (default: current directory)",
     ),
     provider_name: str | None = typer.Option(
         None,
@@ -41,11 +41,7 @@ def populate(
     """
     # Set data directory
     if data_dir is None:
-        data_dir_str = os.getenv("UNITYSVC_DATA_DIR")
-        if data_dir_str:
-            data_dir = Path(data_dir_str)
-        else:
-            data_dir = Path.cwd() / "data"
+        data_dir = Path.cwd()
 
     if not data_dir.is_absolute():
         data_dir = Path.cwd() / data_dir