diff options
Diffstat (limited to 'factory/django.py')
| -rw-r--r-- | factory/django.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/factory/django.py b/factory/django.py index 351cb76..3eabca1 100644 --- a/factory/django.py +++ b/factory/django.py @@ -59,7 +59,9 @@ class DjangoModelFactory(base.Factory): try: return 1 + manager.values_list('pk', flat=True ).order_by('-pk')[0] - except IndexError: + except (IndexError, TypeError): + # IndexError: No instance exist yet + # TypeError: pk isn't an integer type return 1 @classmethod |