pydantic_extra_types_color_and_payment_card_validation.py

1from pydantic import BaseModel
2from pydantic_extra_types.color import Color
3from pydantic_extra_types.payment import PaymentCardNumber
4
5class User(BaseModel):
6    # Validates CSS colors (names, hex, rgb, etc.)
7    favorite_color: Color
8    # Validates payment card numbers using the Luhn algorithm
9    card_number: PaymentCardNumber
10
11# Example usage:
12user = User(
13    favorite_color='red',
14    card_number='4242 4242 4242 4242'
15)
16
17print(user.favorite_color.as_hex())
18# > #ff0000
19
20print(user.card_number.brand)
21# > PaymentCardBrand.visa
22
23print(user.model_dump())
24"""
25{
26    'favorite_color': Color('red', rgb=(255, 0, 0)),
27    'card_number': '4242424242424242'
28}
29"""