Commits

Luke Plant  committed d58df9b

Some cleanups to public interface.

  • Participants
  • Parent commits ff94321

Comments (0)

Files changed (1)

File anonymizer/base.py

         field_vals = set(x[0] for x in field.model._default_manager.values_list(field.name))
         self.init_values[field] = field_vals
 
-
-    def get_allowed_value(self, source, field):
+    def _get_allowed_value(self, source, field):
         retval = source()
 
         # Enforce unique.  Eensure we don't set the same values, as either
         return retval
 
     def __getattr__(self, name):
-        # we delegate all calls to faker, but add checks
+        # we delegate most calls to faker, but add checks
+        source = getattr(self.faker, name)
+
         def func(*args, **kwargs):
-            source = getattr(self.faker, name)
             field = kwargs.get('field', None)
             if field is not None:
-                return self.get_allowed_value(source, field)
+                return self._get_allowed_value(source, field)
             else:
                 return source()
         return func