生成后如何获取工厂的排除属性？

Question

我使用Factory Boy来生成一个假用户。 我需要一个应该被散列的字段password （在 db 中，这是与密码相关的唯一行）和一个clear_password值，以便知道散列的相关明文密码是什么，以供以后在测试中使用。

我尝试了两种类似的方法， Params和元选项exclude 。 仅排除作品以生成稍后散列的清晰密码。 问题：在 object 生成后，我无法访问clear_password 。

要求.txt

passlib[bcrypt]
SQLAlchemy
factory-boy

最小可复制示例（python 3.10 使用最新版本的要求）

import random

import factory
import passlib.context
import sqlalchemy as sa
from sqlalchemy import orm

engine = sa.create_engine(
    'sqlite:///test.db',
    echo=True,
)

Base = orm.declarative_base(bind=engine)
pwd_context = passlib.context.CryptContext(schemes=['bcrypt'], deprecated='auto')

BCRYPT_MAX_LENGTH = 72


class User(Base):
    __tablename__ = 'user'
    id = sa.Column(sa.Integer, primary_key=True, autoincrement=True)
    password = sa.Column(sa.LargeBinary(BCRYPT_MAX_LENGTH), nullable=False)
    ...


class UserFact(factory.Factory):
    class Meta:
        model = User
        exclude = ('clear_password',)

    clear_password = factory.Faker('password', length=random.randint(8, 64))
    password = factory.LazyAttribute(lambda o: bytes(
        pwd_context.hash(o.clear_password), encoding='utf-8'))
    ...


if __name__ == '__main__':
    print(UserFact().clear_password)

追溯

Traceback (most recent call last):
  File "path/to/file.py", line 38, in <module>
    print(User().clear_password)
AttributeError: 'User' object has no attribute 'clear_password'

Answer 1

我知道这可能不是您正在寻找的答案，但是，如何为clear_password分配一个固定（即可预测的）值？

从您的问题来看，您似乎不想测试passlib实现，因此在测试期间为所有User生成相同的密码应该是可以的。

---

对于您最初的问题，您无法访问排除的成员，因为在生成时，工厂返回其底层 model 而不是工厂本身。 但是，您可以像这样对返回的 model 进行猴子修补：

# rest of code omitted
class UserFact(Factory):
    class Meta:
        model = User
        exclude = ('clear_password',)

    clear_password = "azerty"
    password = LazyAttribute(lambda o: bytes(
        pwd_context.hash(o.clear_password), encoding='utf-8'))
    
    @classmethod
    def _generate(cls, create, attrs):
        user = super()._generate(create, attrs)
        user._clear_password = cls.clear_password
        return user

if __name__ == '__main__':
    print(UserFact()._clear_password) # azerty