Implement TimeField; close #7

Author: alexei

adapters/fields.py

@@ -2,14 +2,16 @@
 
 from __future__ import unicode_literals
 
+import datetime
+import dateutil.parser
 from decimal import Decimal
 
 from .base import BaseField
 
 
 __all__ = [
     'BooleanField', 'CharField', 'DecimalField', 'Field', 'FloatField',
-    'IntField']
+    'IntField', 'TimeField']
 
 
 class BooleanField(BaseField):
@@ -39,3 +41,13 @@ def adapt(self, data):
 class IntField(BaseField):
     def adapt(self, data):
         return int(data)
+
+
+class TimeField(BaseField):
+    def adapt(self, data):
+        if isinstance(data, datetime.time):
+            return data
+        elif isinstance(data, (str, unicode)):
+            return dateutil.parser.parse(data).timetz()
+        else:
+            raise ValueError("Invalid time argument")

requirements.txt

@@ -0,0 +1 @@
+python-dateutil==2.6.0

tests/test_fields.py

@@ -2,11 +2,14 @@
 
 from __future__ import unicode_literals
 
+import datetime
 from decimal import Decimal
 import unittest
 
 import adapters
 
+import dateutil.tz
+
 
 class FieldsTest(unittest.TestCase):
     def test_char_field(self):
@@ -58,3 +61,29 @@ def test_boolean_field_from_string(self):
     def test_boolean_field_from_null(self):
         actual = adapters.BooleanField().adapt(None)
         self.assertFalse(actual)
+
+    def test_time_field(self):
+        now = datetime.datetime.today().time()
+        actual = adapters.TimeField().adapt(now)
+        expected = now
+        self.assertEqual(actual, expected)
+
+    def test_time_field_from_string(self):
+        tzoffset = dateutil.tz.tzoffset(None, 7200)
+        time = datetime.time(13, 14, 15, 16, tzinfo=tzoffset)
+
+        actual = adapters.TimeField().adapt('13:14:15.000016+02:00')
+        expected = time
+        self.assertEqual(actual, expected)
+
+        actual = adapters.TimeField().adapt('13:14:15.000016')
+        expected = datetime.time(13, 14, 15, 16)
+        self.assertEqual(actual, expected)
+
+        actual = adapters.TimeField().adapt(time.strftime('%H:%M:%S'))
+        expected = datetime.time(13, 14, 15)
+        self.assertEqual(actual, expected)
+
+        actual = adapters.TimeField().adapt(time.strftime('%H:%M'))
+        expected = datetime.time(13, 14)
+        self.assertEqual(actual, expected)