Stripe订阅：如何将账单周期固定为每月1号

import stripe
import datetime
import calendar
import os

# 设置您的Stripe密钥
# 建议从环境变量中获取，以确保安全
stripe.api_key = os.environ.get("STRIPE_SECRET_KEY", "YOUR_STRIPE_SECRET_KEY")

def create_monthly_anchored_subscription(customer_id, price_id):
    """
    创建Stripe订阅，并将其账单周期固定为每月1号。

    参数:
    customer_id (str): 客户ID
    price_id (str): 按月计费的价格ID

    返回:
    stripe.Subscription: 创建的订阅对象
    """
    # 计算下一个月的1号的Unix时间戳作为billing_cycle_anchor
    # 假设当前日期是2023年10月15日，我们希望从2023年11月1日开始账单

    today = datetime.date.today()

    # 计算下一个月的年份和月份
    if today.month == 12:
        next_month_year = today.year + 1
        next_month = 1
    else:
        next_month_year = today.year
        next_month = today.month + 1

    # 构建下一个月1号的日期对象
    first_day_of_next_month = datetime.datetime(next_month_year, next_month, 1, 0, 0, 0)

    # 将日期对象转换为Unix时间戳
    # 注意：Stripe期望的是秒级Unix时间戳
    billing_anchor_timestamp = int(first_day_of_next_month.timestamp())

    try:
        subscription = stripe.Subscription.create(
            customer=customer_id,
            items=[{"price": price_id}],
            # 设置账单周期锚点为下一个月1号
            billing_cycle_anchor=billing_anchor_timestamp,
            # 可选：如果希望立即开始计费，但账单日期固定，可以设置proration_behavior
            # 如果不设置，Stripe会根据锚点自动调整首次计费，可能包含按比例计费
            # proration_behavior='none' 表示不进行按比例计费，直接从锚点开始
            # proration_behavior='create_prorations' (默认) 会为从现在到锚点之间的时间段生成按比例的账单
        )
        print(f"订阅创建成功: {subscription.id}")
        print(f"账单周期锚点设置为: {datetime.datetime.fromtimestamp(subscription.billing_cycle_anchor)}")
        return subscription
    except stripe.error.StripeError as e:
        print(f"创建订阅失败: {e}")
        return None

# 示例用法 (请替换为您的实际客户ID和价格ID)
if __name__ == "__main__":
    # 假设您已经有了一个客户ID和一个按月计费的价格ID
    # customer_id_example = "cus_YOUR_CUSTOMER_ID"
    # price_id_example = "price_YOUR_PRICE_ID_MONTHLY"

    # 为了运行此示例，您可能需要创建一个测试客户和价格
    # 警告：以下代码会实际创建Stripe资源，请在测试环境中谨慎使用
    try:
        # 创建一个测试客户
        test_customer = stripe.Customer.create(
            email="test_user@example.com",
            description="Test Customer for Monthly Billing Anchor"
        )
        print(f"创建测试客户: {test_customer.id}")

        # 创建一个按月计费的测试价格
        test_product = stripe.Product.create(name="Pro Plan")
        test_price = stripe.Price.create(
            unit_amount=1000,  # 10.00 USD
            currency="usd",
            recurring={"interval": "month"},
            product=test_product.id,
        )
        print(f"创建测试价格: {test_price.id}")

        # 使用测试客户和价格创建订阅
        created_subscription = create_monthly_anchored_subscription(
            test_customer.id,
            test_price.id
        )

        if created_subscription:
            print(f"成功创建的订阅ID: {created_subscription.id}")
            print(f"下一次账单日期将是: {datetime.datetime.fromtimestamp(created_subscription.current_period_end)}")
            # 注意：current_period_end 表示当前计费周期的结束，下一个账单通常在此日期之后不久生成。
            # billing_cycle_anchor 才是固定每月1号的关键。

            # 清理（可选）：删除测试资源
            # stripe.Subscription.delete(created_subscription.id)
            # stripe.Customer.delete(test_customer.id)
            # stripe.Price.delete(test_price.id) # 价格通常不直接删除，而是归档
            # stripe.Product.delete(test_product.id)

    except stripe.error.StripeError as e:
        print(f"示例运行失败: {e}")
    except Exception as e:
        print(f"发生未知错误: {e}")