TIME return datetime.time object

Author: long2ice

CHANGELOG.md

@@ -5,6 +5,7 @@
 ### 0.2.4
 
 - Fix `escape_string` for enum type. (#30)
+- `TIME` return `datetime.time` object.
 
 ### 0.2.3
 

asyncmy/converters.pyx

@@ -102,24 +102,24 @@ cpdef str escape_time(obj, mapping=None):
         fmt = "'{0.hour:02}:{0.minute:02}:{0.second:02}'"
     return fmt.format(obj)
 
-cpdef str  escape_datetime(obj, mapping=None):
+cpdef str escape_datetime(obj, mapping=None):
     if obj.microsecond:
         fmt = "'{0.year:04}-{0.month:02}-{0.day:02} {0.hour:02}:{0.minute:02}:{0.second:02}.{0.microsecond:06}'"
     else:
         fmt = "'{0.year:04}-{0.month:02}-{0.day:02} {0.hour:02}:{0.minute:02}:{0.second:02}'"
     return fmt.format(obj)
 
-cpdef str  escape_date(obj, mapping=None):
+cpdef str escape_date(obj, mapping=None):
     fmt = "'{0.year:04}-{0.month:02}-{0.day:02}'"
     return fmt.format(obj)
 
-cpdef str  escape_struct_time(obj, mapping=None):
+cpdef str escape_struct_time(obj, mapping=None):
     return escape_datetime(datetime.datetime(*obj[:6]))
 
-cpdef str  decimal2literal(o, d):
+cpdef str decimal2literal(o, d):
     return format(o, "f")
 
-cpdef int  _convert_second_fraction(s):
+cpdef int _convert_second_fraction(s):
     if not s:
         return 0
     # Pad zeros to ensure the fraction length in microseconds
@@ -313,7 +313,7 @@ cdef dict decoders = {
     YEAR: int,
     TIMESTAMP: convert_datetime,
     DATETIME: convert_datetime,
-    TIME: convert_timedelta,
+    TIME: convert_time,
     DATE: convert_date,
     BLOB: through,
     TINY_BLOB: through,

conftest.py

@@ -46,6 +46,7 @@ async def initialize_tests(connection):
   `decimal` decimal(10,2) DEFAULT NULL,
   `date` date DEFAULT NULL,
   `datetime` datetime DEFAULT NULL,
+  `time` time DEFAULT NULL,
   `float` float DEFAULT NULL,
   `string` varchar(200) DEFAULT NULL,
   `tinyint` tinyint DEFAULT NULL,

tests/test_cursor.py

@@ -1,3 +1,5 @@
+import datetime
+from decimal import Decimal
 from enum import Enum
 
 import pytest
@@ -33,18 +35,33 @@ async def test_dict_cursor(connection):
 async def test_insert(connection):
     async with connection.cursor(cursor=DictCursor) as cursor:
         rows = await cursor.execute(
-            """INSERT INTO test.asyncmy(id,`decimal`, date, datetime, `float`, string, `tinyint`) VALUES (%s,%s,%s,%s,%s,%s,%s)""",
+            """INSERT INTO test.asyncmy(id,`decimal`, date, datetime, time, `float`, string, `tinyint`) VALUES (%s,%s,%s,%s,%s,%s,%s,%s)""",
             (
-                -1,
+                1,
                 1,
                 "2020-08-08",
                 "2020-08-08 00:00:00",
+                "00:00:00",
                 1,
                 "1",
                 1,
             ),
         )
         assert rows == 1
+        await cursor.execute("select * from test.asyncmy where id = %s", (cursor.lastrowid,))
+        result = await cursor.fetchall()
+        assert result == [
+            {
+                "id": 1,
+                "decimal": Decimal("1.00"),
+                "date": datetime.date(2020, 8, 8),
+                "datetime": datetime.datetime(2020, 8, 8, 0, 0),
+                "time": datetime.time.fromisoformat("00:00"),
+                "float": 1.0,
+                "string": "1",
+                "tinyint": 1,
+            }
+        ]
 
 
 @pytest.mark.asyncio
@@ -58,12 +75,13 @@ async def test_delete(connection):
 async def test_executemany(connection):
     async with connection.cursor(cursor=DictCursor) as cursor:
         rows = await cursor.executemany(
-            """INSERT INTO test.asyncmy(`decimal`, date, datetime, `float`, string, `tinyint`) VALUES (%s,%s,%s,%s,%s,%s)""",
+            """INSERT INTO test.asyncmy(`decimal`, date, datetime, time, `float`, string, `tinyint`) VALUES (%s,%s,%s,%s,%s,%s,%s)""",
             [
                 (
                     1,
                     "2020-08-08",
                     "2020-08-08 00:00:00",
+                    "00:00:00",
                     1,
                     "1",
                     1,
@@ -72,6 +90,7 @@ async def test_executemany(connection):
                     1,
                     "2020-08-08",
                     "2020-08-08 00:00:00",
+                    "00:00:00",
                     1,
                     "1",
                     1,
@@ -105,12 +124,13 @@ class EnumValue(str, Enum):
 async def test_insert_enum(connection):
     async with connection.cursor(cursor=DictCursor) as cursor:
         rows = await cursor.execute(
-            """INSERT INTO test.asyncmy(id,`decimal`, date, datetime, `float`, string, `tinyint`) VALUES (%s,%s,%s,%s,%s,%s,%s)""",
+            """INSERT INTO test.asyncmy(id, `decimal`, date, datetime, time, `float`, string, `tinyint`) VALUES (%s,%s,%s,%s,%s,%s,%s,%s)""",
             (
                 -1,
                 1,
                 "2020-08-08",
                 "2020-08-08 00:00:00",
+                "00:00:00",
                 1,
                 EnumValue.VALUE,
                 1,