Merge pull request #701 from graphql-python/feature/date-improvements

Datetime improvements

Author: syrusakbary

graphene/__init__.py

@@ -11,6 +11,7 @@
     Schema,
     Scalar,
     String, ID, Int, Float, Boolean,
+    Date, DateTime, Time,
     JSONString,
     UUID,
     List, NonNull,
@@ -54,6 +55,9 @@
     'Float',
     'Enum',
     'Boolean',
+    'Date',
+    'DateTime',
+    'Time',
     'JSONString',
     'UUID',
     'List',

graphene/types/__init__.py

@@ -5,6 +5,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 .json import JSONString
 from .uuid import UUID
 from .schema import Schema
@@ -36,6 +37,9 @@
     'ID',
     'Int',
     'Float',
+    'Date',
+    'DateTime',
+    'Time',
     'JSONString',
     'UUID',
     'Boolean',

graphene/types/datetime.py

@@ -3,17 +3,10 @@
 import datetime
 
 from graphql.language import ast
+from aniso8601 import parse_datetime, parse_date, parse_time
 
 from .scalars import Scalar
 
-try:
-    import iso8601
-except ImportError:
-    raise ImportError(
-        "iso8601 package is required for DateTime Scalar.\n"
-        "You can install it using: pip install iso8601."
-    )
-
 
 class Date(Scalar):
     '''
@@ -39,8 +32,8 @@ def parse_literal(cls, node):
     @staticmethod
     def parse_value(value):
         try:
-            return iso8601.parse_date(value).date()
-        except iso8601.ParseError:
+            return parse_date(value)
+        except ValueError:
             return None
 
 
@@ -66,8 +59,8 @@ def parse_literal(cls, node):
     @staticmethod
     def parse_value(value):
         try:
-            return iso8601.parse_date(value)
-        except iso8601.ParseError:
+            return parse_datetime(value)
+        except ValueError:
             return None
 
 
@@ -77,7 +70,6 @@ class Time(Scalar):
     specified by
     [iso8601](https://en.wikipedia.org/wiki/ISO_8601).
     '''
-    epoch_date = '1970-01-01'
 
     @staticmethod
     def serialize(time):
@@ -94,7 +86,6 @@ def parse_literal(cls, node):
     @classmethod
     def parse_value(cls, value):
         try:
-            dt = iso8601.parse_date('{}T{}'.format(cls.epoch_date, value))
-            return datetime.time(dt.hour, dt.minute, dt.second, dt.microsecond, dt.tzinfo)
-        except iso8601.ParseError:
+            return parse_time(value)
+        except ValueError:
             return None

setup.py

@@ -90,6 +90,7 @@ def run_tests(self):
         'graphql-core>=2.0,<3',
         'graphql-relay>=0.4.5,<1',
         'promise>=2.1,<3',
+        'aniso8601>=3,<4',
     ],
     tests_require=tests_require,
     extras_require={