diff options
author | Raphaël Barrois <raphael.barrois@polytechnique.org> | 2015-03-26 22:22:03 +0100 |
---|---|---|
committer | Raphaël Barrois <raphael.barrois@polytechnique.org> | 2015-03-26 22:22:03 +0100 |
commit | 69befae5fde1897cf68c4d44a146db5ba642c814 (patch) | |
tree | 88d696ef1784cd73552a5ecd13bc3f4af5d2ac2a /factory | |
parent | 40d4a4b13d4ca959879d1798f24d510fd7abf4dc (diff) | |
download | factory-boy-69befae5fde1897cf68c4d44a146db5ba642c814.tar factory-boy-69befae5fde1897cf68c4d44a146db5ba642c814.tar.gz |
Allow lazy evaluation of FuzzyChoice's iterators (Closes #184).
This allows the following idiom:
``user = factory.fuzzy.FuzzyChoice(User.objects.all())``
Previously, the ``User.objects.all()`` queryset would have been
evaluated *at import time*; it is now evaluated with the first use of the
``FuzzyChoice``.
Diffstat (limited to 'factory')
-rw-r--r-- | factory/fuzzy.py | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/factory/fuzzy.py b/factory/fuzzy.py index 564264e..4e6a03d 100644 --- a/factory/fuzzy.py +++ b/factory/fuzzy.py @@ -113,13 +113,21 @@ class FuzzyText(BaseFuzzyAttribute): class FuzzyChoice(BaseFuzzyAttribute): - """Handles fuzzy choice of an attribute.""" + """Handles fuzzy choice of an attribute. + + Args: + choices (iterable): An iterable yielding options; will only be unrolled + on the first call. + """ def __init__(self, choices, **kwargs): - self.choices = list(choices) + self.choices = None + self.choices_generator = choices super(FuzzyChoice, self).__init__(**kwargs) def fuzz(self): + if self.choices is None: + self.choices = list(self.choices_generator) return _random.choice(self.choices) |