binalyze-air-sdk 1.0.2__py3-none-any.whl → 1.0.3__py3-none-any.whl

binalyze_air/models/users.py

@@ -1,82 +0,0 @@
-"""
-Users-related data models for the Binalyze AIR SDK.
-"""
-
-from typing import Optional, List, Dict, Any, Union
-from datetime import datetime
-from pydantic import Field
-
-from ..base import AIRBaseModel, Filter
-
-
-class User(AIRBaseModel):
-    """User model."""
-    
-    id: str = Field(alias="_id")
-    username: str
-    email: str
-    organization_ids: Optional[Union[List[int], str]] = Field(default=None, alias="organizationIds")
-    strategy: Optional[str] = None
-    profile: Optional[Dict[str, str]] = None
-    tfa_enabled: Optional[bool] = Field(default=False, alias="tfaEnabled")
-    first_name: Optional[str] = Field(default=None, alias="firstName")
-    last_name: Optional[str] = Field(default=None, alias="lastName")
-    organization_id: Optional[int] = Field(default=None, alias="organizationId")
-    role: Optional[str] = None
-    is_active: bool = Field(default=True, alias="isActive")
-    created_at: Optional[datetime] = Field(default=None, alias="createdAt")
-    updated_at: Optional[datetime] = Field(default=None, alias="updatedAt")
-
-
-class CreateUserRequest(AIRBaseModel):
-    """Create user request model."""
-    
-    username: str
-    email: str
-    password: str
-    firstName: Optional[str] = None
-    lastName: Optional[str] = None
-    organizationId: int
-    role: Optional[str] = None
-
-
-class UpdateUserRequest(AIRBaseModel):
-    """Update user request model."""
-    
-    username: Optional[str] = None
-    email: Optional[str] = None
-    firstName: Optional[str] = None
-    lastName: Optional[str] = None
-    role: Optional[str] = None
-    isActive: Optional[bool] = None
-
-
-class APIUser(AIRBaseModel):
-    """API user model."""
-    
-    id: str
-    name: str
-    description: Optional[str] = None
-    permissions: List[str] = []
-    organizationId: int
-    apiKey: Optional[str] = None
-    isActive: bool = True
-
-
-class CreateAPIUserRequest(AIRBaseModel):
-    """Create API user request model."""
-    
-    name: str
-    description: Optional[str] = None
-    permissions: List[str] = []
-    organizationId: int
-
-
-class UserFilter(Filter):
-    """Filter for user queries."""
-    
-    username: Optional[str] = None
-    email: Optional[str] = None
-    role: Optional[str] = None
-    organizationId: Optional[int] = None
-    isActive: Optional[bool] = None 

binalyze_air/queries/endpoints.py

@@ -1,25 +0,0 @@
-"""
-Endpoints queries for the Binalyze AIR SDK.
-"""
-
-from typing import List, Optional
-
-from ..base import Query
-from ..models.endpoints import EndpointTag, EndpointTagFilter
-from ..http_client import HTTPClient
-
-
-class GetEndpointTagsQuery(Query[List[EndpointTag]]):
-    """Query to get endpoint tags."""
-    
-    def __init__(self, http_client: HTTPClient, filter_params: Optional[EndpointTagFilter] = None):
-        self.http_client = http_client
-        self.filter_params = filter_params or EndpointTagFilter()
-    
-    def execute(self) -> List[EndpointTag]:
-        """Execute the query to get endpoint tags."""
-        params = self.filter_params.to_params()
-        response = self.http_client.get("endpoints/tags", params=params)
-        
-        entities = response.get("result", [])
-        return [EndpointTag(**item) for item in entities] 

binalyze_air/queries/users.py

@@ -1,69 +0,0 @@
-"""
-Users-related queries for the Binalyze AIR SDK.
-"""
-
-from typing import List, Optional, Dict, Any
-
-from ..base import Query
-from ..models.users import User, UserFilter
-from ..http_client import HTTPClient
-
-
-class ListUsersQuery(Query[List[User]]):
-    """Query to list users with pagination."""
-    
-    def __init__(self, http_client: HTTPClient, page_number: int = 1, page_size: int = 10,
-                 sort_by: str = "createdAt", sort_type: str = "ASC"):
-        self.http_client = http_client
-        self.page_number = page_number
-        self.page_size = page_size
-        self.sort_by = sort_by
-        self.sort_type = sort_type
-    
-    def execute(self) -> List[User]:
-        """Execute the list users query."""
-        params = {
-            "pageNumber": self.page_number,
-            "pageSize": self.page_size,
-            "sortBy": self.sort_by,
-            "sortType": self.sort_type,
-            "filter[organizationIds]": "0"  # Add required filter based on API spec
-        }
-        
-        # Use the correct endpoint path from API JSON files
-        response = self.http_client.get("user-management/users", params=params)
-        
-        # Extract entities from response and convert to User objects
-        entities = response.get("result", {}).get("entities", [])
-        
-        users = []
-        for entity_data in entities:
-            try:
-                user = User.model_validate(entity_data)
-                users.append(user)
-            except Exception as e:
-                # Log validation error but continue
-                print(f"Warning: Failed to validate user data: {e}")
-                continue
-        
-        return users
-
-
-class GetUserQuery(Query[User]):
-    """Query to get user by ID."""
-    
-    def __init__(self, http_client: HTTPClient, user_id: str):
-        self.http_client = http_client
-        self.user_id = user_id
-    
-    def execute(self) -> User:
-        """Execute the get user query."""
-        # Use the correct endpoint path from API JSON files
-        response = self.http_client.get(f"user-management/users/{self.user_id}")
-        
-        # Extract user data from response and convert to User object
-        entity_data = response.get("result", {})
-        
-        # Convert to User object using Pydantic model validation
-        user = User.model_validate(entity_data)
-        return user