From 9a5b9879b3511daf6ef506093e119760a80858e7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Adri=C3=A1n=20L=C3=B3pez?= <adrian@undefinedlabs.com>
Date: Tue, 21 Jan 2020 16:08:41 +0100
Subject: [PATCH] Add more generator ranges for other types of integer fields

---
 model_bakery/generators.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/model_bakery/generators.py b/model_bakery/generators.py
index cd498903..cd5406c5 100644
--- a/model_bakery/generators.py
+++ b/model_bakery/generators.py
@@ -34,6 +34,18 @@
 from .gis import default_gis_mapping
 from .utils import import_from_str
 
+try:
+    from django.db.models import AutoField, BigAutoField, SmallAutoField
+except ImportError:
+    AutoField = None
+    BigAutoField = None
+    SmallAutoField = None
+
+try:
+    from django.db.models import PositiveBigIntegerField
+except ImportError:
+    PositiveBigIntegerField = None
+
 try:
     from django.contrib.postgres.fields import ArrayField
 except ImportError:
@@ -113,6 +125,17 @@ def gen_integer():
     default_mapping[CIEmailField] = random_gen.gen_email
 if CITextField:
     default_mapping[CITextField] = random_gen.gen_text
+if AutoField:
+    default_mapping[AutoField] = _make_integer_gen_by_range(AutoField)
+if BigAutoField:
+    default_mapping[BigAutoField] = _make_integer_gen_by_range(BigAutoField)
+if SmallAutoField:
+    default_mapping[SmallAutoField] = _make_integer_gen_by_range(SmallAutoField)
+if PositiveBigIntegerField:
+    default_mapping[PositiveBigIntegerField] = _make_integer_gen_by_range(
+        PositiveBigIntegerField
+    )
+
 
 # Add GIS fields
 default_mapping.update(default_gis_mapping)