File tree Expand file tree Collapse file tree 2 files changed +8
-2
lines changed Expand file tree Collapse file tree 2 files changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -77,7 +77,9 @@ def datetime_to_string(val: datetime.datetime) -> str:
77
77
Returns:
78
78
datetime serialized to iso8601 format.
79
79
"""
80
- return val .isoformat ().replace ('+00:00' , 'Z' )
80
+ if isinstance (val , datetime .datetime ):
81
+ return val .isoformat ().replace ('+00:00' , 'Z' )
82
+ return val
81
83
82
84
def string_to_datetime (string : str ) -> datetime .datetime :
83
85
"""De-serializes string to datetime.
@@ -99,7 +101,9 @@ def date_to_string(val: datetime.date) -> str:
99
101
Returns:
100
102
date serialized to `YYYY-MM-DD` format.
101
103
"""
102
- return str (val )
104
+ if isinstance (val , datetime .date ):
105
+ return str (val )
106
+ return val
103
107
104
108
def string_to_date (string : str ) -> datetime .date :
105
109
"""De-serializes string to date.
Original file line number Diff line number Diff line change @@ -11,12 +11,14 @@ def test_datetime_conversion():
11
11
assert date .day == 6
12
12
res = datetime_to_string (date )
13
13
assert res == '2017-03-06T16:00:04.159338'
14
+ assert datetime_to_string (None ) is None
14
15
15
16
def test_date_conversion ():
16
17
date = string_to_date ('2017-03-06' )
17
18
assert date .day == 6
18
19
res = date_to_string (date )
19
20
assert res == '2017-03-06'
21
+ assert date_to_string (None ) is None
20
22
21
23
def test_convert_model ():
22
24
You can’t perform that action at this time.
0 commit comments