当时间00:00:00时,如何阻止python将mySQL DATETIME转换为datetime.date？

Question

我正在从mySQL数据库中读取各种数据类型.第五列在数据库中具有类型"DATETIME".我将它用作'BloodTraitRecord'对象的entry_date.

import mysql.connector
from datetime import timedelta
from datetime import datetime

show_DB = """select  RUID, test_sname, test_value, units, ref_range, entry_date from %s
             where RUID=%%s and test_sname=%%s order by RUID,
             test_sname, entry_date Limit 5;""" % (tableToUse,)

cursor.execute(show_DB, (ruid, traitPair[0]))
resultsForOneTrait = cursor.fetchall()

for result in resultsForOneTrait:
    ruid = result[0]
    s_name = result[1].decode("UTF-8")
    value = result[2]
    units = result[3].decode("UTF-8")
    ref_range = result[4].decode("UTF-8")

    # Need assistance here
    entryDate = result[5]

    record = BloodTraitRecord(ruid, s_name, value, units, ref_range, entryDate)

BloodTraitRecord类:

class BloodTraitRecord:

def __init__(self, ruid, test_sname, test_value, units, ref_range, entry_date):
    self.RUID = ruid
    self.test_sname = test_sname     
    self.test_value = test_value
    self.units = units               
    self.ref_range = ref_range


    self.entry_date = entry_date

数据库中的DATETIME对象在mySQL服务器中如下所示:

'2008-11-14 13:28:00'

代码按预期运行,除非数据库中的时间是午夜,如下所示:

'2014-05-18 00:00:00'

在那种情况下,仅在这种情况下,我在将记录的entry_date.date()与代码中的另一个datetime.date进行比较时得到此错误:

# 'cancerCutoff' is consistently a datetime.date 
cancerCutoff = firstCancerAnemiaCodeDate[ruidkey] - timedelta(days=180)
if cancerCutoff < record.entry_date.date():
AttributeError: 'datetime.date' object has no attribute 'date'

打印record.entry_date确认此情况的时间属性已消失:

'2014-05-18'

我有办法通过检查对象的类型来解决这个问题,如果对象是一个日期时间,只调用date属性,但我想知道是否有比这更好的修复.

我也不明白为什么当DATETIME时间是00:00:00时,python会立即将MySQL DATETIME转换为datetime.date.

谢谢你的帮助!

Answer 1

我会确保您从数据库中提取日期时间对象后立即拥有它。那么以后就不用再做任何检查了。所以你可以说：

entryDate = ensure_datetime(result[5])

这只是一点额外的代码，而且还有一个优点，如果你的查询发生变化并且你没有正确更新代码，它会立即捕获类型错误。这是一个示例实现：

from datetime import datetime, date

# Thanks to http://stackoverflow.com/a/1937636/2482744
def date_to_datetime(d):
    return datetime.combine(d, datetime.min.time())

def ensure_datetime(d):
    if isinstance(d, datetime):
        return d
    elif isinstance(d, date):
        return date_to_datetime(d)
    else:
        raise TypeError('%s is neither a date nor a datetime' % d)

演示：

for x in [date(2016, 5, 12),
          datetime(2016, 5, 12, 9, 32, 57, 345),
          'a string']:
    print(ensure_datetime(x))

输出：

2016-05-12 00:00:00
2016-05-12 09:32:57.000345
Traceback (most recent call last):
  File "/Users/alexhall/Dropbox/python/sandbox/sandbox.py", line 14, in <module>
    print(ensure_datetime(x))
  File "/Users/alexhall/Dropbox/python/sandbox/sandbox.py", line 9, in ensure_datetime
    raise TypeError('%s is neither a date nor a datetime' % d)
TypeError: a string is neither a date nor a datetime

但我感觉你不想这样做，所以我会按如下方式进行修改：

def clean_types(row):
    new_row = []
    for item in row:
        if isinstance(item, date) and not isinstance(item, datetime):
            item = date_to_datetime(item)
        elif isinstance(item, str):
            item = item.decode("UTF-8")
        new_row.append(item)
    return new_row

# Demo
print(clean_types([3, 'abc', u'def', date.today(), datetime.now()]))
# [3, u'abc', u'def', datetime.datetime(2016, 5, 12, 0, 0), datetime.datetime(2016, 5, 12, 17, 22, 7, 519604)]

现在您的代码可以缩短为：

for result in resultsForOneTrait:
    record = BloodTraitRecord(*clean_types(result))

而且你不需要记住做任何事情。