ai-lls-lib 1.4.0rc2__py3-none-any.whl → 1.4.0rc4__py3-none-any.whl

ai_lls_lib/__init__.py

@@ -13,7 +13,7 @@ from ai_lls_lib.core.verifier import PhoneVerifier
 from ai_lls_lib.core.processor import BulkProcessor
 from ai_lls_lib.core.cache import DynamoDBCache
 
-__version__ = "1.4.0-rc.2"
+__version__ = "1.4.0-rc.4"
 __all__ = [
     "PhoneVerification",
     "BulkJob",

ai_lls_lib/auth/__init__.py

@@ -1,4 +1,4 @@
-"""Auth module for handling authentication and authorization."""
-from .context_parser import get_email_from_event, get_user_from_event
-
-__all__ = ["get_user_from_event", "get_email_from_event"]
+"""Auth module for handling authentication and authorization."""
+from .context_parser import get_email_from_event, get_user_from_event
+
+__all__ = ["get_user_from_event", "get_email_from_event"]

ai_lls_lib/auth/context_parser.py

@@ -1,68 +1,68 @@
-"""Auth context parser for HTTP API v2.0 events."""
-from typing import Any, Dict, Optional
-
-
-def get_user_from_event(event: Dict[str, Any]) -> Optional[str]:
-    """
-    Extract user ID from HTTP API v2.0 event with all possible paths.
-    Handles both JWT and API key authentication contexts.
-
-    This function handles the complexities of AWS API Gateway authorizer contexts,
-    especially when EnableSimpleResponses is set to false, which wraps the
-    context in a 'lambda' key.
-
-    Args:
-        event: The Lambda event from API Gateway HTTP API v2.0
-
-    Returns:
-        User ID string if found, None otherwise
-    """
-    request_context = event.get("requestContext", {})
-    auth = request_context.get("authorizer", {})
-
-    # Handle lambda-wrapped context (EnableSimpleResponses: false)
-    # When EnableSimpleResponses is false, the authorizer context is wrapped
-    lam_ctx = auth.get("lambda", auth) if isinstance(auth.get("lambda"), dict) else auth
-
-    # Try all possible paths for user_id in priority order
-    user_id = (
-        # Lambda authorizer paths (most common with current setup)
-        lam_ctx.get("principal_id") or
-        lam_ctx.get("principalId") or
-        lam_ctx.get("sub") or
-        lam_ctx.get("user_id") or
-        # JWT paths (when using JWT authorizer directly)
-        auth.get("jwt", {}).get("claims", {}).get("sub") or
-        # Direct auth paths (fallback)
-        auth.get("principal_id") or
-        auth.get("principalId") or
-        auth.get("sub")
-    )
-
-    return user_id
-
-
-def get_email_from_event(event: Dict[str, Any]) -> Optional[str]:
-    """
-    Extract email from HTTP API v2.0 event.
-
-    Args:
-        event: The Lambda event from API Gateway HTTP API v2.0
-
-    Returns:
-        Email string if found, None otherwise
-    """
-    request_context = event.get("requestContext", {})
-    auth = request_context.get("authorizer", {})
-
-    # Handle lambda-wrapped context
-    lam_ctx = auth.get("lambda", auth) if isinstance(auth.get("lambda"), dict) else auth
-
-    # Try to get email from various locations
-    email = (
-        lam_ctx.get("email") or
-        auth.get("jwt", {}).get("claims", {}).get("email") or
-        auth.get("email")
-    )
-
-    return email
+"""Auth context parser for HTTP API v2.0 events."""
+from typing import Any, Dict, Optional
+
+
+def get_user_from_event(event: Dict[str, Any]) -> Optional[str]:
+    """
+    Extract user ID from HTTP API v2.0 event with all possible paths.
+    Handles both JWT and API key authentication contexts.
+
+    This function handles the complexities of AWS API Gateway authorizer contexts,
+    especially when EnableSimpleResponses is set to false, which wraps the
+    context in a 'lambda' key.
+
+    Args:
+        event: The Lambda event from API Gateway HTTP API v2.0
+
+    Returns:
+        User ID string if found, None otherwise
+    """
+    request_context = event.get("requestContext", {})
+    auth = request_context.get("authorizer", {})
+
+    # Handle lambda-wrapped context (EnableSimpleResponses: false)
+    # When EnableSimpleResponses is false, the authorizer context is wrapped
+    lam_ctx = auth.get("lambda", auth) if isinstance(auth.get("lambda"), dict) else auth
+
+    # Try all possible paths for user_id in priority order
+    user_id = (
+        # Lambda authorizer paths (most common with current setup)
+        lam_ctx.get("principal_id") or
+        lam_ctx.get("principalId") or
+        lam_ctx.get("sub") or
+        lam_ctx.get("user_id") or
+        # JWT paths (when using JWT authorizer directly)
+        auth.get("jwt", {}).get("claims", {}).get("sub") or
+        # Direct auth paths (fallback)
+        auth.get("principal_id") or
+        auth.get("principalId") or
+        auth.get("sub")
+    )
+
+    return user_id
+
+
+def get_email_from_event(event: Dict[str, Any]) -> Optional[str]:
+    """
+    Extract email from HTTP API v2.0 event.
+
+    Args:
+        event: The Lambda event from API Gateway HTTP API v2.0
+
+    Returns:
+        Email string if found, None otherwise
+    """
+    request_context = event.get("requestContext", {})
+    auth = request_context.get("authorizer", {})
+
+    # Handle lambda-wrapped context
+    lam_ctx = auth.get("lambda", auth) if isinstance(auth.get("lambda"), dict) else auth
+
+    # Try to get email from various locations
+    email = (
+        lam_ctx.get("email") or
+        auth.get("jwt", {}).get("claims", {}).get("email") or
+        auth.get("email")
+    )
+
+    return email

ai_lls_lib/cli/__init__.py

@@ -1,3 +1,3 @@
-"""
-Landline Scrubber CLI - Infrastructure-aware administrative tools
-"""
+"""
+Landline Scrubber CLI - Infrastructure-aware administrative tools
+"""

ai_lls_lib/cli/__main__.py

@@ -1,30 +1,30 @@
-"""
-Landline Scrubber CLI entry point
-"""
-import click
-import sys
-from ai_lls_lib.cli.commands import verify, cache, admin, test_stack, stripe
-
-@click.group()
-@click.version_option(version="0.1.0", prog_name="ai-lls")
-def cli():
-    """Landline Scrubber CLI - Administrative and debugging tools"""
-    pass
-
-# Register command groups
-cli.add_command(verify.verify_group)
-cli.add_command(cache.cache_group)
-cli.add_command(admin.admin_group)
-cli.add_command(test_stack.test_stack_group)
-cli.add_command(stripe.stripe_group)
-
-def main():
-    """Main entry point"""
-    try:
-        cli()
-    except Exception as e:
-        click.echo(f"Error: {e}", err=True)
-        sys.exit(1)
-
-if __name__ == "__main__":
-    main()
+"""
+Landline Scrubber CLI entry point
+"""
+import click
+import sys
+from ai_lls_lib.cli.commands import verify, cache, admin, test_stack, stripe
+
+@click.group()
+@click.version_option(version="0.1.0", prog_name="ai-lls")
+def cli():
+    """Landline Scrubber CLI - Administrative and debugging tools"""
+    pass
+
+# Register command groups
+cli.add_command(verify.verify_group)
+cli.add_command(cache.cache_group)
+cli.add_command(admin.admin_group)
+cli.add_command(test_stack.test_stack_group)
+cli.add_command(stripe.stripe_group)
+
+def main():
+    """Main entry point"""
+    try:
+        cli()
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+
+if __name__ == "__main__":
+    main()

ai_lls_lib/cli/aws_client.py

@@ -1,115 +1,115 @@
-"""
-AWS client utilities for CLI operations
-"""
-import os
-import boto3
-from typing import Optional, Any, Dict
-from botocore.exceptions import ClientError
-import click
-
-class AWSClient:
-    """Wrapper for AWS operations with proper error handling"""
-
-    def __init__(self, region: Optional[str] = None, profile: Optional[str] = None):
-        """Initialize AWS clients"""
-        self.region = region or os.environ.get("AWS_REGION", "us-east-1")
-        self.profile = profile
-
-        # Create session
-        if profile:
-            self.session = boto3.Session(profile_name=profile, region_name=self.region)
-        else:
-            self.session = boto3.Session(region_name=self.region)
-
-        # Lazy-load clients
-        self._dynamodb = None
-        self._s3 = None
-        self._sqs = None
-        self._secretsmanager = None
-        self._cloudformation = None
-
-    @property
-    def dynamodb(self):
-        """Get DynamoDB client"""
-        if not self._dynamodb:
-            self._dynamodb = self.session.resource('dynamodb')
-        return self._dynamodb
-
-    @property
-    def s3(self):
-        """Get S3 client"""
-        if not self._s3:
-            self._s3 = self.session.client('s3')
-        return self._s3
-
-    @property
-    def sqs(self):
-        """Get SQS client"""
-        if not self._sqs:
-            self._sqs = self.session.client('sqs')
-        return self._sqs
-
-    @property
-    def secretsmanager(self):
-        """Get Secrets Manager client"""
-        if not self._secretsmanager:
-            self._secretsmanager = self.session.client('secretsmanager')
-        return self._secretsmanager
-
-    @property
-    def cloudformation(self):
-        """Get CloudFormation client"""
-        if not self._cloudformation:
-            self._cloudformation = self.session.client('cloudformation')
-        return self._cloudformation
-
-    def get_stack_outputs(self, stack_name: str) -> Dict[str, str]:
-        """Get CloudFormation stack outputs"""
-        try:
-            response = self.cloudformation.describe_stacks(StackName=stack_name)
-            stack = response['Stacks'][0]
-            outputs = {}
-            for output in stack.get('Outputs', []):
-                outputs[output['OutputKey']] = output['OutputValue']
-            return outputs
-        except ClientError as e:
-            if e.response['Error']['Code'] == 'ValidationError':
-                raise click.ClickException(f"Stack '{stack_name}' not found")
-            raise
-
-    def get_table_name(self, stack_name: str, logical_name: str) -> str:
-        """Get actual table name from stack"""
-        try:
-            response = self.cloudformation.describe_stack_resource(
-                StackName=stack_name,
-                LogicalResourceId=logical_name
-            )
-            return response['StackResourceDetail']['PhysicalResourceId']
-        except ClientError:
-            # Fallback to conventional naming
-            return f"{stack_name}-{logical_name.lower()}"
-
-    def scan_table(self, table_name: str, limit: int = 100) -> list:
-        """Scan DynamoDB table"""
-        try:
-            table = self.dynamodb.Table(table_name)
-            response = table.scan(Limit=limit)
-            return response.get('Items', [])
-        except ClientError as e:
-            raise click.ClickException(f"Error scanning table: {e}")
-
-    def put_item(self, table_name: str, item: dict) -> None:
-        """Put item to DynamoDB"""
-        try:
-            table = self.dynamodb.Table(table_name)
-            table.put_item(Item=item)
-        except ClientError as e:
-            raise click.ClickException(f"Error putting item: {e}")
-
-    def delete_item(self, table_name: str, key: dict) -> None:
-        """Delete item from DynamoDB"""
-        try:
-            table = self.dynamodb.Table(table_name)
-            table.delete_item(Key=key)
-        except ClientError as e:
-            raise click.ClickException(f"Error deleting item: {e}")
+"""
+AWS client utilities for CLI operations
+"""
+import os
+import boto3
+from typing import Optional, Any, Dict
+from botocore.exceptions import ClientError
+import click
+
+class AWSClient:
+    """Wrapper for AWS operations with proper error handling"""
+
+    def __init__(self, region: Optional[str] = None, profile: Optional[str] = None):
+        """Initialize AWS clients"""
+        self.region = region or os.environ.get("AWS_REGION", "us-east-1")
+        self.profile = profile
+
+        # Create session
+        if profile:
+            self.session = boto3.Session(profile_name=profile, region_name=self.region)
+        else:
+            self.session = boto3.Session(region_name=self.region)
+
+        # Lazy-load clients
+        self._dynamodb = None
+        self._s3 = None
+        self._sqs = None
+        self._secretsmanager = None
+        self._cloudformation = None
+
+    @property
+    def dynamodb(self):
+        """Get DynamoDB client"""
+        if not self._dynamodb:
+            self._dynamodb = self.session.resource('dynamodb')
+        return self._dynamodb
+
+    @property
+    def s3(self):
+        """Get S3 client"""
+        if not self._s3:
+            self._s3 = self.session.client('s3')
+        return self._s3
+
+    @property
+    def sqs(self):
+        """Get SQS client"""
+        if not self._sqs:
+            self._sqs = self.session.client('sqs')
+        return self._sqs
+
+    @property
+    def secretsmanager(self):
+        """Get Secrets Manager client"""
+        if not self._secretsmanager:
+            self._secretsmanager = self.session.client('secretsmanager')
+        return self._secretsmanager
+
+    @property
+    def cloudformation(self):
+        """Get CloudFormation client"""
+        if not self._cloudformation:
+            self._cloudformation = self.session.client('cloudformation')
+        return self._cloudformation
+
+    def get_stack_outputs(self, stack_name: str) -> Dict[str, str]:
+        """Get CloudFormation stack outputs"""
+        try:
+            response = self.cloudformation.describe_stacks(StackName=stack_name)
+            stack = response['Stacks'][0]
+            outputs = {}
+            for output in stack.get('Outputs', []):
+                outputs[output['OutputKey']] = output['OutputValue']
+            return outputs
+        except ClientError as e:
+            if e.response['Error']['Code'] == 'ValidationError':
+                raise click.ClickException(f"Stack '{stack_name}' not found")
+            raise
+
+    def get_table_name(self, stack_name: str, logical_name: str) -> str:
+        """Get actual table name from stack"""
+        try:
+            response = self.cloudformation.describe_stack_resource(
+                StackName=stack_name,
+                LogicalResourceId=logical_name
+            )
+            return response['StackResourceDetail']['PhysicalResourceId']
+        except ClientError:
+            # Fallback to conventional naming
+            return f"{stack_name}-{logical_name.lower()}"
+
+    def scan_table(self, table_name: str, limit: int = 100) -> list:
+        """Scan DynamoDB table"""
+        try:
+            table = self.dynamodb.Table(table_name)
+            response = table.scan(Limit=limit)
+            return response.get('Items', [])
+        except ClientError as e:
+            raise click.ClickException(f"Error scanning table: {e}")
+
+    def put_item(self, table_name: str, item: dict) -> None:
+        """Put item to DynamoDB"""
+        try:
+            table = self.dynamodb.Table(table_name)
+            table.put_item(Item=item)
+        except ClientError as e:
+            raise click.ClickException(f"Error putting item: {e}")
+
+    def delete_item(self, table_name: str, key: dict) -> None:
+        """Delete item from DynamoDB"""
+        try:
+            table = self.dynamodb.Table(table_name)
+            table.delete_item(Key=key)
+        except ClientError as e:
+            raise click.ClickException(f"Error deleting item: {e}")

ai_lls_lib/cli/commands/__init__.py

@@ -1,3 +1,3 @@
-"""
-CLI command modules
-"""
+"""
+CLI command modules
+"""