Zum Inhalt

Geheimnisse

!!! Warnung „🚧 Work in Progress“ Diese Seite ist in Arbeit.

Serialisieren Sie SecretStr und SecretBytes als Klartext

Standardmäßig werden SecretStr und SecretBytes bei der Serialisierung in JSON als ********** serialisiert.

Sie können den [field_serializer][pydantic.Functional_serializers.field_serializer] verwenden, um das Geheimnis bei der Serialisierung in JSON als Klartext auszugeben.

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"}

Erstellen Sie Ihr eigenes Secret-Feld

Pydantic stellt die generische Secret Klasse als Mechanismus zum Erstellen benutzerdefinierter Secret-Typen bereit.

??? API „API-Dokumentation“ pydantic.types.Secret

Pydantic stellt die generische Secret Klasse als Mechanismus zum Erstellen benutzerdefinierter Secret-Typen bereit. Sie können Secret entweder direkt parametrisieren oder eine Unterklasse von einem parametrisierten Secret erstellen, um str() und repr() eines Secret-Typs anzupassen.

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

Sie können Einschränkungen für den zugrunde liegenden Typ durch Anmerkungen erzwingen: Zum Beispiel:

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. Der Eingabewert ist nicht größer als 0 und löst daher einen Validierungsfehler aus.
  2. Der Eingabewert ist keine Ganzzahl und löst daher einen Validierungsfehler aus, da für den Typ SecretPosInt der strikte Modus aktiviert ist.

本文总阅读量