Python2.7 compatible

2 files changed, 4 insertions, 4 deletions

diff --git a/oauthlib/openid/connect/core/grant_types/base.py b/oauthlib/openid/connect/core/grant_types/base.py
index 6272ea2..32a21b6 100644
--- a/oauthlib/openid/connect/core/grant_types/base.py
+++ b/oauthlib/openid/connect/core/grant_types/base.py
@@ -1,9 +1,9 @@
 from .exceptions import OIDCNoPrompt
 
 import base64
-import datetime
 import hashlib
 import logging
+import time
 from json import loads
 
 from oauthlib.oauth2.rfc6749.errors import ConsentRequired, InvalidRequestError, LoginRequired
@@ -108,7 +108,7 @@ class GrantTypeBase(object):
         # Start with technicals fields bound to the specification.
         id_token = {}
         id_token['aud'] = request.client_id
-        id_token['iat'] = int(datetime.datetime.now().timestamp())
+        id_token['iat'] = int(time.time())
 
         # nonce is REQUIRED when response_type value is:
         # - id_token token (Implicit)
diff --git a/tests/openid/connect/core/grant_types/test_base.py b/tests/openid/connect/core/grant_types/test_base.py
index 319904b..76e017f 100644
--- a/tests/openid/connect/core/grant_types/test_base.py
+++ b/tests/openid/connect/core/grant_types/test_base.py
@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
-import datetime
 import mock
+import time
 
 from oauthlib.common import Request
 from oauthlib.openid.connect.core.grant_types.base import GrantTypeBase
@@ -68,7 +68,7 @@ class IDTokenTest(TestCase):
         self.assertEqual(token["id_token"], "eyJ.body.signature")
         id_token = self.mock_validator.finalize_id_token.call_args[0][0]
         self.assertEqual(id_token['aud'], 'abcdef')
-        self.assertGreaterEqual(id_token['iat'], int(datetime.datetime.now().timestamp()))
+        self.assertGreaterEqual(id_token['iat'], int(time.time()))
 
     def test_finalize_id_token_with_nonce(self):
         token = self.grant.add_id_token(self.token, "token_handler_mock", self.request, "my_nonce")