Merge pull request #6 from RegBinder/convert-int-label-column

Convert int label column for SQLAlchemy column_property

Author: syrusakbary

graphene_sqlalchemy/converter.py

@@ -105,9 +105,10 @@ def convert_column_to_datetime(type, column, registry=None):
 @convert_sqlalchemy_type.register(types.Integer)
 def convert_column_to_int_or_id(type, column, registry=None):
     if column.primary_key:
-        return ID(description=column.doc, required=not(column.nullable))
+        return ID(description=column.doc, required=not (column.nullable))
     else:
-        return Int(description=column.doc, required=not(column.nullable))
+        return Int(description=getattr(column, 'doc', None),
+                   required=not (getattr(column, 'nullable', True)))
 
 
 @convert_sqlalchemy_type.register(types.Boolean)

graphene_sqlalchemy/tests/test_converter.py

@@ -114,6 +114,11 @@ def test_should_label_convert_string():
     assert isinstance(graphene_type, graphene.String)
 
 
+def test_should_label_convert_int():
+    label = Label('int_label_test', case([], else_="foo"), type_=types.Integer())
+    graphene_type = convert_sqlalchemy_column(label)
+    assert isinstance(graphene_type, graphene.Int)
+
 def test_should_choice_convert_enum():
     TYPES = [
         (u'es', u'Spanish'),