Skip to content

Typing issue: CreateParams for StripeObjects should allow for Optional values #1287

Open
@tetsujiono

Description

@tetsujiono

Describe the bug

Many CreateParams attributes on StripeObjects are defined to be non-optional. This means that if that keyword argument is passed into the .create( function, it must be a non-null value (that also adheres to the defined typing).
However, the API supports passing null values on many of these attributes and that seems to contextually make sense.

One example is the promotion_code attribute on the Subscription CreateParams class.

The following call succeeds without issue:

import stripe

api_key = "***"  # loaded from env
api_version = "2023-10-16"  # At the time of writing this ticket, I am using this API version

customer_id = "cus_xxx"  # use any customer_id with a payment source in your own env
price_id = "price_xxx". # use an existing active price on a product in your stripe env
promotion_code = None

subscription = stripe.Subscription.create(
    customer=customer_id,
    items=[{"price": price_id}],
    payment_behavior="error_if_incomplete",
    promotion_code=promotion_code,
)

however, mypy will find issues with the fact that promotion_code is None and prefers it not to be passed at all.

The workaround at this current moment (other than ignoring the type issue) would be to add a conditional

if promotion_code:
    subscription = stripe.Subscription.create(
        customer=customer_id,
        items=[{"price": price_id}],
        payment_behavior="error_if_incomplete",
        promotion_code=promotion_code,
    )
else:
    subscription = stripe.Subscription.create(
        customer=customer_id,
        items=[{"price": price_id}],
        payment_behavior="error_if_incomplete",
    )

however, this is quite tedious and becomes unruly if you have more than one attribute you need to check (a permutation of checks will occur)

To Reproduce

  1. Write the follow or similar code
import stripe

api_key = "***"  # loaded from env
api_version = "2023-10-16"  # At the time of writing this ticket, I am using this API version

customer_id = "cus_xxx"  # use any customer_id with a payment source in your own env
price_id = "price_xxx". # use an existing active price on a product in your stripe env
promotion_code = None

subscription = stripe.Subscription.create(
    customer=customer_id,
    items=[{"price": price_id}],
    payment_behavior="error_if_incomplete",
    promotion_code=promotion_code,
)
  1. Run mypy checker
  2. You will see the following error: error: Argument "promotion_code" to "create" of "Subscription" has incompatible type "str | None"; expected "str" [arg-type]

Expected behavior

Using this line as an example, any nullable CreateParams attribute should be noted as Optional

Suggested Solution

promotion_code: NotRequired["str|None"]

^This solution most similarly reflects the way that Optional is implemented on the base RequestOptions class

Code snippets

Sample attribute to modify: https://github.com/stripe/stripe-python/blob/1657f036266412b7ef38d46c77f1b30f5455d83b/stripe/_subscription.py#L594
Existing way Optional is implemented: https://github.com/stripe/stripe-python/blob/1657f036266412b7ef38d46c77f1b30f5455d83b/stripe/_request_options.py#L7

OS

macOS

Language version

Python 3.11.3

Library version

stripe-python v8.9.0

API version

2023-10-16

Additional context

No response

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions