Pass attribute name to PostGenerationDeclaration

1 files changed, 3 insertions, 2 deletions

diff --git a/factory/declarations.py b/factory/declarations.py
index e7c8ed5..037a679 100644
--- a/factory/declarations.py
+++ b/factory/declarations.py
@@ -419,10 +419,11 @@ class List(SubFactory):
 
 class ExtractionContext(object):
     """Private class holding all required context from extraction to postgen."""
-    def __init__(self, value=None, did_extract=False, extra=None):
+    def __init__(self, value=None, did_extract=False, extra=None, for_field=''):
         self.value = value
         self.did_extract = did_extract
         self.extra = extra or {}
+        self.for_field = for_field
 
     def __repr__(self):
         return 'ExtractionContext(%r, %r, %r)' % (
@@ -456,7 +457,7 @@ class PostGenerationDeclaration(object):
             did_extract = False
 
         kwargs = utils.extract_dict(name, attrs)
-        return ExtractionContext(extracted, did_extract, kwargs)
+        return ExtractionContext(extracted, did_extract, kwargs, name)
 
     def call(self, obj, create, extraction_context):  # pragma: no cover
         """Call this hook; no return value is expected.