diff options
author | Raphaël Barrois <raphael.barrois@polytechnique.org> | 2013-03-04 23:18:02 +0100 |
---|---|---|
committer | Raphaël Barrois <raphael.barrois@polytechnique.org> | 2013-03-04 23:18:02 +0100 |
commit | 7d792430e103984a91c102c33da79be2426bc632 (patch) | |
tree | 0044c449e933e93d4fecf1e80e6b361c6c3e8150 /factory/declarations.py | |
parent | 9422cf12516143650f1014f34f996260c00d4c0a (diff) | |
download | factory-boy-7d792430e103984a91c102c33da79be2426bc632.tar factory-boy-7d792430e103984a91c102c33da79be2426bc632.tar.gz |
Add a 'after post_generation' hook to Factory.
Use it in DjangoModelFactory to save objects again if a post_generation hook ran.
Signed-off-by: Raphaël Barrois <raphael.barrois@polytechnique.org>
Diffstat (limited to 'factory/declarations.py')
-rw-r--r-- | factory/declarations.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/factory/declarations.py b/factory/declarations.py index d3d7659..366c2c8 100644 --- a/factory/declarations.py +++ b/factory/declarations.py @@ -412,7 +412,7 @@ class PostGeneration(PostGenerationDeclaration): self.function = function def call(self, obj, create, extracted=None, **kwargs): - self.function(obj, create, extracted, **kwargs) + return self.function(obj, create, extracted, **kwargs) def post_generation(*args, **kwargs): |