Lewati ke isi

Rahasia

!!! peringatan "🚧 Pekerjaan Sedang Berlangsung" Halaman ini sedang dalam proses.

Buat serial SecretStr dan SecretBytes sebagai teks biasa

Secara default, SecretStr dan SecretBytes akan diserialkan sebagai ********** saat membuat serial ke json.

Anda dapat menggunakan [field_serializer][pydantic.function_serializers.field_serializer] untuk membuang rahasia sebagai teks biasa saat membuat serial ke json.

from pydantic import BaseModel, SecretBytes, SecretStr, field_serializer


class Model(BaseModel):
    password: SecretStr
    password_bytes: SecretBytes

    @field_serializer('password', 'password_bytes', when_used='json')
    def dump_secret(self, v):
        return v.get_secret_value()


model = Model(password='IAmSensitive', password_bytes=b'IAmSensitiveBytes')
print(model)
#> password=SecretStr('**********') password_bytes=SecretBytes(b'**********')
print(model.password)
#> **********
print(model.model_dump())
"""
{
    'password': SecretStr('**********'),
    'password_bytes': SecretBytes(b'**********'),
}
"""
print(model.model_dump_json())
#> {"password":"IAmSensitive","password_bytes":"IAmSensitiveBytes"}

Buat bidang Rahasia Anda sendiri

Pydantic menyediakan kelas Secret generik sebagai mekanisme untuk membuat tipe rahasia khusus.

??? api "Dokumentasi API" [pydantic.types.Secret][pydantic.types.Rahasia]

Pydantic menyediakan kelas Secret generik sebagai mekanisme untuk membuat tipe rahasia khusus. Anda dapat secara langsung membuat parameter Secret , atau membuat subkelas dari Secret yang diparametrikan untuk menyesuaikan str() dan repr() dari tipe rahasia.

from datetime import date

from pydantic import BaseModel, Secret

# Using the default representation
SecretDate = Secret[date]


# Overwriting the representation
class SecretSalary(Secret[float]):
    def _display(self) -> str:
        return '$****.**'


class Employee(BaseModel):
    date_of_birth: SecretDate
    salary: SecretSalary


employee = Employee(date_of_birth='1990-01-01', salary=42)

print(employee)
#> date_of_birth=Secret('**********') salary=SecretSalary('$****.**')

print(employee.salary)
#> $****.**

print(employee.salary.get_secret_value())
#> 42.0

print(employee.date_of_birth)
#> **********

print(employee.date_of_birth.get_secret_value())
#> 1990-01-01

Anda dapat menerapkan batasan pada tipe yang mendasarinya melalui anotasi: Misalnya:

from typing_extensions import Annotated

from pydantic import BaseModel, Field, Secret, ValidationError

SecretPosInt = Secret[Annotated[int, Field(gt=0, strict=True)]]


class Model(BaseModel):
    sensitive_int: SecretPosInt


m = Model(sensitive_int=42)
print(m.model_dump())
#> {'sensitive_int': Secret('**********')}

try:
    m = Model(sensitive_int=-42)  # (1)!
except ValidationError as exc_info:
    print(exc_info.errors(include_url=False, include_input=False))
    """
    [
        {
            'type': 'greater_than',
            'loc': ('sensitive_int',),
            'msg': 'Input should be greater than 0',
            'ctx': {'gt': 0},
        }
    ]
    """

try:
    m = Model(sensitive_int='42')  # (2)!
except ValidationError as exc_info:
    print(exc_info.errors(include_url=False, include_input=False))
    """
    [
        {
            'type': 'int_type',
            'loc': ('sensitive_int',),
            'msg': 'Input should be a valid integer',
        }
    ]
    """
  1. Nilai input tidak lebih besar dari 0 sehingga menimbulkan kesalahan validasi.
  2. Nilai inputnya bukan bilangan bulat, sehingga menimbulkan kesalahan validasi karena tipe SecretPosInt mengaktifkan mode ketat.

本文总阅读量