import stripe
import datetime
import calendar
import os

# 設置您的Stripe密鑰
# 建議從環(huán)境變量中獲取，以確保安全
stripe.api_key = os.environ.get("STRIPE_SECRET_KEY", "YOUR_STRIPE_SECRET_KEY")

def create_monthly_anchored_subscription(customer_id, price_id):
    """
    創(chuàng)建Stripe訂閱，并將其賬單周期固定為每月1號。

    參數(shù):
    customer_id (str): 客戶ID
    price_id (str): 按月計費的價格ID

    返回:
    stripe.Subscription: 創(chuàng)建的訂閱對象
    """
    # 計算下一個月的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會根據(jù)錨點自動調整首次計費，可能包含按比例計費
            # proration_behavior='none' 表示不進行按比例計費，直接從錨點開始
            # proration_behavior='create_prorations' (默認) 會為從現(xiàn)在到錨點之間的時間段生成按比例的賬單
        )
        print(f"訂閱創(chuàng)建成功: {subscription.id}")
        print(f"賬單周期錨點設置為: {datetime.datetime.fromtimestamp(subscription.billing_cycle_anchor)}")
        return subscription
    except stripe.error.StripeError as e:
        print(f"創(chuàng)建訂閱失敗: {e}")
        return None

# 示例用法 (請?zhí)鎿Q為您的實際客戶ID和價格ID)
if __name__ == "__main__":
    # 假設您已經(jīng)有了一個客戶ID和一個按月計費的價格ID
    # customer_id_example = "cus_YOUR_CUSTOMER_ID"
    # price_id_example = "price_YOUR_PRICE_ID_MONTHLY"

    # 為了運行此示例，您可能需要創(chuàng)建一個測試客戶和價格
    # 警告：以下代碼會實際創(chuàng)建Stripe資源，請在測試環(huán)境中謹慎使用
    try:
        # 創(chuàng)建一個測試客戶
        test_customer = stripe.Customer.create(
            email="test_user@example.com",
            description="Test Customer for Monthly Billing Anchor"
        )
        print(f"創(chuàng)建測試客戶: {test_customer.id}")

        # 創(chuàng)建一個按月計費的測試價格
        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"創(chuàng)建測試價格: {test_price.id}")

        # 使用測試客戶和價格創(chuàng)建訂閱
        created_subscription = create_monthly_anchored_subscription(
            test_customer.id,
            test_price.id
        )

        if created_subscription:
            print(f"成功創(chuàng)建的訂閱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"發(fā)生未知錯誤: {e}")