Added FileField and ImageField conversion. Fixed #137

graphql-python · Apr 2, 2016 · a1dd2b6 · a1dd2b6
1 parent 1f548f1
commit a1dd2b6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/graphene/contrib/django/converter.py b/graphene/contrib/django/converter.py
@@ -31,6 +31,7 @@ def convert_django_field(field):
 @convert_django_field.register(models.SlugField)
 @convert_django_field.register(models.URLField)
 @convert_django_field.register(models.GenericIPAddressField)
+@convert_django_field.register(models.FileField)
 @convert_django_field.register(UUIDField)
 def convert_field_to_string(field):
     return String(description=field.help_text)

diff --git a/graphene/contrib/django/tests/test_converter.py b/graphene/contrib/django/tests/test_converter.py
@@ -53,6 +53,14 @@ def test_should_ipaddress_convert_string():
     assert_conversion(models.GenericIPAddressField, graphene.String)
 
 
+def test_should_file_convert_string():
+    assert_conversion(models.FileField, graphene.String)
+
+
+def test_should_image_convert_string():
+    assert_conversion(models.ImageField, graphene.String)
+
+
 def test_should_auto_convert_id():
     assert_conversion(models.AutoField, graphene.ID, primary_key=True)