Как получить целочисленный ID объекта Django Factory Boy?

У меня есть фабрика-мальчик, использующая Django, которая мне нужна для доступа к ID сгенерированных объектов. Однако всякий раз, когда я пытаюсь получить этот ID, я получаю TypeErrors. Есть идеи, что я упускаю?

# factories.py
class PartNoFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = PartNo

    id = factory.fuzzy.FuzzyInteger(1, 1000000, step=1)   
    # have also tried id = factory.Sequence(lambda n: n + 1)
    # have also tried id = int(factory.Sequence(lambda n: n + 1)), which results in error "type Sequence cannot be assigned to paramater "_x"

# test_factories.py
def mock_child_part_nos(arg_1: PartNo) -> 'list[int]':
    mock_part_no_one = PartNoFactory()
    mock_part_no_two = PartNoFactory()
    mock_part_no_three = PartNoFactory()

    return [mock_part_no_one.id, mock_part_no_two.id, mock_part_no_three.id]
    # if using FuzzyInteger: "Expression of type "list[FuzzyInteger]" cannot be assigned to return type "list[int]"
    # if using Sequence: "Expression of type "list[Sequence]" cannot be assigned to return type "list[int]"

Вернуться на верх