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)