adds decimal type and associated tests

graphene/__init__.py

@@ -12,6 +12,7 @@ from .types import (
     Scalar,
     String, ID, Int, Float, Boolean,
     Date, DateTime, Time,
+    Decimal,
     JSONString,
     UUID,
     List, NonNull,

graphene/types/__init__.py

@@ -6,6 +6,7 @@ from .interface import Interface
 from .mutation import Mutation
 from .scalars import Scalar, String, ID, Int, Float, Boolean
 from .datetime import Date, DateTime, Time
+from .decimal import Decimal
 from .json import JSONString
 from .uuid import UUID
 from .schema import Schema

graphene/types/decimal.py

@@ -0,0 +1,29 @@
+import decimal
+
+from graphql.language import ast
+
+from .scalars import Scalar
+
+
+class Decimal(Scalar):
+    """
+    The `Decimal` scalar type represents a python Decimal.
+    """
+    @staticmethod
+    def serialize(dec):
+        assert isinstance(dec, decimal.Decimal), (
+            'Received not compatible Decimal "{}"'.format(repr(dec))
+        )
+        return str(dec)
+    
+    @classmethod
+    def parse_literal(cls, node):
+        if isinstance(node, ast.StringValue):
+            return cls.parse_value(node.value)
+
+    @staticmethod
+    def parse_value(value):
+        try:
+            return decimal.Decimal(value)
+        except ValueError:
+            return None

graphene/types/tests/test_decimal.py

@@ -0,0 +1,44 @@
+import decimal
+
+from ..decimal import Decimal
+from ..objecttype import ObjectType
+from ..schema import Schema
+
+
+class Query(ObjectType):
+    decimal = Decimal(input=Decimal())
+
+    def resolve_decimal(self, info, input):
+        return input
+
+schema = Schema(query=Query)
+
+
+def test_decimal_string_query():
+    decimal_value = decimal.Decimal('1969.1974')
+    result = schema.execute('''{ decimal(input: "%s") }''' % decimal_value)
+    assert not result.errors
+    assert result.data == {
+        'decimal': str(decimal_value)
+    }
+
+
+def test_decimal_string_query_variable():
+    decimal_value = decimal.Decimal('1969.1974')
+
+    result = schema.execute(
+        '''query Test($decimal: Decimal){ decimal(input: $decimal) }''',
+        variable_values={'decimal': decimal_value}
+    )
+    assert not result.errors
+    assert result.data == {
+        'decimal': str(decimal_value)
+    }
+
+
+def test_bad_decimal_query():
+    not_a_decimal = "Nobody expects the Spanish Inquisition!"
+    
+    result = schema.execute('''{ decimal(input: "%s") }''' % not_a_decimal)
+    assert len(result.errors) == 1
+    assert result.data == None