Merge branch 'feature/PositiveDecimalGenerator' of https://github.com…

…/fmarco/django-autofixture
gregmuellegger · Mar 2, 2015 · e20f3cb · e20f3cb
2 parents fbf44a2 + 11a3185
commit e20f3cb
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/autofixture/generators.py b/autofixture/generators.py
@@ -314,6 +314,18 @@ def generate(self):
         return value
 
 
+class PositiveDecimalGenerator(DecimalGenerator):
+    def __init__(self, max_digits=None, decimal_places=None, *args, **kwargs):
+        super(PositiveDecimalGenerator, self).__init__(max_digits, decimal_places, *args, **kwargs)
+
+    def generate(self):
+        maxint = 10 ** self.max_digits - 1
+        value = (
+            Decimal(random.randint(0, maxint)) /
+            10 ** self.decimal_places)
+        return value
+
+
 class FirstNameGenerator(Generator):
     """ Generates a first name, either male or female """