ai-lls-lib 1.2.0__py3-none-any.whl → 1.3.1__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.2.0"
+__version__ = "1.3.1"
 __all__ = [
     "PhoneVerification",
     "BulkJob",

ai_lls_lib/cli/commands/stripe.py

@@ -37,6 +37,32 @@ def seed_products(environment: str, api_key: Optional[str], dry_run: bool):
 
     # Define the products and prices to create
     products_config = [
+        {
+            "name": "Landline Scrubber - Variable",
+            "description": "Pay as you go - choose your amount",
+            "metadata": {
+                "product_type": "landline_scrubber",
+                "environment": environment,
+                "tier": "VARIABLE"
+            },
+            "price": {
+                "unit_amount": 500,  # $5.00 minimum
+                "currency": "usd",
+                "metadata": {
+                    "product_type": "landline_scrubber",
+                    "environment": environment,
+                    "plan_type": "prepaid",
+                    "tier": "VARIABLE",
+                    "variable_amount": "true",
+                    "min_amount": "5",
+                    "default_amounts": "5,10,50",
+                    "credits_per_dollar": "285",  # ~$0.00175 per credit
+                    "plan_credits_text": "Variable credits",
+                    "percent_off": "",
+                    "active": "true"
+                }
+            }
+        },
         {
             "name": "Landline Scrubber - STANDARD",
             "description": "One-time purchase",
@@ -53,7 +79,6 @@ def seed_products(environment: str, api_key: Optional[str], dry_run: bool):
                     "environment": environment,
                     "plan_type": "prepaid",
                     "tier": "STANDARD",
-                    "plan_reference": "79541679412215",  # Legacy ID for compatibility
                     "credits": "5000",
                     "plan_credits_text": "5,000 credits",
                     "percent_off": "",
@@ -77,7 +102,6 @@ def seed_products(environment: str, api_key: Optional[str], dry_run: bool):
                     "environment": environment,
                     "plan_type": "prepaid",
                     "tier": "POWER",
-                    "plan_reference": "79541679412216",  # Legacy ID for compatibility
                     "credits": "28500",
                     "plan_credits_text": "28,500 credits",
                     "percent_off": "12.5% OFF",
@@ -101,7 +125,6 @@ def seed_products(environment: str, api_key: Optional[str], dry_run: bool):
                     "environment": environment,
                     "plan_type": "prepaid",
                     "tier": "ELITE",
-                    "plan_reference": "79541679412217",  # Legacy ID for compatibility
                     "credits": "66666",
                     "plan_credits_text": "66,666 credits",
                     "percent_off": "25% OFF",
@@ -126,7 +149,6 @@ def seed_products(environment: str, api_key: Optional[str], dry_run: bool):
                     "environment": environment,
                     "plan_type": "postpaid",
                     "tier": "UNLIMITED",
-                    "plan_reference": "price_unlimited",
                     "credits": "unlimited",
                     "plan_credits_text": "Unlimited",
                     "percent_off": "",

ai_lls_lib/payment/credit_manager.py

@@ -172,3 +172,22 @@ class CreditManager:
         """Check if user has unlimited access via active subscription."""
         info = self.get_user_payment_info(user_id)
         return info.get("subscription_status") == "active"
+
+    def set_stripe_customer_id(self, user_id: str, stripe_customer_id: str) -> None:
+        """Store Stripe customer ID for a user."""
+        if not self.table:
+            return  # Mock for testing
+
+        try:
+            self.table.update_item(
+                Key={"user_id": user_id},
+                UpdateExpression="SET stripe_customer_id = :customer_id, updated_at = :now",
+                ExpressionAttributeValues={
+                    ":customer_id": stripe_customer_id,
+                    ":now": datetime.utcnow().isoformat()
+                }
+            )
+            logger.info(f"Stored Stripe customer ID {stripe_customer_id} for user {user_id}")
+        except ClientError as e:
+            logger.error(f"Error storing Stripe customer ID for {user_id}: {e}")
+            raise

ai_lls_lib/payment/stripe_manager.py

@@ -97,7 +97,8 @@ class StripeManager:
 
             return {
                 "client_secret": setup_intent.client_secret,
-                "setup_intent_id": setup_intent.id
+                "setup_intent_id": setup_intent.id,
+                "customer_id": customer.id
             }
 
         except stripe.error.StripeError as e:
@@ -186,49 +187,62 @@ class StripeManager:
     def charge_prepaid(self, user_id: str, reference_code: str, amount: Optional[float] = None) -> Dict[str, Any]:
         """
         Charge saved payment method for credit purchase.
-        Supports both fixed-price and legacy variable-amount plans.
+        Supports both fixed-price and metadata-based variable-amount plans.
         """
         try:
             customer = self._get_or_create_customer(user_id)
 
-            # Legacy variable-amount plan references
-            VARIABLE_AMOUNT_REFS = ["79541679412215", "79541679412216"]
+            # Look up price from Stripe
+            prices = stripe.Price.list(active=True, limit=100, expand=["data.product"])
+            price = None
+
+            for p in prices.data:
+                metadata = p.metadata or {}
+                # Match by price ID or plan_reference in metadata
+                if (p.id == reference_code or
+                    metadata.get("plan_reference") == reference_code or
+                    (metadata.get("tier") == reference_code and
+                     metadata.get("environment") == self.environment)):
+                    price = p
+                    break
+
+            if not price:
+                raise ValueError(f"Invalid plan reference: {reference_code}")
 
-            if reference_code in VARIABLE_AMOUNT_REFS:
+            price_metadata = price.metadata or {}
+
+            # Check if this is a variable amount plan
+            if price_metadata.get("variable_amount") == "true":
                 # Variable amount plan - validate amount
                 if not amount:
                     raise ValueError("Amount required for variable-amount plan")
 
-                # Validate amount against allowed values from metadata
-                # For now, allow common amounts
-                allowed_amounts = [5, 10, 20, 25, 50, 75, 100, 150, 200, 250, 500]
-                if amount not in allowed_amounts:
-                    raise ValueError(f"Invalid amount ${amount}. Allowed amounts: {allowed_amounts}")
-
-                # Calculate credits based on tier
-                if reference_code == "79541679412215":  # STANDARD
-                    credits_per_dollar = 500
-                else:  # POWER
-                    credits_per_dollar = 570
-
+                # Get validation rules from metadata
+                min_amount = float(price_metadata.get("min_amount", "5"))
+                if amount < min_amount:
+                    raise ValueError(f"Amount ${amount} is below minimum ${min_amount}")
+
+                # Check against default amounts if specified
+                default_amounts_str = price_metadata.get("default_amounts", "")
+                if default_amounts_str:
+                    allowed_amounts = [float(x.strip()) for x in default_amounts_str.split(",")]
+                    # Allow default amounts OR any amount >= minimum
+                    if amount not in allowed_amounts and amount < max(allowed_amounts):
+                        logger.info(f"Amount ${amount} not in defaults {allowed_amounts}, but allowed as >= ${min_amount}")
+
+                # Calculate credits based on credits_per_dollar
+                credits_per_dollar = float(price_metadata.get("credits_per_dollar", "285"))
                 credits_to_add = int(amount * credits_per_dollar)
                 charge_amount = int(amount * 100)  # Convert to cents
 
             else:
-                # Fixed price plan - look up from Stripe
-                prices = stripe.Price.list(active=True, limit=100)
-                price = None
-
-                for p in prices.data:
-                    if p.id == reference_code or p.metadata.get("plan_reference") == reference_code:
-                        price = p
-                        break
-
-                if not price:
-                    raise ValueError(f"Invalid plan reference: {reference_code}")
-
+                # Fixed price plan
                 charge_amount = price.unit_amount
-                credits_to_add = int(price.metadata.get("credits", 0))
+                credits_str = price_metadata.get("credits", "0")
+                if credits_str.lower() == "unlimited":
+                    credits_to_add = 0  # Subscription handles this differently
+                else:
+                    credits_to_add = int(credits_str)
 
             # Get default payment method
             default_pm = customer.invoice_settings.get("default_payment_method")
@@ -431,7 +445,7 @@ class StripeManager:
         """Return mock plans for development/testing."""
         return [
             Plan(
-                plan_reference="79541679412215",
+                plan_reference="price_standard_mock",
                 plan_type="prepaid",
                 plan_name="STANDARD",
                 plan_subtitle="One-time purchase",
@@ -441,7 +455,7 @@ class StripeManager:
                 percent_off=""
             ),
             Plan(
-                plan_reference="79541679412216",
+                plan_reference="price_power_mock",
                 plan_type="prepaid",
                 plan_name="POWER",
                 plan_subtitle="Best value",
@@ -451,7 +465,7 @@ class StripeManager:
                 percent_off="12.5% OFF"
             ),
             Plan(
-                plan_reference="79541679412217",
+                plan_reference="price_elite_mock",
                 plan_type="prepaid",
                 plan_name="ELITE",
                 plan_subtitle="Maximum savings",
@@ -461,7 +475,7 @@ class StripeManager:
                 percent_off="25% OFF"
             ),
             Plan(
-                plan_reference="price_unlimited",
+                plan_reference="price_unlimited_mock",
                 plan_type="postpaid",
                 plan_name="UNLIMITED",
                 plan_subtitle="Monthly subscription",

ai_lls_lib/payment/webhook_processor.py

@@ -49,7 +49,10 @@ class WebhookProcessor:
 
         logger.info(f"Processing webhook event: {event_type}")
 
-        if event_type == "checkout.session.completed":
+        if event_type == "payment_intent.succeeded":
+            return self._handle_payment_intent_succeeded(event_data)
+
+        elif event_type == "checkout.session.completed":
             return self._handle_checkout_completed(event_data)
 
         elif event_type == "customer.subscription.created":
@@ -67,6 +70,9 @@ class WebhookProcessor:
         elif event_type == "invoice.payment_failed":
             return self._handle_invoice_failed(event_data)
 
+        elif event_type == "charge.dispute.created":
+            return self._handle_dispute_created(event_data)
+
         else:
             logger.info(f"Unhandled event type: {event_type}")
             return {"message": f"Event {event_type} received but not processed"}
@@ -161,3 +167,49 @@ class WebhookProcessor:
         logger.warning(f"Invoice payment failed for customer {customer_id}")
         # Could pause subscription or send notification here
         return {"status": "payment_failed"}
+
+    def _handle_payment_intent_succeeded(self, payment_intent: Dict[str, Any]) -> Dict[str, Any]:
+        """Handle successful payment intent (credit purchase)."""
+        metadata = payment_intent.get("metadata", {})
+        user_id = metadata.get("user_id")
+
+        if not user_id:
+            logger.error("No user_id in payment_intent metadata")
+            return {"error": "Missing user_id"}
+
+        # Check if this is a verification charge ($1)
+        if metadata.get("type") == "verification":
+            # This was the $1 verification, credits already added in payment_setup handler
+            logger.info(f"Verification charge completed for user {user_id}")
+            return {"type": "verification", "status": "completed"}
+
+        # Get credits from metadata (set during payment creation)
+        credits = int(metadata.get("credits", 0))
+
+        if credits > 0:
+            new_balance = self.credit_manager.add_credits(user_id, credits)
+            logger.info(f"Added {credits} credits to user {user_id}, new balance: {new_balance}")
+            return {"credits_added": credits, "new_balance": new_balance}
+
+        return {"message": "Payment processed"}
+
+    def _handle_dispute_created(self, dispute: Dict[str, Any]) -> Dict[str, Any]:
+        """Handle charge dispute (mark account as disputed)."""
+        # Get the charge and its metadata
+        charge_id = dispute.get("charge")
+
+        if not charge_id:
+            logger.error("No charge_id in dispute")
+            return {"error": "Missing charge_id"}
+
+        # In production, would fetch the charge from Stripe to get metadata
+        # For now, log the dispute for manual handling
+        amount = dispute.get("amount", 0) / 100.0
+        reason = dispute.get("reason", "unknown")
+
+        logger.warning(f"Dispute created for charge {charge_id}: ${amount}, reason: {reason}")
+
+        # TODO: Mark user account as disputed in CreditsTable
+        # This would prevent new purchases until resolved
+
+        return {"dispute_id": dispute.get("id"), "status": "created", "amount": amount}

{ai_lls_lib-1.2.0.dist-info → ai_lls_lib-1.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: ai-lls-lib
-Version: 1.2.0
+Version: 1.3.1
 Summary: Landline Scrubber core library - phone verification and DNC checking
 Author: LandlineScrubber Team
 Requires-Python: >=3.12,<4.0

{ai_lls_lib-1.2.0.dist-info → ai_lls_lib-1.3.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ai_lls_lib/__init__.py,sha256=vX28RZKgWt_4PDkQ_H69AwCCp7Jt9nHwhxt4RhKOotU,584
+ai_lls_lib/__init__.py,sha256=nkkaFByPHCiQNRa0kvk3DVE4Yh8eicP2KHfuSuE4tlc,584
 ai_lls_lib/auth/__init__.py,sha256=c6zomHSB6y9Seakf84ciGsD3XgWarIty9xty6P8fxVw,194
 ai_lls_lib/auth/context_parser.py,sha256=8I0vGbtykNLWqm8ldedxXjE-E3nqsCy113JgeyuiJoM,2222
 ai_lls_lib/cli/__init__.py,sha256=m9qjZTW1jpENwXAUeuRrlP0b66BWRcqSO28MSjvOyCs,74
@@ -7,7 +7,7 @@ ai_lls_lib/cli/aws_client.py,sha256=YcCWCpTNOW9JPLxSNLRy5-F5HPKguJJk7dPNrPqhJv0,
 ai_lls_lib/cli/commands/__init__.py,sha256=_kROrYuR_p2i110c0OvNeArfEFQbn15zR1c3pdeZOoo,28
 ai_lls_lib/cli/commands/admin.py,sha256=bNBJi2fZBP0J40JQP6HP7NYadNmI214iII1TeLhooyE,6687
 ai_lls_lib/cli/commands/cache.py,sha256=vWt0vy4L9CEgUEWUzfdehU6u43PE8vUvHx7xxg4e5tw,5680
-ai_lls_lib/cli/commands/stripe.py,sha256=x5QtirbCPiBrjdMRVTR0aeQN2gy0_p6jcGzELObFfzg,12006
+ai_lls_lib/cli/commands/stripe.py,sha256=Iccp8ZmNE18q9rvoox-Mo7dZgByEhIVP7otRklgr4uI,12707
 ai_lls_lib/cli/commands/test_stack.py,sha256=rNq4mhRXX7Ixo67kSoEPWlxqgXCzM9e2PR96qTAG7pE,7378
 ai_lls_lib/cli/commands/verify.py,sha256=V5ucjmjCUxqN8_AeEJWWgrmYin8BNV3h4WbZ-iX3loU,4238
 ai_lls_lib/cli/env_loader.py,sha256=YTCB6QDyknOuedPbQsW4bezB5SgHzJMGjhpSPArHFVc,3762
@@ -17,17 +17,17 @@ ai_lls_lib/core/models.py,sha256=ABRYaeMCWahQh4WdbkCxt3AO0-EvPWZwlL-JITQSnEM,238
 ai_lls_lib/core/processor.py,sha256=6752IPDQ-Mz5i_CU7aBM0UjvV7IjyZFl35LKVPkHMpc,9974
 ai_lls_lib/core/verifier.py,sha256=6uB_jawWoIqsNYAadTKr0lSolIWygg2gK6ykf8lrul0,2716
 ai_lls_lib/payment/__init__.py,sha256=xhUWgfLnk3syXXQItswmDXdfXUyJTXTQAA0zIUuCVII,295
-ai_lls_lib/payment/credit_manager.py,sha256=TDrdW7BYjhtlg8MX_EJEnMNct1SU_6Tdb_k19_e8kpo,6343
+ai_lls_lib/payment/credit_manager.py,sha256=ynjweRkbdHI-A6fROUoqlazNDmXOugXsIaMco7k62S0,7147
 ai_lls_lib/payment/models.py,sha256=JjSmWKwpuFF85Jzmabj6y7UyolJlJlsh5CmOWRg21B8,3339
-ai_lls_lib/payment/stripe_manager.py,sha256=-L329stuu-6V_QKzRdkm9qnoi74SbWzST0fuf2WAJXA,17111
-ai_lls_lib/payment/webhook_processor.py,sha256=UVngeAGcou5ieRAp-49pqdWh0wsJQVwpiRoLmyl5Adc,6448
+ai_lls_lib/payment/stripe_manager.py,sha256=R_M4Bii9BGianL_STqvz4UU8ww8mlCByGo66V-e2C6Y,18027
+ai_lls_lib/payment/webhook_processor.py,sha256=cIZqCS98Q305SCI3-4AJ1h-IJ-l-7fMby1FFy1ckP6k,8765
 ai_lls_lib/providers/__init__.py,sha256=AEv3ARenWDwDo5PLCoszP2fQ70RgSHkrSLSUz7xHJDk,179
 ai_lls_lib/providers/base.py,sha256=344XYOg7bxDQMWJ6Lle8U7NOHpabnCp0XYbZpeWpPAk,681
 ai_lls_lib/providers/external.py,sha256=-Hhnlm8lQWRcWr5vG0dmD3sca2rohURrx0usOR2y1MM,2623
 ai_lls_lib/providers/stub.py,sha256=847Tmw522B3HQ2j38BH1sdcZQy--RdtDcXsrIrFKNBQ,1150
 ai_lls_lib/testing/__init__.py,sha256=RUxRYBzzPCPS15Umb6bUrE6rL5BQXBQf4SJM2E3ffrg,39
 ai_lls_lib/testing/fixtures.py,sha256=_n6bbr95LnQf9Dvu1qKs2HsvHEA7AAbe59B75qxE10w,3310
-ai_lls_lib-1.2.0.dist-info/METADATA,sha256=o76nenDApQ5pITFYnlmX7IS-maEi39JgU75iUyI9udk,7248
-ai_lls_lib-1.2.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-ai_lls_lib-1.2.0.dist-info/entry_points.txt,sha256=Pi0V_HBViEKGFbNQKatl5lhhnHHBXlxaom-5gH9gXZ0,55
-ai_lls_lib-1.2.0.dist-info/RECORD,,
+ai_lls_lib-1.3.1.dist-info/METADATA,sha256=AxasNt6DhVulTID8lw_C7FFuDCJMs_dPLFicO7fPK6s,7248
+ai_lls_lib-1.3.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+ai_lls_lib-1.3.1.dist-info/entry_points.txt,sha256=Pi0V_HBViEKGFbNQKatl5lhhnHHBXlxaom-5gH9gXZ0,55
+ai_lls_lib-1.3.1.dist-info/RECORD,,