factories.py

__copyright__ = "Copyright © Stichting SciPost (SciPost Foundation)"
__license__ = "AGPL v3"


import factory

from common.helpers import random_arxiv_identifier_without_version_number

from .models import Preprint


class PreprintFactory(factory.django.DjangoModelFactory):
    """
    Generate random Preprint instances.
    """

    vn_nr = 1
    identifier_wo_vn_nr = factory.Sequence(
        lambda n: random_arxiv_identifier_without_version_number())
    identifier_w_vn_nr = factory.lazy_attribute(lambda o: '%sv%i' % (
        o.identifier_wo_vn_nr, o.vn_nr))
    url = factory.lazy_attribute(lambda o: (
        'https://arxiv.org/abs/%s' % o.identifier_wo_vn_nr))
    scipost_preprint_identifier = factory.Sequence(lambda n: Preprint.objects.count() + 1)

    class Meta:
        model = Preprint