From a13549407845c166b2750cbd35e2304ca70a763c Mon Sep 17 00:00:00 2001 From: Paul Bailey Date: Mon, 7 Jun 2021 16:23:41 -0500 Subject: [PATCH] make small auto optional --- graphene_django/converter.py | 10 +++++++++- graphene_django/tests/test_converter.py | 2 +- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/graphene_django/converter.py b/graphene_django/converter.py index 29adc50..1f6fbec 100644 --- a/graphene_django/converter.py +++ b/graphene_django/converter.py @@ -166,13 +166,21 @@ def convert_field_to_string(field, registry=None): ) -#@convert_django_field.register(models.SmallAutoField) @convert_django_field.register(models.BigAutoField) @convert_django_field.register(models.AutoField) def convert_field_to_id(field, registry=None): return ID(description=get_django_field_description(field), required=not field.null) +if hasattr(models, "SmallAutoField"): + + @convert_django_field.register(models.SmallAutoField) + def convert_field_to_id(field, registry=None): + return ID( + description=get_django_field_description(field), required=not field.null + ) + + @convert_django_field.register(models.UUIDField) def convert_field_to_uuid(field, registry=None): return UUID( diff --git a/graphene_django/tests/test_converter.py b/graphene_django/tests/test_converter.py index 29a5a59..afd744f 100644 --- a/graphene_django/tests/test_converter.py +++ b/graphene_django/tests/test_converter.py @@ -116,7 +116,7 @@ def test_should_big_auto_convert_id(): def test_should_small_auto_convert_id(): - if hasattr(models, 'SmallAutoField'): + if hasattr(models, "SmallAutoField"): assert_conversion(models.SmallAutoField, graphene.ID, primary_key=True)