从 pydantic 数据模型的打印和调用中隐藏变量（例如“密码”）

Question

我正在将 pydantic 用于某些用户/密码数据模型。打印模型时，我想用其他值替换密码的值（***例如），以防止密码意外写入日志文件或控制台。

from pydantic import BaseModel

class myUserClass(BaseModel):
    User = 'foo'
    Password = 'bar'

    def __str__(self):
        return "Hidden parameters."

U = myUserClass()

print(U)
# > Hidden parameters. # Should be User='foo', Password='***' or similar

U # This will still show my password.
#> myUserClass(User='foo', Password='bar')

如何访问通常打印的字符串并仅将“bar”替换为“***”，但保留所有参数？

我怎样才能在打电话时也做到这一点U？这可能并不像通常print调用的日志记录和控制台输出那么重要。

Answer 1

Pydantic 提供SecretStr数据类型。转换为str时（如打印）被替换为***，实际值可以通过get_secret_value()方法获取：

class Foobar(BaseModel):
    password: SecretStr
    empty_password: SecretStr

# Initialize the model.
f = Foobar(password='1234', empty_password='')


# Assert str and repr are correct.
assert str(f.password) == '**********'
assert str(f.empty_password) == ''

# Assert retrieval of secret value is correct
assert f.password.get_secret_value() == '1234'
assert f.empty_password.get_secret_value() == ''