Source code for sqlobject.tests.test_datetime

from datetime import datetime, date, time
import pytest

from sqlobject import SQLObject
from sqlobject import col
from sqlobject.col import DateCol, DateTimeCol, TimeCol, use_microseconds, \
    DATETIME_IMPLEMENTATION, MXDATETIME_IMPLEMENTATION, mxdatetime_available, \
    ZOPE_DATETIME_IMPLEMENTATION, zope_datetime_available
from sqlobject.tests.dbtest import getConnection, setupClass
from sqlobject.converters import pendulumDateTimeType

########################################
# Date/time columns
########################################


col.default_datetime_implementation = DATETIME_IMPLEMENTATION


[docs]class DateTime1(SQLObject): col1 = DateTimeCol() col2 = DateCol() col3 = TimeCol()
[docs]def test_dateTime(): setupClass(DateTime1) _now = datetime.now().replace(microsecond=0) dt1 = DateTime1(col1=_now, col2=_now, col3=_now.time()) assert isinstance(dt1.col1, datetime) assert dt1.col1.year == _now.year assert dt1.col1.month == _now.month assert dt1.col1.day == _now.day assert dt1.col1.hour == _now.hour assert dt1.col1.minute == _now.minute assert dt1.col1.second == _now.second assert isinstance(dt1.col2, date) assert not isinstance(dt1.col2, datetime) assert dt1.col2.year == _now.year assert dt1.col2.month == _now.month assert dt1.col2.day == _now.day assert isinstance(dt1.col3, time) assert dt1.col3.hour == _now.hour assert dt1.col3.minute == _now.minute assert dt1.col3.second == _now.second
[docs]def test_microseconds(): connection = getConnection() if not hasattr(connection, 'can_use_microseconds') or \ not connection.can_use_microseconds(): pytest.skip( "The database doesn't support microseconds; " "microseconds are supported by MariaDB since version 5.3.0, " "by MySQL since version 5.6.4, " "by MSSQL since MS SQL Server 2008.") setupClass(DateTime1) _now = datetime.now() dt1 = DateTime1(col1=_now, col2=_now, col3=_now.time()) assert dt1.col1.microsecond == _now.microsecond assert dt1.col3.microsecond == _now.microsecond use_microseconds(False) setupClass(DateTime1) _now = datetime.now() dt1 = DateTime1(col1=_now, col2=_now, col3=_now.time()) assert dt1.col1.microsecond == 0 assert dt1.col3.microsecond == 0 use_microseconds(True) setupClass(DateTime1) _now = datetime.now() dt1 = DateTime1(col1=_now, col2=_now, col3=_now.time()) assert dt1.col1.microsecond == _now.microsecond assert dt1.col3.microsecond == _now.microsecond
if mxdatetime_available: col.default_datetime_implementation = MXDATETIME_IMPLEMENTATION from mx.DateTime import now as mx_now, Time as mxTime dateFormat = None # use default try: connection = getConnection() except AttributeError: # The module was imported during documentation building pass else: if connection.dbName == "sqlite": if connection.using_sqlite2: # mxDateTime sends and PySQLite2 returns # full date/time for dates dateFormat = "%Y-%m-%d %H:%M:%S.%f" class DateTime2(SQLObject): col1 = DateTimeCol() col2 = DateCol(dateFormat=dateFormat) col3 = TimeCol() def test_mxDateTime(): setupClass(DateTime2) _now = mx_now() dt2 = DateTime2(col1=_now, col2=_now.pydate(), col3=mxTime(_now.hour, _now.minute, _now.second)) assert isinstance(dt2.col1, col.mxDateTimeType) assert dt2.col1.year == _now.year assert dt2.col1.month == _now.month assert dt2.col1.day == _now.day assert dt2.col1.hour == _now.hour assert dt2.col1.minute == _now.minute assert dt2.col1.second == int(_now.second) assert isinstance(dt2.col2, col.mxDateTimeType) assert dt2.col2.year == _now.year assert dt2.col2.month == _now.month assert dt2.col2.day == _now.day assert dt2.col2.hour == 0 assert dt2.col2.minute == 0 assert dt2.col2.second == 0 assert isinstance(dt2.col3, (col.mxDateTimeType, col.mxTimeType)) assert dt2.col3.hour == _now.hour assert dt2.col3.minute == _now.minute assert dt2.col3.second == int(_now.second) if pendulumDateTimeType: col.default_datetime_implementation = DATETIME_IMPLEMENTATION import pendulum class DateTimePendulum(SQLObject): col1 = DateTimeCol() def test_PendulumDateTime(): setupClass(DateTimePendulum) _now = pendulum.now() dtp = DateTimePendulum(col1=_now) assert isinstance(dtp.col1, datetime) assert dtp.col1.year == _now.year assert dtp.col1.month == _now.month assert dtp.col1.day == _now.day assert dtp.col1.hour == _now.hour assert dtp.col1.minute == _now.minute assert int(dtp.col1.second) == int(_now.second) if zope_datetime_available: col.default_datetime_implementation = ZOPE_DATETIME_IMPLEMENTATION from DateTime import DateTime as zopeDateTime class DateTime3(SQLObject): col1 = DateTimeCol() def test_ZopeDateTime(): setupClass(DateTime3) _now = zopeDateTime() dt3 = DateTime3(col1=_now) assert isinstance(dt3.col1, col.zopeDateTimeType) assert dt3.col1.year() == _now.year() assert dt3.col1.month() == _now.month() assert dt3.col1.day() == _now.day() assert dt3.col1.hour() == _now.hour() assert dt3.col1.minute() == _now.minute() assert int(dt3.col1.second()) == int(_now.second())