Added information on function name added to LogRecord, and the 'extra' keyword parameter.
[python.git] / Modules / datetimemodule.c
blob50f47d45f46b129ca356adee5a6197ff6e2ce83d
1 /* C implementation for the date/time type documented at
2 * http://www.zope.org/Members/fdrake/DateTimeWiki/FrontPage
3 */
5 #include "Python.h"
6 #include "modsupport.h"
7 #include "structmember.h"
9 #include <time.h>
11 #include "timefuncs.h"
13 /* Differentiate between building the core module and building extension
14 * modules.
16 #define Py_BUILD_CORE
17 #include "datetime.h"
18 #undef Py_BUILD_CORE
20 /* We require that C int be at least 32 bits, and use int virtually
21 * everywhere. In just a few cases we use a temp long, where a Python
22 * API returns a C long. In such cases, we have to ensure that the
23 * final result fits in a C int (this can be an issue on 64-bit boxes).
25 #if SIZEOF_INT < 4
26 # error "datetime.c requires that C int have at least 32 bits"
27 #endif
29 #define MINYEAR 1
30 #define MAXYEAR 9999
32 /* Nine decimal digits is easy to communicate, and leaves enough room
33 * so that two delta days can be added w/o fear of overflowing a signed
34 * 32-bit int, and with plenty of room left over to absorb any possible
35 * carries from adding seconds.
37 #define MAX_DELTA_DAYS 999999999
39 /* Rename the long macros in datetime.h to more reasonable short names. */
40 #define GET_YEAR PyDateTime_GET_YEAR
41 #define GET_MONTH PyDateTime_GET_MONTH
42 #define GET_DAY PyDateTime_GET_DAY
43 #define DATE_GET_HOUR PyDateTime_DATE_GET_HOUR
44 #define DATE_GET_MINUTE PyDateTime_DATE_GET_MINUTE
45 #define DATE_GET_SECOND PyDateTime_DATE_GET_SECOND
46 #define DATE_GET_MICROSECOND PyDateTime_DATE_GET_MICROSECOND
48 /* Date accessors for date and datetime. */
49 #define SET_YEAR(o, v) (((o)->data[0] = ((v) & 0xff00) >> 8), \
50 ((o)->data[1] = ((v) & 0x00ff)))
51 #define SET_MONTH(o, v) (PyDateTime_GET_MONTH(o) = (v))
52 #define SET_DAY(o, v) (PyDateTime_GET_DAY(o) = (v))
54 /* Date/Time accessors for datetime. */
55 #define DATE_SET_HOUR(o, v) (PyDateTime_DATE_GET_HOUR(o) = (v))
56 #define DATE_SET_MINUTE(o, v) (PyDateTime_DATE_GET_MINUTE(o) = (v))
57 #define DATE_SET_SECOND(o, v) (PyDateTime_DATE_GET_SECOND(o) = (v))
58 #define DATE_SET_MICROSECOND(o, v) \
59 (((o)->data[7] = ((v) & 0xff0000) >> 16), \
60 ((o)->data[8] = ((v) & 0x00ff00) >> 8), \
61 ((o)->data[9] = ((v) & 0x0000ff)))
63 /* Time accessors for time. */
64 #define TIME_GET_HOUR PyDateTime_TIME_GET_HOUR
65 #define TIME_GET_MINUTE PyDateTime_TIME_GET_MINUTE
66 #define TIME_GET_SECOND PyDateTime_TIME_GET_SECOND
67 #define TIME_GET_MICROSECOND PyDateTime_TIME_GET_MICROSECOND
68 #define TIME_SET_HOUR(o, v) (PyDateTime_TIME_GET_HOUR(o) = (v))
69 #define TIME_SET_MINUTE(o, v) (PyDateTime_TIME_GET_MINUTE(o) = (v))
70 #define TIME_SET_SECOND(o, v) (PyDateTime_TIME_GET_SECOND(o) = (v))
71 #define TIME_SET_MICROSECOND(o, v) \
72 (((o)->data[3] = ((v) & 0xff0000) >> 16), \
73 ((o)->data[4] = ((v) & 0x00ff00) >> 8), \
74 ((o)->data[5] = ((v) & 0x0000ff)))
76 /* Delta accessors for timedelta. */
77 #define GET_TD_DAYS(o) (((PyDateTime_Delta *)(o))->days)
78 #define GET_TD_SECONDS(o) (((PyDateTime_Delta *)(o))->seconds)
79 #define GET_TD_MICROSECONDS(o) (((PyDateTime_Delta *)(o))->microseconds)
81 #define SET_TD_DAYS(o, v) ((o)->days = (v))
82 #define SET_TD_SECONDS(o, v) ((o)->seconds = (v))
83 #define SET_TD_MICROSECONDS(o, v) ((o)->microseconds = (v))
85 /* p is a pointer to a time or a datetime object; HASTZINFO(p) returns
86 * p->hastzinfo.
88 #define HASTZINFO(p) (((_PyDateTime_BaseTZInfo *)(p))->hastzinfo)
90 /* M is a char or int claiming to be a valid month. The macro is equivalent
91 * to the two-sided Python test
92 * 1 <= M <= 12
94 #define MONTH_IS_SANE(M) ((unsigned int)(M) - 1 < 12)
96 /* Forward declarations. */
97 static PyTypeObject PyDateTime_DateType;
98 static PyTypeObject PyDateTime_DateTimeType;
99 static PyTypeObject PyDateTime_DeltaType;
100 static PyTypeObject PyDateTime_TimeType;
101 static PyTypeObject PyDateTime_TZInfoType;
103 /* ---------------------------------------------------------------------------
104 * Math utilities.
107 /* k = i+j overflows iff k differs in sign from both inputs,
108 * iff k^i has sign bit set and k^j has sign bit set,
109 * iff (k^i)&(k^j) has sign bit set.
111 #define SIGNED_ADD_OVERFLOWED(RESULT, I, J) \
112 ((((RESULT) ^ (I)) & ((RESULT) ^ (J))) < 0)
114 /* Compute Python divmod(x, y), returning the quotient and storing the
115 * remainder into *r. The quotient is the floor of x/y, and that's
116 * the real point of this. C will probably truncate instead (C99
117 * requires truncation; C89 left it implementation-defined).
118 * Simplification: we *require* that y > 0 here. That's appropriate
119 * for all the uses made of it. This simplifies the code and makes
120 * the overflow case impossible (divmod(LONG_MIN, -1) is the only
121 * overflow case).
123 static int
124 divmod(int x, int y, int *r)
126 int quo;
128 assert(y > 0);
129 quo = x / y;
130 *r = x - quo * y;
131 if (*r < 0) {
132 --quo;
133 *r += y;
135 assert(0 <= *r && *r < y);
136 return quo;
139 /* Round a double to the nearest long. |x| must be small enough to fit
140 * in a C long; this is not checked.
142 static long
143 round_to_long(double x)
145 if (x >= 0.0)
146 x = floor(x + 0.5);
147 else
148 x = ceil(x - 0.5);
149 return (long)x;
152 /* ---------------------------------------------------------------------------
153 * General calendrical helper functions
156 /* For each month ordinal in 1..12, the number of days in that month,
157 * and the number of days before that month in the same year. These
158 * are correct for non-leap years only.
160 static int _days_in_month[] = {
161 0, /* unused; this vector uses 1-based indexing */
162 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31
165 static int _days_before_month[] = {
166 0, /* unused; this vector uses 1-based indexing */
167 0, 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334
170 /* year -> 1 if leap year, else 0. */
171 static int
172 is_leap(int year)
174 /* Cast year to unsigned. The result is the same either way, but
175 * C can generate faster code for unsigned mod than for signed
176 * mod (especially for % 4 -- a good compiler should just grab
177 * the last 2 bits when the LHS is unsigned).
179 const unsigned int ayear = (unsigned int)year;
180 return ayear % 4 == 0 && (ayear % 100 != 0 || ayear % 400 == 0);
183 /* year, month -> number of days in that month in that year */
184 static int
185 days_in_month(int year, int month)
187 assert(month >= 1);
188 assert(month <= 12);
189 if (month == 2 && is_leap(year))
190 return 29;
191 else
192 return _days_in_month[month];
195 /* year, month -> number of days in year preceeding first day of month */
196 static int
197 days_before_month(int year, int month)
199 int days;
201 assert(month >= 1);
202 assert(month <= 12);
203 days = _days_before_month[month];
204 if (month > 2 && is_leap(year))
205 ++days;
206 return days;
209 /* year -> number of days before January 1st of year. Remember that we
210 * start with year 1, so days_before_year(1) == 0.
212 static int
213 days_before_year(int year)
215 int y = year - 1;
216 /* This is incorrect if year <= 0; we really want the floor
217 * here. But so long as MINYEAR is 1, the smallest year this
218 * can see is 0 (this can happen in some normalization endcases),
219 * so we'll just special-case that.
221 assert (year >= 0);
222 if (y >= 0)
223 return y*365 + y/4 - y/100 + y/400;
224 else {
225 assert(y == -1);
226 return -366;
230 /* Number of days in 4, 100, and 400 year cycles. That these have
231 * the correct values is asserted in the module init function.
233 #define DI4Y 1461 /* days_before_year(5); days in 4 years */
234 #define DI100Y 36524 /* days_before_year(101); days in 100 years */
235 #define DI400Y 146097 /* days_before_year(401); days in 400 years */
237 /* ordinal -> year, month, day, considering 01-Jan-0001 as day 1. */
238 static void
239 ord_to_ymd(int ordinal, int *year, int *month, int *day)
241 int n, n1, n4, n100, n400, leapyear, preceding;
243 /* ordinal is a 1-based index, starting at 1-Jan-1. The pattern of
244 * leap years repeats exactly every 400 years. The basic strategy is
245 * to find the closest 400-year boundary at or before ordinal, then
246 * work with the offset from that boundary to ordinal. Life is much
247 * clearer if we subtract 1 from ordinal first -- then the values
248 * of ordinal at 400-year boundaries are exactly those divisible
249 * by DI400Y:
251 * D M Y n n-1
252 * -- --- ---- ---------- ----------------
253 * 31 Dec -400 -DI400Y -DI400Y -1
254 * 1 Jan -399 -DI400Y +1 -DI400Y 400-year boundary
255 * ...
256 * 30 Dec 000 -1 -2
257 * 31 Dec 000 0 -1
258 * 1 Jan 001 1 0 400-year boundary
259 * 2 Jan 001 2 1
260 * 3 Jan 001 3 2
261 * ...
262 * 31 Dec 400 DI400Y DI400Y -1
263 * 1 Jan 401 DI400Y +1 DI400Y 400-year boundary
265 assert(ordinal >= 1);
266 --ordinal;
267 n400 = ordinal / DI400Y;
268 n = ordinal % DI400Y;
269 *year = n400 * 400 + 1;
271 /* Now n is the (non-negative) offset, in days, from January 1 of
272 * year, to the desired date. Now compute how many 100-year cycles
273 * precede n.
274 * Note that it's possible for n100 to equal 4! In that case 4 full
275 * 100-year cycles precede the desired day, which implies the
276 * desired day is December 31 at the end of a 400-year cycle.
278 n100 = n / DI100Y;
279 n = n % DI100Y;
281 /* Now compute how many 4-year cycles precede it. */
282 n4 = n / DI4Y;
283 n = n % DI4Y;
285 /* And now how many single years. Again n1 can be 4, and again
286 * meaning that the desired day is December 31 at the end of the
287 * 4-year cycle.
289 n1 = n / 365;
290 n = n % 365;
292 *year += n100 * 100 + n4 * 4 + n1;
293 if (n1 == 4 || n100 == 4) {
294 assert(n == 0);
295 *year -= 1;
296 *month = 12;
297 *day = 31;
298 return;
301 /* Now the year is correct, and n is the offset from January 1. We
302 * find the month via an estimate that's either exact or one too
303 * large.
305 leapyear = n1 == 3 && (n4 != 24 || n100 == 3);
306 assert(leapyear == is_leap(*year));
307 *month = (n + 50) >> 5;
308 preceding = (_days_before_month[*month] + (*month > 2 && leapyear));
309 if (preceding > n) {
310 /* estimate is too large */
311 *month -= 1;
312 preceding -= days_in_month(*year, *month);
314 n -= preceding;
315 assert(0 <= n);
316 assert(n < days_in_month(*year, *month));
318 *day = n + 1;
321 /* year, month, day -> ordinal, considering 01-Jan-0001 as day 1. */
322 static int
323 ymd_to_ord(int year, int month, int day)
325 return days_before_year(year) + days_before_month(year, month) + day;
328 /* Day of week, where Monday==0, ..., Sunday==6. 1/1/1 was a Monday. */
329 static int
330 weekday(int year, int month, int day)
332 return (ymd_to_ord(year, month, day) + 6) % 7;
335 /* Ordinal of the Monday starting week 1 of the ISO year. Week 1 is the
336 * first calendar week containing a Thursday.
338 static int
339 iso_week1_monday(int year)
341 int first_day = ymd_to_ord(year, 1, 1); /* ord of 1/1 */
342 /* 0 if 1/1 is a Monday, 1 if a Tue, etc. */
343 int first_weekday = (first_day + 6) % 7;
344 /* ordinal of closest Monday at or before 1/1 */
345 int week1_monday = first_day - first_weekday;
347 if (first_weekday > 3) /* if 1/1 was Fri, Sat, Sun */
348 week1_monday += 7;
349 return week1_monday;
352 /* ---------------------------------------------------------------------------
353 * Range checkers.
356 /* Check that -MAX_DELTA_DAYS <= days <= MAX_DELTA_DAYS. If so, return 0.
357 * If not, raise OverflowError and return -1.
359 static int
360 check_delta_day_range(int days)
362 if (-MAX_DELTA_DAYS <= days && days <= MAX_DELTA_DAYS)
363 return 0;
364 PyErr_Format(PyExc_OverflowError,
365 "days=%d; must have magnitude <= %d",
366 days, MAX_DELTA_DAYS);
367 return -1;
370 /* Check that date arguments are in range. Return 0 if they are. If they
371 * aren't, raise ValueError and return -1.
373 static int
374 check_date_args(int year, int month, int day)
377 if (year < MINYEAR || year > MAXYEAR) {
378 PyErr_SetString(PyExc_ValueError,
379 "year is out of range");
380 return -1;
382 if (month < 1 || month > 12) {
383 PyErr_SetString(PyExc_ValueError,
384 "month must be in 1..12");
385 return -1;
387 if (day < 1 || day > days_in_month(year, month)) {
388 PyErr_SetString(PyExc_ValueError,
389 "day is out of range for month");
390 return -1;
392 return 0;
395 /* Check that time arguments are in range. Return 0 if they are. If they
396 * aren't, raise ValueError and return -1.
398 static int
399 check_time_args(int h, int m, int s, int us)
401 if (h < 0 || h > 23) {
402 PyErr_SetString(PyExc_ValueError,
403 "hour must be in 0..23");
404 return -1;
406 if (m < 0 || m > 59) {
407 PyErr_SetString(PyExc_ValueError,
408 "minute must be in 0..59");
409 return -1;
411 if (s < 0 || s > 59) {
412 PyErr_SetString(PyExc_ValueError,
413 "second must be in 0..59");
414 return -1;
416 if (us < 0 || us > 999999) {
417 PyErr_SetString(PyExc_ValueError,
418 "microsecond must be in 0..999999");
419 return -1;
421 return 0;
424 /* ---------------------------------------------------------------------------
425 * Normalization utilities.
428 /* One step of a mixed-radix conversion. A "hi" unit is equivalent to
429 * factor "lo" units. factor must be > 0. If *lo is less than 0, or
430 * at least factor, enough of *lo is converted into "hi" units so that
431 * 0 <= *lo < factor. The input values must be such that int overflow
432 * is impossible.
434 static void
435 normalize_pair(int *hi, int *lo, int factor)
437 assert(factor > 0);
438 assert(lo != hi);
439 if (*lo < 0 || *lo >= factor) {
440 const int num_hi = divmod(*lo, factor, lo);
441 const int new_hi = *hi + num_hi;
442 assert(! SIGNED_ADD_OVERFLOWED(new_hi, *hi, num_hi));
443 *hi = new_hi;
445 assert(0 <= *lo && *lo < factor);
448 /* Fiddle days (d), seconds (s), and microseconds (us) so that
449 * 0 <= *s < 24*3600
450 * 0 <= *us < 1000000
451 * The input values must be such that the internals don't overflow.
452 * The way this routine is used, we don't get close.
454 static void
455 normalize_d_s_us(int *d, int *s, int *us)
457 if (*us < 0 || *us >= 1000000) {
458 normalize_pair(s, us, 1000000);
459 /* |s| can't be bigger than about
460 * |original s| + |original us|/1000000 now.
464 if (*s < 0 || *s >= 24*3600) {
465 normalize_pair(d, s, 24*3600);
466 /* |d| can't be bigger than about
467 * |original d| +
468 * (|original s| + |original us|/1000000) / (24*3600) now.
471 assert(0 <= *s && *s < 24*3600);
472 assert(0 <= *us && *us < 1000000);
475 /* Fiddle years (y), months (m), and days (d) so that
476 * 1 <= *m <= 12
477 * 1 <= *d <= days_in_month(*y, *m)
478 * The input values must be such that the internals don't overflow.
479 * The way this routine is used, we don't get close.
481 static void
482 normalize_y_m_d(int *y, int *m, int *d)
484 int dim; /* # of days in month */
486 /* This gets muddy: the proper range for day can't be determined
487 * without knowing the correct month and year, but if day is, e.g.,
488 * plus or minus a million, the current month and year values make
489 * no sense (and may also be out of bounds themselves).
490 * Saying 12 months == 1 year should be non-controversial.
492 if (*m < 1 || *m > 12) {
493 --*m;
494 normalize_pair(y, m, 12);
495 ++*m;
496 /* |y| can't be bigger than about
497 * |original y| + |original m|/12 now.
500 assert(1 <= *m && *m <= 12);
502 /* Now only day can be out of bounds (year may also be out of bounds
503 * for a datetime object, but we don't care about that here).
504 * If day is out of bounds, what to do is arguable, but at least the
505 * method here is principled and explainable.
507 dim = days_in_month(*y, *m);
508 if (*d < 1 || *d > dim) {
509 /* Move day-1 days from the first of the month. First try to
510 * get off cheap if we're only one day out of range
511 * (adjustments for timezone alone can't be worse than that).
513 if (*d == 0) {
514 --*m;
515 if (*m > 0)
516 *d = days_in_month(*y, *m);
517 else {
518 --*y;
519 *m = 12;
520 *d = 31;
523 else if (*d == dim + 1) {
524 /* move forward a day */
525 ++*m;
526 *d = 1;
527 if (*m > 12) {
528 *m = 1;
529 ++*y;
532 else {
533 int ordinal = ymd_to_ord(*y, *m, 1) +
534 *d - 1;
535 ord_to_ymd(ordinal, y, m, d);
538 assert(*m > 0);
539 assert(*d > 0);
542 /* Fiddle out-of-bounds months and days so that the result makes some kind
543 * of sense. The parameters are both inputs and outputs. Returns < 0 on
544 * failure, where failure means the adjusted year is out of bounds.
546 static int
547 normalize_date(int *year, int *month, int *day)
549 int result;
551 normalize_y_m_d(year, month, day);
552 if (MINYEAR <= *year && *year <= MAXYEAR)
553 result = 0;
554 else {
555 PyErr_SetString(PyExc_OverflowError,
556 "date value out of range");
557 result = -1;
559 return result;
562 /* Force all the datetime fields into range. The parameters are both
563 * inputs and outputs. Returns < 0 on error.
565 static int
566 normalize_datetime(int *year, int *month, int *day,
567 int *hour, int *minute, int *second,
568 int *microsecond)
570 normalize_pair(second, microsecond, 1000000);
571 normalize_pair(minute, second, 60);
572 normalize_pair(hour, minute, 60);
573 normalize_pair(day, hour, 24);
574 return normalize_date(year, month, day);
577 /* ---------------------------------------------------------------------------
578 * Basic object allocation: tp_alloc implementations. These allocate
579 * Python objects of the right size and type, and do the Python object-
580 * initialization bit. If there's not enough memory, they return NULL after
581 * setting MemoryError. All data members remain uninitialized trash.
583 * We abuse the tp_alloc "nitems" argument to communicate whether a tzinfo
584 * member is needed. This is ugly, imprecise, and possibly insecure.
585 * tp_basicsize for the time and datetime types is set to the size of the
586 * struct that has room for the tzinfo member, so subclasses in Python will
587 * allocate enough space for a tzinfo member whether or not one is actually
588 * needed. That's the "ugly and imprecise" parts. The "possibly insecure"
589 * part is that PyType_GenericAlloc() (which subclasses in Python end up
590 * using) just happens today to effectively ignore the nitems argument
591 * when tp_itemsize is 0, which it is for these type objects. If that
592 * changes, perhaps the callers of tp_alloc slots in this file should
593 * be changed to force a 0 nitems argument unless the type being allocated
594 * is a base type implemented in this file (so that tp_alloc is time_alloc
595 * or datetime_alloc below, which know about the nitems abuse).
598 static PyObject *
599 time_alloc(PyTypeObject *type, int aware)
601 PyObject *self;
603 self = (PyObject *)
604 PyObject_MALLOC(aware ?
605 sizeof(PyDateTime_Time) :
606 sizeof(_PyDateTime_BaseTime));
607 if (self == NULL)
608 return (PyObject *)PyErr_NoMemory();
609 PyObject_INIT(self, type);
610 return self;
613 static PyObject *
614 datetime_alloc(PyTypeObject *type, int aware)
616 PyObject *self;
618 self = (PyObject *)
619 PyObject_MALLOC(aware ?
620 sizeof(PyDateTime_DateTime) :
621 sizeof(_PyDateTime_BaseDateTime));
622 if (self == NULL)
623 return (PyObject *)PyErr_NoMemory();
624 PyObject_INIT(self, type);
625 return self;
628 /* ---------------------------------------------------------------------------
629 * Helpers for setting object fields. These work on pointers to the
630 * appropriate base class.
633 /* For date and datetime. */
634 static void
635 set_date_fields(PyDateTime_Date *self, int y, int m, int d)
637 self->hashcode = -1;
638 SET_YEAR(self, y);
639 SET_MONTH(self, m);
640 SET_DAY(self, d);
643 /* ---------------------------------------------------------------------------
644 * Create various objects, mostly without range checking.
647 /* Create a date instance with no range checking. */
648 static PyObject *
649 new_date_ex(int year, int month, int day, PyTypeObject *type)
651 PyDateTime_Date *self;
653 self = (PyDateTime_Date *) (type->tp_alloc(type, 0));
654 if (self != NULL)
655 set_date_fields(self, year, month, day);
656 return (PyObject *) self;
659 #define new_date(year, month, day) \
660 new_date_ex(year, month, day, &PyDateTime_DateType)
662 /* Create a datetime instance with no range checking. */
663 static PyObject *
664 new_datetime_ex(int year, int month, int day, int hour, int minute,
665 int second, int usecond, PyObject *tzinfo, PyTypeObject *type)
667 PyDateTime_DateTime *self;
668 char aware = tzinfo != Py_None;
670 self = (PyDateTime_DateTime *) (type->tp_alloc(type, aware));
671 if (self != NULL) {
672 self->hastzinfo = aware;
673 set_date_fields((PyDateTime_Date *)self, year, month, day);
674 DATE_SET_HOUR(self, hour);
675 DATE_SET_MINUTE(self, minute);
676 DATE_SET_SECOND(self, second);
677 DATE_SET_MICROSECOND(self, usecond);
678 if (aware) {
679 Py_INCREF(tzinfo);
680 self->tzinfo = tzinfo;
683 return (PyObject *)self;
686 #define new_datetime(y, m, d, hh, mm, ss, us, tzinfo) \
687 new_datetime_ex(y, m, d, hh, mm, ss, us, tzinfo, \
688 &PyDateTime_DateTimeType)
690 /* Create a time instance with no range checking. */
691 static PyObject *
692 new_time_ex(int hour, int minute, int second, int usecond,
693 PyObject *tzinfo, PyTypeObject *type)
695 PyDateTime_Time *self;
696 char aware = tzinfo != Py_None;
698 self = (PyDateTime_Time *) (type->tp_alloc(type, aware));
699 if (self != NULL) {
700 self->hastzinfo = aware;
701 self->hashcode = -1;
702 TIME_SET_HOUR(self, hour);
703 TIME_SET_MINUTE(self, minute);
704 TIME_SET_SECOND(self, second);
705 TIME_SET_MICROSECOND(self, usecond);
706 if (aware) {
707 Py_INCREF(tzinfo);
708 self->tzinfo = tzinfo;
711 return (PyObject *)self;
714 #define new_time(hh, mm, ss, us, tzinfo) \
715 new_time_ex(hh, mm, ss, us, tzinfo, &PyDateTime_TimeType)
717 /* Create a timedelta instance. Normalize the members iff normalize is
718 * true. Passing false is a speed optimization, if you know for sure
719 * that seconds and microseconds are already in their proper ranges. In any
720 * case, raises OverflowError and returns NULL if the normalized days is out
721 * of range).
723 static PyObject *
724 new_delta_ex(int days, int seconds, int microseconds, int normalize,
725 PyTypeObject *type)
727 PyDateTime_Delta *self;
729 if (normalize)
730 normalize_d_s_us(&days, &seconds, &microseconds);
731 assert(0 <= seconds && seconds < 24*3600);
732 assert(0 <= microseconds && microseconds < 1000000);
734 if (check_delta_day_range(days) < 0)
735 return NULL;
737 self = (PyDateTime_Delta *) (type->tp_alloc(type, 0));
738 if (self != NULL) {
739 self->hashcode = -1;
740 SET_TD_DAYS(self, days);
741 SET_TD_SECONDS(self, seconds);
742 SET_TD_MICROSECONDS(self, microseconds);
744 return (PyObject *) self;
747 #define new_delta(d, s, us, normalize) \
748 new_delta_ex(d, s, us, normalize, &PyDateTime_DeltaType)
750 /* ---------------------------------------------------------------------------
751 * tzinfo helpers.
754 /* Ensure that p is None or of a tzinfo subclass. Return 0 if OK; if not
755 * raise TypeError and return -1.
757 static int
758 check_tzinfo_subclass(PyObject *p)
760 if (p == Py_None || PyTZInfo_Check(p))
761 return 0;
762 PyErr_Format(PyExc_TypeError,
763 "tzinfo argument must be None or of a tzinfo subclass, "
764 "not type '%s'",
765 p->ob_type->tp_name);
766 return -1;
769 /* Return tzinfo.methname(tzinfoarg), without any checking of results.
770 * If tzinfo is None, returns None.
772 static PyObject *
773 call_tzinfo_method(PyObject *tzinfo, char *methname, PyObject *tzinfoarg)
775 PyObject *result;
777 assert(tzinfo && methname && tzinfoarg);
778 assert(check_tzinfo_subclass(tzinfo) >= 0);
779 if (tzinfo == Py_None) {
780 result = Py_None;
781 Py_INCREF(result);
783 else
784 result = PyObject_CallMethod(tzinfo, methname, "O", tzinfoarg);
785 return result;
788 /* If self has a tzinfo member, return a BORROWED reference to it. Else
789 * return NULL, which is NOT AN ERROR. There are no error returns here,
790 * and the caller must not decref the result.
792 static PyObject *
793 get_tzinfo_member(PyObject *self)
795 PyObject *tzinfo = NULL;
797 if (PyDateTime_Check(self) && HASTZINFO(self))
798 tzinfo = ((PyDateTime_DateTime *)self)->tzinfo;
799 else if (PyTime_Check(self) && HASTZINFO(self))
800 tzinfo = ((PyDateTime_Time *)self)->tzinfo;
802 return tzinfo;
805 /* Call getattr(tzinfo, name)(tzinfoarg), and extract an int from the
806 * result. tzinfo must be an instance of the tzinfo class. If the method
807 * returns None, this returns 0 and sets *none to 1. If the method doesn't
808 * return None or timedelta, TypeError is raised and this returns -1. If it
809 * returnsa timedelta and the value is out of range or isn't a whole number
810 * of minutes, ValueError is raised and this returns -1.
811 * Else *none is set to 0 and the integer method result is returned.
813 static int
814 call_utc_tzinfo_method(PyObject *tzinfo, char *name, PyObject *tzinfoarg,
815 int *none)
817 PyObject *u;
818 int result = -1;
820 assert(tzinfo != NULL);
821 assert(PyTZInfo_Check(tzinfo));
822 assert(tzinfoarg != NULL);
824 *none = 0;
825 u = call_tzinfo_method(tzinfo, name, tzinfoarg);
826 if (u == NULL)
827 return -1;
829 else if (u == Py_None) {
830 result = 0;
831 *none = 1;
833 else if (PyDelta_Check(u)) {
834 const int days = GET_TD_DAYS(u);
835 if (days < -1 || days > 0)
836 result = 24*60; /* trigger ValueError below */
837 else {
838 /* next line can't overflow because we know days
839 * is -1 or 0 now
841 int ss = days * 24 * 3600 + GET_TD_SECONDS(u);
842 result = divmod(ss, 60, &ss);
843 if (ss || GET_TD_MICROSECONDS(u)) {
844 PyErr_Format(PyExc_ValueError,
845 "tzinfo.%s() must return a "
846 "whole number of minutes",
847 name);
848 result = -1;
852 else {
853 PyErr_Format(PyExc_TypeError,
854 "tzinfo.%s() must return None or "
855 "timedelta, not '%s'",
856 name, u->ob_type->tp_name);
859 Py_DECREF(u);
860 if (result < -1439 || result > 1439) {
861 PyErr_Format(PyExc_ValueError,
862 "tzinfo.%s() returned %d; must be in "
863 "-1439 .. 1439",
864 name, result);
865 result = -1;
867 return result;
870 /* Call tzinfo.utcoffset(tzinfoarg), and extract an integer from the
871 * result. tzinfo must be an instance of the tzinfo class. If utcoffset()
872 * returns None, call_utcoffset returns 0 and sets *none to 1. If uctoffset()
873 * doesn't return None or timedelta, TypeError is raised and this returns -1.
874 * If utcoffset() returns an invalid timedelta (out of range, or not a whole
875 * # of minutes), ValueError is raised and this returns -1. Else *none is
876 * set to 0 and the offset is returned (as int # of minutes east of UTC).
878 static int
879 call_utcoffset(PyObject *tzinfo, PyObject *tzinfoarg, int *none)
881 return call_utc_tzinfo_method(tzinfo, "utcoffset", tzinfoarg, none);
884 /* Call tzinfo.name(tzinfoarg), and return the offset as a timedelta or None.
886 static PyObject *
887 offset_as_timedelta(PyObject *tzinfo, char *name, PyObject *tzinfoarg) {
888 PyObject *result;
890 assert(tzinfo && name && tzinfoarg);
891 if (tzinfo == Py_None) {
892 result = Py_None;
893 Py_INCREF(result);
895 else {
896 int none;
897 int offset = call_utc_tzinfo_method(tzinfo, name, tzinfoarg,
898 &none);
899 if (offset < 0 && PyErr_Occurred())
900 return NULL;
901 if (none) {
902 result = Py_None;
903 Py_INCREF(result);
905 else
906 result = new_delta(0, offset * 60, 0, 1);
908 return result;
911 /* Call tzinfo.dst(tzinfoarg), and extract an integer from the
912 * result. tzinfo must be an instance of the tzinfo class. If dst()
913 * returns None, call_dst returns 0 and sets *none to 1. If dst()
914 & doesn't return None or timedelta, TypeError is raised and this
915 * returns -1. If dst() returns an invalid timedelta for a UTC offset,
916 * ValueError is raised and this returns -1. Else *none is set to 0 and
917 * the offset is returned (as an int # of minutes east of UTC).
919 static int
920 call_dst(PyObject *tzinfo, PyObject *tzinfoarg, int *none)
922 return call_utc_tzinfo_method(tzinfo, "dst", tzinfoarg, none);
925 /* Call tzinfo.tzname(tzinfoarg), and return the result. tzinfo must be
926 * an instance of the tzinfo class or None. If tzinfo isn't None, and
927 * tzname() doesn't return None or a string, TypeError is raised and this
928 * returns NULL.
930 static PyObject *
931 call_tzname(PyObject *tzinfo, PyObject *tzinfoarg)
933 PyObject *result;
935 assert(tzinfo != NULL);
936 assert(check_tzinfo_subclass(tzinfo) >= 0);
937 assert(tzinfoarg != NULL);
939 if (tzinfo == Py_None) {
940 result = Py_None;
941 Py_INCREF(result);
943 else
944 result = PyObject_CallMethod(tzinfo, "tzname", "O", tzinfoarg);
946 if (result != NULL && result != Py_None && ! PyString_Check(result)) {
947 PyErr_Format(PyExc_TypeError, "tzinfo.tzname() must "
948 "return None or a string, not '%s'",
949 result->ob_type->tp_name);
950 Py_DECREF(result);
951 result = NULL;
953 return result;
956 typedef enum {
957 /* an exception has been set; the caller should pass it on */
958 OFFSET_ERROR,
960 /* type isn't date, datetime, or time subclass */
961 OFFSET_UNKNOWN,
963 /* date,
964 * datetime with !hastzinfo
965 * datetime with None tzinfo,
966 * datetime where utcoffset() returns None
967 * time with !hastzinfo
968 * time with None tzinfo,
969 * time where utcoffset() returns None
971 OFFSET_NAIVE,
973 /* time or datetime where utcoffset() doesn't return None */
974 OFFSET_AWARE,
975 } naivety;
977 /* Classify an object as to whether it's naive or offset-aware. See
978 * the "naivety" typedef for details. If the type is aware, *offset is set
979 * to minutes east of UTC (as returned by the tzinfo.utcoffset() method).
980 * If the type is offset-naive (or unknown, or error), *offset is set to 0.
981 * tzinfoarg is the argument to pass to the tzinfo.utcoffset() method.
983 static naivety
984 classify_utcoffset(PyObject *op, PyObject *tzinfoarg, int *offset)
986 int none;
987 PyObject *tzinfo;
989 assert(tzinfoarg != NULL);
990 *offset = 0;
991 tzinfo = get_tzinfo_member(op); /* NULL means no tzinfo, not error */
992 if (tzinfo == Py_None)
993 return OFFSET_NAIVE;
994 if (tzinfo == NULL) {
995 /* note that a datetime passes the PyDate_Check test */
996 return (PyTime_Check(op) || PyDate_Check(op)) ?
997 OFFSET_NAIVE : OFFSET_UNKNOWN;
999 *offset = call_utcoffset(tzinfo, tzinfoarg, &none);
1000 if (*offset == -1 && PyErr_Occurred())
1001 return OFFSET_ERROR;
1002 return none ? OFFSET_NAIVE : OFFSET_AWARE;
1005 /* Classify two objects as to whether they're naive or offset-aware.
1006 * This isn't quite the same as calling classify_utcoffset() twice: for
1007 * binary operations (comparison and subtraction), we generally want to
1008 * ignore the tzinfo members if they're identical. This is by design,
1009 * so that results match "naive" expectations when mixing objects from a
1010 * single timezone. So in that case, this sets both offsets to 0 and
1011 * both naiveties to OFFSET_NAIVE.
1012 * The function returns 0 if everything's OK, and -1 on error.
1014 static int
1015 classify_two_utcoffsets(PyObject *o1, int *offset1, naivety *n1,
1016 PyObject *tzinfoarg1,
1017 PyObject *o2, int *offset2, naivety *n2,
1018 PyObject *tzinfoarg2)
1020 if (get_tzinfo_member(o1) == get_tzinfo_member(o2)) {
1021 *offset1 = *offset2 = 0;
1022 *n1 = *n2 = OFFSET_NAIVE;
1024 else {
1025 *n1 = classify_utcoffset(o1, tzinfoarg1, offset1);
1026 if (*n1 == OFFSET_ERROR)
1027 return -1;
1028 *n2 = classify_utcoffset(o2, tzinfoarg2, offset2);
1029 if (*n2 == OFFSET_ERROR)
1030 return -1;
1032 return 0;
1035 /* repr is like "someclass(arg1, arg2)". If tzinfo isn't None,
1036 * stuff
1037 * ", tzinfo=" + repr(tzinfo)
1038 * before the closing ")".
1040 static PyObject *
1041 append_keyword_tzinfo(PyObject *repr, PyObject *tzinfo)
1043 PyObject *temp;
1045 assert(PyString_Check(repr));
1046 assert(tzinfo);
1047 if (tzinfo == Py_None)
1048 return repr;
1049 /* Get rid of the trailing ')'. */
1050 assert(PyString_AsString(repr)[PyString_Size(repr)-1] == ')');
1051 temp = PyString_FromStringAndSize(PyString_AsString(repr),
1052 PyString_Size(repr) - 1);
1053 Py_DECREF(repr);
1054 if (temp == NULL)
1055 return NULL;
1056 repr = temp;
1058 /* Append ", tzinfo=". */
1059 PyString_ConcatAndDel(&repr, PyString_FromString(", tzinfo="));
1061 /* Append repr(tzinfo). */
1062 PyString_ConcatAndDel(&repr, PyObject_Repr(tzinfo));
1064 /* Add a closing paren. */
1065 PyString_ConcatAndDel(&repr, PyString_FromString(")"));
1066 return repr;
1069 /* ---------------------------------------------------------------------------
1070 * String format helpers.
1073 static PyObject *
1074 format_ctime(PyDateTime_Date *date, int hours, int minutes, int seconds)
1076 static const char *DayNames[] = {
1077 "Mon", "Tue", "Wed", "Thu", "Fri", "Sat", "Sun"
1079 static const char *MonthNames[] = {
1080 "Jan", "Feb", "Mar", "Apr", "May", "Jun",
1081 "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
1084 char buffer[128];
1085 int wday = weekday(GET_YEAR(date), GET_MONTH(date), GET_DAY(date));
1087 PyOS_snprintf(buffer, sizeof(buffer), "%s %s %2d %02d:%02d:%02d %04d",
1088 DayNames[wday], MonthNames[GET_MONTH(date) - 1],
1089 GET_DAY(date), hours, minutes, seconds,
1090 GET_YEAR(date));
1091 return PyString_FromString(buffer);
1094 /* Add an hours & minutes UTC offset string to buf. buf has no more than
1095 * buflen bytes remaining. The UTC offset is gotten by calling
1096 * tzinfo.uctoffset(tzinfoarg). If that returns None, \0 is stored into
1097 * *buf, and that's all. Else the returned value is checked for sanity (an
1098 * integer in range), and if that's OK it's converted to an hours & minutes
1099 * string of the form
1100 * sign HH sep MM
1101 * Returns 0 if everything is OK. If the return value from utcoffset() is
1102 * bogus, an appropriate exception is set and -1 is returned.
1104 static int
1105 format_utcoffset(char *buf, size_t buflen, const char *sep,
1106 PyObject *tzinfo, PyObject *tzinfoarg)
1108 int offset;
1109 int hours;
1110 int minutes;
1111 char sign;
1112 int none;
1114 offset = call_utcoffset(tzinfo, tzinfoarg, &none);
1115 if (offset == -1 && PyErr_Occurred())
1116 return -1;
1117 if (none) {
1118 *buf = '\0';
1119 return 0;
1121 sign = '+';
1122 if (offset < 0) {
1123 sign = '-';
1124 offset = - offset;
1126 hours = divmod(offset, 60, &minutes);
1127 PyOS_snprintf(buf, buflen, "%c%02d%s%02d", sign, hours, sep, minutes);
1128 return 0;
1131 /* I sure don't want to reproduce the strftime code from the time module,
1132 * so this imports the module and calls it. All the hair is due to
1133 * giving special meanings to the %z and %Z format codes via a preprocessing
1134 * step on the format string.
1135 * tzinfoarg is the argument to pass to the object's tzinfo method, if
1136 * needed.
1138 static PyObject *
1139 wrap_strftime(PyObject *object, PyObject *format, PyObject *timetuple,
1140 PyObject *tzinfoarg)
1142 PyObject *result = NULL; /* guilty until proved innocent */
1144 PyObject *zreplacement = NULL; /* py string, replacement for %z */
1145 PyObject *Zreplacement = NULL; /* py string, replacement for %Z */
1147 char *pin; /* pointer to next char in input format */
1148 char ch; /* next char in input format */
1150 PyObject *newfmt = NULL; /* py string, the output format */
1151 char *pnew; /* pointer to available byte in output format */
1152 char totalnew; /* number bytes total in output format buffer,
1153 exclusive of trailing \0 */
1154 char usednew; /* number bytes used so far in output format buffer */
1156 char *ptoappend; /* pointer to string to append to output buffer */
1157 int ntoappend; /* # of bytes to append to output buffer */
1159 assert(object && format && timetuple);
1160 assert(PyString_Check(format));
1162 /* Give up if the year is before 1900.
1163 * Python strftime() plays games with the year, and different
1164 * games depending on whether envar PYTHON2K is set. This makes
1165 * years before 1900 a nightmare, even if the platform strftime
1166 * supports them (and not all do).
1167 * We could get a lot farther here by avoiding Python's strftime
1168 * wrapper and calling the C strftime() directly, but that isn't
1169 * an option in the Python implementation of this module.
1172 long year;
1173 PyObject *pyyear = PySequence_GetItem(timetuple, 0);
1174 if (pyyear == NULL) return NULL;
1175 assert(PyInt_Check(pyyear));
1176 year = PyInt_AsLong(pyyear);
1177 Py_DECREF(pyyear);
1178 if (year < 1900) {
1179 PyErr_Format(PyExc_ValueError, "year=%ld is before "
1180 "1900; the datetime strftime() "
1181 "methods require year >= 1900",
1182 year);
1183 return NULL;
1187 /* Scan the input format, looking for %z and %Z escapes, building
1188 * a new format. Since computing the replacements for those codes
1189 * is expensive, don't unless they're actually used.
1191 totalnew = PyString_Size(format) + 1; /* realistic if no %z/%Z */
1192 newfmt = PyString_FromStringAndSize(NULL, totalnew);
1193 if (newfmt == NULL) goto Done;
1194 pnew = PyString_AsString(newfmt);
1195 usednew = 0;
1197 pin = PyString_AsString(format);
1198 while ((ch = *pin++) != '\0') {
1199 if (ch != '%') {
1200 ptoappend = pin - 1;
1201 ntoappend = 1;
1203 else if ((ch = *pin++) == '\0') {
1204 /* There's a lone trailing %; doesn't make sense. */
1205 PyErr_SetString(PyExc_ValueError, "strftime format "
1206 "ends with raw %");
1207 goto Done;
1209 /* A % has been seen and ch is the character after it. */
1210 else if (ch == 'z') {
1211 if (zreplacement == NULL) {
1212 /* format utcoffset */
1213 char buf[100];
1214 PyObject *tzinfo = get_tzinfo_member(object);
1215 zreplacement = PyString_FromString("");
1216 if (zreplacement == NULL) goto Done;
1217 if (tzinfo != Py_None && tzinfo != NULL) {
1218 assert(tzinfoarg != NULL);
1219 if (format_utcoffset(buf,
1220 sizeof(buf),
1222 tzinfo,
1223 tzinfoarg) < 0)
1224 goto Done;
1225 Py_DECREF(zreplacement);
1226 zreplacement = PyString_FromString(buf);
1227 if (zreplacement == NULL) goto Done;
1230 assert(zreplacement != NULL);
1231 ptoappend = PyString_AsString(zreplacement);
1232 ntoappend = PyString_Size(zreplacement);
1234 else if (ch == 'Z') {
1235 /* format tzname */
1236 if (Zreplacement == NULL) {
1237 PyObject *tzinfo = get_tzinfo_member(object);
1238 Zreplacement = PyString_FromString("");
1239 if (Zreplacement == NULL) goto Done;
1240 if (tzinfo != Py_None && tzinfo != NULL) {
1241 PyObject *temp;
1242 assert(tzinfoarg != NULL);
1243 temp = call_tzname(tzinfo, tzinfoarg);
1244 if (temp == NULL) goto Done;
1245 if (temp != Py_None) {
1246 assert(PyString_Check(temp));
1247 /* Since the tzname is getting
1248 * stuffed into the format, we
1249 * have to double any % signs
1250 * so that strftime doesn't
1251 * treat them as format codes.
1253 Py_DECREF(Zreplacement);
1254 Zreplacement = PyObject_CallMethod(
1255 temp, "replace",
1256 "ss", "%", "%%");
1257 Py_DECREF(temp);
1258 if (Zreplacement == NULL)
1259 goto Done;
1261 else
1262 Py_DECREF(temp);
1265 assert(Zreplacement != NULL);
1266 ptoappend = PyString_AsString(Zreplacement);
1267 ntoappend = PyString_Size(Zreplacement);
1269 else {
1270 /* percent followed by neither z nor Z */
1271 ptoappend = pin - 2;
1272 ntoappend = 2;
1275 /* Append the ntoappend chars starting at ptoappend to
1276 * the new format.
1278 assert(ntoappend >= 0);
1279 if (ntoappend == 0)
1280 continue;
1281 while (usednew + ntoappend > totalnew) {
1282 int bigger = totalnew << 1;
1283 if ((bigger >> 1) != totalnew) { /* overflow */
1284 PyErr_NoMemory();
1285 goto Done;
1287 if (_PyString_Resize(&newfmt, bigger) < 0)
1288 goto Done;
1289 totalnew = bigger;
1290 pnew = PyString_AsString(newfmt) + usednew;
1292 memcpy(pnew, ptoappend, ntoappend);
1293 pnew += ntoappend;
1294 usednew += ntoappend;
1295 assert(usednew <= totalnew);
1296 } /* end while() */
1298 if (_PyString_Resize(&newfmt, usednew) < 0)
1299 goto Done;
1301 PyObject *time = PyImport_ImportModule("time");
1302 if (time == NULL)
1303 goto Done;
1304 result = PyObject_CallMethod(time, "strftime", "OO",
1305 newfmt, timetuple);
1306 Py_DECREF(time);
1308 Done:
1309 Py_XDECREF(zreplacement);
1310 Py_XDECREF(Zreplacement);
1311 Py_XDECREF(newfmt);
1312 return result;
1315 static char *
1316 isoformat_date(PyDateTime_Date *dt, char buffer[], int bufflen)
1318 int x;
1319 x = PyOS_snprintf(buffer, bufflen,
1320 "%04d-%02d-%02d",
1321 GET_YEAR(dt), GET_MONTH(dt), GET_DAY(dt));
1322 return buffer + x;
1325 static void
1326 isoformat_time(PyDateTime_DateTime *dt, char buffer[], int bufflen)
1328 int us = DATE_GET_MICROSECOND(dt);
1330 PyOS_snprintf(buffer, bufflen,
1331 "%02d:%02d:%02d", /* 8 characters */
1332 DATE_GET_HOUR(dt),
1333 DATE_GET_MINUTE(dt),
1334 DATE_GET_SECOND(dt));
1335 if (us)
1336 PyOS_snprintf(buffer + 8, bufflen - 8, ".%06d", us);
1339 /* ---------------------------------------------------------------------------
1340 * Wrap functions from the time module. These aren't directly available
1341 * from C. Perhaps they should be.
1344 /* Call time.time() and return its result (a Python float). */
1345 static PyObject *
1346 time_time(void)
1348 PyObject *result = NULL;
1349 PyObject *time = PyImport_ImportModule("time");
1351 if (time != NULL) {
1352 result = PyObject_CallMethod(time, "time", "()");
1353 Py_DECREF(time);
1355 return result;
1358 /* Build a time.struct_time. The weekday and day number are automatically
1359 * computed from the y,m,d args.
1361 static PyObject *
1362 build_struct_time(int y, int m, int d, int hh, int mm, int ss, int dstflag)
1364 PyObject *time;
1365 PyObject *result = NULL;
1367 time = PyImport_ImportModule("time");
1368 if (time != NULL) {
1369 result = PyObject_CallMethod(time, "struct_time",
1370 "((iiiiiiiii))",
1371 y, m, d,
1372 hh, mm, ss,
1373 weekday(y, m, d),
1374 days_before_month(y, m) + d,
1375 dstflag);
1376 Py_DECREF(time);
1378 return result;
1381 /* ---------------------------------------------------------------------------
1382 * Miscellaneous helpers.
1385 /* For obscure reasons, we need to use tp_richcompare instead of tp_compare.
1386 * The comparisons here all most naturally compute a cmp()-like result.
1387 * This little helper turns that into a bool result for rich comparisons.
1389 static PyObject *
1390 diff_to_bool(int diff, int op)
1392 PyObject *result;
1393 int istrue;
1395 switch (op) {
1396 case Py_EQ: istrue = diff == 0; break;
1397 case Py_NE: istrue = diff != 0; break;
1398 case Py_LE: istrue = diff <= 0; break;
1399 case Py_GE: istrue = diff >= 0; break;
1400 case Py_LT: istrue = diff < 0; break;
1401 case Py_GT: istrue = diff > 0; break;
1402 default:
1403 assert(! "op unknown");
1404 istrue = 0; /* To shut up compiler */
1406 result = istrue ? Py_True : Py_False;
1407 Py_INCREF(result);
1408 return result;
1411 /* Raises a "can't compare" TypeError and returns NULL. */
1412 static PyObject *
1413 cmperror(PyObject *a, PyObject *b)
1415 PyErr_Format(PyExc_TypeError,
1416 "can't compare %s to %s",
1417 a->ob_type->tp_name, b->ob_type->tp_name);
1418 return NULL;
1421 /* ---------------------------------------------------------------------------
1422 * Cached Python objects; these are set by the module init function.
1425 /* Conversion factors. */
1426 static PyObject *us_per_us = NULL; /* 1 */
1427 static PyObject *us_per_ms = NULL; /* 1000 */
1428 static PyObject *us_per_second = NULL; /* 1000000 */
1429 static PyObject *us_per_minute = NULL; /* 1e6 * 60 as Python int */
1430 static PyObject *us_per_hour = NULL; /* 1e6 * 3600 as Python long */
1431 static PyObject *us_per_day = NULL; /* 1e6 * 3600 * 24 as Python long */
1432 static PyObject *us_per_week = NULL; /* 1e6*3600*24*7 as Python long */
1433 static PyObject *seconds_per_day = NULL; /* 3600*24 as Python int */
1435 /* ---------------------------------------------------------------------------
1436 * Class implementations.
1440 * PyDateTime_Delta implementation.
1443 /* Convert a timedelta to a number of us,
1444 * (24*3600*self.days + self.seconds)*1000000 + self.microseconds
1445 * as a Python int or long.
1446 * Doing mixed-radix arithmetic by hand instead is excruciating in C,
1447 * due to ubiquitous overflow possibilities.
1449 static PyObject *
1450 delta_to_microseconds(PyDateTime_Delta *self)
1452 PyObject *x1 = NULL;
1453 PyObject *x2 = NULL;
1454 PyObject *x3 = NULL;
1455 PyObject *result = NULL;
1457 x1 = PyInt_FromLong(GET_TD_DAYS(self));
1458 if (x1 == NULL)
1459 goto Done;
1460 x2 = PyNumber_Multiply(x1, seconds_per_day); /* days in seconds */
1461 if (x2 == NULL)
1462 goto Done;
1463 Py_DECREF(x1);
1464 x1 = NULL;
1466 /* x2 has days in seconds */
1467 x1 = PyInt_FromLong(GET_TD_SECONDS(self)); /* seconds */
1468 if (x1 == NULL)
1469 goto Done;
1470 x3 = PyNumber_Add(x1, x2); /* days and seconds in seconds */
1471 if (x3 == NULL)
1472 goto Done;
1473 Py_DECREF(x1);
1474 Py_DECREF(x2);
1475 x1 = x2 = NULL;
1477 /* x3 has days+seconds in seconds */
1478 x1 = PyNumber_Multiply(x3, us_per_second); /* us */
1479 if (x1 == NULL)
1480 goto Done;
1481 Py_DECREF(x3);
1482 x3 = NULL;
1484 /* x1 has days+seconds in us */
1485 x2 = PyInt_FromLong(GET_TD_MICROSECONDS(self));
1486 if (x2 == NULL)
1487 goto Done;
1488 result = PyNumber_Add(x1, x2);
1490 Done:
1491 Py_XDECREF(x1);
1492 Py_XDECREF(x2);
1493 Py_XDECREF(x3);
1494 return result;
1497 /* Convert a number of us (as a Python int or long) to a timedelta.
1499 static PyObject *
1500 microseconds_to_delta_ex(PyObject *pyus, PyTypeObject *type)
1502 int us;
1503 int s;
1504 int d;
1505 long temp;
1507 PyObject *tuple = NULL;
1508 PyObject *num = NULL;
1509 PyObject *result = NULL;
1511 tuple = PyNumber_Divmod(pyus, us_per_second);
1512 if (tuple == NULL)
1513 goto Done;
1515 num = PyTuple_GetItem(tuple, 1); /* us */
1516 if (num == NULL)
1517 goto Done;
1518 temp = PyLong_AsLong(num);
1519 num = NULL;
1520 if (temp == -1 && PyErr_Occurred())
1521 goto Done;
1522 assert(0 <= temp && temp < 1000000);
1523 us = (int)temp;
1524 if (us < 0) {
1525 /* The divisor was positive, so this must be an error. */
1526 assert(PyErr_Occurred());
1527 goto Done;
1530 num = PyTuple_GetItem(tuple, 0); /* leftover seconds */
1531 if (num == NULL)
1532 goto Done;
1533 Py_INCREF(num);
1534 Py_DECREF(tuple);
1536 tuple = PyNumber_Divmod(num, seconds_per_day);
1537 if (tuple == NULL)
1538 goto Done;
1539 Py_DECREF(num);
1541 num = PyTuple_GetItem(tuple, 1); /* seconds */
1542 if (num == NULL)
1543 goto Done;
1544 temp = PyLong_AsLong(num);
1545 num = NULL;
1546 if (temp == -1 && PyErr_Occurred())
1547 goto Done;
1548 assert(0 <= temp && temp < 24*3600);
1549 s = (int)temp;
1551 if (s < 0) {
1552 /* The divisor was positive, so this must be an error. */
1553 assert(PyErr_Occurred());
1554 goto Done;
1557 num = PyTuple_GetItem(tuple, 0); /* leftover days */
1558 if (num == NULL)
1559 goto Done;
1560 Py_INCREF(num);
1561 temp = PyLong_AsLong(num);
1562 if (temp == -1 && PyErr_Occurred())
1563 goto Done;
1564 d = (int)temp;
1565 if ((long)d != temp) {
1566 PyErr_SetString(PyExc_OverflowError, "normalized days too "
1567 "large to fit in a C int");
1568 goto Done;
1570 result = new_delta_ex(d, s, us, 0, type);
1572 Done:
1573 Py_XDECREF(tuple);
1574 Py_XDECREF(num);
1575 return result;
1578 #define microseconds_to_delta(pymicros) \
1579 microseconds_to_delta_ex(pymicros, &PyDateTime_DeltaType)
1581 static PyObject *
1582 multiply_int_timedelta(PyObject *intobj, PyDateTime_Delta *delta)
1584 PyObject *pyus_in;
1585 PyObject *pyus_out;
1586 PyObject *result;
1588 pyus_in = delta_to_microseconds(delta);
1589 if (pyus_in == NULL)
1590 return NULL;
1592 pyus_out = PyNumber_Multiply(pyus_in, intobj);
1593 Py_DECREF(pyus_in);
1594 if (pyus_out == NULL)
1595 return NULL;
1597 result = microseconds_to_delta(pyus_out);
1598 Py_DECREF(pyus_out);
1599 return result;
1602 static PyObject *
1603 divide_timedelta_int(PyDateTime_Delta *delta, PyObject *intobj)
1605 PyObject *pyus_in;
1606 PyObject *pyus_out;
1607 PyObject *result;
1609 pyus_in = delta_to_microseconds(delta);
1610 if (pyus_in == NULL)
1611 return NULL;
1613 pyus_out = PyNumber_FloorDivide(pyus_in, intobj);
1614 Py_DECREF(pyus_in);
1615 if (pyus_out == NULL)
1616 return NULL;
1618 result = microseconds_to_delta(pyus_out);
1619 Py_DECREF(pyus_out);
1620 return result;
1623 static PyObject *
1624 delta_add(PyObject *left, PyObject *right)
1626 PyObject *result = Py_NotImplemented;
1628 if (PyDelta_Check(left) && PyDelta_Check(right)) {
1629 /* delta + delta */
1630 /* The C-level additions can't overflow because of the
1631 * invariant bounds.
1633 int days = GET_TD_DAYS(left) + GET_TD_DAYS(right);
1634 int seconds = GET_TD_SECONDS(left) + GET_TD_SECONDS(right);
1635 int microseconds = GET_TD_MICROSECONDS(left) +
1636 GET_TD_MICROSECONDS(right);
1637 result = new_delta(days, seconds, microseconds, 1);
1640 if (result == Py_NotImplemented)
1641 Py_INCREF(result);
1642 return result;
1645 static PyObject *
1646 delta_negative(PyDateTime_Delta *self)
1648 return new_delta(-GET_TD_DAYS(self),
1649 -GET_TD_SECONDS(self),
1650 -GET_TD_MICROSECONDS(self),
1654 static PyObject *
1655 delta_positive(PyDateTime_Delta *self)
1657 /* Could optimize this (by returning self) if this isn't a
1658 * subclass -- but who uses unary + ? Approximately nobody.
1660 return new_delta(GET_TD_DAYS(self),
1661 GET_TD_SECONDS(self),
1662 GET_TD_MICROSECONDS(self),
1666 static PyObject *
1667 delta_abs(PyDateTime_Delta *self)
1669 PyObject *result;
1671 assert(GET_TD_MICROSECONDS(self) >= 0);
1672 assert(GET_TD_SECONDS(self) >= 0);
1674 if (GET_TD_DAYS(self) < 0)
1675 result = delta_negative(self);
1676 else
1677 result = delta_positive(self);
1679 return result;
1682 static PyObject *
1683 delta_subtract(PyObject *left, PyObject *right)
1685 PyObject *result = Py_NotImplemented;
1687 if (PyDelta_Check(left) && PyDelta_Check(right)) {
1688 /* delta - delta */
1689 PyObject *minus_right = PyNumber_Negative(right);
1690 if (minus_right) {
1691 result = delta_add(left, minus_right);
1692 Py_DECREF(minus_right);
1694 else
1695 result = NULL;
1698 if (result == Py_NotImplemented)
1699 Py_INCREF(result);
1700 return result;
1703 /* This is more natural as a tp_compare, but doesn't work then: for whatever
1704 * reason, Python's try_3way_compare ignores tp_compare unless
1705 * PyInstance_Check returns true, but these aren't old-style classes.
1707 static PyObject *
1708 delta_richcompare(PyDateTime_Delta *self, PyObject *other, int op)
1710 int diff = 42; /* nonsense */
1712 if (PyDelta_Check(other)) {
1713 diff = GET_TD_DAYS(self) - GET_TD_DAYS(other);
1714 if (diff == 0) {
1715 diff = GET_TD_SECONDS(self) - GET_TD_SECONDS(other);
1716 if (diff == 0)
1717 diff = GET_TD_MICROSECONDS(self) -
1718 GET_TD_MICROSECONDS(other);
1721 else if (op == Py_EQ || op == Py_NE)
1722 diff = 1; /* any non-zero value will do */
1724 else /* stop this from falling back to address comparison */
1725 return cmperror((PyObject *)self, other);
1727 return diff_to_bool(diff, op);
1730 static PyObject *delta_getstate(PyDateTime_Delta *self);
1732 static long
1733 delta_hash(PyDateTime_Delta *self)
1735 if (self->hashcode == -1) {
1736 PyObject *temp = delta_getstate(self);
1737 if (temp != NULL) {
1738 self->hashcode = PyObject_Hash(temp);
1739 Py_DECREF(temp);
1742 return self->hashcode;
1745 static PyObject *
1746 delta_multiply(PyObject *left, PyObject *right)
1748 PyObject *result = Py_NotImplemented;
1750 if (PyDelta_Check(left)) {
1751 /* delta * ??? */
1752 if (PyInt_Check(right) || PyLong_Check(right))
1753 result = multiply_int_timedelta(right,
1754 (PyDateTime_Delta *) left);
1756 else if (PyInt_Check(left) || PyLong_Check(left))
1757 result = multiply_int_timedelta(left,
1758 (PyDateTime_Delta *) right);
1760 if (result == Py_NotImplemented)
1761 Py_INCREF(result);
1762 return result;
1765 static PyObject *
1766 delta_divide(PyObject *left, PyObject *right)
1768 PyObject *result = Py_NotImplemented;
1770 if (PyDelta_Check(left)) {
1771 /* delta * ??? */
1772 if (PyInt_Check(right) || PyLong_Check(right))
1773 result = divide_timedelta_int(
1774 (PyDateTime_Delta *)left,
1775 right);
1778 if (result == Py_NotImplemented)
1779 Py_INCREF(result);
1780 return result;
1783 /* Fold in the value of the tag ("seconds", "weeks", etc) component of a
1784 * timedelta constructor. sofar is the # of microseconds accounted for
1785 * so far, and there are factor microseconds per current unit, the number
1786 * of which is given by num. num * factor is added to sofar in a
1787 * numerically careful way, and that's the result. Any fractional
1788 * microseconds left over (this can happen if num is a float type) are
1789 * added into *leftover.
1790 * Note that there are many ways this can give an error (NULL) return.
1792 static PyObject *
1793 accum(const char* tag, PyObject *sofar, PyObject *num, PyObject *factor,
1794 double *leftover)
1796 PyObject *prod;
1797 PyObject *sum;
1799 assert(num != NULL);
1801 if (PyInt_Check(num) || PyLong_Check(num)) {
1802 prod = PyNumber_Multiply(num, factor);
1803 if (prod == NULL)
1804 return NULL;
1805 sum = PyNumber_Add(sofar, prod);
1806 Py_DECREF(prod);
1807 return sum;
1810 if (PyFloat_Check(num)) {
1811 double dnum;
1812 double fracpart;
1813 double intpart;
1814 PyObject *x;
1815 PyObject *y;
1817 /* The Plan: decompose num into an integer part and a
1818 * fractional part, num = intpart + fracpart.
1819 * Then num * factor ==
1820 * intpart * factor + fracpart * factor
1821 * and the LHS can be computed exactly in long arithmetic.
1822 * The RHS is again broken into an int part and frac part.
1823 * and the frac part is added into *leftover.
1825 dnum = PyFloat_AsDouble(num);
1826 if (dnum == -1.0 && PyErr_Occurred())
1827 return NULL;
1828 fracpart = modf(dnum, &intpart);
1829 x = PyLong_FromDouble(intpart);
1830 if (x == NULL)
1831 return NULL;
1833 prod = PyNumber_Multiply(x, factor);
1834 Py_DECREF(x);
1835 if (prod == NULL)
1836 return NULL;
1838 sum = PyNumber_Add(sofar, prod);
1839 Py_DECREF(prod);
1840 if (sum == NULL)
1841 return NULL;
1843 if (fracpart == 0.0)
1844 return sum;
1845 /* So far we've lost no information. Dealing with the
1846 * fractional part requires float arithmetic, and may
1847 * lose a little info.
1849 assert(PyInt_Check(factor) || PyLong_Check(factor));
1850 if (PyInt_Check(factor))
1851 dnum = (double)PyInt_AsLong(factor);
1852 else
1853 dnum = PyLong_AsDouble(factor);
1855 dnum *= fracpart;
1856 fracpart = modf(dnum, &intpart);
1857 x = PyLong_FromDouble(intpart);
1858 if (x == NULL) {
1859 Py_DECREF(sum);
1860 return NULL;
1863 y = PyNumber_Add(sum, x);
1864 Py_DECREF(sum);
1865 Py_DECREF(x);
1866 *leftover += fracpart;
1867 return y;
1870 PyErr_Format(PyExc_TypeError,
1871 "unsupported type for timedelta %s component: %s",
1872 tag, num->ob_type->tp_name);
1873 return NULL;
1876 static PyObject *
1877 delta_new(PyTypeObject *type, PyObject *args, PyObject *kw)
1879 PyObject *self = NULL;
1881 /* Argument objects. */
1882 PyObject *day = NULL;
1883 PyObject *second = NULL;
1884 PyObject *us = NULL;
1885 PyObject *ms = NULL;
1886 PyObject *minute = NULL;
1887 PyObject *hour = NULL;
1888 PyObject *week = NULL;
1890 PyObject *x = NULL; /* running sum of microseconds */
1891 PyObject *y = NULL; /* temp sum of microseconds */
1892 double leftover_us = 0.0;
1894 static const char *keywords[] = {
1895 "days", "seconds", "microseconds", "milliseconds",
1896 "minutes", "hours", "weeks", NULL
1899 if (PyArg_ParseTupleAndKeywords(args, kw, "|OOOOOOO:__new__",
1900 keywords,
1901 &day, &second, &us,
1902 &ms, &minute, &hour, &week) == 0)
1903 goto Done;
1905 x = PyInt_FromLong(0);
1906 if (x == NULL)
1907 goto Done;
1909 #define CLEANUP \
1910 Py_DECREF(x); \
1911 x = y; \
1912 if (x == NULL) \
1913 goto Done
1915 if (us) {
1916 y = accum("microseconds", x, us, us_per_us, &leftover_us);
1917 CLEANUP;
1919 if (ms) {
1920 y = accum("milliseconds", x, ms, us_per_ms, &leftover_us);
1921 CLEANUP;
1923 if (second) {
1924 y = accum("seconds", x, second, us_per_second, &leftover_us);
1925 CLEANUP;
1927 if (minute) {
1928 y = accum("minutes", x, minute, us_per_minute, &leftover_us);
1929 CLEANUP;
1931 if (hour) {
1932 y = accum("hours", x, hour, us_per_hour, &leftover_us);
1933 CLEANUP;
1935 if (day) {
1936 y = accum("days", x, day, us_per_day, &leftover_us);
1937 CLEANUP;
1939 if (week) {
1940 y = accum("weeks", x, week, us_per_week, &leftover_us);
1941 CLEANUP;
1943 if (leftover_us) {
1944 /* Round to nearest whole # of us, and add into x. */
1945 PyObject *temp = PyLong_FromLong(round_to_long(leftover_us));
1946 if (temp == NULL) {
1947 Py_DECREF(x);
1948 goto Done;
1950 y = PyNumber_Add(x, temp);
1951 Py_DECREF(temp);
1952 CLEANUP;
1955 self = microseconds_to_delta_ex(x, type);
1956 Py_DECREF(x);
1957 Done:
1958 return self;
1960 #undef CLEANUP
1963 static int
1964 delta_nonzero(PyDateTime_Delta *self)
1966 return (GET_TD_DAYS(self) != 0
1967 || GET_TD_SECONDS(self) != 0
1968 || GET_TD_MICROSECONDS(self) != 0);
1971 static PyObject *
1972 delta_repr(PyDateTime_Delta *self)
1974 if (GET_TD_MICROSECONDS(self) != 0)
1975 return PyString_FromFormat("%s(%d, %d, %d)",
1976 self->ob_type->tp_name,
1977 GET_TD_DAYS(self),
1978 GET_TD_SECONDS(self),
1979 GET_TD_MICROSECONDS(self));
1980 if (GET_TD_SECONDS(self) != 0)
1981 return PyString_FromFormat("%s(%d, %d)",
1982 self->ob_type->tp_name,
1983 GET_TD_DAYS(self),
1984 GET_TD_SECONDS(self));
1986 return PyString_FromFormat("%s(%d)",
1987 self->ob_type->tp_name,
1988 GET_TD_DAYS(self));
1991 static PyObject *
1992 delta_str(PyDateTime_Delta *self)
1994 int days = GET_TD_DAYS(self);
1995 int seconds = GET_TD_SECONDS(self);
1996 int us = GET_TD_MICROSECONDS(self);
1997 int hours;
1998 int minutes;
1999 char buf[100];
2000 char *pbuf = buf;
2001 size_t buflen = sizeof(buf);
2002 int n;
2004 minutes = divmod(seconds, 60, &seconds);
2005 hours = divmod(minutes, 60, &minutes);
2007 if (days) {
2008 n = PyOS_snprintf(pbuf, buflen, "%d day%s, ", days,
2009 (days == 1 || days == -1) ? "" : "s");
2010 if (n < 0 || (size_t)n >= buflen)
2011 goto Fail;
2012 pbuf += n;
2013 buflen -= (size_t)n;
2016 n = PyOS_snprintf(pbuf, buflen, "%d:%02d:%02d",
2017 hours, minutes, seconds);
2018 if (n < 0 || (size_t)n >= buflen)
2019 goto Fail;
2020 pbuf += n;
2021 buflen -= (size_t)n;
2023 if (us) {
2024 n = PyOS_snprintf(pbuf, buflen, ".%06d", us);
2025 if (n < 0 || (size_t)n >= buflen)
2026 goto Fail;
2027 pbuf += n;
2030 return PyString_FromStringAndSize(buf, pbuf - buf);
2032 Fail:
2033 PyErr_SetString(PyExc_SystemError, "goofy result from PyOS_snprintf");
2034 return NULL;
2037 /* Pickle support, a simple use of __reduce__. */
2039 /* __getstate__ isn't exposed */
2040 static PyObject *
2041 delta_getstate(PyDateTime_Delta *self)
2043 return Py_BuildValue("iii", GET_TD_DAYS(self),
2044 GET_TD_SECONDS(self),
2045 GET_TD_MICROSECONDS(self));
2048 static PyObject *
2049 delta_reduce(PyDateTime_Delta* self)
2051 return Py_BuildValue("ON", self->ob_type, delta_getstate(self));
2054 #define OFFSET(field) offsetof(PyDateTime_Delta, field)
2056 static PyMemberDef delta_members[] = {
2058 {"days", T_INT, OFFSET(days), READONLY,
2059 PyDoc_STR("Number of days.")},
2061 {"seconds", T_INT, OFFSET(seconds), READONLY,
2062 PyDoc_STR("Number of seconds (>= 0 and less than 1 day).")},
2064 {"microseconds", T_INT, OFFSET(microseconds), READONLY,
2065 PyDoc_STR("Number of microseconds (>= 0 and less than 1 second).")},
2066 {NULL}
2069 static PyMethodDef delta_methods[] = {
2070 {"__reduce__", (PyCFunction)delta_reduce, METH_NOARGS,
2071 PyDoc_STR("__reduce__() -> (cls, state)")},
2073 {NULL, NULL},
2076 static char delta_doc[] =
2077 PyDoc_STR("Difference between two datetime values.");
2079 static PyNumberMethods delta_as_number = {
2080 delta_add, /* nb_add */
2081 delta_subtract, /* nb_subtract */
2082 delta_multiply, /* nb_multiply */
2083 delta_divide, /* nb_divide */
2084 0, /* nb_remainder */
2085 0, /* nb_divmod */
2086 0, /* nb_power */
2087 (unaryfunc)delta_negative, /* nb_negative */
2088 (unaryfunc)delta_positive, /* nb_positive */
2089 (unaryfunc)delta_abs, /* nb_absolute */
2090 (inquiry)delta_nonzero, /* nb_nonzero */
2091 0, /*nb_invert*/
2092 0, /*nb_lshift*/
2093 0, /*nb_rshift*/
2094 0, /*nb_and*/
2095 0, /*nb_xor*/
2096 0, /*nb_or*/
2097 0, /*nb_coerce*/
2098 0, /*nb_int*/
2099 0, /*nb_long*/
2100 0, /*nb_float*/
2101 0, /*nb_oct*/
2102 0, /*nb_hex*/
2103 0, /*nb_inplace_add*/
2104 0, /*nb_inplace_subtract*/
2105 0, /*nb_inplace_multiply*/
2106 0, /*nb_inplace_divide*/
2107 0, /*nb_inplace_remainder*/
2108 0, /*nb_inplace_power*/
2109 0, /*nb_inplace_lshift*/
2110 0, /*nb_inplace_rshift*/
2111 0, /*nb_inplace_and*/
2112 0, /*nb_inplace_xor*/
2113 0, /*nb_inplace_or*/
2114 delta_divide, /* nb_floor_divide */
2115 0, /* nb_true_divide */
2116 0, /* nb_inplace_floor_divide */
2117 0, /* nb_inplace_true_divide */
2120 static PyTypeObject PyDateTime_DeltaType = {
2121 PyObject_HEAD_INIT(NULL)
2122 0, /* ob_size */
2123 "datetime.timedelta", /* tp_name */
2124 sizeof(PyDateTime_Delta), /* tp_basicsize */
2125 0, /* tp_itemsize */
2126 0, /* tp_dealloc */
2127 0, /* tp_print */
2128 0, /* tp_getattr */
2129 0, /* tp_setattr */
2130 0, /* tp_compare */
2131 (reprfunc)delta_repr, /* tp_repr */
2132 &delta_as_number, /* tp_as_number */
2133 0, /* tp_as_sequence */
2134 0, /* tp_as_mapping */
2135 (hashfunc)delta_hash, /* tp_hash */
2136 0, /* tp_call */
2137 (reprfunc)delta_str, /* tp_str */
2138 PyObject_GenericGetAttr, /* tp_getattro */
2139 0, /* tp_setattro */
2140 0, /* tp_as_buffer */
2141 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_CHECKTYPES |
2142 Py_TPFLAGS_BASETYPE, /* tp_flags */
2143 delta_doc, /* tp_doc */
2144 0, /* tp_traverse */
2145 0, /* tp_clear */
2146 (richcmpfunc)delta_richcompare, /* tp_richcompare */
2147 0, /* tp_weaklistoffset */
2148 0, /* tp_iter */
2149 0, /* tp_iternext */
2150 delta_methods, /* tp_methods */
2151 delta_members, /* tp_members */
2152 0, /* tp_getset */
2153 0, /* tp_base */
2154 0, /* tp_dict */
2155 0, /* tp_descr_get */
2156 0, /* tp_descr_set */
2157 0, /* tp_dictoffset */
2158 0, /* tp_init */
2159 0, /* tp_alloc */
2160 delta_new, /* tp_new */
2161 0, /* tp_free */
2165 * PyDateTime_Date implementation.
2168 /* Accessor properties. */
2170 static PyObject *
2171 date_year(PyDateTime_Date *self, void *unused)
2173 return PyInt_FromLong(GET_YEAR(self));
2176 static PyObject *
2177 date_month(PyDateTime_Date *self, void *unused)
2179 return PyInt_FromLong(GET_MONTH(self));
2182 static PyObject *
2183 date_day(PyDateTime_Date *self, void *unused)
2185 return PyInt_FromLong(GET_DAY(self));
2188 static PyGetSetDef date_getset[] = {
2189 {"year", (getter)date_year},
2190 {"month", (getter)date_month},
2191 {"day", (getter)date_day},
2192 {NULL}
2195 /* Constructors. */
2197 static const char *date_kws[] = {"year", "month", "day", NULL};
2199 static PyObject *
2200 date_new(PyTypeObject *type, PyObject *args, PyObject *kw)
2202 PyObject *self = NULL;
2203 PyObject *state;
2204 int year;
2205 int month;
2206 int day;
2208 /* Check for invocation from pickle with __getstate__ state */
2209 if (PyTuple_GET_SIZE(args) == 1 &&
2210 PyString_Check(state = PyTuple_GET_ITEM(args, 0)) &&
2211 PyString_GET_SIZE(state) == _PyDateTime_DATE_DATASIZE &&
2212 MONTH_IS_SANE(PyString_AS_STRING(state)[2]))
2214 PyDateTime_Date *me;
2216 me = (PyDateTime_Date *) (type->tp_alloc(type, 0));
2217 if (me != NULL) {
2218 char *pdata = PyString_AS_STRING(state);
2219 memcpy(me->data, pdata, _PyDateTime_DATE_DATASIZE);
2220 me->hashcode = -1;
2222 return (PyObject *)me;
2225 if (PyArg_ParseTupleAndKeywords(args, kw, "iii", date_kws,
2226 &year, &month, &day)) {
2227 if (check_date_args(year, month, day) < 0)
2228 return NULL;
2229 self = new_date_ex(year, month, day, type);
2231 return self;
2234 /* Return new date from localtime(t). */
2235 static PyObject *
2236 date_local_from_time_t(PyObject *cls, double ts)
2238 struct tm *tm;
2239 time_t t;
2240 PyObject *result = NULL;
2242 t = _PyTime_DoubleToTimet(ts);
2243 if (t == (time_t)-1 && PyErr_Occurred())
2244 return NULL;
2245 tm = localtime(&t);
2246 if (tm)
2247 result = PyObject_CallFunction(cls, "iii",
2248 tm->tm_year + 1900,
2249 tm->tm_mon + 1,
2250 tm->tm_mday);
2251 else
2252 PyErr_SetString(PyExc_ValueError,
2253 "timestamp out of range for "
2254 "platform localtime() function");
2255 return result;
2258 /* Return new date from current time.
2259 * We say this is equivalent to fromtimestamp(time.time()), and the
2260 * only way to be sure of that is to *call* time.time(). That's not
2261 * generally the same as calling C's time.
2263 static PyObject *
2264 date_today(PyObject *cls, PyObject *dummy)
2266 PyObject *time;
2267 PyObject *result;
2269 time = time_time();
2270 if (time == NULL)
2271 return NULL;
2273 /* Note well: today() is a class method, so this may not call
2274 * date.fromtimestamp. For example, it may call
2275 * datetime.fromtimestamp. That's why we need all the accuracy
2276 * time.time() delivers; if someone were gonzo about optimization,
2277 * date.today() could get away with plain C time().
2279 result = PyObject_CallMethod(cls, "fromtimestamp", "O", time);
2280 Py_DECREF(time);
2281 return result;
2284 /* Return new date from given timestamp (Python timestamp -- a double). */
2285 static PyObject *
2286 date_fromtimestamp(PyObject *cls, PyObject *args)
2288 double timestamp;
2289 PyObject *result = NULL;
2291 if (PyArg_ParseTuple(args, "d:fromtimestamp", &timestamp))
2292 result = date_local_from_time_t(cls, timestamp);
2293 return result;
2296 /* Return new date from proleptic Gregorian ordinal. Raises ValueError if
2297 * the ordinal is out of range.
2299 static PyObject *
2300 date_fromordinal(PyObject *cls, PyObject *args)
2302 PyObject *result = NULL;
2303 int ordinal;
2305 if (PyArg_ParseTuple(args, "i:fromordinal", &ordinal)) {
2306 int year;
2307 int month;
2308 int day;
2310 if (ordinal < 1)
2311 PyErr_SetString(PyExc_ValueError, "ordinal must be "
2312 ">= 1");
2313 else {
2314 ord_to_ymd(ordinal, &year, &month, &day);
2315 result = PyObject_CallFunction(cls, "iii",
2316 year, month, day);
2319 return result;
2323 * Date arithmetic.
2326 /* date + timedelta -> date. If arg negate is true, subtract the timedelta
2327 * instead.
2329 static PyObject *
2330 add_date_timedelta(PyDateTime_Date *date, PyDateTime_Delta *delta, int negate)
2332 PyObject *result = NULL;
2333 int year = GET_YEAR(date);
2334 int month = GET_MONTH(date);
2335 int deltadays = GET_TD_DAYS(delta);
2336 /* C-level overflow is impossible because |deltadays| < 1e9. */
2337 int day = GET_DAY(date) + (negate ? -deltadays : deltadays);
2339 if (normalize_date(&year, &month, &day) >= 0)
2340 result = new_date(year, month, day);
2341 return result;
2344 static PyObject *
2345 date_add(PyObject *left, PyObject *right)
2347 if (PyDateTime_Check(left) || PyDateTime_Check(right)) {
2348 Py_INCREF(Py_NotImplemented);
2349 return Py_NotImplemented;
2351 if (PyDate_Check(left)) {
2352 /* date + ??? */
2353 if (PyDelta_Check(right))
2354 /* date + delta */
2355 return add_date_timedelta((PyDateTime_Date *) left,
2356 (PyDateTime_Delta *) right,
2359 else {
2360 /* ??? + date
2361 * 'right' must be one of us, or we wouldn't have been called
2363 if (PyDelta_Check(left))
2364 /* delta + date */
2365 return add_date_timedelta((PyDateTime_Date *) right,
2366 (PyDateTime_Delta *) left,
2369 Py_INCREF(Py_NotImplemented);
2370 return Py_NotImplemented;
2373 static PyObject *
2374 date_subtract(PyObject *left, PyObject *right)
2376 if (PyDateTime_Check(left) || PyDateTime_Check(right)) {
2377 Py_INCREF(Py_NotImplemented);
2378 return Py_NotImplemented;
2380 if (PyDate_Check(left)) {
2381 if (PyDate_Check(right)) {
2382 /* date - date */
2383 int left_ord = ymd_to_ord(GET_YEAR(left),
2384 GET_MONTH(left),
2385 GET_DAY(left));
2386 int right_ord = ymd_to_ord(GET_YEAR(right),
2387 GET_MONTH(right),
2388 GET_DAY(right));
2389 return new_delta(left_ord - right_ord, 0, 0, 0);
2391 if (PyDelta_Check(right)) {
2392 /* date - delta */
2393 return add_date_timedelta((PyDateTime_Date *) left,
2394 (PyDateTime_Delta *) right,
2398 Py_INCREF(Py_NotImplemented);
2399 return Py_NotImplemented;
2403 /* Various ways to turn a date into a string. */
2405 static PyObject *
2406 date_repr(PyDateTime_Date *self)
2408 char buffer[1028];
2409 const char *typename;
2411 typename = self->ob_type->tp_name;
2412 PyOS_snprintf(buffer, sizeof(buffer), "%s(%d, %d, %d)",
2413 typename,
2414 GET_YEAR(self), GET_MONTH(self), GET_DAY(self));
2416 return PyString_FromString(buffer);
2419 static PyObject *
2420 date_isoformat(PyDateTime_Date *self)
2422 char buffer[128];
2424 isoformat_date(self, buffer, sizeof(buffer));
2425 return PyString_FromString(buffer);
2428 /* str() calls the appropriate isoformat() method. */
2429 static PyObject *
2430 date_str(PyDateTime_Date *self)
2432 return PyObject_CallMethod((PyObject *)self, "isoformat", "()");
2436 static PyObject *
2437 date_ctime(PyDateTime_Date *self)
2439 return format_ctime(self, 0, 0, 0);
2442 static PyObject *
2443 date_strftime(PyDateTime_Date *self, PyObject *args, PyObject *kw)
2445 /* This method can be inherited, and needs to call the
2446 * timetuple() method appropriate to self's class.
2448 PyObject *result;
2449 PyObject *format;
2450 PyObject *tuple;
2451 static const char *keywords[] = {"format", NULL};
2453 if (! PyArg_ParseTupleAndKeywords(args, kw, "O!:strftime", keywords,
2454 &PyString_Type, &format))
2455 return NULL;
2457 tuple = PyObject_CallMethod((PyObject *)self, "timetuple", "()");
2458 if (tuple == NULL)
2459 return NULL;
2460 result = wrap_strftime((PyObject *)self, format, tuple,
2461 (PyObject *)self);
2462 Py_DECREF(tuple);
2463 return result;
2466 /* ISO methods. */
2468 static PyObject *
2469 date_isoweekday(PyDateTime_Date *self)
2471 int dow = weekday(GET_YEAR(self), GET_MONTH(self), GET_DAY(self));
2473 return PyInt_FromLong(dow + 1);
2476 static PyObject *
2477 date_isocalendar(PyDateTime_Date *self)
2479 int year = GET_YEAR(self);
2480 int week1_monday = iso_week1_monday(year);
2481 int today = ymd_to_ord(year, GET_MONTH(self), GET_DAY(self));
2482 int week;
2483 int day;
2485 week = divmod(today - week1_monday, 7, &day);
2486 if (week < 0) {
2487 --year;
2488 week1_monday = iso_week1_monday(year);
2489 week = divmod(today - week1_monday, 7, &day);
2491 else if (week >= 52 && today >= iso_week1_monday(year + 1)) {
2492 ++year;
2493 week = 0;
2495 return Py_BuildValue("iii", year, week + 1, day + 1);
2498 /* Miscellaneous methods. */
2500 /* This is more natural as a tp_compare, but doesn't work then: for whatever
2501 * reason, Python's try_3way_compare ignores tp_compare unless
2502 * PyInstance_Check returns true, but these aren't old-style classes.
2504 static PyObject *
2505 date_richcompare(PyDateTime_Date *self, PyObject *other, int op)
2507 int diff = 42; /* nonsense */
2509 if (PyDate_Check(other))
2510 diff = memcmp(self->data, ((PyDateTime_Date *)other)->data,
2511 _PyDateTime_DATE_DATASIZE);
2513 else if (PyObject_HasAttrString(other, "timetuple")) {
2514 /* A hook for other kinds of date objects. */
2515 Py_INCREF(Py_NotImplemented);
2516 return Py_NotImplemented;
2518 else if (op == Py_EQ || op == Py_NE)
2519 diff = 1; /* any non-zero value will do */
2521 else /* stop this from falling back to address comparison */
2522 return cmperror((PyObject *)self, other);
2524 return diff_to_bool(diff, op);
2527 static PyObject *
2528 date_timetuple(PyDateTime_Date *self)
2530 return build_struct_time(GET_YEAR(self),
2531 GET_MONTH(self),
2532 GET_DAY(self),
2533 0, 0, 0, -1);
2536 static PyObject *
2537 date_replace(PyDateTime_Date *self, PyObject *args, PyObject *kw)
2539 PyObject *clone;
2540 PyObject *tuple;
2541 int year = GET_YEAR(self);
2542 int month = GET_MONTH(self);
2543 int day = GET_DAY(self);
2545 if (! PyArg_ParseTupleAndKeywords(args, kw, "|iii:replace", date_kws,
2546 &year, &month, &day))
2547 return NULL;
2548 tuple = Py_BuildValue("iii", year, month, day);
2549 if (tuple == NULL)
2550 return NULL;
2551 clone = date_new(self->ob_type, tuple, NULL);
2552 Py_DECREF(tuple);
2553 return clone;
2556 static PyObject *date_getstate(PyDateTime_Date *self);
2558 static long
2559 date_hash(PyDateTime_Date *self)
2561 if (self->hashcode == -1) {
2562 PyObject *temp = date_getstate(self);
2563 if (temp != NULL) {
2564 self->hashcode = PyObject_Hash(temp);
2565 Py_DECREF(temp);
2568 return self->hashcode;
2571 static PyObject *
2572 date_toordinal(PyDateTime_Date *self)
2574 return PyInt_FromLong(ymd_to_ord(GET_YEAR(self), GET_MONTH(self),
2575 GET_DAY(self)));
2578 static PyObject *
2579 date_weekday(PyDateTime_Date *self)
2581 int dow = weekday(GET_YEAR(self), GET_MONTH(self), GET_DAY(self));
2583 return PyInt_FromLong(dow);
2586 /* Pickle support, a simple use of __reduce__. */
2588 /* __getstate__ isn't exposed */
2589 static PyObject *
2590 date_getstate(PyDateTime_Date *self)
2592 return Py_BuildValue(
2593 "(N)",
2594 PyString_FromStringAndSize((char *)self->data,
2595 _PyDateTime_DATE_DATASIZE));
2598 static PyObject *
2599 date_reduce(PyDateTime_Date *self, PyObject *arg)
2601 return Py_BuildValue("(ON)", self->ob_type, date_getstate(self));
2604 static PyMethodDef date_methods[] = {
2606 /* Class methods: */
2608 {"fromtimestamp", (PyCFunction)date_fromtimestamp, METH_VARARGS |
2609 METH_CLASS,
2610 PyDoc_STR("timestamp -> local date from a POSIX timestamp (like "
2611 "time.time()).")},
2613 {"fromordinal", (PyCFunction)date_fromordinal, METH_VARARGS |
2614 METH_CLASS,
2615 PyDoc_STR("int -> date corresponding to a proleptic Gregorian "
2616 "ordinal.")},
2618 {"today", (PyCFunction)date_today, METH_NOARGS | METH_CLASS,
2619 PyDoc_STR("Current date or datetime: same as "
2620 "self.__class__.fromtimestamp(time.time()).")},
2622 /* Instance methods: */
2624 {"ctime", (PyCFunction)date_ctime, METH_NOARGS,
2625 PyDoc_STR("Return ctime() style string.")},
2627 {"strftime", (PyCFunction)date_strftime, METH_KEYWORDS,
2628 PyDoc_STR("format -> strftime() style string.")},
2630 {"timetuple", (PyCFunction)date_timetuple, METH_NOARGS,
2631 PyDoc_STR("Return time tuple, compatible with time.localtime().")},
2633 {"isocalendar", (PyCFunction)date_isocalendar, METH_NOARGS,
2634 PyDoc_STR("Return a 3-tuple containing ISO year, week number, and "
2635 "weekday.")},
2637 {"isoformat", (PyCFunction)date_isoformat, METH_NOARGS,
2638 PyDoc_STR("Return string in ISO 8601 format, YYYY-MM-DD.")},
2640 {"isoweekday", (PyCFunction)date_isoweekday, METH_NOARGS,
2641 PyDoc_STR("Return the day of the week represented by the date.\n"
2642 "Monday == 1 ... Sunday == 7")},
2644 {"toordinal", (PyCFunction)date_toordinal, METH_NOARGS,
2645 PyDoc_STR("Return proleptic Gregorian ordinal. January 1 of year "
2646 "1 is day 1.")},
2648 {"weekday", (PyCFunction)date_weekday, METH_NOARGS,
2649 PyDoc_STR("Return the day of the week represented by the date.\n"
2650 "Monday == 0 ... Sunday == 6")},
2652 {"replace", (PyCFunction)date_replace, METH_KEYWORDS,
2653 PyDoc_STR("Return date with new specified fields.")},
2655 {"__reduce__", (PyCFunction)date_reduce, METH_NOARGS,
2656 PyDoc_STR("__reduce__() -> (cls, state)")},
2658 {NULL, NULL}
2661 static char date_doc[] =
2662 PyDoc_STR("date(year, month, day) --> date object");
2664 static PyNumberMethods date_as_number = {
2665 date_add, /* nb_add */
2666 date_subtract, /* nb_subtract */
2667 0, /* nb_multiply */
2668 0, /* nb_divide */
2669 0, /* nb_remainder */
2670 0, /* nb_divmod */
2671 0, /* nb_power */
2672 0, /* nb_negative */
2673 0, /* nb_positive */
2674 0, /* nb_absolute */
2675 0, /* nb_nonzero */
2678 static PyTypeObject PyDateTime_DateType = {
2679 PyObject_HEAD_INIT(NULL)
2680 0, /* ob_size */
2681 "datetime.date", /* tp_name */
2682 sizeof(PyDateTime_Date), /* tp_basicsize */
2683 0, /* tp_itemsize */
2684 0, /* tp_dealloc */
2685 0, /* tp_print */
2686 0, /* tp_getattr */
2687 0, /* tp_setattr */
2688 0, /* tp_compare */
2689 (reprfunc)date_repr, /* tp_repr */
2690 &date_as_number, /* tp_as_number */
2691 0, /* tp_as_sequence */
2692 0, /* tp_as_mapping */
2693 (hashfunc)date_hash, /* tp_hash */
2694 0, /* tp_call */
2695 (reprfunc)date_str, /* tp_str */
2696 PyObject_GenericGetAttr, /* tp_getattro */
2697 0, /* tp_setattro */
2698 0, /* tp_as_buffer */
2699 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_CHECKTYPES |
2700 Py_TPFLAGS_BASETYPE, /* tp_flags */
2701 date_doc, /* tp_doc */
2702 0, /* tp_traverse */
2703 0, /* tp_clear */
2704 (richcmpfunc)date_richcompare, /* tp_richcompare */
2705 0, /* tp_weaklistoffset */
2706 0, /* tp_iter */
2707 0, /* tp_iternext */
2708 date_methods, /* tp_methods */
2709 0, /* tp_members */
2710 date_getset, /* tp_getset */
2711 0, /* tp_base */
2712 0, /* tp_dict */
2713 0, /* tp_descr_get */
2714 0, /* tp_descr_set */
2715 0, /* tp_dictoffset */
2716 0, /* tp_init */
2717 0, /* tp_alloc */
2718 date_new, /* tp_new */
2719 0, /* tp_free */
2723 * PyDateTime_TZInfo implementation.
2726 /* This is a pure abstract base class, so doesn't do anything beyond
2727 * raising NotImplemented exceptions. Real tzinfo classes need
2728 * to derive from this. This is mostly for clarity, and for efficiency in
2729 * datetime and time constructors (their tzinfo arguments need to
2730 * be subclasses of this tzinfo class, which is easy and quick to check).
2732 * Note: For reasons having to do with pickling of subclasses, we have
2733 * to allow tzinfo objects to be instantiated. This wasn't an issue
2734 * in the Python implementation (__init__() could raise NotImplementedError
2735 * there without ill effect), but doing so in the C implementation hit a
2736 * brick wall.
2739 static PyObject *
2740 tzinfo_nogo(const char* methodname)
2742 PyErr_Format(PyExc_NotImplementedError,
2743 "a tzinfo subclass must implement %s()",
2744 methodname);
2745 return NULL;
2748 /* Methods. A subclass must implement these. */
2750 static PyObject *
2751 tzinfo_tzname(PyDateTime_TZInfo *self, PyObject *dt)
2753 return tzinfo_nogo("tzname");
2756 static PyObject *
2757 tzinfo_utcoffset(PyDateTime_TZInfo *self, PyObject *dt)
2759 return tzinfo_nogo("utcoffset");
2762 static PyObject *
2763 tzinfo_dst(PyDateTime_TZInfo *self, PyObject *dt)
2765 return tzinfo_nogo("dst");
2768 static PyObject *
2769 tzinfo_fromutc(PyDateTime_TZInfo *self, PyDateTime_DateTime *dt)
2771 int y, m, d, hh, mm, ss, us;
2773 PyObject *result;
2774 int off, dst;
2775 int none;
2776 int delta;
2778 if (! PyDateTime_Check(dt)) {
2779 PyErr_SetString(PyExc_TypeError,
2780 "fromutc: argument must be a datetime");
2781 return NULL;
2783 if (! HASTZINFO(dt) || dt->tzinfo != (PyObject *)self) {
2784 PyErr_SetString(PyExc_ValueError, "fromutc: dt.tzinfo "
2785 "is not self");
2786 return NULL;
2789 off = call_utcoffset(dt->tzinfo, (PyObject *)dt, &none);
2790 if (off == -1 && PyErr_Occurred())
2791 return NULL;
2792 if (none) {
2793 PyErr_SetString(PyExc_ValueError, "fromutc: non-None "
2794 "utcoffset() result required");
2795 return NULL;
2798 dst = call_dst(dt->tzinfo, (PyObject *)dt, &none);
2799 if (dst == -1 && PyErr_Occurred())
2800 return NULL;
2801 if (none) {
2802 PyErr_SetString(PyExc_ValueError, "fromutc: non-None "
2803 "dst() result required");
2804 return NULL;
2807 y = GET_YEAR(dt);
2808 m = GET_MONTH(dt);
2809 d = GET_DAY(dt);
2810 hh = DATE_GET_HOUR(dt);
2811 mm = DATE_GET_MINUTE(dt);
2812 ss = DATE_GET_SECOND(dt);
2813 us = DATE_GET_MICROSECOND(dt);
2815 delta = off - dst;
2816 mm += delta;
2817 if ((mm < 0 || mm >= 60) &&
2818 normalize_datetime(&y, &m, &d, &hh, &mm, &ss, &us) < 0)
2819 return NULL;
2820 result = new_datetime(y, m, d, hh, mm, ss, us, dt->tzinfo);
2821 if (result == NULL)
2822 return result;
2824 dst = call_dst(dt->tzinfo, result, &none);
2825 if (dst == -1 && PyErr_Occurred())
2826 goto Fail;
2827 if (none)
2828 goto Inconsistent;
2829 if (dst == 0)
2830 return result;
2832 mm += dst;
2833 if ((mm < 0 || mm >= 60) &&
2834 normalize_datetime(&y, &m, &d, &hh, &mm, &ss, &us) < 0)
2835 goto Fail;
2836 Py_DECREF(result);
2837 result = new_datetime(y, m, d, hh, mm, ss, us, dt->tzinfo);
2838 return result;
2840 Inconsistent:
2841 PyErr_SetString(PyExc_ValueError, "fromutc: tz.dst() gave"
2842 "inconsistent results; cannot convert");
2844 /* fall thru to failure */
2845 Fail:
2846 Py_DECREF(result);
2847 return NULL;
2851 * Pickle support. This is solely so that tzinfo subclasses can use
2852 * pickling -- tzinfo itself is supposed to be uninstantiable.
2855 static PyObject *
2856 tzinfo_reduce(PyObject *self)
2858 PyObject *args, *state, *tmp;
2859 PyObject *getinitargs, *getstate;
2861 tmp = PyTuple_New(0);
2862 if (tmp == NULL)
2863 return NULL;
2865 getinitargs = PyObject_GetAttrString(self, "__getinitargs__");
2866 if (getinitargs != NULL) {
2867 args = PyObject_CallObject(getinitargs, tmp);
2868 Py_DECREF(getinitargs);
2869 if (args == NULL) {
2870 Py_DECREF(tmp);
2871 return NULL;
2874 else {
2875 PyErr_Clear();
2876 args = tmp;
2877 Py_INCREF(args);
2880 getstate = PyObject_GetAttrString(self, "__getstate__");
2881 if (getstate != NULL) {
2882 state = PyObject_CallObject(getstate, tmp);
2883 Py_DECREF(getstate);
2884 if (state == NULL) {
2885 Py_DECREF(args);
2886 Py_DECREF(tmp);
2887 return NULL;
2890 else {
2891 PyObject **dictptr;
2892 PyErr_Clear();
2893 state = Py_None;
2894 dictptr = _PyObject_GetDictPtr(self);
2895 if (dictptr && *dictptr && PyDict_Size(*dictptr))
2896 state = *dictptr;
2897 Py_INCREF(state);
2900 Py_DECREF(tmp);
2902 if (state == Py_None) {
2903 Py_DECREF(state);
2904 return Py_BuildValue("(ON)", self->ob_type, args);
2906 else
2907 return Py_BuildValue("(ONN)", self->ob_type, args, state);
2910 static PyMethodDef tzinfo_methods[] = {
2912 {"tzname", (PyCFunction)tzinfo_tzname, METH_O,
2913 PyDoc_STR("datetime -> string name of time zone.")},
2915 {"utcoffset", (PyCFunction)tzinfo_utcoffset, METH_O,
2916 PyDoc_STR("datetime -> minutes east of UTC (negative for "
2917 "west of UTC).")},
2919 {"dst", (PyCFunction)tzinfo_dst, METH_O,
2920 PyDoc_STR("datetime -> DST offset in minutes east of UTC.")},
2922 {"fromutc", (PyCFunction)tzinfo_fromutc, METH_O,
2923 PyDoc_STR("datetime in UTC -> datetime in local time.")},
2925 {"__reduce__", (PyCFunction)tzinfo_reduce, METH_NOARGS,
2926 PyDoc_STR("-> (cls, state)")},
2928 {NULL, NULL}
2931 static char tzinfo_doc[] =
2932 PyDoc_STR("Abstract base class for time zone info objects.");
2934 statichere PyTypeObject PyDateTime_TZInfoType = {
2935 PyObject_HEAD_INIT(NULL)
2936 0, /* ob_size */
2937 "datetime.tzinfo", /* tp_name */
2938 sizeof(PyDateTime_TZInfo), /* tp_basicsize */
2939 0, /* tp_itemsize */
2940 0, /* tp_dealloc */
2941 0, /* tp_print */
2942 0, /* tp_getattr */
2943 0, /* tp_setattr */
2944 0, /* tp_compare */
2945 0, /* tp_repr */
2946 0, /* tp_as_number */
2947 0, /* tp_as_sequence */
2948 0, /* tp_as_mapping */
2949 0, /* tp_hash */
2950 0, /* tp_call */
2951 0, /* tp_str */
2952 PyObject_GenericGetAttr, /* tp_getattro */
2953 0, /* tp_setattro */
2954 0, /* tp_as_buffer */
2955 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_CHECKTYPES |
2956 Py_TPFLAGS_BASETYPE, /* tp_flags */
2957 tzinfo_doc, /* tp_doc */
2958 0, /* tp_traverse */
2959 0, /* tp_clear */
2960 0, /* tp_richcompare */
2961 0, /* tp_weaklistoffset */
2962 0, /* tp_iter */
2963 0, /* tp_iternext */
2964 tzinfo_methods, /* tp_methods */
2965 0, /* tp_members */
2966 0, /* tp_getset */
2967 0, /* tp_base */
2968 0, /* tp_dict */
2969 0, /* tp_descr_get */
2970 0, /* tp_descr_set */
2971 0, /* tp_dictoffset */
2972 0, /* tp_init */
2973 0, /* tp_alloc */
2974 PyType_GenericNew, /* tp_new */
2975 0, /* tp_free */
2979 * PyDateTime_Time implementation.
2982 /* Accessor properties.
2985 static PyObject *
2986 time_hour(PyDateTime_Time *self, void *unused)
2988 return PyInt_FromLong(TIME_GET_HOUR(self));
2991 static PyObject *
2992 time_minute(PyDateTime_Time *self, void *unused)
2994 return PyInt_FromLong(TIME_GET_MINUTE(self));
2997 /* The name time_second conflicted with some platform header file. */
2998 static PyObject *
2999 py_time_second(PyDateTime_Time *self, void *unused)
3001 return PyInt_FromLong(TIME_GET_SECOND(self));
3004 static PyObject *
3005 time_microsecond(PyDateTime_Time *self, void *unused)
3007 return PyInt_FromLong(TIME_GET_MICROSECOND(self));
3010 static PyObject *
3011 time_tzinfo(PyDateTime_Time *self, void *unused)
3013 PyObject *result = HASTZINFO(self) ? self->tzinfo : Py_None;
3014 Py_INCREF(result);
3015 return result;
3018 static PyGetSetDef time_getset[] = {
3019 {"hour", (getter)time_hour},
3020 {"minute", (getter)time_minute},
3021 {"second", (getter)py_time_second},
3022 {"microsecond", (getter)time_microsecond},
3023 {"tzinfo", (getter)time_tzinfo},
3024 {NULL}
3028 * Constructors.
3031 static const char *time_kws[] = {"hour", "minute", "second", "microsecond",
3032 "tzinfo", NULL};
3034 static PyObject *
3035 time_new(PyTypeObject *type, PyObject *args, PyObject *kw)
3037 PyObject *self = NULL;
3038 PyObject *state;
3039 int hour = 0;
3040 int minute = 0;
3041 int second = 0;
3042 int usecond = 0;
3043 PyObject *tzinfo = Py_None;
3045 /* Check for invocation from pickle with __getstate__ state */
3046 if (PyTuple_GET_SIZE(args) >= 1 &&
3047 PyTuple_GET_SIZE(args) <= 2 &&
3048 PyString_Check(state = PyTuple_GET_ITEM(args, 0)) &&
3049 PyString_GET_SIZE(state) == _PyDateTime_TIME_DATASIZE &&
3050 ((unsigned char) (PyString_AS_STRING(state)[0])) < 24)
3052 PyDateTime_Time *me;
3053 char aware;
3055 if (PyTuple_GET_SIZE(args) == 2) {
3056 tzinfo = PyTuple_GET_ITEM(args, 1);
3057 if (check_tzinfo_subclass(tzinfo) < 0) {
3058 PyErr_SetString(PyExc_TypeError, "bad "
3059 "tzinfo state arg");
3060 return NULL;
3063 aware = (char)(tzinfo != Py_None);
3064 me = (PyDateTime_Time *) (type->tp_alloc(type, aware));
3065 if (me != NULL) {
3066 char *pdata = PyString_AS_STRING(state);
3068 memcpy(me->data, pdata, _PyDateTime_TIME_DATASIZE);
3069 me->hashcode = -1;
3070 me->hastzinfo = aware;
3071 if (aware) {
3072 Py_INCREF(tzinfo);
3073 me->tzinfo = tzinfo;
3076 return (PyObject *)me;
3079 if (PyArg_ParseTupleAndKeywords(args, kw, "|iiiiO", time_kws,
3080 &hour, &minute, &second, &usecond,
3081 &tzinfo)) {
3082 if (check_time_args(hour, minute, second, usecond) < 0)
3083 return NULL;
3084 if (check_tzinfo_subclass(tzinfo) < 0)
3085 return NULL;
3086 self = new_time_ex(hour, minute, second, usecond, tzinfo,
3087 type);
3089 return self;
3093 * Destructor.
3096 static void
3097 time_dealloc(PyDateTime_Time *self)
3099 if (HASTZINFO(self)) {
3100 Py_XDECREF(self->tzinfo);
3102 self->ob_type->tp_free((PyObject *)self);
3106 * Indirect access to tzinfo methods.
3109 /* These are all METH_NOARGS, so don't need to check the arglist. */
3110 static PyObject *
3111 time_utcoffset(PyDateTime_Time *self, PyObject *unused) {
3112 return offset_as_timedelta(HASTZINFO(self) ? self->tzinfo : Py_None,
3113 "utcoffset", Py_None);
3116 static PyObject *
3117 time_dst(PyDateTime_Time *self, PyObject *unused) {
3118 return offset_as_timedelta(HASTZINFO(self) ? self->tzinfo : Py_None,
3119 "dst", Py_None);
3122 static PyObject *
3123 time_tzname(PyDateTime_Time *self, PyObject *unused) {
3124 return call_tzname(HASTZINFO(self) ? self->tzinfo : Py_None,
3125 Py_None);
3129 * Various ways to turn a time into a string.
3132 static PyObject *
3133 time_repr(PyDateTime_Time *self)
3135 char buffer[100];
3136 const char *typename = self->ob_type->tp_name;
3137 int h = TIME_GET_HOUR(self);
3138 int m = TIME_GET_MINUTE(self);
3139 int s = TIME_GET_SECOND(self);
3140 int us = TIME_GET_MICROSECOND(self);
3141 PyObject *result = NULL;
3143 if (us)
3144 PyOS_snprintf(buffer, sizeof(buffer),
3145 "%s(%d, %d, %d, %d)", typename, h, m, s, us);
3146 else if (s)
3147 PyOS_snprintf(buffer, sizeof(buffer),
3148 "%s(%d, %d, %d)", typename, h, m, s);
3149 else
3150 PyOS_snprintf(buffer, sizeof(buffer),
3151 "%s(%d, %d)", typename, h, m);
3152 result = PyString_FromString(buffer);
3153 if (result != NULL && HASTZINFO(self))
3154 result = append_keyword_tzinfo(result, self->tzinfo);
3155 return result;
3158 static PyObject *
3159 time_str(PyDateTime_Time *self)
3161 return PyObject_CallMethod((PyObject *)self, "isoformat", "()");
3164 static PyObject *
3165 time_isoformat(PyDateTime_Time *self)
3167 char buf[100];
3168 PyObject *result;
3169 /* Reuse the time format code from the datetime type. */
3170 PyDateTime_DateTime datetime;
3171 PyDateTime_DateTime *pdatetime = &datetime;
3173 /* Copy over just the time bytes. */
3174 memcpy(pdatetime->data + _PyDateTime_DATE_DATASIZE,
3175 self->data,
3176 _PyDateTime_TIME_DATASIZE);
3178 isoformat_time(pdatetime, buf, sizeof(buf));
3179 result = PyString_FromString(buf);
3180 if (result == NULL || ! HASTZINFO(self) || self->tzinfo == Py_None)
3181 return result;
3183 /* We need to append the UTC offset. */
3184 if (format_utcoffset(buf, sizeof(buf), ":", self->tzinfo,
3185 Py_None) < 0) {
3186 Py_DECREF(result);
3187 return NULL;
3189 PyString_ConcatAndDel(&result, PyString_FromString(buf));
3190 return result;
3193 static PyObject *
3194 time_strftime(PyDateTime_Time *self, PyObject *args, PyObject *kw)
3196 PyObject *result;
3197 PyObject *format;
3198 PyObject *tuple;
3199 static const char *keywords[] = {"format", NULL};
3201 if (! PyArg_ParseTupleAndKeywords(args, kw, "O!:strftime", keywords,
3202 &PyString_Type, &format))
3203 return NULL;
3205 /* Python's strftime does insane things with the year part of the
3206 * timetuple. The year is forced to (the otherwise nonsensical)
3207 * 1900 to worm around that.
3209 tuple = Py_BuildValue("iiiiiiiii",
3210 1900, 1, 1, /* year, month, day */
3211 TIME_GET_HOUR(self),
3212 TIME_GET_MINUTE(self),
3213 TIME_GET_SECOND(self),
3214 0, 1, -1); /* weekday, daynum, dst */
3215 if (tuple == NULL)
3216 return NULL;
3217 assert(PyTuple_Size(tuple) == 9);
3218 result = wrap_strftime((PyObject *)self, format, tuple, Py_None);
3219 Py_DECREF(tuple);
3220 return result;
3224 * Miscellaneous methods.
3227 /* This is more natural as a tp_compare, but doesn't work then: for whatever
3228 * reason, Python's try_3way_compare ignores tp_compare unless
3229 * PyInstance_Check returns true, but these aren't old-style classes.
3231 static PyObject *
3232 time_richcompare(PyDateTime_Time *self, PyObject *other, int op)
3234 int diff;
3235 naivety n1, n2;
3236 int offset1, offset2;
3238 if (! PyTime_Check(other)) {
3239 if (op == Py_EQ || op == Py_NE) {
3240 PyObject *result = op == Py_EQ ? Py_False : Py_True;
3241 Py_INCREF(result);
3242 return result;
3244 /* Stop this from falling back to address comparison. */
3245 return cmperror((PyObject *)self, other);
3247 if (classify_two_utcoffsets((PyObject *)self, &offset1, &n1, Py_None,
3248 other, &offset2, &n2, Py_None) < 0)
3249 return NULL;
3250 assert(n1 != OFFSET_UNKNOWN && n2 != OFFSET_UNKNOWN);
3251 /* If they're both naive, or both aware and have the same offsets,
3252 * we get off cheap. Note that if they're both naive, offset1 ==
3253 * offset2 == 0 at this point.
3255 if (n1 == n2 && offset1 == offset2) {
3256 diff = memcmp(self->data, ((PyDateTime_Time *)other)->data,
3257 _PyDateTime_TIME_DATASIZE);
3258 return diff_to_bool(diff, op);
3261 if (n1 == OFFSET_AWARE && n2 == OFFSET_AWARE) {
3262 assert(offset1 != offset2); /* else last "if" handled it */
3263 /* Convert everything except microseconds to seconds. These
3264 * can't overflow (no more than the # of seconds in 2 days).
3266 offset1 = TIME_GET_HOUR(self) * 3600 +
3267 (TIME_GET_MINUTE(self) - offset1) * 60 +
3268 TIME_GET_SECOND(self);
3269 offset2 = TIME_GET_HOUR(other) * 3600 +
3270 (TIME_GET_MINUTE(other) - offset2) * 60 +
3271 TIME_GET_SECOND(other);
3272 diff = offset1 - offset2;
3273 if (diff == 0)
3274 diff = TIME_GET_MICROSECOND(self) -
3275 TIME_GET_MICROSECOND(other);
3276 return diff_to_bool(diff, op);
3279 assert(n1 != n2);
3280 PyErr_SetString(PyExc_TypeError,
3281 "can't compare offset-naive and "
3282 "offset-aware times");
3283 return NULL;
3286 static long
3287 time_hash(PyDateTime_Time *self)
3289 if (self->hashcode == -1) {
3290 naivety n;
3291 int offset;
3292 PyObject *temp;
3294 n = classify_utcoffset((PyObject *)self, Py_None, &offset);
3295 assert(n != OFFSET_UNKNOWN);
3296 if (n == OFFSET_ERROR)
3297 return -1;
3299 /* Reduce this to a hash of another object. */
3300 if (offset == 0)
3301 temp = PyString_FromStringAndSize((char *)self->data,
3302 _PyDateTime_TIME_DATASIZE);
3303 else {
3304 int hour;
3305 int minute;
3307 assert(n == OFFSET_AWARE);
3308 assert(HASTZINFO(self));
3309 hour = divmod(TIME_GET_HOUR(self) * 60 +
3310 TIME_GET_MINUTE(self) - offset,
3312 &minute);
3313 if (0 <= hour && hour < 24)
3314 temp = new_time(hour, minute,
3315 TIME_GET_SECOND(self),
3316 TIME_GET_MICROSECOND(self),
3317 Py_None);
3318 else
3319 temp = Py_BuildValue("iiii",
3320 hour, minute,
3321 TIME_GET_SECOND(self),
3322 TIME_GET_MICROSECOND(self));
3324 if (temp != NULL) {
3325 self->hashcode = PyObject_Hash(temp);
3326 Py_DECREF(temp);
3329 return self->hashcode;
3332 static PyObject *
3333 time_replace(PyDateTime_Time *self, PyObject *args, PyObject *kw)
3335 PyObject *clone;
3336 PyObject *tuple;
3337 int hh = TIME_GET_HOUR(self);
3338 int mm = TIME_GET_MINUTE(self);
3339 int ss = TIME_GET_SECOND(self);
3340 int us = TIME_GET_MICROSECOND(self);
3341 PyObject *tzinfo = HASTZINFO(self) ? self->tzinfo : Py_None;
3343 if (! PyArg_ParseTupleAndKeywords(args, kw, "|iiiiO:replace",
3344 time_kws,
3345 &hh, &mm, &ss, &us, &tzinfo))
3346 return NULL;
3347 tuple = Py_BuildValue("iiiiO", hh, mm, ss, us, tzinfo);
3348 if (tuple == NULL)
3349 return NULL;
3350 clone = time_new(self->ob_type, tuple, NULL);
3351 Py_DECREF(tuple);
3352 return clone;
3355 static int
3356 time_nonzero(PyDateTime_Time *self)
3358 int offset;
3359 int none;
3361 if (TIME_GET_SECOND(self) || TIME_GET_MICROSECOND(self)) {
3362 /* Since utcoffset is in whole minutes, nothing can
3363 * alter the conclusion that this is nonzero.
3365 return 1;
3367 offset = 0;
3368 if (HASTZINFO(self) && self->tzinfo != Py_None) {
3369 offset = call_utcoffset(self->tzinfo, Py_None, &none);
3370 if (offset == -1 && PyErr_Occurred())
3371 return -1;
3373 return (TIME_GET_MINUTE(self) - offset + TIME_GET_HOUR(self)*60) != 0;
3376 /* Pickle support, a simple use of __reduce__. */
3378 /* Let basestate be the non-tzinfo data string.
3379 * If tzinfo is None, this returns (basestate,), else (basestate, tzinfo).
3380 * So it's a tuple in any (non-error) case.
3381 * __getstate__ isn't exposed.
3383 static PyObject *
3384 time_getstate(PyDateTime_Time *self)
3386 PyObject *basestate;
3387 PyObject *result = NULL;
3389 basestate = PyString_FromStringAndSize((char *)self->data,
3390 _PyDateTime_TIME_DATASIZE);
3391 if (basestate != NULL) {
3392 if (! HASTZINFO(self) || self->tzinfo == Py_None)
3393 result = PyTuple_Pack(1, basestate);
3394 else
3395 result = PyTuple_Pack(2, basestate, self->tzinfo);
3396 Py_DECREF(basestate);
3398 return result;
3401 static PyObject *
3402 time_reduce(PyDateTime_Time *self, PyObject *arg)
3404 return Py_BuildValue("(ON)", self->ob_type, time_getstate(self));
3407 static PyMethodDef time_methods[] = {
3409 {"isoformat", (PyCFunction)time_isoformat, METH_KEYWORDS,
3410 PyDoc_STR("Return string in ISO 8601 format, HH:MM:SS[.mmmmmm]"
3411 "[+HH:MM].")},
3413 {"strftime", (PyCFunction)time_strftime, METH_KEYWORDS,
3414 PyDoc_STR("format -> strftime() style string.")},
3416 {"utcoffset", (PyCFunction)time_utcoffset, METH_NOARGS,
3417 PyDoc_STR("Return self.tzinfo.utcoffset(self).")},
3419 {"tzname", (PyCFunction)time_tzname, METH_NOARGS,
3420 PyDoc_STR("Return self.tzinfo.tzname(self).")},
3422 {"dst", (PyCFunction)time_dst, METH_NOARGS,
3423 PyDoc_STR("Return self.tzinfo.dst(self).")},
3425 {"replace", (PyCFunction)time_replace, METH_KEYWORDS,
3426 PyDoc_STR("Return time with new specified fields.")},
3428 {"__reduce__", (PyCFunction)time_reduce, METH_NOARGS,
3429 PyDoc_STR("__reduce__() -> (cls, state)")},
3431 {NULL, NULL}
3434 static char time_doc[] =
3435 PyDoc_STR("time([hour[, minute[, second[, microsecond[, tzinfo]]]]]) --> a time object\n\
3437 All arguments are optional. tzinfo may be None, or an instance of\n\
3438 a tzinfo subclass. The remaining arguments may be ints or longs.\n");
3440 static PyNumberMethods time_as_number = {
3441 0, /* nb_add */
3442 0, /* nb_subtract */
3443 0, /* nb_multiply */
3444 0, /* nb_divide */
3445 0, /* nb_remainder */
3446 0, /* nb_divmod */
3447 0, /* nb_power */
3448 0, /* nb_negative */
3449 0, /* nb_positive */
3450 0, /* nb_absolute */
3451 (inquiry)time_nonzero, /* nb_nonzero */
3454 statichere PyTypeObject PyDateTime_TimeType = {
3455 PyObject_HEAD_INIT(NULL)
3456 0, /* ob_size */
3457 "datetime.time", /* tp_name */
3458 sizeof(PyDateTime_Time), /* tp_basicsize */
3459 0, /* tp_itemsize */
3460 (destructor)time_dealloc, /* tp_dealloc */
3461 0, /* tp_print */
3462 0, /* tp_getattr */
3463 0, /* tp_setattr */
3464 0, /* tp_compare */
3465 (reprfunc)time_repr, /* tp_repr */
3466 &time_as_number, /* tp_as_number */
3467 0, /* tp_as_sequence */
3468 0, /* tp_as_mapping */
3469 (hashfunc)time_hash, /* tp_hash */
3470 0, /* tp_call */
3471 (reprfunc)time_str, /* tp_str */
3472 PyObject_GenericGetAttr, /* tp_getattro */
3473 0, /* tp_setattro */
3474 0, /* tp_as_buffer */
3475 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_CHECKTYPES |
3476 Py_TPFLAGS_BASETYPE, /* tp_flags */
3477 time_doc, /* tp_doc */
3478 0, /* tp_traverse */
3479 0, /* tp_clear */
3480 (richcmpfunc)time_richcompare, /* tp_richcompare */
3481 0, /* tp_weaklistoffset */
3482 0, /* tp_iter */
3483 0, /* tp_iternext */
3484 time_methods, /* tp_methods */
3485 0, /* tp_members */
3486 time_getset, /* tp_getset */
3487 0, /* tp_base */
3488 0, /* tp_dict */
3489 0, /* tp_descr_get */
3490 0, /* tp_descr_set */
3491 0, /* tp_dictoffset */
3492 0, /* tp_init */
3493 time_alloc, /* tp_alloc */
3494 time_new, /* tp_new */
3495 0, /* tp_free */
3499 * PyDateTime_DateTime implementation.
3502 /* Accessor properties. Properties for day, month, and year are inherited
3503 * from date.
3506 static PyObject *
3507 datetime_hour(PyDateTime_DateTime *self, void *unused)
3509 return PyInt_FromLong(DATE_GET_HOUR(self));
3512 static PyObject *
3513 datetime_minute(PyDateTime_DateTime *self, void *unused)
3515 return PyInt_FromLong(DATE_GET_MINUTE(self));
3518 static PyObject *
3519 datetime_second(PyDateTime_DateTime *self, void *unused)
3521 return PyInt_FromLong(DATE_GET_SECOND(self));
3524 static PyObject *
3525 datetime_microsecond(PyDateTime_DateTime *self, void *unused)
3527 return PyInt_FromLong(DATE_GET_MICROSECOND(self));
3530 static PyObject *
3531 datetime_tzinfo(PyDateTime_DateTime *self, void *unused)
3533 PyObject *result = HASTZINFO(self) ? self->tzinfo : Py_None;
3534 Py_INCREF(result);
3535 return result;
3538 static PyGetSetDef datetime_getset[] = {
3539 {"hour", (getter)datetime_hour},
3540 {"minute", (getter)datetime_minute},
3541 {"second", (getter)datetime_second},
3542 {"microsecond", (getter)datetime_microsecond},
3543 {"tzinfo", (getter)datetime_tzinfo},
3544 {NULL}
3548 * Constructors.
3551 static const char *datetime_kws[] = {
3552 "year", "month", "day", "hour", "minute", "second",
3553 "microsecond", "tzinfo", NULL
3556 static PyObject *
3557 datetime_new(PyTypeObject *type, PyObject *args, PyObject *kw)
3559 PyObject *self = NULL;
3560 PyObject *state;
3561 int year;
3562 int month;
3563 int day;
3564 int hour = 0;
3565 int minute = 0;
3566 int second = 0;
3567 int usecond = 0;
3568 PyObject *tzinfo = Py_None;
3570 /* Check for invocation from pickle with __getstate__ state */
3571 if (PyTuple_GET_SIZE(args) >= 1 &&
3572 PyTuple_GET_SIZE(args) <= 2 &&
3573 PyString_Check(state = PyTuple_GET_ITEM(args, 0)) &&
3574 PyString_GET_SIZE(state) == _PyDateTime_DATETIME_DATASIZE &&
3575 MONTH_IS_SANE(PyString_AS_STRING(state)[2]))
3577 PyDateTime_DateTime *me;
3578 char aware;
3580 if (PyTuple_GET_SIZE(args) == 2) {
3581 tzinfo = PyTuple_GET_ITEM(args, 1);
3582 if (check_tzinfo_subclass(tzinfo) < 0) {
3583 PyErr_SetString(PyExc_TypeError, "bad "
3584 "tzinfo state arg");
3585 return NULL;
3588 aware = (char)(tzinfo != Py_None);
3589 me = (PyDateTime_DateTime *) (type->tp_alloc(type , aware));
3590 if (me != NULL) {
3591 char *pdata = PyString_AS_STRING(state);
3593 memcpy(me->data, pdata, _PyDateTime_DATETIME_DATASIZE);
3594 me->hashcode = -1;
3595 me->hastzinfo = aware;
3596 if (aware) {
3597 Py_INCREF(tzinfo);
3598 me->tzinfo = tzinfo;
3601 return (PyObject *)me;
3604 if (PyArg_ParseTupleAndKeywords(args, kw, "iii|iiiiO", datetime_kws,
3605 &year, &month, &day, &hour, &minute,
3606 &second, &usecond, &tzinfo)) {
3607 if (check_date_args(year, month, day) < 0)
3608 return NULL;
3609 if (check_time_args(hour, minute, second, usecond) < 0)
3610 return NULL;
3611 if (check_tzinfo_subclass(tzinfo) < 0)
3612 return NULL;
3613 self = new_datetime_ex(year, month, day,
3614 hour, minute, second, usecond,
3615 tzinfo, type);
3617 return self;
3620 /* TM_FUNC is the shared type of localtime() and gmtime(). */
3621 typedef struct tm *(*TM_FUNC)(const time_t *timer);
3623 /* Internal helper.
3624 * Build datetime from a time_t and a distinct count of microseconds.
3625 * Pass localtime or gmtime for f, to control the interpretation of timet.
3627 static PyObject *
3628 datetime_from_timet_and_us(PyObject *cls, TM_FUNC f, time_t timet, int us,
3629 PyObject *tzinfo)
3631 struct tm *tm;
3632 PyObject *result = NULL;
3634 tm = f(&timet);
3635 if (tm) {
3636 /* The platform localtime/gmtime may insert leap seconds,
3637 * indicated by tm->tm_sec > 59. We don't care about them,
3638 * except to the extent that passing them on to the datetime
3639 * constructor would raise ValueError for a reason that
3640 * made no sense to the user.
3642 if (tm->tm_sec > 59)
3643 tm->tm_sec = 59;
3644 result = PyObject_CallFunction(cls, "iiiiiiiO",
3645 tm->tm_year + 1900,
3646 tm->tm_mon + 1,
3647 tm->tm_mday,
3648 tm->tm_hour,
3649 tm->tm_min,
3650 tm->tm_sec,
3652 tzinfo);
3654 else
3655 PyErr_SetString(PyExc_ValueError,
3656 "timestamp out of range for "
3657 "platform localtime()/gmtime() function");
3658 return result;
3661 /* Internal helper.
3662 * Build datetime from a Python timestamp. Pass localtime or gmtime for f,
3663 * to control the interpretation of the timestamp. Since a double doesn't
3664 * have enough bits to cover a datetime's full range of precision, it's
3665 * better to call datetime_from_timet_and_us provided you have a way
3666 * to get that much precision (e.g., C time() isn't good enough).
3668 static PyObject *
3669 datetime_from_timestamp(PyObject *cls, TM_FUNC f, double timestamp,
3670 PyObject *tzinfo)
3672 time_t timet;
3673 double fraction;
3674 int us;
3676 timet = _PyTime_DoubleToTimet(timestamp);
3677 if (timet == (time_t)-1 && PyErr_Occurred())
3678 return NULL;
3679 fraction = timestamp - (double)timet;
3680 us = (int)round_to_long(fraction * 1e6);
3681 return datetime_from_timet_and_us(cls, f, timet, us, tzinfo);
3684 /* Internal helper.
3685 * Build most accurate possible datetime for current time. Pass localtime or
3686 * gmtime for f as appropriate.
3688 static PyObject *
3689 datetime_best_possible(PyObject *cls, TM_FUNC f, PyObject *tzinfo)
3691 #ifdef HAVE_GETTIMEOFDAY
3692 struct timeval t;
3694 #ifdef GETTIMEOFDAY_NO_TZ
3695 gettimeofday(&t);
3696 #else
3697 gettimeofday(&t, (struct timezone *)NULL);
3698 #endif
3699 return datetime_from_timet_and_us(cls, f, t.tv_sec, (int)t.tv_usec,
3700 tzinfo);
3702 #else /* ! HAVE_GETTIMEOFDAY */
3703 /* No flavor of gettimeofday exists on this platform. Python's
3704 * time.time() does a lot of other platform tricks to get the
3705 * best time it can on the platform, and we're not going to do
3706 * better than that (if we could, the better code would belong
3707 * in time.time()!) We're limited by the precision of a double,
3708 * though.
3710 PyObject *time;
3711 double dtime;
3713 time = time_time();
3714 if (time == NULL)
3715 return NULL;
3716 dtime = PyFloat_AsDouble(time);
3717 Py_DECREF(time);
3718 if (dtime == -1.0 && PyErr_Occurred())
3719 return NULL;
3720 return datetime_from_timestamp(cls, f, dtime, tzinfo);
3721 #endif /* ! HAVE_GETTIMEOFDAY */
3724 /* Return best possible local time -- this isn't constrained by the
3725 * precision of a timestamp.
3727 static PyObject *
3728 datetime_now(PyObject *cls, PyObject *args, PyObject *kw)
3730 PyObject *self;
3731 PyObject *tzinfo = Py_None;
3732 static const char *keywords[] = {"tz", NULL};
3734 if (! PyArg_ParseTupleAndKeywords(args, kw, "|O:now", keywords,
3735 &tzinfo))
3736 return NULL;
3737 if (check_tzinfo_subclass(tzinfo) < 0)
3738 return NULL;
3740 self = datetime_best_possible(cls,
3741 tzinfo == Py_None ? localtime : gmtime,
3742 tzinfo);
3743 if (self != NULL && tzinfo != Py_None) {
3744 /* Convert UTC to tzinfo's zone. */
3745 PyObject *temp = self;
3746 self = PyObject_CallMethod(tzinfo, "fromutc", "O", self);
3747 Py_DECREF(temp);
3749 return self;
3752 /* Return best possible UTC time -- this isn't constrained by the
3753 * precision of a timestamp.
3755 static PyObject *
3756 datetime_utcnow(PyObject *cls, PyObject *dummy)
3758 return datetime_best_possible(cls, gmtime, Py_None);
3761 /* Return new local datetime from timestamp (Python timestamp -- a double). */
3762 static PyObject *
3763 datetime_fromtimestamp(PyObject *cls, PyObject *args, PyObject *kw)
3765 PyObject *self;
3766 double timestamp;
3767 PyObject *tzinfo = Py_None;
3768 static const char *keywords[] = {"timestamp", "tz", NULL};
3770 if (! PyArg_ParseTupleAndKeywords(args, kw, "d|O:fromtimestamp",
3771 keywords, &timestamp, &tzinfo))
3772 return NULL;
3773 if (check_tzinfo_subclass(tzinfo) < 0)
3774 return NULL;
3776 self = datetime_from_timestamp(cls,
3777 tzinfo == Py_None ? localtime : gmtime,
3778 timestamp,
3779 tzinfo);
3780 if (self != NULL && tzinfo != Py_None) {
3781 /* Convert UTC to tzinfo's zone. */
3782 PyObject *temp = self;
3783 self = PyObject_CallMethod(tzinfo, "fromutc", "O", self);
3784 Py_DECREF(temp);
3786 return self;
3789 /* Return new UTC datetime from timestamp (Python timestamp -- a double). */
3790 static PyObject *
3791 datetime_utcfromtimestamp(PyObject *cls, PyObject *args)
3793 double timestamp;
3794 PyObject *result = NULL;
3796 if (PyArg_ParseTuple(args, "d:utcfromtimestamp", &timestamp))
3797 result = datetime_from_timestamp(cls, gmtime, timestamp,
3798 Py_None);
3799 return result;
3802 /* Return new datetime from time.strptime(). */
3803 static PyObject *
3804 datetime_strptime(PyObject *cls, PyObject *args)
3806 PyObject *result = NULL, *obj, *module;
3807 const char *string, *format;
3809 if (!PyArg_ParseTuple(args, "ss:strptime", &string, &format))
3810 return NULL;
3812 if ((module = PyImport_ImportModule("time")) == NULL)
3813 return NULL;
3814 obj = PyObject_CallMethod(module, "strptime", "ss", string, format);
3815 Py_DECREF(module);
3817 if (obj != NULL) {
3818 int i, good_timetuple = 1;
3819 long int ia[6];
3820 if (PySequence_Check(obj) && PySequence_Size(obj) >= 6)
3821 for (i=0; i < 6; i++) {
3822 PyObject *p = PySequence_GetItem(obj, i);
3823 if (PyInt_Check(p))
3824 ia[i] = PyInt_AsLong(p);
3825 else
3826 good_timetuple = 0;
3827 Py_DECREF(p);
3829 else
3830 good_timetuple = 0;
3831 if (good_timetuple)
3832 result = PyObject_CallFunction(cls, "iiiiii",
3833 ia[0], ia[1], ia[2], ia[3], ia[4], ia[5]);
3834 else
3835 PyErr_SetString(PyExc_ValueError,
3836 "unexpected value from time.strptime");
3837 Py_DECREF(obj);
3839 return result;
3842 /* Return new datetime from date/datetime and time arguments. */
3843 static PyObject *
3844 datetime_combine(PyObject *cls, PyObject *args, PyObject *kw)
3846 static const char *keywords[] = {"date", "time", NULL};
3847 PyObject *date;
3848 PyObject *time;
3849 PyObject *result = NULL;
3851 if (PyArg_ParseTupleAndKeywords(args, kw, "O!O!:combine", keywords,
3852 &PyDateTime_DateType, &date,
3853 &PyDateTime_TimeType, &time)) {
3854 PyObject *tzinfo = Py_None;
3856 if (HASTZINFO(time))
3857 tzinfo = ((PyDateTime_Time *)time)->tzinfo;
3858 result = PyObject_CallFunction(cls, "iiiiiiiO",
3859 GET_YEAR(date),
3860 GET_MONTH(date),
3861 GET_DAY(date),
3862 TIME_GET_HOUR(time),
3863 TIME_GET_MINUTE(time),
3864 TIME_GET_SECOND(time),
3865 TIME_GET_MICROSECOND(time),
3866 tzinfo);
3868 return result;
3872 * Destructor.
3875 static void
3876 datetime_dealloc(PyDateTime_DateTime *self)
3878 if (HASTZINFO(self)) {
3879 Py_XDECREF(self->tzinfo);
3881 self->ob_type->tp_free((PyObject *)self);
3885 * Indirect access to tzinfo methods.
3888 /* These are all METH_NOARGS, so don't need to check the arglist. */
3889 static PyObject *
3890 datetime_utcoffset(PyDateTime_DateTime *self, PyObject *unused) {
3891 return offset_as_timedelta(HASTZINFO(self) ? self->tzinfo : Py_None,
3892 "utcoffset", (PyObject *)self);
3895 static PyObject *
3896 datetime_dst(PyDateTime_DateTime *self, PyObject *unused) {
3897 return offset_as_timedelta(HASTZINFO(self) ? self->tzinfo : Py_None,
3898 "dst", (PyObject *)self);
3901 static PyObject *
3902 datetime_tzname(PyDateTime_DateTime *self, PyObject *unused) {
3903 return call_tzname(HASTZINFO(self) ? self->tzinfo : Py_None,
3904 (PyObject *)self);
3908 * datetime arithmetic.
3911 /* factor must be 1 (to add) or -1 (to subtract). The result inherits
3912 * the tzinfo state of date.
3914 static PyObject *
3915 add_datetime_timedelta(PyDateTime_DateTime *date, PyDateTime_Delta *delta,
3916 int factor)
3918 /* Note that the C-level additions can't overflow, because of
3919 * invariant bounds on the member values.
3921 int year = GET_YEAR(date);
3922 int month = GET_MONTH(date);
3923 int day = GET_DAY(date) + GET_TD_DAYS(delta) * factor;
3924 int hour = DATE_GET_HOUR(date);
3925 int minute = DATE_GET_MINUTE(date);
3926 int second = DATE_GET_SECOND(date) + GET_TD_SECONDS(delta) * factor;
3927 int microsecond = DATE_GET_MICROSECOND(date) +
3928 GET_TD_MICROSECONDS(delta) * factor;
3930 assert(factor == 1 || factor == -1);
3931 if (normalize_datetime(&year, &month, &day,
3932 &hour, &minute, &second, &microsecond) < 0)
3933 return NULL;
3934 else
3935 return new_datetime(year, month, day,
3936 hour, minute, second, microsecond,
3937 HASTZINFO(date) ? date->tzinfo : Py_None);
3940 static PyObject *
3941 datetime_add(PyObject *left, PyObject *right)
3943 if (PyDateTime_Check(left)) {
3944 /* datetime + ??? */
3945 if (PyDelta_Check(right))
3946 /* datetime + delta */
3947 return add_datetime_timedelta(
3948 (PyDateTime_DateTime *)left,
3949 (PyDateTime_Delta *)right,
3952 else if (PyDelta_Check(left)) {
3953 /* delta + datetime */
3954 return add_datetime_timedelta((PyDateTime_DateTime *) right,
3955 (PyDateTime_Delta *) left,
3958 Py_INCREF(Py_NotImplemented);
3959 return Py_NotImplemented;
3962 static PyObject *
3963 datetime_subtract(PyObject *left, PyObject *right)
3965 PyObject *result = Py_NotImplemented;
3967 if (PyDateTime_Check(left)) {
3968 /* datetime - ??? */
3969 if (PyDateTime_Check(right)) {
3970 /* datetime - datetime */
3971 naivety n1, n2;
3972 int offset1, offset2;
3973 int delta_d, delta_s, delta_us;
3975 if (classify_two_utcoffsets(left, &offset1, &n1, left,
3976 right, &offset2, &n2,
3977 right) < 0)
3978 return NULL;
3979 assert(n1 != OFFSET_UNKNOWN && n2 != OFFSET_UNKNOWN);
3980 if (n1 != n2) {
3981 PyErr_SetString(PyExc_TypeError,
3982 "can't subtract offset-naive and "
3983 "offset-aware datetimes");
3984 return NULL;
3986 delta_d = ymd_to_ord(GET_YEAR(left),
3987 GET_MONTH(left),
3988 GET_DAY(left)) -
3989 ymd_to_ord(GET_YEAR(right),
3990 GET_MONTH(right),
3991 GET_DAY(right));
3992 /* These can't overflow, since the values are
3993 * normalized. At most this gives the number of
3994 * seconds in one day.
3996 delta_s = (DATE_GET_HOUR(left) -
3997 DATE_GET_HOUR(right)) * 3600 +
3998 (DATE_GET_MINUTE(left) -
3999 DATE_GET_MINUTE(right)) * 60 +
4000 (DATE_GET_SECOND(left) -
4001 DATE_GET_SECOND(right));
4002 delta_us = DATE_GET_MICROSECOND(left) -
4003 DATE_GET_MICROSECOND(right);
4004 /* (left - offset1) - (right - offset2) =
4005 * (left - right) + (offset2 - offset1)
4007 delta_s += (offset2 - offset1) * 60;
4008 result = new_delta(delta_d, delta_s, delta_us, 1);
4010 else if (PyDelta_Check(right)) {
4011 /* datetime - delta */
4012 result = add_datetime_timedelta(
4013 (PyDateTime_DateTime *)left,
4014 (PyDateTime_Delta *)right,
4015 -1);
4019 if (result == Py_NotImplemented)
4020 Py_INCREF(result);
4021 return result;
4024 /* Various ways to turn a datetime into a string. */
4026 static PyObject *
4027 datetime_repr(PyDateTime_DateTime *self)
4029 char buffer[1000];
4030 const char *typename = self->ob_type->tp_name;
4031 PyObject *baserepr;
4033 if (DATE_GET_MICROSECOND(self)) {
4034 PyOS_snprintf(buffer, sizeof(buffer),
4035 "%s(%d, %d, %d, %d, %d, %d, %d)",
4036 typename,
4037 GET_YEAR(self), GET_MONTH(self), GET_DAY(self),
4038 DATE_GET_HOUR(self), DATE_GET_MINUTE(self),
4039 DATE_GET_SECOND(self),
4040 DATE_GET_MICROSECOND(self));
4042 else if (DATE_GET_SECOND(self)) {
4043 PyOS_snprintf(buffer, sizeof(buffer),
4044 "%s(%d, %d, %d, %d, %d, %d)",
4045 typename,
4046 GET_YEAR(self), GET_MONTH(self), GET_DAY(self),
4047 DATE_GET_HOUR(self), DATE_GET_MINUTE(self),
4048 DATE_GET_SECOND(self));
4050 else {
4051 PyOS_snprintf(buffer, sizeof(buffer),
4052 "%s(%d, %d, %d, %d, %d)",
4053 typename,
4054 GET_YEAR(self), GET_MONTH(self), GET_DAY(self),
4055 DATE_GET_HOUR(self), DATE_GET_MINUTE(self));
4057 baserepr = PyString_FromString(buffer);
4058 if (baserepr == NULL || ! HASTZINFO(self))
4059 return baserepr;
4060 return append_keyword_tzinfo(baserepr, self->tzinfo);
4063 static PyObject *
4064 datetime_str(PyDateTime_DateTime *self)
4066 return PyObject_CallMethod((PyObject *)self, "isoformat", "(s)", " ");
4069 static PyObject *
4070 datetime_isoformat(PyDateTime_DateTime *self, PyObject *args, PyObject *kw)
4072 char sep = 'T';
4073 static const char *keywords[] = {"sep", NULL};
4074 char buffer[100];
4075 char *cp;
4076 PyObject *result;
4078 if (!PyArg_ParseTupleAndKeywords(args, kw, "|c:isoformat", keywords,
4079 &sep))
4080 return NULL;
4081 cp = isoformat_date((PyDateTime_Date *)self, buffer, sizeof(buffer));
4082 assert(cp != NULL);
4083 *cp++ = sep;
4084 isoformat_time(self, cp, sizeof(buffer) - (cp - buffer));
4085 result = PyString_FromString(buffer);
4086 if (result == NULL || ! HASTZINFO(self))
4087 return result;
4089 /* We need to append the UTC offset. */
4090 if (format_utcoffset(buffer, sizeof(buffer), ":", self->tzinfo,
4091 (PyObject *)self) < 0) {
4092 Py_DECREF(result);
4093 return NULL;
4095 PyString_ConcatAndDel(&result, PyString_FromString(buffer));
4096 return result;
4099 static PyObject *
4100 datetime_ctime(PyDateTime_DateTime *self)
4102 return format_ctime((PyDateTime_Date *)self,
4103 DATE_GET_HOUR(self),
4104 DATE_GET_MINUTE(self),
4105 DATE_GET_SECOND(self));
4108 /* Miscellaneous methods. */
4110 /* This is more natural as a tp_compare, but doesn't work then: for whatever
4111 * reason, Python's try_3way_compare ignores tp_compare unless
4112 * PyInstance_Check returns true, but these aren't old-style classes.
4114 static PyObject *
4115 datetime_richcompare(PyDateTime_DateTime *self, PyObject *other, int op)
4117 int diff;
4118 naivety n1, n2;
4119 int offset1, offset2;
4121 if (! PyDateTime_Check(other)) {
4122 /* If other has a "timetuple" attr, that's an advertised
4123 * hook for other classes to ask to get comparison control.
4124 * However, date instances have a timetuple attr, and we
4125 * don't want to allow that comparison. Because datetime
4126 * is a subclass of date, when mixing date and datetime
4127 * in a comparison, Python gives datetime the first shot
4128 * (it's the more specific subtype). So we can stop that
4129 * combination here reliably.
4131 if (PyObject_HasAttrString(other, "timetuple") &&
4132 ! PyDate_Check(other)) {
4133 /* A hook for other kinds of datetime objects. */
4134 Py_INCREF(Py_NotImplemented);
4135 return Py_NotImplemented;
4137 if (op == Py_EQ || op == Py_NE) {
4138 PyObject *result = op == Py_EQ ? Py_False : Py_True;
4139 Py_INCREF(result);
4140 return result;
4142 /* Stop this from falling back to address comparison. */
4143 return cmperror((PyObject *)self, other);
4146 if (classify_two_utcoffsets((PyObject *)self, &offset1, &n1,
4147 (PyObject *)self,
4148 other, &offset2, &n2,
4149 other) < 0)
4150 return NULL;
4151 assert(n1 != OFFSET_UNKNOWN && n2 != OFFSET_UNKNOWN);
4152 /* If they're both naive, or both aware and have the same offsets,
4153 * we get off cheap. Note that if they're both naive, offset1 ==
4154 * offset2 == 0 at this point.
4156 if (n1 == n2 && offset1 == offset2) {
4157 diff = memcmp(self->data, ((PyDateTime_DateTime *)other)->data,
4158 _PyDateTime_DATETIME_DATASIZE);
4159 return diff_to_bool(diff, op);
4162 if (n1 == OFFSET_AWARE && n2 == OFFSET_AWARE) {
4163 PyDateTime_Delta *delta;
4165 assert(offset1 != offset2); /* else last "if" handled it */
4166 delta = (PyDateTime_Delta *)datetime_subtract((PyObject *)self,
4167 other);
4168 if (delta == NULL)
4169 return NULL;
4170 diff = GET_TD_DAYS(delta);
4171 if (diff == 0)
4172 diff = GET_TD_SECONDS(delta) |
4173 GET_TD_MICROSECONDS(delta);
4174 Py_DECREF(delta);
4175 return diff_to_bool(diff, op);
4178 assert(n1 != n2);
4179 PyErr_SetString(PyExc_TypeError,
4180 "can't compare offset-naive and "
4181 "offset-aware datetimes");
4182 return NULL;
4185 static long
4186 datetime_hash(PyDateTime_DateTime *self)
4188 if (self->hashcode == -1) {
4189 naivety n;
4190 int offset;
4191 PyObject *temp;
4193 n = classify_utcoffset((PyObject *)self, (PyObject *)self,
4194 &offset);
4195 assert(n != OFFSET_UNKNOWN);
4196 if (n == OFFSET_ERROR)
4197 return -1;
4199 /* Reduce this to a hash of another object. */
4200 if (n == OFFSET_NAIVE)
4201 temp = PyString_FromStringAndSize(
4202 (char *)self->data,
4203 _PyDateTime_DATETIME_DATASIZE);
4204 else {
4205 int days;
4206 int seconds;
4208 assert(n == OFFSET_AWARE);
4209 assert(HASTZINFO(self));
4210 days = ymd_to_ord(GET_YEAR(self),
4211 GET_MONTH(self),
4212 GET_DAY(self));
4213 seconds = DATE_GET_HOUR(self) * 3600 +
4214 (DATE_GET_MINUTE(self) - offset) * 60 +
4215 DATE_GET_SECOND(self);
4216 temp = new_delta(days,
4217 seconds,
4218 DATE_GET_MICROSECOND(self),
4221 if (temp != NULL) {
4222 self->hashcode = PyObject_Hash(temp);
4223 Py_DECREF(temp);
4226 return self->hashcode;
4229 static PyObject *
4230 datetime_replace(PyDateTime_DateTime *self, PyObject *args, PyObject *kw)
4232 PyObject *clone;
4233 PyObject *tuple;
4234 int y = GET_YEAR(self);
4235 int m = GET_MONTH(self);
4236 int d = GET_DAY(self);
4237 int hh = DATE_GET_HOUR(self);
4238 int mm = DATE_GET_MINUTE(self);
4239 int ss = DATE_GET_SECOND(self);
4240 int us = DATE_GET_MICROSECOND(self);
4241 PyObject *tzinfo = HASTZINFO(self) ? self->tzinfo : Py_None;
4243 if (! PyArg_ParseTupleAndKeywords(args, kw, "|iiiiiiiO:replace",
4244 datetime_kws,
4245 &y, &m, &d, &hh, &mm, &ss, &us,
4246 &tzinfo))
4247 return NULL;
4248 tuple = Py_BuildValue("iiiiiiiO", y, m, d, hh, mm, ss, us, tzinfo);
4249 if (tuple == NULL)
4250 return NULL;
4251 clone = datetime_new(self->ob_type, tuple, NULL);
4252 Py_DECREF(tuple);
4253 return clone;
4256 static PyObject *
4257 datetime_astimezone(PyDateTime_DateTime *self, PyObject *args, PyObject *kw)
4259 int y, m, d, hh, mm, ss, us;
4260 PyObject *result;
4261 int offset, none;
4263 PyObject *tzinfo;
4264 static const char *keywords[] = {"tz", NULL};
4266 if (! PyArg_ParseTupleAndKeywords(args, kw, "O!:astimezone", keywords,
4267 &PyDateTime_TZInfoType, &tzinfo))
4268 return NULL;
4270 if (!HASTZINFO(self) || self->tzinfo == Py_None)
4271 goto NeedAware;
4273 /* Conversion to self's own time zone is a NOP. */
4274 if (self->tzinfo == tzinfo) {
4275 Py_INCREF(self);
4276 return (PyObject *)self;
4279 /* Convert self to UTC. */
4280 offset = call_utcoffset(self->tzinfo, (PyObject *)self, &none);
4281 if (offset == -1 && PyErr_Occurred())
4282 return NULL;
4283 if (none)
4284 goto NeedAware;
4286 y = GET_YEAR(self);
4287 m = GET_MONTH(self);
4288 d = GET_DAY(self);
4289 hh = DATE_GET_HOUR(self);
4290 mm = DATE_GET_MINUTE(self);
4291 ss = DATE_GET_SECOND(self);
4292 us = DATE_GET_MICROSECOND(self);
4294 mm -= offset;
4295 if ((mm < 0 || mm >= 60) &&
4296 normalize_datetime(&y, &m, &d, &hh, &mm, &ss, &us) < 0)
4297 return NULL;
4299 /* Attach new tzinfo and let fromutc() do the rest. */
4300 result = new_datetime(y, m, d, hh, mm, ss, us, tzinfo);
4301 if (result != NULL) {
4302 PyObject *temp = result;
4304 result = PyObject_CallMethod(tzinfo, "fromutc", "O", temp);
4305 Py_DECREF(temp);
4307 return result;
4309 NeedAware:
4310 PyErr_SetString(PyExc_ValueError, "astimezone() cannot be applied to "
4311 "a naive datetime");
4312 return NULL;
4315 static PyObject *
4316 datetime_timetuple(PyDateTime_DateTime *self)
4318 int dstflag = -1;
4320 if (HASTZINFO(self) && self->tzinfo != Py_None) {
4321 int none;
4323 dstflag = call_dst(self->tzinfo, (PyObject *)self, &none);
4324 if (dstflag == -1 && PyErr_Occurred())
4325 return NULL;
4327 if (none)
4328 dstflag = -1;
4329 else if (dstflag != 0)
4330 dstflag = 1;
4333 return build_struct_time(GET_YEAR(self),
4334 GET_MONTH(self),
4335 GET_DAY(self),
4336 DATE_GET_HOUR(self),
4337 DATE_GET_MINUTE(self),
4338 DATE_GET_SECOND(self),
4339 dstflag);
4342 static PyObject *
4343 datetime_getdate(PyDateTime_DateTime *self)
4345 return new_date(GET_YEAR(self),
4346 GET_MONTH(self),
4347 GET_DAY(self));
4350 static PyObject *
4351 datetime_gettime(PyDateTime_DateTime *self)
4353 return new_time(DATE_GET_HOUR(self),
4354 DATE_GET_MINUTE(self),
4355 DATE_GET_SECOND(self),
4356 DATE_GET_MICROSECOND(self),
4357 Py_None);
4360 static PyObject *
4361 datetime_gettimetz(PyDateTime_DateTime *self)
4363 return new_time(DATE_GET_HOUR(self),
4364 DATE_GET_MINUTE(self),
4365 DATE_GET_SECOND(self),
4366 DATE_GET_MICROSECOND(self),
4367 HASTZINFO(self) ? self->tzinfo : Py_None);
4370 static PyObject *
4371 datetime_utctimetuple(PyDateTime_DateTime *self)
4373 int y = GET_YEAR(self);
4374 int m = GET_MONTH(self);
4375 int d = GET_DAY(self);
4376 int hh = DATE_GET_HOUR(self);
4377 int mm = DATE_GET_MINUTE(self);
4378 int ss = DATE_GET_SECOND(self);
4379 int us = 0; /* microseconds are ignored in a timetuple */
4380 int offset = 0;
4382 if (HASTZINFO(self) && self->tzinfo != Py_None) {
4383 int none;
4385 offset = call_utcoffset(self->tzinfo, (PyObject *)self, &none);
4386 if (offset == -1 && PyErr_Occurred())
4387 return NULL;
4389 /* Even if offset is 0, don't call timetuple() -- tm_isdst should be
4390 * 0 in a UTC timetuple regardless of what dst() says.
4392 if (offset) {
4393 /* Subtract offset minutes & normalize. */
4394 int stat;
4396 mm -= offset;
4397 stat = normalize_datetime(&y, &m, &d, &hh, &mm, &ss, &us);
4398 if (stat < 0) {
4399 /* At the edges, it's possible we overflowed
4400 * beyond MINYEAR or MAXYEAR.
4402 if (PyErr_ExceptionMatches(PyExc_OverflowError))
4403 PyErr_Clear();
4404 else
4405 return NULL;
4408 return build_struct_time(y, m, d, hh, mm, ss, 0);
4411 /* Pickle support, a simple use of __reduce__. */
4413 /* Let basestate be the non-tzinfo data string.
4414 * If tzinfo is None, this returns (basestate,), else (basestate, tzinfo).
4415 * So it's a tuple in any (non-error) case.
4416 * __getstate__ isn't exposed.
4418 static PyObject *
4419 datetime_getstate(PyDateTime_DateTime *self)
4421 PyObject *basestate;
4422 PyObject *result = NULL;
4424 basestate = PyString_FromStringAndSize((char *)self->data,
4425 _PyDateTime_DATETIME_DATASIZE);
4426 if (basestate != NULL) {
4427 if (! HASTZINFO(self) || self->tzinfo == Py_None)
4428 result = PyTuple_Pack(1, basestate);
4429 else
4430 result = PyTuple_Pack(2, basestate, self->tzinfo);
4431 Py_DECREF(basestate);
4433 return result;
4436 static PyObject *
4437 datetime_reduce(PyDateTime_DateTime *self, PyObject *arg)
4439 return Py_BuildValue("(ON)", self->ob_type, datetime_getstate(self));
4442 static PyMethodDef datetime_methods[] = {
4444 /* Class methods: */
4446 {"now", (PyCFunction)datetime_now,
4447 METH_KEYWORDS | METH_CLASS,
4448 PyDoc_STR("[tz] -> new datetime with tz's local day and time.")},
4450 {"utcnow", (PyCFunction)datetime_utcnow,
4451 METH_NOARGS | METH_CLASS,
4452 PyDoc_STR("Return a new datetime representing UTC day and time.")},
4454 {"fromtimestamp", (PyCFunction)datetime_fromtimestamp,
4455 METH_KEYWORDS | METH_CLASS,
4456 PyDoc_STR("timestamp[, tz] -> tz's local time from POSIX timestamp.")},
4458 {"utcfromtimestamp", (PyCFunction)datetime_utcfromtimestamp,
4459 METH_VARARGS | METH_CLASS,
4460 PyDoc_STR("timestamp -> UTC datetime from a POSIX timestamp "
4461 "(like time.time()).")},
4463 {"strptime", (PyCFunction)datetime_strptime,
4464 METH_VARARGS | METH_CLASS,
4465 PyDoc_STR("string, format -> new datetime parsed from a string "
4466 "(like time.strptime()).")},
4468 {"combine", (PyCFunction)datetime_combine,
4469 METH_VARARGS | METH_KEYWORDS | METH_CLASS,
4470 PyDoc_STR("date, time -> datetime with same date and time fields")},
4472 /* Instance methods: */
4474 {"date", (PyCFunction)datetime_getdate, METH_NOARGS,
4475 PyDoc_STR("Return date object with same year, month and day.")},
4477 {"time", (PyCFunction)datetime_gettime, METH_NOARGS,
4478 PyDoc_STR("Return time object with same time but with tzinfo=None.")},
4480 {"timetz", (PyCFunction)datetime_gettimetz, METH_NOARGS,
4481 PyDoc_STR("Return time object with same time and tzinfo.")},
4483 {"ctime", (PyCFunction)datetime_ctime, METH_NOARGS,
4484 PyDoc_STR("Return ctime() style string.")},
4486 {"timetuple", (PyCFunction)datetime_timetuple, METH_NOARGS,
4487 PyDoc_STR("Return time tuple, compatible with time.localtime().")},
4489 {"utctimetuple", (PyCFunction)datetime_utctimetuple, METH_NOARGS,
4490 PyDoc_STR("Return UTC time tuple, compatible with time.localtime().")},
4492 {"isoformat", (PyCFunction)datetime_isoformat, METH_KEYWORDS,
4493 PyDoc_STR("[sep] -> string in ISO 8601 format, "
4494 "YYYY-MM-DDTHH:MM:SS[.mmmmmm][+HH:MM].\n\n"
4495 "sep is used to separate the year from the time, and "
4496 "defaults to 'T'.")},
4498 {"utcoffset", (PyCFunction)datetime_utcoffset, METH_NOARGS,
4499 PyDoc_STR("Return self.tzinfo.utcoffset(self).")},
4501 {"tzname", (PyCFunction)datetime_tzname, METH_NOARGS,
4502 PyDoc_STR("Return self.tzinfo.tzname(self).")},
4504 {"dst", (PyCFunction)datetime_dst, METH_NOARGS,
4505 PyDoc_STR("Return self.tzinfo.dst(self).")},
4507 {"replace", (PyCFunction)datetime_replace, METH_KEYWORDS,
4508 PyDoc_STR("Return datetime with new specified fields.")},
4510 {"astimezone", (PyCFunction)datetime_astimezone, METH_KEYWORDS,
4511 PyDoc_STR("tz -> convert to local time in new timezone tz\n")},
4513 {"__reduce__", (PyCFunction)datetime_reduce, METH_NOARGS,
4514 PyDoc_STR("__reduce__() -> (cls, state)")},
4516 {NULL, NULL}
4519 static char datetime_doc[] =
4520 PyDoc_STR("datetime(year, month, day[, hour[, minute[, second[, microsecond[,tzinfo]]]]])\n\
4522 The year, month and day arguments are required. tzinfo may be None, or an\n\
4523 instance of a tzinfo subclass. The remaining arguments may be ints or longs.\n");
4525 static PyNumberMethods datetime_as_number = {
4526 datetime_add, /* nb_add */
4527 datetime_subtract, /* nb_subtract */
4528 0, /* nb_multiply */
4529 0, /* nb_divide */
4530 0, /* nb_remainder */
4531 0, /* nb_divmod */
4532 0, /* nb_power */
4533 0, /* nb_negative */
4534 0, /* nb_positive */
4535 0, /* nb_absolute */
4536 0, /* nb_nonzero */
4539 statichere PyTypeObject PyDateTime_DateTimeType = {
4540 PyObject_HEAD_INIT(NULL)
4541 0, /* ob_size */
4542 "datetime.datetime", /* tp_name */
4543 sizeof(PyDateTime_DateTime), /* tp_basicsize */
4544 0, /* tp_itemsize */
4545 (destructor)datetime_dealloc, /* tp_dealloc */
4546 0, /* tp_print */
4547 0, /* tp_getattr */
4548 0, /* tp_setattr */
4549 0, /* tp_compare */
4550 (reprfunc)datetime_repr, /* tp_repr */
4551 &datetime_as_number, /* tp_as_number */
4552 0, /* tp_as_sequence */
4553 0, /* tp_as_mapping */
4554 (hashfunc)datetime_hash, /* tp_hash */
4555 0, /* tp_call */
4556 (reprfunc)datetime_str, /* tp_str */
4557 PyObject_GenericGetAttr, /* tp_getattro */
4558 0, /* tp_setattro */
4559 0, /* tp_as_buffer */
4560 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_CHECKTYPES |
4561 Py_TPFLAGS_BASETYPE, /* tp_flags */
4562 datetime_doc, /* tp_doc */
4563 0, /* tp_traverse */
4564 0, /* tp_clear */
4565 (richcmpfunc)datetime_richcompare, /* tp_richcompare */
4566 0, /* tp_weaklistoffset */
4567 0, /* tp_iter */
4568 0, /* tp_iternext */
4569 datetime_methods, /* tp_methods */
4570 0, /* tp_members */
4571 datetime_getset, /* tp_getset */
4572 &PyDateTime_DateType, /* tp_base */
4573 0, /* tp_dict */
4574 0, /* tp_descr_get */
4575 0, /* tp_descr_set */
4576 0, /* tp_dictoffset */
4577 0, /* tp_init */
4578 datetime_alloc, /* tp_alloc */
4579 datetime_new, /* tp_new */
4580 0, /* tp_free */
4583 /* ---------------------------------------------------------------------------
4584 * Module methods and initialization.
4587 static PyMethodDef module_methods[] = {
4588 {NULL, NULL}
4591 /* C API. Clients get at this via PyDateTime_IMPORT, defined in
4592 * datetime.h.
4594 static PyDateTime_CAPI CAPI = {
4595 &PyDateTime_DateType,
4596 &PyDateTime_DateTimeType,
4597 &PyDateTime_TimeType,
4598 &PyDateTime_DeltaType,
4599 &PyDateTime_TZInfoType,
4600 new_date_ex,
4601 new_datetime_ex,
4602 new_time_ex,
4603 new_delta_ex,
4604 datetime_fromtimestamp,
4605 date_fromtimestamp
4609 PyMODINIT_FUNC
4610 initdatetime(void)
4612 PyObject *m; /* a module object */
4613 PyObject *d; /* its dict */
4614 PyObject *x;
4616 m = Py_InitModule3("datetime", module_methods,
4617 "Fast implementation of the datetime type.");
4618 if (m == NULL)
4619 return;
4621 if (PyType_Ready(&PyDateTime_DateType) < 0)
4622 return;
4623 if (PyType_Ready(&PyDateTime_DateTimeType) < 0)
4624 return;
4625 if (PyType_Ready(&PyDateTime_DeltaType) < 0)
4626 return;
4627 if (PyType_Ready(&PyDateTime_TimeType) < 0)
4628 return;
4629 if (PyType_Ready(&PyDateTime_TZInfoType) < 0)
4630 return;
4632 /* timedelta values */
4633 d = PyDateTime_DeltaType.tp_dict;
4635 x = new_delta(0, 0, 1, 0);
4636 if (x == NULL || PyDict_SetItemString(d, "resolution", x) < 0)
4637 return;
4638 Py_DECREF(x);
4640 x = new_delta(-MAX_DELTA_DAYS, 0, 0, 0);
4641 if (x == NULL || PyDict_SetItemString(d, "min", x) < 0)
4642 return;
4643 Py_DECREF(x);
4645 x = new_delta(MAX_DELTA_DAYS, 24*3600-1, 1000000-1, 0);
4646 if (x == NULL || PyDict_SetItemString(d, "max", x) < 0)
4647 return;
4648 Py_DECREF(x);
4650 /* date values */
4651 d = PyDateTime_DateType.tp_dict;
4653 x = new_date(1, 1, 1);
4654 if (x == NULL || PyDict_SetItemString(d, "min", x) < 0)
4655 return;
4656 Py_DECREF(x);
4658 x = new_date(MAXYEAR, 12, 31);
4659 if (x == NULL || PyDict_SetItemString(d, "max", x) < 0)
4660 return;
4661 Py_DECREF(x);
4663 x = new_delta(1, 0, 0, 0);
4664 if (x == NULL || PyDict_SetItemString(d, "resolution", x) < 0)
4665 return;
4666 Py_DECREF(x);
4668 /* time values */
4669 d = PyDateTime_TimeType.tp_dict;
4671 x = new_time(0, 0, 0, 0, Py_None);
4672 if (x == NULL || PyDict_SetItemString(d, "min", x) < 0)
4673 return;
4674 Py_DECREF(x);
4676 x = new_time(23, 59, 59, 999999, Py_None);
4677 if (x == NULL || PyDict_SetItemString(d, "max", x) < 0)
4678 return;
4679 Py_DECREF(x);
4681 x = new_delta(0, 0, 1, 0);
4682 if (x == NULL || PyDict_SetItemString(d, "resolution", x) < 0)
4683 return;
4684 Py_DECREF(x);
4686 /* datetime values */
4687 d = PyDateTime_DateTimeType.tp_dict;
4689 x = new_datetime(1, 1, 1, 0, 0, 0, 0, Py_None);
4690 if (x == NULL || PyDict_SetItemString(d, "min", x) < 0)
4691 return;
4692 Py_DECREF(x);
4694 x = new_datetime(MAXYEAR, 12, 31, 23, 59, 59, 999999, Py_None);
4695 if (x == NULL || PyDict_SetItemString(d, "max", x) < 0)
4696 return;
4697 Py_DECREF(x);
4699 x = new_delta(0, 0, 1, 0);
4700 if (x == NULL || PyDict_SetItemString(d, "resolution", x) < 0)
4701 return;
4702 Py_DECREF(x);
4704 /* module initialization */
4705 PyModule_AddIntConstant(m, "MINYEAR", MINYEAR);
4706 PyModule_AddIntConstant(m, "MAXYEAR", MAXYEAR);
4708 Py_INCREF(&PyDateTime_DateType);
4709 PyModule_AddObject(m, "date", (PyObject *) &PyDateTime_DateType);
4711 Py_INCREF(&PyDateTime_DateTimeType);
4712 PyModule_AddObject(m, "datetime",
4713 (PyObject *)&PyDateTime_DateTimeType);
4715 Py_INCREF(&PyDateTime_TimeType);
4716 PyModule_AddObject(m, "time", (PyObject *) &PyDateTime_TimeType);
4718 Py_INCREF(&PyDateTime_DeltaType);
4719 PyModule_AddObject(m, "timedelta", (PyObject *) &PyDateTime_DeltaType);
4721 Py_INCREF(&PyDateTime_TZInfoType);
4722 PyModule_AddObject(m, "tzinfo", (PyObject *) &PyDateTime_TZInfoType);
4724 x = PyCObject_FromVoidPtrAndDesc(&CAPI, (void*) DATETIME_API_MAGIC,
4725 NULL);
4726 if (x == NULL)
4727 return;
4728 PyModule_AddObject(m, "datetime_CAPI", x);
4730 /* A 4-year cycle has an extra leap day over what we'd get from
4731 * pasting together 4 single years.
4733 assert(DI4Y == 4 * 365 + 1);
4734 assert(DI4Y == days_before_year(4+1));
4736 /* Similarly, a 400-year cycle has an extra leap day over what we'd
4737 * get from pasting together 4 100-year cycles.
4739 assert(DI400Y == 4 * DI100Y + 1);
4740 assert(DI400Y == days_before_year(400+1));
4742 /* OTOH, a 100-year cycle has one fewer leap day than we'd get from
4743 * pasting together 25 4-year cycles.
4745 assert(DI100Y == 25 * DI4Y - 1);
4746 assert(DI100Y == days_before_year(100+1));
4748 us_per_us = PyInt_FromLong(1);
4749 us_per_ms = PyInt_FromLong(1000);
4750 us_per_second = PyInt_FromLong(1000000);
4751 us_per_minute = PyInt_FromLong(60000000);
4752 seconds_per_day = PyInt_FromLong(24 * 3600);
4753 if (us_per_us == NULL || us_per_ms == NULL || us_per_second == NULL ||
4754 us_per_minute == NULL || seconds_per_day == NULL)
4755 return;
4757 /* The rest are too big for 32-bit ints, but even
4758 * us_per_week fits in 40 bits, so doubles should be exact.
4760 us_per_hour = PyLong_FromDouble(3600000000.0);
4761 us_per_day = PyLong_FromDouble(86400000000.0);
4762 us_per_week = PyLong_FromDouble(604800000000.0);
4763 if (us_per_hour == NULL || us_per_day == NULL || us_per_week == NULL)
4764 return;
4767 /* ---------------------------------------------------------------------------
4768 Some time zone algebra. For a datetime x, let
4769 x.n = x stripped of its timezone -- its naive time.
4770 x.o = x.utcoffset(), and assuming that doesn't raise an exception or
4771 return None
4772 x.d = x.dst(), and assuming that doesn't raise an exception or
4773 return None
4774 x.s = x's standard offset, x.o - x.d
4776 Now some derived rules, where k is a duration (timedelta).
4778 1. x.o = x.s + x.d
4779 This follows from the definition of x.s.
4781 2. If x and y have the same tzinfo member, x.s = y.s.
4782 This is actually a requirement, an assumption we need to make about
4783 sane tzinfo classes.
4785 3. The naive UTC time corresponding to x is x.n - x.o.
4786 This is again a requirement for a sane tzinfo class.
4788 4. (x+k).s = x.s
4789 This follows from #2, and that datimetimetz+timedelta preserves tzinfo.
4791 5. (x+k).n = x.n + k
4792 Again follows from how arithmetic is defined.
4794 Now we can explain tz.fromutc(x). Let's assume it's an interesting case
4795 (meaning that the various tzinfo methods exist, and don't blow up or return
4796 None when called).
4798 The function wants to return a datetime y with timezone tz, equivalent to x.
4799 x is already in UTC.
4801 By #3, we want
4803 y.n - y.o = x.n [1]
4805 The algorithm starts by attaching tz to x.n, and calling that y. So
4806 x.n = y.n at the start. Then it wants to add a duration k to y, so that [1]
4807 becomes true; in effect, we want to solve [2] for k:
4809 (y+k).n - (y+k).o = x.n [2]
4811 By #1, this is the same as
4813 (y+k).n - ((y+k).s + (y+k).d) = x.n [3]
4815 By #5, (y+k).n = y.n + k, which equals x.n + k because x.n=y.n at the start.
4816 Substituting that into [3],
4818 x.n + k - (y+k).s - (y+k).d = x.n; the x.n terms cancel, leaving
4819 k - (y+k).s - (y+k).d = 0; rearranging,
4820 k = (y+k).s - (y+k).d; by #4, (y+k).s == y.s, so
4821 k = y.s - (y+k).d
4823 On the RHS, (y+k).d can't be computed directly, but y.s can be, and we
4824 approximate k by ignoring the (y+k).d term at first. Note that k can't be
4825 very large, since all offset-returning methods return a duration of magnitude
4826 less than 24 hours. For that reason, if y is firmly in std time, (y+k).d must
4827 be 0, so ignoring it has no consequence then.
4829 In any case, the new value is
4831 z = y + y.s [4]
4833 It's helpful to step back at look at [4] from a higher level: it's simply
4834 mapping from UTC to tz's standard time.
4836 At this point, if
4838 z.n - z.o = x.n [5]
4840 we have an equivalent time, and are almost done. The insecurity here is
4841 at the start of daylight time. Picture US Eastern for concreteness. The wall
4842 time jumps from 1:59 to 3:00, and wall hours of the form 2:MM don't make good
4843 sense then. The docs ask that an Eastern tzinfo class consider such a time to
4844 be EDT (because it's "after 2"), which is a redundant spelling of 1:MM EST
4845 on the day DST starts. We want to return the 1:MM EST spelling because that's
4846 the only spelling that makes sense on the local wall clock.
4848 In fact, if [5] holds at this point, we do have the standard-time spelling,
4849 but that takes a bit of proof. We first prove a stronger result. What's the
4850 difference between the LHS and RHS of [5]? Let
4852 diff = x.n - (z.n - z.o) [6]
4855 z.n = by [4]
4856 (y + y.s).n = by #5
4857 y.n + y.s = since y.n = x.n
4858 x.n + y.s = since z and y are have the same tzinfo member,
4859 y.s = z.s by #2
4860 x.n + z.s
4862 Plugging that back into [6] gives
4864 diff =
4865 x.n - ((x.n + z.s) - z.o) = expanding
4866 x.n - x.n - z.s + z.o = cancelling
4867 - z.s + z.o = by #2
4870 So diff = z.d.
4872 If [5] is true now, diff = 0, so z.d = 0 too, and we have the standard-time
4873 spelling we wanted in the endcase described above. We're done. Contrarily,
4874 if z.d = 0, then we have a UTC equivalent, and are also done.
4876 If [5] is not true now, diff = z.d != 0, and z.d is the offset we need to
4877 add to z (in effect, z is in tz's standard time, and we need to shift the
4878 local clock into tz's daylight time).
4882 z' = z + z.d = z + diff [7]
4884 and we can again ask whether
4886 z'.n - z'.o = x.n [8]
4888 If so, we're done. If not, the tzinfo class is insane, according to the
4889 assumptions we've made. This also requires a bit of proof. As before, let's
4890 compute the difference between the LHS and RHS of [8] (and skipping some of
4891 the justifications for the kinds of substitutions we've done several times
4892 already):
4894 diff' = x.n - (z'.n - z'.o) = replacing z'.n via [7]
4895 x.n - (z.n + diff - z'.o) = replacing diff via [6]
4896 x.n - (z.n + x.n - (z.n - z.o) - z'.o) =
4897 x.n - z.n - x.n + z.n - z.o + z'.o = cancel x.n
4898 - z.n + z.n - z.o + z'.o = cancel z.n
4899 - z.o + z'.o = #1 twice
4900 -z.s - z.d + z'.s + z'.d = z and z' have same tzinfo
4901 z'.d - z.d
4903 So z' is UTC-equivalent to x iff z'.d = z.d at this point. If they are equal,
4904 we've found the UTC-equivalent so are done. In fact, we stop with [7] and
4905 return z', not bothering to compute z'.d.
4907 How could z.d and z'd differ? z' = z + z.d [7], so merely moving z' by
4908 a dst() offset, and starting *from* a time already in DST (we know z.d != 0),
4909 would have to change the result dst() returns: we start in DST, and moving
4910 a little further into it takes us out of DST.
4912 There isn't a sane case where this can happen. The closest it gets is at
4913 the end of DST, where there's an hour in UTC with no spelling in a hybrid
4914 tzinfo class. In US Eastern, that's 5:MM UTC = 0:MM EST = 1:MM EDT. During
4915 that hour, on an Eastern clock 1:MM is taken as being in standard time (6:MM
4916 UTC) because the docs insist on that, but 0:MM is taken as being in daylight
4917 time (4:MM UTC). There is no local time mapping to 5:MM UTC. The local
4918 clock jumps from 1:59 back to 1:00 again, and repeats the 1:MM hour in
4919 standard time. Since that's what the local clock *does*, we want to map both
4920 UTC hours 5:MM and 6:MM to 1:MM Eastern. The result is ambiguous
4921 in local time, but so it goes -- it's the way the local clock works.
4923 When x = 5:MM UTC is the input to this algorithm, x.o=0, y.o=-5 and y.d=0,
4924 so z=0:MM. z.d=60 (minutes) then, so [5] doesn't hold and we keep going.
4925 z' = z + z.d = 1:MM then, and z'.d=0, and z'.d - z.d = -60 != 0 so [8]
4926 (correctly) concludes that z' is not UTC-equivalent to x.
4928 Because we know z.d said z was in daylight time (else [5] would have held and
4929 we would have stopped then), and we know z.d != z'.d (else [8] would have held
4930 and we would have stopped then), and there are only 2 possible values dst() can
4931 return in Eastern, it follows that z'.d must be 0 (which it is in the example,
4932 but the reasoning doesn't depend on the example -- it depends on there being
4933 two possible dst() outcomes, one zero and the other non-zero). Therefore
4934 z' must be in standard time, and is the spelling we want in this case.
4936 Note again that z' is not UTC-equivalent as far as the hybrid tzinfo class is
4937 concerned (because it takes z' as being in standard time rather than the
4938 daylight time we intend here), but returning it gives the real-life "local
4939 clock repeats an hour" behavior when mapping the "unspellable" UTC hour into
4942 When the input is 6:MM, z=1:MM and z.d=0, and we stop at once, again with
4943 the 1:MM standard time spelling we want.
4945 So how can this break? One of the assumptions must be violated. Two
4946 possibilities:
4948 1) [2] effectively says that y.s is invariant across all y belong to a given
4949 time zone. This isn't true if, for political reasons or continental drift,
4950 a region decides to change its base offset from UTC.
4952 2) There may be versions of "double daylight" time where the tail end of
4953 the analysis gives up a step too early. I haven't thought about that
4954 enough to say.
4956 In any case, it's clear that the default fromutc() is strong enough to handle
4957 "almost all" time zones: so long as the standard offset is invariant, it
4958 doesn't matter if daylight time transition points change from year to year, or
4959 if daylight time is skipped in some years; it doesn't matter how large or
4960 small dst() may get within its bounds; and it doesn't even matter if some
4961 perverse time zone returns a negative dst()). So a breaking case must be
4962 pretty bizarre, and a tzinfo subclass can override fromutc() if it is.
4963 --------------------------------------------------------------------------- */