piotrkilczuk

class PlayerFactory(DjangoModelFactory):
    # other declarations removed for brevity

    @lazy_attribute_sequence
    def government_id(self, n):
        dob_part = int(self.dob.strftime('%y%m%d')) * 10000

        # lets start at
        # last known government_id for a date + 1 if last known government_id for a date is odd
        # last known government_id for a date + 2 if last known government_id for a date is even

piotrkilczuk

>>> from championship.factories import *
>>> players = PlayerFactory.build_batch(10)
>>> club = ClubFactory(players=players)
>>> club.players.count()
10
>>> players[0].club == club
True

piotrkilczuk

class ClubFactory(DjangoModelFactory):
    name = Faker('company')

    class Meta:
        model = Club

    @post_generation
    def players(self, create, extracted, **kwargs):
        if not create:
            return

piotrkilczuk

>>> from championship.factories import *
>>> club = ClubFactory()
>>> ten_players = PlayerFactory.create_batch(10, club=club)
>>> tournament = TournamentFactory(players=ten_players)
>>> tournament.players.count()
10
>>> ten_players[0].tournaments.count()
1

piotrkilczuk

class ClubFactory(DjangoModelFactory):
    name = Faker('company')

    class Meta:
        model = Club


class PlayerFactory(DjangoModelFactory):
    # other declarations removed for brevity
    

piotrkilczuk

>>> player = Player.objects.first()
>>> tournament = Tournament.objects.last()
>>> tournament in player.tournaments.all()
False
>>> player.tournaments.add(tournament)
>>> tournament in player.tournaments.all()
True
>>> player in tournament.players.all()
True

piotrkilczuk

>>> class ClubFactory(DjangoModelFactory):
...     name = Faker('company')
...
...     class Meta:
...         model = Club
...
>>> club = ClubFactory()
>>> club.pk
3
>>> club = ClubFactory.create()

piotrkilczuk

class Club(models.Model):
    name = models.CharField(max_length=100)

    def __str__(self):
        return self.name

    
def picture_upload_path(self, filename):
    basename = os.path.basename(filename)
    return f'player/picture/{self.id}/{basename}'

piotrkilczuk

piotrkilczuk

def qux(): 
    try: 
        fun() 
    except Exception as exc: 
        raise RuntimeError('Qux') from exc 
                                                                                                  

def fun(): 
    raise ValueError('Foo bar baz')