Function main

app/scripts/sunset_legacy_subscriptions.py:23–147

()

def main():
    parser = argparse.ArgumentParser(description='Sunset all legacy subscriptions')
    parser.add_argument(
        '--dry-run', action='store_true', help='Show what would be done without making changes'
    )
    parser.add_argument('--limit', type=int, default=100, help='Number of subscriptions to process per page')
    args = parser.parse_args()

    # Get environment variables
    stripe_secret_key = os.getenv('STRIPE_SECRET_KEY')
    current_price_id = os.getenv('STRIPE_SUBSCRIPTION_PRICE_ID')

    if not stripe_secret_key:
        logger.error("STRIPE_SECRET_KEY environment variable not set")
        sys.exit(1)

    if not current_price_id:
        logger.error("STRIPE_SUBSCRIPTION_PRICE_ID environment variable not set")
        sys.exit(1)

    stripe.api_key = stripe_secret_key

    logger.info(f"Current price ID: {current_price_id}")
    logger.info(f"Dry run: {args.dry_run}")

    sunset_count = 0
    already_cancelled_count = 0
    current_pricing_count = 0
    error_count = 0

    try:
        # Get all active subscriptions
        has_more = True
        starting_after = None

        while has_more:
            params = {
                'status': 'active',
                'limit': args.limit,
            }
            if starting_after:
                params['starting_after'] = starting_after

            subscriptions = stripe.Subscription.list(**params)

            for sub in subscriptions.data:
                try:
                    # Check if it's using the old price
                    is_legacy = True
                    legacy_price_id = None

                    for item in sub.get('items', {}).get('data', []):
                        price_id = item.get('price', {}).get('id')
                        if price_id == current_price_id:
                            is_legacy = False
                            break
                        else:
                            legacy_price_id = price_id