from datetime import date, datetime, timedelta, timezone

from django.forms import DateTimeField, ValidationError
from django.test import SimpleTestCase
from django.utils.timezone import utc


class DateTimeFieldTest(SimpleTestCase):

    def test_datetimefield_clean_defaults(self):
        f = DateTimeField()
        self.assertEqual(datetime(2006, 10, 25, 0, 0), f.clean(date(2006, 10, 25)))
        self.assertEqual(datetime(2006, 10, 25, 14, 30), f.clean(datetime(2006, 10, 25, 14, 30)))
        self.assertEqual(
            datetime(2006, 10, 25, 14, 30, 59),
            f.clean(datetime(2006, 10, 25, 14, 30, 59))
        )
        self.assertEqual(
            datetime(2006, 10, 25, 14, 30, 59, 200),
            f.clean(datetime(2006, 10, 25, 14, 30, 59, 200))
        )
        self.assertEqual(datetime(2006, 10, 25, 14, 30, 45, 200), f.clean('2006-10-25 14:30:45.000200'))
        self.assertEqual(datetime(2006, 10, 25, 14, 30, 45, 200), f.clean('2006-10-25 14:30:45.0002'))
        self.assertEqual(datetime(2006, 10, 25, 14, 30, 45), f.clean('2006-10-25 14:30:45'))
        self.assertEqual(datetime(2006, 10, 25, 14, 30), f.clean('2006-10-25 14:30:00'))
        self.assertEqual(datetime(2006, 10, 25, 14, 30), f.clean('2006-10-25 14:30'))
        self.assertEqual(datetime(2006, 10, 25, 0, 0), f.clean('2006-10-25'))
        self.assertEqual(datetime(2006, 10, 25, 14, 30, 45, 200), f.clean('10/25/2006 14:30:45.000200'))
        self.assertEqual(datetime(2006, 10, 25, 14, 30, 45), f.clean('10/25/2006 14:30:45'))
        self.assertEqual(datetime(2006, 10, 25, 14, 30), f.clean('10/25/2006 14:30:00'))
        self.assertEqual(datetime(2006, 10, 25, 14, 30), f.clean('10/25/2006 14:30'))
        self.assertEqual(datetime(2006, 10, 25, 0, 0), f.clean('10/25/2006'))
        self.assertEqual(datetime(2006, 10, 25, 14, 30, 45, 200), f.clean('10/25/06 14:30:45.000200'))
        self.assertEqual(datetime(2006, 10, 25, 14, 30, 45), f.clean('10/25/06 14:30:45'))
        self.assertEqual(datetime(2006, 10, 25, 14, 30), f.clean('10/25/06 14:30:00'))
        self.assertEqual(datetime(2006, 10, 25, 14, 30), f.clean('10/25/06 14:30'))
        self.assertEqual(datetime(2006, 10, 25, 0, 0), f.clean('10/25/06'))
        with self.assertRaisesMessage(ValidationError, "'Enter a valid date/time.'"):
            f.clean('hello')
        with self.assertRaisesMessage(ValidationError, "'Enter a valid date/time.'"):
            f.clean('2006-10-25 4:30 p.m.')

    def test_datetimefield_iso8601(self):
        f = DateTimeField()
        self.assertEqual(
            datetime(2014, 9, 23, 22, 34, 41, 614804),
            f.clean('2014-09-23T22:34:41.614804')
        )
        self.assertEqual(datetime(2014, 9, 23, 22, 34, 41), f.clean('2014-09-23T22:34:41'))
        self.assertEqual(datetime(2014, 9, 23, 22, 34), f.clean('2014-09-23T22:34'))
        self.assertEqual(
            datetime(2014, 9, 23, 22, 34, tzinfo=utc),
            f.clean('2014-09-23T22:34Z')
        )
        self.assertEqual(
            datetime(2014, 9, 23, 22, 34, tzinfo=timezone(timedelta(hours=7))),
            f.clean('2014-09-23T22:34+07:00')
        )
        with self.assertRaisesMessage(ValidationError, "'Enter a valid date/time.'"):
            f.clean('2014-09-23T28:23')

    def test_datetimefield_clean_input_formats(self):
        f = DateTimeField(input_formats=['%Y %m %d %I:%M %p'])
        self.assertEqual(datetime(2006, 10, 25, 0, 0), f.clean(date(2006, 10, 25)))
        self.assertEqual(datetime(2006, 10, 25, 14, 30), f.clean(datetime(2006, 10, 25, 14, 30)))
        self.assertEqual(
            datetime(2006, 10, 25, 14, 30, 59),
            f.clean(datetime(2006, 10, 25, 14, 30, 59))
        )
        self.assertEqual(
            datetime(2006, 10, 25, 14, 30, 59, 200),
            f.clean(datetime(2006, 10, 25, 14, 30, 59, 200))
        )
        self.assertEqual(datetime(2006, 10, 25, 14, 30), f.clean('2006 10 25 2:30 PM'))
        # ISO-like formats are always accepted, even when not in input_formats.
        self.assertEqual(datetime(2006, 10, 25, 14, 30, 45), f.clean('2006-10-25 14:30:45'))
        with self.assertRaisesMessage(ValidationError, "'Enter a valid date/time.'"):
            f.clean('2006.10.25 14:30:45')

        f = DateTimeField(input_formats=['%Y.%m.%d %H:%M:%S.%f'])
        self.assertEqual(
            datetime(2006, 10, 25, 14, 30, 45, 200),
            f.clean('2006.10.25 14:30:45.0002')
        )

    def test_datetimefield_not_required(self):
        f = DateTimeField(required=False)
        self.assertIsNone(f.clean(None))
        self.assertEqual('None', repr(f.clean(None)))
        self.assertIsNone(f.clean(''))
        self.assertEqual('None', repr(f.clean('')))

    def test_datetimefield_whitespace_stripping(self):
        f = DateTimeField()
        # Test whitespace stripping behavior (#5714)
        self.assertEqual(datetime(2006, 10, 25, 14, 30, 45), f.clean(' 2006-10-25   14:30:45 '))
        self.assertEqual(datetime(2006, 10, 25, 0, 0), f.clean(' 2006-10-25 '))
        self.assertEqual(datetime(2006, 10, 25, 14, 30, 45), f.clean(' 10/25/2006 14:30:45 '))
        self.assertEqual(datetime(2006, 10, 25, 14, 30), f.clean(' 10/25/2006 14:30 '))
        self.assertEqual(datetime(2006, 10, 25, 0, 0), f.clean(' 10/25/2006 '))
        self.assertEqual(datetime(2006, 10, 25, 14, 30, 45), f.clean(' 10/25/06 14:30:45 '))
        self.assertEqual(datetime(2006, 10, 25, 0, 0), f.clean(' 10/25/06 '))
        with self.assertRaisesMessage(ValidationError, "'Enter a valid date/time.'"):
            f.clean('   ')

    def test_datetimefield_changed(self):
        format = '%Y %m %d %I:%M %p'
        f = DateTimeField(input_formats=[format])
        d = datetime(2006, 9, 17, 14, 30, 0)
        self.assertFalse(f.has_changed(d, '2006 09 17 2:30 PM'))