blob: e64b230a432fa50b2b9c4e11d8dbce3838c93aa5 [file] [log] [blame]
Tim Peters2a799bf2002-12-16 20:18:38 +00001/* C implementation for the date/time type documented at
2 * http://www.zope.org/Members/fdrake/DateTimeWiki/FrontPage
3 */
4
5#include "Python.h"
6#include "modsupport.h"
7#include "structmember.h"
8
9#include <time.h>
10
Tim Peters1b6f7a92004-06-20 02:50:16 +000011#include "timefuncs.h"
Tim Peters9ddf40b2004-06-20 22:41:32 +000012
13/* Differentiate between building the core module and building extension
14 * modules.
15 */
Guido van Rossum360e4b82007-05-14 22:51:27 +000016#ifndef Py_BUILD_CORE
Tim Peters9ddf40b2004-06-20 22:41:32 +000017#define Py_BUILD_CORE
Guido van Rossum360e4b82007-05-14 22:51:27 +000018#endif
Tim Peters2a799bf2002-12-16 20:18:38 +000019#include "datetime.h"
Tim Peters9ddf40b2004-06-20 22:41:32 +000020#undef Py_BUILD_CORE
Tim Peters2a799bf2002-12-16 20:18:38 +000021
22/* We require that C int be at least 32 bits, and use int virtually
23 * everywhere. In just a few cases we use a temp long, where a Python
24 * API returns a C long. In such cases, we have to ensure that the
25 * final result fits in a C int (this can be an issue on 64-bit boxes).
26 */
27#if SIZEOF_INT < 4
28# error "datetime.c requires that C int have at least 32 bits"
29#endif
30
31#define MINYEAR 1
32#define MAXYEAR 9999
33
34/* Nine decimal digits is easy to communicate, and leaves enough room
35 * so that two delta days can be added w/o fear of overflowing a signed
36 * 32-bit int, and with plenty of room left over to absorb any possible
37 * carries from adding seconds.
38 */
39#define MAX_DELTA_DAYS 999999999
40
41/* Rename the long macros in datetime.h to more reasonable short names. */
42#define GET_YEAR PyDateTime_GET_YEAR
43#define GET_MONTH PyDateTime_GET_MONTH
44#define GET_DAY PyDateTime_GET_DAY
45#define DATE_GET_HOUR PyDateTime_DATE_GET_HOUR
46#define DATE_GET_MINUTE PyDateTime_DATE_GET_MINUTE
47#define DATE_GET_SECOND PyDateTime_DATE_GET_SECOND
48#define DATE_GET_MICROSECOND PyDateTime_DATE_GET_MICROSECOND
49
50/* Date accessors for date and datetime. */
51#define SET_YEAR(o, v) (((o)->data[0] = ((v) & 0xff00) >> 8), \
52 ((o)->data[1] = ((v) & 0x00ff)))
53#define SET_MONTH(o, v) (PyDateTime_GET_MONTH(o) = (v))
54#define SET_DAY(o, v) (PyDateTime_GET_DAY(o) = (v))
55
56/* Date/Time accessors for datetime. */
57#define DATE_SET_HOUR(o, v) (PyDateTime_DATE_GET_HOUR(o) = (v))
58#define DATE_SET_MINUTE(o, v) (PyDateTime_DATE_GET_MINUTE(o) = (v))
59#define DATE_SET_SECOND(o, v) (PyDateTime_DATE_GET_SECOND(o) = (v))
60#define DATE_SET_MICROSECOND(o, v) \
61 (((o)->data[7] = ((v) & 0xff0000) >> 16), \
62 ((o)->data[8] = ((v) & 0x00ff00) >> 8), \
63 ((o)->data[9] = ((v) & 0x0000ff)))
64
65/* Time accessors for time. */
66#define TIME_GET_HOUR PyDateTime_TIME_GET_HOUR
67#define TIME_GET_MINUTE PyDateTime_TIME_GET_MINUTE
68#define TIME_GET_SECOND PyDateTime_TIME_GET_SECOND
69#define TIME_GET_MICROSECOND PyDateTime_TIME_GET_MICROSECOND
70#define TIME_SET_HOUR(o, v) (PyDateTime_TIME_GET_HOUR(o) = (v))
71#define TIME_SET_MINUTE(o, v) (PyDateTime_TIME_GET_MINUTE(o) = (v))
72#define TIME_SET_SECOND(o, v) (PyDateTime_TIME_GET_SECOND(o) = (v))
73#define TIME_SET_MICROSECOND(o, v) \
74 (((o)->data[3] = ((v) & 0xff0000) >> 16), \
75 ((o)->data[4] = ((v) & 0x00ff00) >> 8), \
76 ((o)->data[5] = ((v) & 0x0000ff)))
77
78/* Delta accessors for timedelta. */
79#define GET_TD_DAYS(o) (((PyDateTime_Delta *)(o))->days)
80#define GET_TD_SECONDS(o) (((PyDateTime_Delta *)(o))->seconds)
81#define GET_TD_MICROSECONDS(o) (((PyDateTime_Delta *)(o))->microseconds)
82
83#define SET_TD_DAYS(o, v) ((o)->days = (v))
84#define SET_TD_SECONDS(o, v) ((o)->seconds = (v))
85#define SET_TD_MICROSECONDS(o, v) ((o)->microseconds = (v))
86
Tim Petersa032d2e2003-01-11 00:15:54 +000087/* p is a pointer to a time or a datetime object; HASTZINFO(p) returns
88 * p->hastzinfo.
89 */
90#define HASTZINFO(p) (((_PyDateTime_BaseTZInfo *)(p))->hastzinfo)
91
Tim Peters3f606292004-03-21 23:38:41 +000092/* M is a char or int claiming to be a valid month. The macro is equivalent
93 * to the two-sided Python test
94 * 1 <= M <= 12
95 */
96#define MONTH_IS_SANE(M) ((unsigned int)(M) - 1 < 12)
97
Tim Peters2a799bf2002-12-16 20:18:38 +000098/* Forward declarations. */
99static PyTypeObject PyDateTime_DateType;
100static PyTypeObject PyDateTime_DateTimeType;
Tim Peters2a799bf2002-12-16 20:18:38 +0000101static PyTypeObject PyDateTime_DeltaType;
102static PyTypeObject PyDateTime_TimeType;
103static PyTypeObject PyDateTime_TZInfoType;
Tim Peters2a799bf2002-12-16 20:18:38 +0000104
105/* ---------------------------------------------------------------------------
106 * Math utilities.
107 */
108
109/* k = i+j overflows iff k differs in sign from both inputs,
110 * iff k^i has sign bit set and k^j has sign bit set,
111 * iff (k^i)&(k^j) has sign bit set.
112 */
113#define SIGNED_ADD_OVERFLOWED(RESULT, I, J) \
114 ((((RESULT) ^ (I)) & ((RESULT) ^ (J))) < 0)
115
116/* Compute Python divmod(x, y), returning the quotient and storing the
117 * remainder into *r. The quotient is the floor of x/y, and that's
118 * the real point of this. C will probably truncate instead (C99
119 * requires truncation; C89 left it implementation-defined).
120 * Simplification: we *require* that y > 0 here. That's appropriate
121 * for all the uses made of it. This simplifies the code and makes
122 * the overflow case impossible (divmod(LONG_MIN, -1) is the only
123 * overflow case).
124 */
125static int
126divmod(int x, int y, int *r)
127{
128 int quo;
129
130 assert(y > 0);
131 quo = x / y;
132 *r = x - quo * y;
133 if (*r < 0) {
134 --quo;
135 *r += y;
136 }
137 assert(0 <= *r && *r < y);
138 return quo;
139}
140
Tim Peters5d644dd2003-01-02 16:32:54 +0000141/* Round a double to the nearest long. |x| must be small enough to fit
142 * in a C long; this is not checked.
143 */
144static long
145round_to_long(double x)
146{
147 if (x >= 0.0)
148 x = floor(x + 0.5);
149 else
150 x = ceil(x - 0.5);
151 return (long)x;
152}
153
Tim Peters2a799bf2002-12-16 20:18:38 +0000154/* ---------------------------------------------------------------------------
155 * General calendrical helper functions
156 */
157
158/* For each month ordinal in 1..12, the number of days in that month,
159 * and the number of days before that month in the same year. These
160 * are correct for non-leap years only.
161 */
162static int _days_in_month[] = {
163 0, /* unused; this vector uses 1-based indexing */
164 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31
165};
166
167static int _days_before_month[] = {
168 0, /* unused; this vector uses 1-based indexing */
169 0, 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334
170};
171
172/* year -> 1 if leap year, else 0. */
173static int
174is_leap(int year)
175{
176 /* Cast year to unsigned. The result is the same either way, but
177 * C can generate faster code for unsigned mod than for signed
178 * mod (especially for % 4 -- a good compiler should just grab
179 * the last 2 bits when the LHS is unsigned).
180 */
181 const unsigned int ayear = (unsigned int)year;
182 return ayear % 4 == 0 && (ayear % 100 != 0 || ayear % 400 == 0);
183}
184
185/* year, month -> number of days in that month in that year */
186static int
187days_in_month(int year, int month)
188{
189 assert(month >= 1);
190 assert(month <= 12);
191 if (month == 2 && is_leap(year))
192 return 29;
193 else
194 return _days_in_month[month];
195}
196
197/* year, month -> number of days in year preceeding first day of month */
198static int
199days_before_month(int year, int month)
200{
201 int days;
202
203 assert(month >= 1);
204 assert(month <= 12);
205 days = _days_before_month[month];
206 if (month > 2 && is_leap(year))
207 ++days;
208 return days;
209}
210
211/* year -> number of days before January 1st of year. Remember that we
212 * start with year 1, so days_before_year(1) == 0.
213 */
214static int
215days_before_year(int year)
216{
217 int y = year - 1;
218 /* This is incorrect if year <= 0; we really want the floor
219 * here. But so long as MINYEAR is 1, the smallest year this
220 * can see is 0 (this can happen in some normalization endcases),
221 * so we'll just special-case that.
222 */
223 assert (year >= 0);
224 if (y >= 0)
225 return y*365 + y/4 - y/100 + y/400;
226 else {
227 assert(y == -1);
228 return -366;
229 }
230}
231
232/* Number of days in 4, 100, and 400 year cycles. That these have
233 * the correct values is asserted in the module init function.
234 */
235#define DI4Y 1461 /* days_before_year(5); days in 4 years */
236#define DI100Y 36524 /* days_before_year(101); days in 100 years */
237#define DI400Y 146097 /* days_before_year(401); days in 400 years */
238
239/* ordinal -> year, month, day, considering 01-Jan-0001 as day 1. */
240static void
241ord_to_ymd(int ordinal, int *year, int *month, int *day)
242{
243 int n, n1, n4, n100, n400, leapyear, preceding;
244
245 /* ordinal is a 1-based index, starting at 1-Jan-1. The pattern of
246 * leap years repeats exactly every 400 years. The basic strategy is
247 * to find the closest 400-year boundary at or before ordinal, then
248 * work with the offset from that boundary to ordinal. Life is much
249 * clearer if we subtract 1 from ordinal first -- then the values
250 * of ordinal at 400-year boundaries are exactly those divisible
251 * by DI400Y:
252 *
253 * D M Y n n-1
254 * -- --- ---- ---------- ----------------
255 * 31 Dec -400 -DI400Y -DI400Y -1
256 * 1 Jan -399 -DI400Y +1 -DI400Y 400-year boundary
257 * ...
258 * 30 Dec 000 -1 -2
259 * 31 Dec 000 0 -1
260 * 1 Jan 001 1 0 400-year boundary
261 * 2 Jan 001 2 1
262 * 3 Jan 001 3 2
263 * ...
264 * 31 Dec 400 DI400Y DI400Y -1
265 * 1 Jan 401 DI400Y +1 DI400Y 400-year boundary
266 */
267 assert(ordinal >= 1);
268 --ordinal;
269 n400 = ordinal / DI400Y;
270 n = ordinal % DI400Y;
271 *year = n400 * 400 + 1;
272
273 /* Now n is the (non-negative) offset, in days, from January 1 of
274 * year, to the desired date. Now compute how many 100-year cycles
275 * precede n.
276 * Note that it's possible for n100 to equal 4! In that case 4 full
277 * 100-year cycles precede the desired day, which implies the
278 * desired day is December 31 at the end of a 400-year cycle.
279 */
280 n100 = n / DI100Y;
281 n = n % DI100Y;
282
283 /* Now compute how many 4-year cycles precede it. */
284 n4 = n / DI4Y;
285 n = n % DI4Y;
286
287 /* And now how many single years. Again n1 can be 4, and again
288 * meaning that the desired day is December 31 at the end of the
289 * 4-year cycle.
290 */
291 n1 = n / 365;
292 n = n % 365;
293
294 *year += n100 * 100 + n4 * 4 + n1;
295 if (n1 == 4 || n100 == 4) {
296 assert(n == 0);
297 *year -= 1;
298 *month = 12;
299 *day = 31;
300 return;
301 }
302
303 /* Now the year is correct, and n is the offset from January 1. We
304 * find the month via an estimate that's either exact or one too
305 * large.
306 */
307 leapyear = n1 == 3 && (n4 != 24 || n100 == 3);
308 assert(leapyear == is_leap(*year));
309 *month = (n + 50) >> 5;
310 preceding = (_days_before_month[*month] + (*month > 2 && leapyear));
311 if (preceding > n) {
312 /* estimate is too large */
313 *month -= 1;
314 preceding -= days_in_month(*year, *month);
315 }
316 n -= preceding;
317 assert(0 <= n);
318 assert(n < days_in_month(*year, *month));
319
320 *day = n + 1;
321}
322
323/* year, month, day -> ordinal, considering 01-Jan-0001 as day 1. */
324static int
325ymd_to_ord(int year, int month, int day)
326{
327 return days_before_year(year) + days_before_month(year, month) + day;
328}
329
330/* Day of week, where Monday==0, ..., Sunday==6. 1/1/1 was a Monday. */
331static int
332weekday(int year, int month, int day)
333{
334 return (ymd_to_ord(year, month, day) + 6) % 7;
335}
336
337/* Ordinal of the Monday starting week 1 of the ISO year. Week 1 is the
338 * first calendar week containing a Thursday.
339 */
340static int
341iso_week1_monday(int year)
342{
343 int first_day = ymd_to_ord(year, 1, 1); /* ord of 1/1 */
344 /* 0 if 1/1 is a Monday, 1 if a Tue, etc. */
345 int first_weekday = (first_day + 6) % 7;
346 /* ordinal of closest Monday at or before 1/1 */
347 int week1_monday = first_day - first_weekday;
348
349 if (first_weekday > 3) /* if 1/1 was Fri, Sat, Sun */
350 week1_monday += 7;
351 return week1_monday;
352}
353
354/* ---------------------------------------------------------------------------
355 * Range checkers.
356 */
357
358/* Check that -MAX_DELTA_DAYS <= days <= MAX_DELTA_DAYS. If so, return 0.
359 * If not, raise OverflowError and return -1.
360 */
361static int
362check_delta_day_range(int days)
363{
364 if (-MAX_DELTA_DAYS <= days && days <= MAX_DELTA_DAYS)
365 return 0;
366 PyErr_Format(PyExc_OverflowError,
367 "days=%d; must have magnitude <= %d",
Guido van Rossumbd43e912002-12-16 20:34:55 +0000368 days, MAX_DELTA_DAYS);
Tim Peters2a799bf2002-12-16 20:18:38 +0000369 return -1;
370}
371
372/* Check that date arguments are in range. Return 0 if they are. If they
373 * aren't, raise ValueError and return -1.
374 */
375static int
376check_date_args(int year, int month, int day)
377{
378
379 if (year < MINYEAR || year > MAXYEAR) {
380 PyErr_SetString(PyExc_ValueError,
381 "year is out of range");
382 return -1;
383 }
384 if (month < 1 || month > 12) {
385 PyErr_SetString(PyExc_ValueError,
386 "month must be in 1..12");
387 return -1;
388 }
389 if (day < 1 || day > days_in_month(year, month)) {
390 PyErr_SetString(PyExc_ValueError,
391 "day is out of range for month");
392 return -1;
393 }
394 return 0;
395}
396
397/* Check that time arguments are in range. Return 0 if they are. If they
398 * aren't, raise ValueError and return -1.
399 */
400static int
401check_time_args(int h, int m, int s, int us)
402{
403 if (h < 0 || h > 23) {
404 PyErr_SetString(PyExc_ValueError,
405 "hour must be in 0..23");
406 return -1;
407 }
408 if (m < 0 || m > 59) {
409 PyErr_SetString(PyExc_ValueError,
410 "minute must be in 0..59");
411 return -1;
412 }
413 if (s < 0 || s > 59) {
414 PyErr_SetString(PyExc_ValueError,
415 "second must be in 0..59");
416 return -1;
417 }
418 if (us < 0 || us > 999999) {
419 PyErr_SetString(PyExc_ValueError,
420 "microsecond must be in 0..999999");
421 return -1;
422 }
423 return 0;
424}
425
426/* ---------------------------------------------------------------------------
427 * Normalization utilities.
428 */
429
430/* One step of a mixed-radix conversion. A "hi" unit is equivalent to
431 * factor "lo" units. factor must be > 0. If *lo is less than 0, or
432 * at least factor, enough of *lo is converted into "hi" units so that
433 * 0 <= *lo < factor. The input values must be such that int overflow
434 * is impossible.
435 */
436static void
437normalize_pair(int *hi, int *lo, int factor)
438{
439 assert(factor > 0);
440 assert(lo != hi);
441 if (*lo < 0 || *lo >= factor) {
442 const int num_hi = divmod(*lo, factor, lo);
443 const int new_hi = *hi + num_hi;
444 assert(! SIGNED_ADD_OVERFLOWED(new_hi, *hi, num_hi));
445 *hi = new_hi;
446 }
447 assert(0 <= *lo && *lo < factor);
448}
449
450/* Fiddle days (d), seconds (s), and microseconds (us) so that
451 * 0 <= *s < 24*3600
452 * 0 <= *us < 1000000
453 * The input values must be such that the internals don't overflow.
454 * The way this routine is used, we don't get close.
455 */
456static void
457normalize_d_s_us(int *d, int *s, int *us)
458{
459 if (*us < 0 || *us >= 1000000) {
460 normalize_pair(s, us, 1000000);
461 /* |s| can't be bigger than about
462 * |original s| + |original us|/1000000 now.
463 */
464
465 }
466 if (*s < 0 || *s >= 24*3600) {
467 normalize_pair(d, s, 24*3600);
468 /* |d| can't be bigger than about
469 * |original d| +
470 * (|original s| + |original us|/1000000) / (24*3600) now.
471 */
472 }
473 assert(0 <= *s && *s < 24*3600);
474 assert(0 <= *us && *us < 1000000);
475}
476
477/* Fiddle years (y), months (m), and days (d) so that
478 * 1 <= *m <= 12
479 * 1 <= *d <= days_in_month(*y, *m)
480 * The input values must be such that the internals don't overflow.
481 * The way this routine is used, we don't get close.
482 */
483static void
484normalize_y_m_d(int *y, int *m, int *d)
485{
486 int dim; /* # of days in month */
487
488 /* This gets muddy: the proper range for day can't be determined
489 * without knowing the correct month and year, but if day is, e.g.,
490 * plus or minus a million, the current month and year values make
491 * no sense (and may also be out of bounds themselves).
492 * Saying 12 months == 1 year should be non-controversial.
493 */
494 if (*m < 1 || *m > 12) {
495 --*m;
496 normalize_pair(y, m, 12);
497 ++*m;
498 /* |y| can't be bigger than about
499 * |original y| + |original m|/12 now.
500 */
501 }
502 assert(1 <= *m && *m <= 12);
503
504 /* Now only day can be out of bounds (year may also be out of bounds
505 * for a datetime object, but we don't care about that here).
506 * If day is out of bounds, what to do is arguable, but at least the
507 * method here is principled and explainable.
508 */
509 dim = days_in_month(*y, *m);
510 if (*d < 1 || *d > dim) {
511 /* Move day-1 days from the first of the month. First try to
512 * get off cheap if we're only one day out of range
513 * (adjustments for timezone alone can't be worse than that).
514 */
515 if (*d == 0) {
516 --*m;
517 if (*m > 0)
518 *d = days_in_month(*y, *m);
519 else {
520 --*y;
521 *m = 12;
522 *d = 31;
523 }
524 }
525 else if (*d == dim + 1) {
526 /* move forward a day */
527 ++*m;
528 *d = 1;
529 if (*m > 12) {
530 *m = 1;
531 ++*y;
532 }
533 }
534 else {
535 int ordinal = ymd_to_ord(*y, *m, 1) +
536 *d - 1;
537 ord_to_ymd(ordinal, y, m, d);
538 }
539 }
540 assert(*m > 0);
541 assert(*d > 0);
542}
543
544/* Fiddle out-of-bounds months and days so that the result makes some kind
545 * of sense. The parameters are both inputs and outputs. Returns < 0 on
546 * failure, where failure means the adjusted year is out of bounds.
547 */
548static int
549normalize_date(int *year, int *month, int *day)
550{
551 int result;
552
553 normalize_y_m_d(year, month, day);
554 if (MINYEAR <= *year && *year <= MAXYEAR)
555 result = 0;
556 else {
557 PyErr_SetString(PyExc_OverflowError,
558 "date value out of range");
559 result = -1;
560 }
561 return result;
562}
563
564/* Force all the datetime fields into range. The parameters are both
565 * inputs and outputs. Returns < 0 on error.
566 */
567static int
568normalize_datetime(int *year, int *month, int *day,
569 int *hour, int *minute, int *second,
570 int *microsecond)
571{
572 normalize_pair(second, microsecond, 1000000);
573 normalize_pair(minute, second, 60);
574 normalize_pair(hour, minute, 60);
575 normalize_pair(day, hour, 24);
576 return normalize_date(year, month, day);
577}
578
579/* ---------------------------------------------------------------------------
Tim Petersb0c854d2003-05-17 15:57:00 +0000580 * Basic object allocation: tp_alloc implementations. These allocate
581 * Python objects of the right size and type, and do the Python object-
582 * initialization bit. If there's not enough memory, they return NULL after
583 * setting MemoryError. All data members remain uninitialized trash.
584 *
585 * We abuse the tp_alloc "nitems" argument to communicate whether a tzinfo
Tim Peters03eaf8b2003-05-18 02:24:46 +0000586 * member is needed. This is ugly, imprecise, and possibly insecure.
587 * tp_basicsize for the time and datetime types is set to the size of the
588 * struct that has room for the tzinfo member, so subclasses in Python will
589 * allocate enough space for a tzinfo member whether or not one is actually
590 * needed. That's the "ugly and imprecise" parts. The "possibly insecure"
591 * part is that PyType_GenericAlloc() (which subclasses in Python end up
592 * using) just happens today to effectively ignore the nitems argument
593 * when tp_itemsize is 0, which it is for these type objects. If that
594 * changes, perhaps the callers of tp_alloc slots in this file should
595 * be changed to force a 0 nitems argument unless the type being allocated
596 * is a base type implemented in this file (so that tp_alloc is time_alloc
597 * or datetime_alloc below, which know about the nitems abuse).
Tim Petersb0c854d2003-05-17 15:57:00 +0000598 */
599
600static PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +0000601time_alloc(PyTypeObject *type, Py_ssize_t aware)
Tim Petersb0c854d2003-05-17 15:57:00 +0000602{
603 PyObject *self;
604
605 self = (PyObject *)
606 PyObject_MALLOC(aware ?
607 sizeof(PyDateTime_Time) :
608 sizeof(_PyDateTime_BaseTime));
609 if (self == NULL)
610 return (PyObject *)PyErr_NoMemory();
611 PyObject_INIT(self, type);
612 return self;
613}
614
615static PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +0000616datetime_alloc(PyTypeObject *type, Py_ssize_t aware)
Tim Petersb0c854d2003-05-17 15:57:00 +0000617{
618 PyObject *self;
619
620 self = (PyObject *)
621 PyObject_MALLOC(aware ?
622 sizeof(PyDateTime_DateTime) :
623 sizeof(_PyDateTime_BaseDateTime));
624 if (self == NULL)
625 return (PyObject *)PyErr_NoMemory();
626 PyObject_INIT(self, type);
627 return self;
628}
629
630/* ---------------------------------------------------------------------------
631 * Helpers for setting object fields. These work on pointers to the
632 * appropriate base class.
633 */
634
635/* For date and datetime. */
636static void
637set_date_fields(PyDateTime_Date *self, int y, int m, int d)
638{
639 self->hashcode = -1;
640 SET_YEAR(self, y);
641 SET_MONTH(self, m);
642 SET_DAY(self, d);
643}
644
645/* ---------------------------------------------------------------------------
646 * Create various objects, mostly without range checking.
647 */
648
649/* Create a date instance with no range checking. */
650static PyObject *
651new_date_ex(int year, int month, int day, PyTypeObject *type)
652{
653 PyDateTime_Date *self;
654
655 self = (PyDateTime_Date *) (type->tp_alloc(type, 0));
656 if (self != NULL)
657 set_date_fields(self, year, month, day);
658 return (PyObject *) self;
659}
660
661#define new_date(year, month, day) \
662 new_date_ex(year, month, day, &PyDateTime_DateType)
663
664/* Create a datetime instance with no range checking. */
665static PyObject *
666new_datetime_ex(int year, int month, int day, int hour, int minute,
667 int second, int usecond, PyObject *tzinfo, PyTypeObject *type)
668{
669 PyDateTime_DateTime *self;
670 char aware = tzinfo != Py_None;
671
672 self = (PyDateTime_DateTime *) (type->tp_alloc(type, aware));
673 if (self != NULL) {
674 self->hastzinfo = aware;
675 set_date_fields((PyDateTime_Date *)self, year, month, day);
676 DATE_SET_HOUR(self, hour);
677 DATE_SET_MINUTE(self, minute);
678 DATE_SET_SECOND(self, second);
679 DATE_SET_MICROSECOND(self, usecond);
680 if (aware) {
681 Py_INCREF(tzinfo);
682 self->tzinfo = tzinfo;
683 }
684 }
685 return (PyObject *)self;
686}
687
688#define new_datetime(y, m, d, hh, mm, ss, us, tzinfo) \
689 new_datetime_ex(y, m, d, hh, mm, ss, us, tzinfo, \
690 &PyDateTime_DateTimeType)
691
692/* Create a time instance with no range checking. */
693static PyObject *
694new_time_ex(int hour, int minute, int second, int usecond,
695 PyObject *tzinfo, PyTypeObject *type)
696{
697 PyDateTime_Time *self;
698 char aware = tzinfo != Py_None;
699
700 self = (PyDateTime_Time *) (type->tp_alloc(type, aware));
701 if (self != NULL) {
702 self->hastzinfo = aware;
703 self->hashcode = -1;
704 TIME_SET_HOUR(self, hour);
705 TIME_SET_MINUTE(self, minute);
706 TIME_SET_SECOND(self, second);
707 TIME_SET_MICROSECOND(self, usecond);
708 if (aware) {
709 Py_INCREF(tzinfo);
710 self->tzinfo = tzinfo;
711 }
712 }
713 return (PyObject *)self;
714}
715
716#define new_time(hh, mm, ss, us, tzinfo) \
717 new_time_ex(hh, mm, ss, us, tzinfo, &PyDateTime_TimeType)
718
719/* Create a timedelta instance. Normalize the members iff normalize is
720 * true. Passing false is a speed optimization, if you know for sure
721 * that seconds and microseconds are already in their proper ranges. In any
722 * case, raises OverflowError and returns NULL if the normalized days is out
723 * of range).
724 */
725static PyObject *
726new_delta_ex(int days, int seconds, int microseconds, int normalize,
727 PyTypeObject *type)
728{
729 PyDateTime_Delta *self;
730
731 if (normalize)
732 normalize_d_s_us(&days, &seconds, &microseconds);
733 assert(0 <= seconds && seconds < 24*3600);
734 assert(0 <= microseconds && microseconds < 1000000);
735
736 if (check_delta_day_range(days) < 0)
737 return NULL;
738
739 self = (PyDateTime_Delta *) (type->tp_alloc(type, 0));
740 if (self != NULL) {
741 self->hashcode = -1;
742 SET_TD_DAYS(self, days);
743 SET_TD_SECONDS(self, seconds);
744 SET_TD_MICROSECONDS(self, microseconds);
745 }
746 return (PyObject *) self;
747}
748
749#define new_delta(d, s, us, normalize) \
750 new_delta_ex(d, s, us, normalize, &PyDateTime_DeltaType)
751
752/* ---------------------------------------------------------------------------
Tim Peters2a799bf2002-12-16 20:18:38 +0000753 * tzinfo helpers.
754 */
755
Tim Peters855fe882002-12-22 03:43:39 +0000756/* Ensure that p is None or of a tzinfo subclass. Return 0 if OK; if not
757 * raise TypeError and return -1.
758 */
759static int
760check_tzinfo_subclass(PyObject *p)
761{
762 if (p == Py_None || PyTZInfo_Check(p))
763 return 0;
764 PyErr_Format(PyExc_TypeError,
765 "tzinfo argument must be None or of a tzinfo subclass, "
766 "not type '%s'",
Christian Heimes90aa7642007-12-19 02:45:37 +0000767 Py_TYPE(p)->tp_name);
Tim Peters855fe882002-12-22 03:43:39 +0000768 return -1;
769}
770
Tim Petersbad8ff02002-12-30 20:52:32 +0000771/* Return tzinfo.methname(tzinfoarg), without any checking of results.
Tim Peters855fe882002-12-22 03:43:39 +0000772 * If tzinfo is None, returns None.
773 */
774static PyObject *
Tim Petersbad8ff02002-12-30 20:52:32 +0000775call_tzinfo_method(PyObject *tzinfo, char *methname, PyObject *tzinfoarg)
Tim Peters855fe882002-12-22 03:43:39 +0000776{
777 PyObject *result;
778
Tim Petersbad8ff02002-12-30 20:52:32 +0000779 assert(tzinfo && methname && tzinfoarg);
Tim Peters855fe882002-12-22 03:43:39 +0000780 assert(check_tzinfo_subclass(tzinfo) >= 0);
781 if (tzinfo == Py_None) {
782 result = Py_None;
783 Py_INCREF(result);
784 }
785 else
Tim Petersbad8ff02002-12-30 20:52:32 +0000786 result = PyObject_CallMethod(tzinfo, methname, "O", tzinfoarg);
Tim Peters855fe882002-12-22 03:43:39 +0000787 return result;
788}
789
Tim Peters2a799bf2002-12-16 20:18:38 +0000790/* If self has a tzinfo member, return a BORROWED reference to it. Else
791 * return NULL, which is NOT AN ERROR. There are no error returns here,
792 * and the caller must not decref the result.
793 */
794static PyObject *
795get_tzinfo_member(PyObject *self)
796{
797 PyObject *tzinfo = NULL;
798
Tim Petersa9bc1682003-01-11 03:39:11 +0000799 if (PyDateTime_Check(self) && HASTZINFO(self))
800 tzinfo = ((PyDateTime_DateTime *)self)->tzinfo;
Tim Petersa032d2e2003-01-11 00:15:54 +0000801 else if (PyTime_Check(self) && HASTZINFO(self))
Tim Peters37f39822003-01-10 03:49:02 +0000802 tzinfo = ((PyDateTime_Time *)self)->tzinfo;
Tim Peters2a799bf2002-12-16 20:18:38 +0000803
804 return tzinfo;
805}
806
Tim Petersbad8ff02002-12-30 20:52:32 +0000807/* Call getattr(tzinfo, name)(tzinfoarg), and extract an int from the
Tim Peters2a799bf2002-12-16 20:18:38 +0000808 * result. tzinfo must be an instance of the tzinfo class. If the method
809 * returns None, this returns 0 and sets *none to 1. If the method doesn't
Tim Peters397301e2003-01-02 21:28:08 +0000810 * return None or timedelta, TypeError is raised and this returns -1. If it
811 * returnsa timedelta and the value is out of range or isn't a whole number
812 * of minutes, ValueError is raised and this returns -1.
Tim Peters2a799bf2002-12-16 20:18:38 +0000813 * Else *none is set to 0 and the integer method result is returned.
814 */
815static int
816call_utc_tzinfo_method(PyObject *tzinfo, char *name, PyObject *tzinfoarg,
817 int *none)
818{
819 PyObject *u;
Tim Peters397301e2003-01-02 21:28:08 +0000820 int result = -1;
Tim Peters2a799bf2002-12-16 20:18:38 +0000821
822 assert(tzinfo != NULL);
823 assert(PyTZInfo_Check(tzinfo));
824 assert(tzinfoarg != NULL);
825
826 *none = 0;
Tim Petersbad8ff02002-12-30 20:52:32 +0000827 u = call_tzinfo_method(tzinfo, name, tzinfoarg);
Tim Peters2a799bf2002-12-16 20:18:38 +0000828 if (u == NULL)
829 return -1;
830
Tim Peters27362852002-12-23 16:17:39 +0000831 else if (u == Py_None) {
Tim Peters2a799bf2002-12-16 20:18:38 +0000832 result = 0;
833 *none = 1;
Tim Peters2a799bf2002-12-16 20:18:38 +0000834 }
Tim Peters855fe882002-12-22 03:43:39 +0000835 else if (PyDelta_Check(u)) {
836 const int days = GET_TD_DAYS(u);
837 if (days < -1 || days > 0)
838 result = 24*60; /* trigger ValueError below */
839 else {
840 /* next line can't overflow because we know days
841 * is -1 or 0 now
842 */
843 int ss = days * 24 * 3600 + GET_TD_SECONDS(u);
844 result = divmod(ss, 60, &ss);
845 if (ss || GET_TD_MICROSECONDS(u)) {
846 PyErr_Format(PyExc_ValueError,
847 "tzinfo.%s() must return a "
848 "whole number of minutes",
849 name);
850 result = -1;
Tim Peters855fe882002-12-22 03:43:39 +0000851 }
852 }
853 }
Tim Peters2a799bf2002-12-16 20:18:38 +0000854 else {
855 PyErr_Format(PyExc_TypeError,
Tim Peters397301e2003-01-02 21:28:08 +0000856 "tzinfo.%s() must return None or "
Tim Peters855fe882002-12-22 03:43:39 +0000857 "timedelta, not '%s'",
Christian Heimes90aa7642007-12-19 02:45:37 +0000858 name, Py_TYPE(u)->tp_name);
Tim Peters2a799bf2002-12-16 20:18:38 +0000859 }
860
Tim Peters2a799bf2002-12-16 20:18:38 +0000861 Py_DECREF(u);
862 if (result < -1439 || result > 1439) {
863 PyErr_Format(PyExc_ValueError,
Neal Norwitz506a2242003-01-04 01:02:25 +0000864 "tzinfo.%s() returned %d; must be in "
Tim Peters2a799bf2002-12-16 20:18:38 +0000865 "-1439 .. 1439",
866 name, result);
867 result = -1;
868 }
Tim Peters397301e2003-01-02 21:28:08 +0000869 return result;
Tim Peters2a799bf2002-12-16 20:18:38 +0000870}
871
872/* Call tzinfo.utcoffset(tzinfoarg), and extract an integer from the
873 * result. tzinfo must be an instance of the tzinfo class. If utcoffset()
874 * returns None, call_utcoffset returns 0 and sets *none to 1. If uctoffset()
Tim Peters397301e2003-01-02 21:28:08 +0000875 * doesn't return None or timedelta, TypeError is raised and this returns -1.
876 * If utcoffset() returns an invalid timedelta (out of range, or not a whole
877 * # of minutes), ValueError is raised and this returns -1. Else *none is
878 * set to 0 and the offset is returned (as int # of minutes east of UTC).
Tim Peters2a799bf2002-12-16 20:18:38 +0000879 */
880static int
881call_utcoffset(PyObject *tzinfo, PyObject *tzinfoarg, int *none)
882{
883 return call_utc_tzinfo_method(tzinfo, "utcoffset", tzinfoarg, none);
884}
885
Tim Petersbad8ff02002-12-30 20:52:32 +0000886/* Call tzinfo.name(tzinfoarg), and return the offset as a timedelta or None.
887 */
Tim Peters855fe882002-12-22 03:43:39 +0000888static PyObject *
Tim Petersbad8ff02002-12-30 20:52:32 +0000889offset_as_timedelta(PyObject *tzinfo, char *name, PyObject *tzinfoarg) {
Tim Peters855fe882002-12-22 03:43:39 +0000890 PyObject *result;
891
Tim Petersbad8ff02002-12-30 20:52:32 +0000892 assert(tzinfo && name && tzinfoarg);
Tim Peters855fe882002-12-22 03:43:39 +0000893 if (tzinfo == Py_None) {
894 result = Py_None;
895 Py_INCREF(result);
896 }
897 else {
898 int none;
Tim Petersbad8ff02002-12-30 20:52:32 +0000899 int offset = call_utc_tzinfo_method(tzinfo, name, tzinfoarg,
900 &none);
Tim Peters855fe882002-12-22 03:43:39 +0000901 if (offset < 0 && PyErr_Occurred())
902 return NULL;
903 if (none) {
904 result = Py_None;
905 Py_INCREF(result);
906 }
907 else
908 result = new_delta(0, offset * 60, 0, 1);
909 }
910 return result;
911}
912
Tim Peters2a799bf2002-12-16 20:18:38 +0000913/* Call tzinfo.dst(tzinfoarg), and extract an integer from the
914 * result. tzinfo must be an instance of the tzinfo class. If dst()
915 * returns None, call_dst returns 0 and sets *none to 1. If dst()
Tim Peters397301e2003-01-02 21:28:08 +0000916 & doesn't return None or timedelta, TypeError is raised and this
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +0000917 * returns -1. If dst() returns an invalid timedelta for a UTC offset,
Tim Peters397301e2003-01-02 21:28:08 +0000918 * ValueError is raised and this returns -1. Else *none is set to 0 and
919 * the offset is returned (as an int # of minutes east of UTC).
Tim Peters2a799bf2002-12-16 20:18:38 +0000920 */
921static int
922call_dst(PyObject *tzinfo, PyObject *tzinfoarg, int *none)
923{
924 return call_utc_tzinfo_method(tzinfo, "dst", tzinfoarg, none);
925}
926
Tim Petersbad8ff02002-12-30 20:52:32 +0000927/* Call tzinfo.tzname(tzinfoarg), and return the result. tzinfo must be
Tim Peters855fe882002-12-22 03:43:39 +0000928 * an instance of the tzinfo class or None. If tzinfo isn't None, and
Tim Petersbad8ff02002-12-30 20:52:32 +0000929 * tzname() doesn't return None or a string, TypeError is raised and this
Guido van Rossume3d1d412007-05-23 21:24:35 +0000930 * returns NULL. If the result is a string, we ensure it is a Unicode
931 * string.
Tim Peters2a799bf2002-12-16 20:18:38 +0000932 */
933static PyObject *
Tim Petersbad8ff02002-12-30 20:52:32 +0000934call_tzname(PyObject *tzinfo, PyObject *tzinfoarg)
Tim Peters2a799bf2002-12-16 20:18:38 +0000935{
936 PyObject *result;
937
938 assert(tzinfo != NULL);
Tim Peters855fe882002-12-22 03:43:39 +0000939 assert(check_tzinfo_subclass(tzinfo) >= 0);
Tim Petersbad8ff02002-12-30 20:52:32 +0000940 assert(tzinfoarg != NULL);
Tim Peters2a799bf2002-12-16 20:18:38 +0000941
Tim Peters855fe882002-12-22 03:43:39 +0000942 if (tzinfo == Py_None) {
943 result = Py_None;
944 Py_INCREF(result);
945 }
946 else
Tim Petersbad8ff02002-12-30 20:52:32 +0000947 result = PyObject_CallMethod(tzinfo, "tzname", "O", tzinfoarg);
Tim Peters855fe882002-12-22 03:43:39 +0000948
Guido van Rossume3d1d412007-05-23 21:24:35 +0000949 if (result != NULL && result != Py_None) {
Guido van Rossumfd53fd62007-08-24 04:05:13 +0000950 if (!PyUnicode_Check(result)) {
Guido van Rossume3d1d412007-05-23 21:24:35 +0000951 PyErr_Format(PyExc_TypeError, "tzinfo.tzname() must "
952 "return None or a string, not '%s'",
Christian Heimes90aa7642007-12-19 02:45:37 +0000953 Py_TYPE(result)->tp_name);
Guido van Rossume3d1d412007-05-23 21:24:35 +0000954 Py_DECREF(result);
955 result = NULL;
956 }
957 else if (!PyUnicode_Check(result)) {
958 PyObject *temp = PyUnicode_FromObject(result);
959 Py_DECREF(result);
960 result = temp;
961 }
Tim Peters2a799bf2002-12-16 20:18:38 +0000962 }
963 return result;
964}
965
966typedef enum {
967 /* an exception has been set; the caller should pass it on */
968 OFFSET_ERROR,
969
Tim Petersa9bc1682003-01-11 03:39:11 +0000970 /* type isn't date, datetime, or time subclass */
Tim Peters2a799bf2002-12-16 20:18:38 +0000971 OFFSET_UNKNOWN,
972
973 /* date,
Tim Petersa9bc1682003-01-11 03:39:11 +0000974 * datetime with !hastzinfo
975 * datetime with None tzinfo,
976 * datetime where utcoffset() returns None
Tim Peters37f39822003-01-10 03:49:02 +0000977 * time with !hastzinfo
978 * time with None tzinfo,
979 * time where utcoffset() returns None
Tim Peters2a799bf2002-12-16 20:18:38 +0000980 */
981 OFFSET_NAIVE,
982
Tim Petersa9bc1682003-01-11 03:39:11 +0000983 /* time or datetime where utcoffset() doesn't return None */
Georg Brandle810fe22006-02-19 15:28:47 +0000984 OFFSET_AWARE
Tim Peters2a799bf2002-12-16 20:18:38 +0000985} naivety;
986
Tim Peters14b69412002-12-22 18:10:22 +0000987/* Classify an object as to whether it's naive or offset-aware. See
Tim Peters2a799bf2002-12-16 20:18:38 +0000988 * the "naivety" typedef for details. If the type is aware, *offset is set
989 * to minutes east of UTC (as returned by the tzinfo.utcoffset() method).
Tim Peters14b69412002-12-22 18:10:22 +0000990 * If the type is offset-naive (or unknown, or error), *offset is set to 0.
Tim Peterse39a80c2002-12-30 21:28:52 +0000991 * tzinfoarg is the argument to pass to the tzinfo.utcoffset() method.
Tim Peters2a799bf2002-12-16 20:18:38 +0000992 */
993static naivety
Tim Peterse39a80c2002-12-30 21:28:52 +0000994classify_utcoffset(PyObject *op, PyObject *tzinfoarg, int *offset)
Tim Peters2a799bf2002-12-16 20:18:38 +0000995{
996 int none;
997 PyObject *tzinfo;
998
Tim Peterse39a80c2002-12-30 21:28:52 +0000999 assert(tzinfoarg != NULL);
Tim Peters2a799bf2002-12-16 20:18:38 +00001000 *offset = 0;
Tim Peters14b69412002-12-22 18:10:22 +00001001 tzinfo = get_tzinfo_member(op); /* NULL means no tzinfo, not error */
Tim Peters2a799bf2002-12-16 20:18:38 +00001002 if (tzinfo == Py_None)
1003 return OFFSET_NAIVE;
Tim Peters14b69412002-12-22 18:10:22 +00001004 if (tzinfo == NULL) {
1005 /* note that a datetime passes the PyDate_Check test */
1006 return (PyTime_Check(op) || PyDate_Check(op)) ?
1007 OFFSET_NAIVE : OFFSET_UNKNOWN;
1008 }
Tim Peterse39a80c2002-12-30 21:28:52 +00001009 *offset = call_utcoffset(tzinfo, tzinfoarg, &none);
Tim Peters2a799bf2002-12-16 20:18:38 +00001010 if (*offset == -1 && PyErr_Occurred())
1011 return OFFSET_ERROR;
1012 return none ? OFFSET_NAIVE : OFFSET_AWARE;
1013}
1014
Tim Peters00237032002-12-27 02:21:51 +00001015/* Classify two objects as to whether they're naive or offset-aware.
1016 * This isn't quite the same as calling classify_utcoffset() twice: for
1017 * binary operations (comparison and subtraction), we generally want to
1018 * ignore the tzinfo members if they're identical. This is by design,
1019 * so that results match "naive" expectations when mixing objects from a
1020 * single timezone. So in that case, this sets both offsets to 0 and
1021 * both naiveties to OFFSET_NAIVE.
1022 * The function returns 0 if everything's OK, and -1 on error.
1023 */
1024static int
1025classify_two_utcoffsets(PyObject *o1, int *offset1, naivety *n1,
Tim Peterse39a80c2002-12-30 21:28:52 +00001026 PyObject *tzinfoarg1,
1027 PyObject *o2, int *offset2, naivety *n2,
1028 PyObject *tzinfoarg2)
Tim Peters00237032002-12-27 02:21:51 +00001029{
1030 if (get_tzinfo_member(o1) == get_tzinfo_member(o2)) {
1031 *offset1 = *offset2 = 0;
1032 *n1 = *n2 = OFFSET_NAIVE;
1033 }
1034 else {
Tim Peterse39a80c2002-12-30 21:28:52 +00001035 *n1 = classify_utcoffset(o1, tzinfoarg1, offset1);
Tim Peters00237032002-12-27 02:21:51 +00001036 if (*n1 == OFFSET_ERROR)
1037 return -1;
Tim Peterse39a80c2002-12-30 21:28:52 +00001038 *n2 = classify_utcoffset(o2, tzinfoarg2, offset2);
Tim Peters00237032002-12-27 02:21:51 +00001039 if (*n2 == OFFSET_ERROR)
1040 return -1;
1041 }
1042 return 0;
1043}
1044
Tim Peters2a799bf2002-12-16 20:18:38 +00001045/* repr is like "someclass(arg1, arg2)". If tzinfo isn't None,
1046 * stuff
1047 * ", tzinfo=" + repr(tzinfo)
1048 * before the closing ")".
1049 */
1050static PyObject *
1051append_keyword_tzinfo(PyObject *repr, PyObject *tzinfo)
1052{
1053 PyObject *temp;
1054
Walter Dörwald1ab83302007-05-18 17:15:44 +00001055 assert(PyUnicode_Check(repr));
Tim Peters2a799bf2002-12-16 20:18:38 +00001056 assert(tzinfo);
1057 if (tzinfo == Py_None)
1058 return repr;
1059 /* Get rid of the trailing ')'. */
Walter Dörwald1ab83302007-05-18 17:15:44 +00001060 assert(PyUnicode_AS_UNICODE(repr)[PyUnicode_GET_SIZE(repr)-1] == ')');
1061 temp = PyUnicode_FromUnicode(PyUnicode_AS_UNICODE(repr),
1062 PyUnicode_GET_SIZE(repr) - 1);
Tim Peters2a799bf2002-12-16 20:18:38 +00001063 Py_DECREF(repr);
1064 if (temp == NULL)
1065 return NULL;
Walter Dörwald517bcfe2007-05-23 20:45:05 +00001066 repr = PyUnicode_FromFormat("%U, tzinfo=%R)", temp, tzinfo);
1067 Py_DECREF(temp);
Tim Peters2a799bf2002-12-16 20:18:38 +00001068 return repr;
1069}
1070
1071/* ---------------------------------------------------------------------------
1072 * String format helpers.
1073 */
1074
1075static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00001076format_ctime(PyDateTime_Date *date, int hours, int minutes, int seconds)
Tim Peters2a799bf2002-12-16 20:18:38 +00001077{
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00001078 static const char *DayNames[] = {
Tim Peters2a799bf2002-12-16 20:18:38 +00001079 "Mon", "Tue", "Wed", "Thu", "Fri", "Sat", "Sun"
1080 };
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00001081 static const char *MonthNames[] = {
Tim Peters2a799bf2002-12-16 20:18:38 +00001082 "Jan", "Feb", "Mar", "Apr", "May", "Jun",
1083 "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
1084 };
1085
Tim Peters2a799bf2002-12-16 20:18:38 +00001086 int wday = weekday(GET_YEAR(date), GET_MONTH(date), GET_DAY(date));
1087
Walter Dörwald4af32b32007-05-31 16:19:50 +00001088 return PyUnicode_FromFormat("%s %s %2d %02d:%02d:%02d %04d",
1089 DayNames[wday], MonthNames[GET_MONTH(date)-1],
1090 GET_DAY(date), hours, minutes, seconds,
1091 GET_YEAR(date));
Tim Peters2a799bf2002-12-16 20:18:38 +00001092}
1093
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.
1103 */
1104static int
Tim Peters328fff72002-12-20 01:31:27 +00001105format_utcoffset(char *buf, size_t buflen, const char *sep,
Tim Peters2a799bf2002-12-16 20:18:38 +00001106 PyObject *tzinfo, PyObject *tzinfoarg)
1107{
1108 int offset;
1109 int hours;
1110 int minutes;
1111 char sign;
1112 int none;
1113
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;
1120 }
1121 sign = '+';
1122 if (offset < 0) {
1123 sign = '-';
1124 offset = - offset;
1125 }
1126 hours = divmod(offset, 60, &minutes);
1127 PyOS_snprintf(buf, buflen, "%c%02d%s%02d", sign, hours, sep, minutes);
1128 return 0;
1129}
1130
Guido van Rossumd8595fe2007-05-23 21:36:49 +00001131static PyObject *
1132make_Zreplacement(PyObject *object, PyObject *tzinfoarg)
1133{
Neal Norwitzaea70e02007-08-12 04:32:26 +00001134 PyObject *temp;
Guido van Rossumd8595fe2007-05-23 21:36:49 +00001135 PyObject *tzinfo = get_tzinfo_member(object);
Guido van Rossum98297ee2007-11-06 21:34:58 +00001136 PyObject *Zreplacement = PyUnicode_FromStringAndSize(NULL, 0);
Guido van Rossumd8595fe2007-05-23 21:36:49 +00001137 if (Zreplacement == NULL)
1138 return NULL;
Neal Norwitzaea70e02007-08-12 04:32:26 +00001139 if (tzinfo == Py_None || tzinfo == NULL)
1140 return Zreplacement;
1141
1142 assert(tzinfoarg != NULL);
1143 temp = call_tzname(tzinfo, tzinfoarg);
1144 if (temp == NULL)
1145 goto Error;
1146 if (temp == Py_None) {
1147 Py_DECREF(temp);
1148 return Zreplacement;
1149 }
1150
1151 assert(PyUnicode_Check(temp));
1152 /* Since the tzname is getting stuffed into the
1153 * format, we have to double any % signs so that
1154 * strftime doesn't treat them as format codes.
1155 */
1156 Py_DECREF(Zreplacement);
1157 Zreplacement = PyObject_CallMethod(temp, "replace", "ss", "%", "%%");
1158 Py_DECREF(temp);
1159 if (Zreplacement == NULL)
1160 return NULL;
Guido van Rossum98297ee2007-11-06 21:34:58 +00001161 if (!PyUnicode_Check(Zreplacement)) {
Neal Norwitzaea70e02007-08-12 04:32:26 +00001162 PyErr_SetString(PyExc_TypeError,
1163 "tzname.replace() did not return a string");
1164 goto Error;
Guido van Rossumd8595fe2007-05-23 21:36:49 +00001165 }
1166 return Zreplacement;
1167
1168 Error:
1169 Py_DECREF(Zreplacement);
1170 return NULL;
1171}
1172
Christian Heimesdd15f6c2008-03-16 00:07:10 +00001173static PyObject *
1174make_freplacement(PyObject *object)
1175{
Christian Heimesb186d002008-03-18 15:15:01 +00001176 char freplacement[64];
Christian Heimesdd15f6c2008-03-16 00:07:10 +00001177 if (PyTime_Check(object))
1178 sprintf(freplacement, "%06d", TIME_GET_MICROSECOND(object));
1179 else if (PyDateTime_Check(object))
1180 sprintf(freplacement, "%06d", DATE_GET_MICROSECOND(object));
1181 else
1182 sprintf(freplacement, "%06d", 0);
1183
1184 return PyString_FromStringAndSize(freplacement, strlen(freplacement));
1185}
1186
Tim Peters2a799bf2002-12-16 20:18:38 +00001187/* I sure don't want to reproduce the strftime code from the time module,
1188 * so this imports the module and calls it. All the hair is due to
Christian Heimesdd15f6c2008-03-16 00:07:10 +00001189 * giving special meanings to the %z, %Z and %f format codes via a
1190 * preprocessing step on the format string.
Tim Petersbad8ff02002-12-30 20:52:32 +00001191 * tzinfoarg is the argument to pass to the object's tzinfo method, if
1192 * needed.
Tim Peters2a799bf2002-12-16 20:18:38 +00001193 */
1194static PyObject *
Tim Petersbad8ff02002-12-30 20:52:32 +00001195wrap_strftime(PyObject *object, PyObject *format, PyObject *timetuple,
1196 PyObject *tzinfoarg)
Tim Peters2a799bf2002-12-16 20:18:38 +00001197{
1198 PyObject *result = NULL; /* guilty until proved innocent */
1199
1200 PyObject *zreplacement = NULL; /* py string, replacement for %z */
1201 PyObject *Zreplacement = NULL; /* py string, replacement for %Z */
Christian Heimesdd15f6c2008-03-16 00:07:10 +00001202 PyObject *freplacement = NULL; /* py string, replacement for %f */
Tim Peters2a799bf2002-12-16 20:18:38 +00001203
Guido van Rossumbce56a62007-05-10 18:04:33 +00001204 const char *pin;/* pointer to next char in input format */
1205 Py_ssize_t flen;/* length of input format */
Tim Peters2a799bf2002-12-16 20:18:38 +00001206 char ch; /* next char in input format */
1207
1208 PyObject *newfmt = NULL; /* py string, the output format */
1209 char *pnew; /* pointer to available byte in output format */
Thomas Wouters89f507f2006-12-13 04:49:30 +00001210 int totalnew; /* number bytes total in output format buffer,
Tim Peters2a799bf2002-12-16 20:18:38 +00001211 exclusive of trailing \0 */
Thomas Wouters89f507f2006-12-13 04:49:30 +00001212 int usednew; /* number bytes used so far in output format buffer */
Tim Peters2a799bf2002-12-16 20:18:38 +00001213
Guido van Rossumbce56a62007-05-10 18:04:33 +00001214 const char *ptoappend;/* pointer to string to append to output buffer */
Brett Cannon27da8122007-11-06 23:15:11 +00001215 Py_ssize_t ntoappend; /* # of bytes to append to output buffer */
Tim Peters2a799bf2002-12-16 20:18:38 +00001216
Tim Peters2a799bf2002-12-16 20:18:38 +00001217 assert(object && format && timetuple);
Guido van Rossumfd53fd62007-08-24 04:05:13 +00001218 assert(PyUnicode_Check(format));
Neal Norwitz908c8712007-08-27 04:58:38 +00001219 /* Convert the input format to a C string and size */
Alexandre Vassalottia85998a2008-05-03 18:24:43 +00001220 pin = PyUnicode_AsStringAndSize(format, &flen);
Neal Norwitz908c8712007-08-27 04:58:38 +00001221 if (!pin)
1222 return NULL;
Tim Peters2a799bf2002-12-16 20:18:38 +00001223
Tim Petersd6844152002-12-22 20:58:42 +00001224 /* Give up if the year is before 1900.
1225 * Python strftime() plays games with the year, and different
1226 * games depending on whether envar PYTHON2K is set. This makes
1227 * years before 1900 a nightmare, even if the platform strftime
1228 * supports them (and not all do).
1229 * We could get a lot farther here by avoiding Python's strftime
1230 * wrapper and calling the C strftime() directly, but that isn't
1231 * an option in the Python implementation of this module.
1232 */
1233 {
1234 long year;
1235 PyObject *pyyear = PySequence_GetItem(timetuple, 0);
1236 if (pyyear == NULL) return NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00001237 assert(PyLong_Check(pyyear));
1238 year = PyLong_AsLong(pyyear);
Tim Petersd6844152002-12-22 20:58:42 +00001239 Py_DECREF(pyyear);
1240 if (year < 1900) {
1241 PyErr_Format(PyExc_ValueError, "year=%ld is before "
1242 "1900; the datetime strftime() "
1243 "methods require year >= 1900",
1244 year);
1245 return NULL;
1246 }
1247 }
1248
Christian Heimesdd15f6c2008-03-16 00:07:10 +00001249 /* Scan the input format, looking for %z/%Z/%f escapes, building
Tim Peters328fff72002-12-20 01:31:27 +00001250 * a new format. Since computing the replacements for those codes
1251 * is expensive, don't unless they're actually used.
Tim Peters2a799bf2002-12-16 20:18:38 +00001252 */
Guido van Rossumbce56a62007-05-10 18:04:33 +00001253 totalnew = flen + 1; /* realistic if no %z/%Z */
Walter Dörwald6bd238c2007-11-22 09:38:52 +00001254 newfmt = PyString_FromStringAndSize(NULL, totalnew);
Tim Peters2a799bf2002-12-16 20:18:38 +00001255 if (newfmt == NULL) goto Done;
Walter Dörwald6bd238c2007-11-22 09:38:52 +00001256 pnew = PyString_AsString(newfmt);
Tim Peters2a799bf2002-12-16 20:18:38 +00001257 usednew = 0;
1258
Tim Peters2a799bf2002-12-16 20:18:38 +00001259 while ((ch = *pin++) != '\0') {
1260 if (ch != '%') {
Tim Peters328fff72002-12-20 01:31:27 +00001261 ptoappend = pin - 1;
Tim Peters2a799bf2002-12-16 20:18:38 +00001262 ntoappend = 1;
1263 }
1264 else if ((ch = *pin++) == '\0') {
1265 /* There's a lone trailing %; doesn't make sense. */
1266 PyErr_SetString(PyExc_ValueError, "strftime format "
1267 "ends with raw %");
1268 goto Done;
1269 }
1270 /* A % has been seen and ch is the character after it. */
1271 else if (ch == 'z') {
1272 if (zreplacement == NULL) {
1273 /* format utcoffset */
Tim Peters328fff72002-12-20 01:31:27 +00001274 char buf[100];
Tim Peters2a799bf2002-12-16 20:18:38 +00001275 PyObject *tzinfo = get_tzinfo_member(object);
Walter Dörwald6bd238c2007-11-22 09:38:52 +00001276 zreplacement = PyString_FromStringAndSize("", 0);
Tim Peters2a799bf2002-12-16 20:18:38 +00001277 if (zreplacement == NULL) goto Done;
1278 if (tzinfo != Py_None && tzinfo != NULL) {
Tim Petersbad8ff02002-12-30 20:52:32 +00001279 assert(tzinfoarg != NULL);
Tim Peters2a799bf2002-12-16 20:18:38 +00001280 if (format_utcoffset(buf,
Tim Peters328fff72002-12-20 01:31:27 +00001281 sizeof(buf),
Tim Peters2a799bf2002-12-16 20:18:38 +00001282 "",
1283 tzinfo,
Tim Petersbad8ff02002-12-30 20:52:32 +00001284 tzinfoarg) < 0)
Tim Peters2a799bf2002-12-16 20:18:38 +00001285 goto Done;
1286 Py_DECREF(zreplacement);
Guido van Rossumfd53fd62007-08-24 04:05:13 +00001287 zreplacement =
Walter Dörwald6bd238c2007-11-22 09:38:52 +00001288 PyString_FromStringAndSize(buf,
Guido van Rossumfd53fd62007-08-24 04:05:13 +00001289 strlen(buf));
1290 if (zreplacement == NULL)
1291 goto Done;
Tim Peters2a799bf2002-12-16 20:18:38 +00001292 }
1293 }
1294 assert(zreplacement != NULL);
Walter Dörwald6bd238c2007-11-22 09:38:52 +00001295 ptoappend = PyString_AS_STRING(zreplacement);
1296 ntoappend = PyString_GET_SIZE(zreplacement);
Tim Peters2a799bf2002-12-16 20:18:38 +00001297 }
1298 else if (ch == 'Z') {
1299 /* format tzname */
1300 if (Zreplacement == NULL) {
Guido van Rossumd8595fe2007-05-23 21:36:49 +00001301 Zreplacement = make_Zreplacement(object,
1302 tzinfoarg);
1303 if (Zreplacement == NULL)
1304 goto Done;
Tim Peters2a799bf2002-12-16 20:18:38 +00001305 }
1306 assert(Zreplacement != NULL);
Guido van Rossum98297ee2007-11-06 21:34:58 +00001307 assert(PyUnicode_Check(Zreplacement));
1308 ptoappend = PyUnicode_AsStringAndSize(Zreplacement,
1309 &ntoappend);
Christian Heimes90aa7642007-12-19 02:45:37 +00001310 ntoappend = Py_SIZE(Zreplacement);
Tim Peters2a799bf2002-12-16 20:18:38 +00001311 }
Christian Heimesdd15f6c2008-03-16 00:07:10 +00001312 else if (ch == 'f') {
1313 /* format microseconds */
1314 if (freplacement == NULL) {
1315 freplacement = make_freplacement(object);
1316 if (freplacement == NULL)
1317 goto Done;
1318 }
1319 assert(freplacement != NULL);
1320 assert(PyString_Check(freplacement));
1321 ptoappend = PyString_AS_STRING(freplacement);
1322 ntoappend = PyString_GET_SIZE(freplacement);
1323 }
Tim Peters2a799bf2002-12-16 20:18:38 +00001324 else {
Tim Peters328fff72002-12-20 01:31:27 +00001325 /* percent followed by neither z nor Z */
1326 ptoappend = pin - 2;
Tim Peters2a799bf2002-12-16 20:18:38 +00001327 ntoappend = 2;
1328 }
1329
1330 /* Append the ntoappend chars starting at ptoappend to
1331 * the new format.
1332 */
Tim Peters2a799bf2002-12-16 20:18:38 +00001333 if (ntoappend == 0)
1334 continue;
Guido van Rossumfd53fd62007-08-24 04:05:13 +00001335 assert(ptoappend != NULL);
1336 assert(ntoappend > 0);
Tim Peters2a799bf2002-12-16 20:18:38 +00001337 while (usednew + ntoappend > totalnew) {
1338 int bigger = totalnew << 1;
1339 if ((bigger >> 1) != totalnew) { /* overflow */
1340 PyErr_NoMemory();
1341 goto Done;
1342 }
Walter Dörwald6bd238c2007-11-22 09:38:52 +00001343 if (_PyString_Resize(&newfmt, bigger) < 0)
Tim Peters2a799bf2002-12-16 20:18:38 +00001344 goto Done;
1345 totalnew = bigger;
Walter Dörwald6bd238c2007-11-22 09:38:52 +00001346 pnew = PyString_AsString(newfmt) + usednew;
Tim Peters2a799bf2002-12-16 20:18:38 +00001347 }
1348 memcpy(pnew, ptoappend, ntoappend);
1349 pnew += ntoappend;
1350 usednew += ntoappend;
1351 assert(usednew <= totalnew);
1352 } /* end while() */
1353
Walter Dörwald6bd238c2007-11-22 09:38:52 +00001354 if (_PyString_Resize(&newfmt, usednew) < 0)
Tim Peters2a799bf2002-12-16 20:18:38 +00001355 goto Done;
1356 {
Neal Norwitz908c8712007-08-27 04:58:38 +00001357 PyObject *format;
Christian Heimes072c0f12008-01-03 23:01:04 +00001358 PyObject *time = PyImport_ImportModuleNoBlock("time");
Tim Peters2a799bf2002-12-16 20:18:38 +00001359 if (time == NULL)
1360 goto Done;
Walter Dörwald6bd238c2007-11-22 09:38:52 +00001361 format = PyUnicode_FromString(PyString_AS_STRING(newfmt));
Neal Norwitz908c8712007-08-27 04:58:38 +00001362 if (format != NULL) {
1363 result = PyObject_CallMethod(time, "strftime", "OO",
1364 format, timetuple);
1365 Py_DECREF(format);
1366 }
Tim Peters2a799bf2002-12-16 20:18:38 +00001367 Py_DECREF(time);
1368 }
1369 Done:
Christian Heimesdd15f6c2008-03-16 00:07:10 +00001370 Py_XDECREF(freplacement);
Tim Peters2a799bf2002-12-16 20:18:38 +00001371 Py_XDECREF(zreplacement);
1372 Py_XDECREF(Zreplacement);
1373 Py_XDECREF(newfmt);
1374 return result;
1375}
1376
Tim Peters2a799bf2002-12-16 20:18:38 +00001377/* ---------------------------------------------------------------------------
1378 * Wrap functions from the time module. These aren't directly available
1379 * from C. Perhaps they should be.
1380 */
1381
1382/* Call time.time() and return its result (a Python float). */
1383static PyObject *
Guido van Rossumbd43e912002-12-16 20:34:55 +00001384time_time(void)
Tim Peters2a799bf2002-12-16 20:18:38 +00001385{
1386 PyObject *result = NULL;
Christian Heimes072c0f12008-01-03 23:01:04 +00001387 PyObject *time = PyImport_ImportModuleNoBlock("time");
Tim Peters2a799bf2002-12-16 20:18:38 +00001388
1389 if (time != NULL) {
1390 result = PyObject_CallMethod(time, "time", "()");
1391 Py_DECREF(time);
1392 }
1393 return result;
1394}
1395
1396/* Build a time.struct_time. The weekday and day number are automatically
1397 * computed from the y,m,d args.
1398 */
1399static PyObject *
1400build_struct_time(int y, int m, int d, int hh, int mm, int ss, int dstflag)
1401{
1402 PyObject *time;
1403 PyObject *result = NULL;
1404
Christian Heimes072c0f12008-01-03 23:01:04 +00001405 time = PyImport_ImportModuleNoBlock("time");
Tim Peters2a799bf2002-12-16 20:18:38 +00001406 if (time != NULL) {
1407 result = PyObject_CallMethod(time, "struct_time",
1408 "((iiiiiiiii))",
1409 y, m, d,
1410 hh, mm, ss,
1411 weekday(y, m, d),
1412 days_before_month(y, m) + d,
1413 dstflag);
1414 Py_DECREF(time);
1415 }
1416 return result;
1417}
1418
1419/* ---------------------------------------------------------------------------
1420 * Miscellaneous helpers.
1421 */
1422
Guido van Rossum19960592006-08-24 17:29:38 +00001423/* For various reasons, we need to use tp_richcompare instead of tp_compare.
Tim Peters2a799bf2002-12-16 20:18:38 +00001424 * The comparisons here all most naturally compute a cmp()-like result.
1425 * This little helper turns that into a bool result for rich comparisons.
1426 */
1427static PyObject *
1428diff_to_bool(int diff, int op)
1429{
1430 PyObject *result;
1431 int istrue;
1432
1433 switch (op) {
1434 case Py_EQ: istrue = diff == 0; break;
1435 case Py_NE: istrue = diff != 0; break;
1436 case Py_LE: istrue = diff <= 0; break;
1437 case Py_GE: istrue = diff >= 0; break;
1438 case Py_LT: istrue = diff < 0; break;
1439 case Py_GT: istrue = diff > 0; break;
1440 default:
1441 assert(! "op unknown");
1442 istrue = 0; /* To shut up compiler */
1443 }
1444 result = istrue ? Py_True : Py_False;
1445 Py_INCREF(result);
1446 return result;
1447}
1448
Tim Peters07534a62003-02-07 22:50:28 +00001449/* Raises a "can't compare" TypeError and returns NULL. */
1450static PyObject *
1451cmperror(PyObject *a, PyObject *b)
1452{
1453 PyErr_Format(PyExc_TypeError,
1454 "can't compare %s to %s",
Christian Heimes90aa7642007-12-19 02:45:37 +00001455 Py_TYPE(a)->tp_name, Py_TYPE(b)->tp_name);
Tim Peters07534a62003-02-07 22:50:28 +00001456 return NULL;
1457}
1458
Tim Peters2a799bf2002-12-16 20:18:38 +00001459/* ---------------------------------------------------------------------------
Tim Peters2a799bf2002-12-16 20:18:38 +00001460 * Cached Python objects; these are set by the module init function.
1461 */
1462
1463/* Conversion factors. */
1464static PyObject *us_per_us = NULL; /* 1 */
1465static PyObject *us_per_ms = NULL; /* 1000 */
1466static PyObject *us_per_second = NULL; /* 1000000 */
1467static PyObject *us_per_minute = NULL; /* 1e6 * 60 as Python int */
1468static PyObject *us_per_hour = NULL; /* 1e6 * 3600 as Python long */
1469static PyObject *us_per_day = NULL; /* 1e6 * 3600 * 24 as Python long */
1470static PyObject *us_per_week = NULL; /* 1e6*3600*24*7 as Python long */
1471static PyObject *seconds_per_day = NULL; /* 3600*24 as Python int */
1472
Tim Peters2a799bf2002-12-16 20:18:38 +00001473/* ---------------------------------------------------------------------------
1474 * Class implementations.
1475 */
1476
1477/*
1478 * PyDateTime_Delta implementation.
1479 */
1480
1481/* Convert a timedelta to a number of us,
1482 * (24*3600*self.days + self.seconds)*1000000 + self.microseconds
1483 * as a Python int or long.
1484 * Doing mixed-radix arithmetic by hand instead is excruciating in C,
1485 * due to ubiquitous overflow possibilities.
1486 */
1487static PyObject *
1488delta_to_microseconds(PyDateTime_Delta *self)
1489{
1490 PyObject *x1 = NULL;
1491 PyObject *x2 = NULL;
1492 PyObject *x3 = NULL;
1493 PyObject *result = NULL;
1494
Christian Heimes217cfd12007-12-02 14:31:20 +00001495 x1 = PyLong_FromLong(GET_TD_DAYS(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00001496 if (x1 == NULL)
1497 goto Done;
1498 x2 = PyNumber_Multiply(x1, seconds_per_day); /* days in seconds */
1499 if (x2 == NULL)
1500 goto Done;
1501 Py_DECREF(x1);
1502 x1 = NULL;
1503
1504 /* x2 has days in seconds */
Christian Heimes217cfd12007-12-02 14:31:20 +00001505 x1 = PyLong_FromLong(GET_TD_SECONDS(self)); /* seconds */
Tim Peters2a799bf2002-12-16 20:18:38 +00001506 if (x1 == NULL)
1507 goto Done;
1508 x3 = PyNumber_Add(x1, x2); /* days and seconds in seconds */
1509 if (x3 == NULL)
1510 goto Done;
1511 Py_DECREF(x1);
1512 Py_DECREF(x2);
1513 x1 = x2 = NULL;
1514
1515 /* x3 has days+seconds in seconds */
1516 x1 = PyNumber_Multiply(x3, us_per_second); /* us */
1517 if (x1 == NULL)
1518 goto Done;
1519 Py_DECREF(x3);
1520 x3 = NULL;
1521
1522 /* x1 has days+seconds in us */
Christian Heimes217cfd12007-12-02 14:31:20 +00001523 x2 = PyLong_FromLong(GET_TD_MICROSECONDS(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00001524 if (x2 == NULL)
1525 goto Done;
1526 result = PyNumber_Add(x1, x2);
1527
1528Done:
1529 Py_XDECREF(x1);
1530 Py_XDECREF(x2);
1531 Py_XDECREF(x3);
1532 return result;
1533}
1534
1535/* Convert a number of us (as a Python int or long) to a timedelta.
1536 */
1537static PyObject *
Tim Petersb0c854d2003-05-17 15:57:00 +00001538microseconds_to_delta_ex(PyObject *pyus, PyTypeObject *type)
Tim Peters2a799bf2002-12-16 20:18:38 +00001539{
1540 int us;
1541 int s;
1542 int d;
Tim Peters0b0f41c2002-12-19 01:44:38 +00001543 long temp;
Tim Peters2a799bf2002-12-16 20:18:38 +00001544
1545 PyObject *tuple = NULL;
1546 PyObject *num = NULL;
1547 PyObject *result = NULL;
1548
1549 tuple = PyNumber_Divmod(pyus, us_per_second);
1550 if (tuple == NULL)
1551 goto Done;
1552
1553 num = PyTuple_GetItem(tuple, 1); /* us */
1554 if (num == NULL)
1555 goto Done;
Tim Peters0b0f41c2002-12-19 01:44:38 +00001556 temp = PyLong_AsLong(num);
Tim Peters2a799bf2002-12-16 20:18:38 +00001557 num = NULL;
Tim Peters0b0f41c2002-12-19 01:44:38 +00001558 if (temp == -1 && PyErr_Occurred())
1559 goto Done;
1560 assert(0 <= temp && temp < 1000000);
1561 us = (int)temp;
Tim Peters2a799bf2002-12-16 20:18:38 +00001562 if (us < 0) {
1563 /* The divisor was positive, so this must be an error. */
1564 assert(PyErr_Occurred());
1565 goto Done;
1566 }
1567
1568 num = PyTuple_GetItem(tuple, 0); /* leftover seconds */
1569 if (num == NULL)
1570 goto Done;
1571 Py_INCREF(num);
1572 Py_DECREF(tuple);
1573
1574 tuple = PyNumber_Divmod(num, seconds_per_day);
1575 if (tuple == NULL)
1576 goto Done;
1577 Py_DECREF(num);
1578
1579 num = PyTuple_GetItem(tuple, 1); /* seconds */
1580 if (num == NULL)
1581 goto Done;
Tim Peters0b0f41c2002-12-19 01:44:38 +00001582 temp = PyLong_AsLong(num);
Tim Peters2a799bf2002-12-16 20:18:38 +00001583 num = NULL;
Tim Peters0b0f41c2002-12-19 01:44:38 +00001584 if (temp == -1 && PyErr_Occurred())
1585 goto Done;
1586 assert(0 <= temp && temp < 24*3600);
1587 s = (int)temp;
1588
Tim Peters2a799bf2002-12-16 20:18:38 +00001589 if (s < 0) {
1590 /* The divisor was positive, so this must be an error. */
1591 assert(PyErr_Occurred());
1592 goto Done;
1593 }
1594
1595 num = PyTuple_GetItem(tuple, 0); /* leftover days */
1596 if (num == NULL)
1597 goto Done;
1598 Py_INCREF(num);
Tim Peters0b0f41c2002-12-19 01:44:38 +00001599 temp = PyLong_AsLong(num);
1600 if (temp == -1 && PyErr_Occurred())
Tim Peters2a799bf2002-12-16 20:18:38 +00001601 goto Done;
Tim Peters0b0f41c2002-12-19 01:44:38 +00001602 d = (int)temp;
1603 if ((long)d != temp) {
1604 PyErr_SetString(PyExc_OverflowError, "normalized days too "
1605 "large to fit in a C int");
1606 goto Done;
1607 }
Tim Petersb0c854d2003-05-17 15:57:00 +00001608 result = new_delta_ex(d, s, us, 0, type);
Tim Peters2a799bf2002-12-16 20:18:38 +00001609
1610Done:
1611 Py_XDECREF(tuple);
1612 Py_XDECREF(num);
1613 return result;
1614}
1615
Tim Petersb0c854d2003-05-17 15:57:00 +00001616#define microseconds_to_delta(pymicros) \
1617 microseconds_to_delta_ex(pymicros, &PyDateTime_DeltaType)
1618
Tim Peters2a799bf2002-12-16 20:18:38 +00001619static PyObject *
1620multiply_int_timedelta(PyObject *intobj, PyDateTime_Delta *delta)
1621{
1622 PyObject *pyus_in;
1623 PyObject *pyus_out;
1624 PyObject *result;
1625
1626 pyus_in = delta_to_microseconds(delta);
1627 if (pyus_in == NULL)
1628 return NULL;
1629
1630 pyus_out = PyNumber_Multiply(pyus_in, intobj);
1631 Py_DECREF(pyus_in);
1632 if (pyus_out == NULL)
1633 return NULL;
1634
1635 result = microseconds_to_delta(pyus_out);
1636 Py_DECREF(pyus_out);
1637 return result;
1638}
1639
1640static PyObject *
1641divide_timedelta_int(PyDateTime_Delta *delta, PyObject *intobj)
1642{
1643 PyObject *pyus_in;
1644 PyObject *pyus_out;
1645 PyObject *result;
1646
1647 pyus_in = delta_to_microseconds(delta);
1648 if (pyus_in == NULL)
1649 return NULL;
1650
1651 pyus_out = PyNumber_FloorDivide(pyus_in, intobj);
1652 Py_DECREF(pyus_in);
1653 if (pyus_out == NULL)
1654 return NULL;
1655
1656 result = microseconds_to_delta(pyus_out);
1657 Py_DECREF(pyus_out);
1658 return result;
1659}
1660
1661static PyObject *
1662delta_add(PyObject *left, PyObject *right)
1663{
1664 PyObject *result = Py_NotImplemented;
1665
1666 if (PyDelta_Check(left) && PyDelta_Check(right)) {
1667 /* delta + delta */
1668 /* The C-level additions can't overflow because of the
1669 * invariant bounds.
1670 */
1671 int days = GET_TD_DAYS(left) + GET_TD_DAYS(right);
1672 int seconds = GET_TD_SECONDS(left) + GET_TD_SECONDS(right);
1673 int microseconds = GET_TD_MICROSECONDS(left) +
1674 GET_TD_MICROSECONDS(right);
1675 result = new_delta(days, seconds, microseconds, 1);
1676 }
1677
1678 if (result == Py_NotImplemented)
1679 Py_INCREF(result);
1680 return result;
1681}
1682
1683static PyObject *
1684delta_negative(PyDateTime_Delta *self)
1685{
1686 return new_delta(-GET_TD_DAYS(self),
1687 -GET_TD_SECONDS(self),
1688 -GET_TD_MICROSECONDS(self),
1689 1);
1690}
1691
1692static PyObject *
1693delta_positive(PyDateTime_Delta *self)
1694{
1695 /* Could optimize this (by returning self) if this isn't a
1696 * subclass -- but who uses unary + ? Approximately nobody.
1697 */
1698 return new_delta(GET_TD_DAYS(self),
1699 GET_TD_SECONDS(self),
1700 GET_TD_MICROSECONDS(self),
1701 0);
1702}
1703
1704static PyObject *
1705delta_abs(PyDateTime_Delta *self)
1706{
1707 PyObject *result;
1708
1709 assert(GET_TD_MICROSECONDS(self) >= 0);
1710 assert(GET_TD_SECONDS(self) >= 0);
1711
1712 if (GET_TD_DAYS(self) < 0)
1713 result = delta_negative(self);
1714 else
1715 result = delta_positive(self);
1716
1717 return result;
1718}
1719
1720static PyObject *
1721delta_subtract(PyObject *left, PyObject *right)
1722{
1723 PyObject *result = Py_NotImplemented;
1724
1725 if (PyDelta_Check(left) && PyDelta_Check(right)) {
1726 /* delta - delta */
1727 PyObject *minus_right = PyNumber_Negative(right);
1728 if (minus_right) {
1729 result = delta_add(left, minus_right);
1730 Py_DECREF(minus_right);
1731 }
1732 else
1733 result = NULL;
1734 }
1735
1736 if (result == Py_NotImplemented)
1737 Py_INCREF(result);
1738 return result;
1739}
1740
Tim Peters2a799bf2002-12-16 20:18:38 +00001741static PyObject *
Guido van Rossum19960592006-08-24 17:29:38 +00001742delta_richcompare(PyObject *self, PyObject *other, int op)
Tim Peters2a799bf2002-12-16 20:18:38 +00001743{
Tim Petersaa7d8492003-02-08 03:28:59 +00001744 if (PyDelta_Check(other)) {
Guido van Rossum19960592006-08-24 17:29:38 +00001745 int diff = GET_TD_DAYS(self) - GET_TD_DAYS(other);
Tim Peters07534a62003-02-07 22:50:28 +00001746 if (diff == 0) {
1747 diff = GET_TD_SECONDS(self) - GET_TD_SECONDS(other);
1748 if (diff == 0)
1749 diff = GET_TD_MICROSECONDS(self) -
1750 GET_TD_MICROSECONDS(other);
1751 }
Guido van Rossum19960592006-08-24 17:29:38 +00001752 return diff_to_bool(diff, op);
Tim Peters2a799bf2002-12-16 20:18:38 +00001753 }
Guido van Rossum19960592006-08-24 17:29:38 +00001754 else {
1755 Py_INCREF(Py_NotImplemented);
1756 return Py_NotImplemented;
1757 }
Tim Peters2a799bf2002-12-16 20:18:38 +00001758}
1759
1760static PyObject *delta_getstate(PyDateTime_Delta *self);
1761
1762static long
1763delta_hash(PyDateTime_Delta *self)
1764{
1765 if (self->hashcode == -1) {
1766 PyObject *temp = delta_getstate(self);
1767 if (temp != NULL) {
1768 self->hashcode = PyObject_Hash(temp);
1769 Py_DECREF(temp);
1770 }
1771 }
1772 return self->hashcode;
1773}
1774
1775static PyObject *
1776delta_multiply(PyObject *left, PyObject *right)
1777{
1778 PyObject *result = Py_NotImplemented;
1779
1780 if (PyDelta_Check(left)) {
1781 /* delta * ??? */
Neal Norwitz1fe5f382007-08-31 04:32:55 +00001782 if (PyLong_Check(right))
Tim Peters2a799bf2002-12-16 20:18:38 +00001783 result = multiply_int_timedelta(right,
1784 (PyDateTime_Delta *) left);
1785 }
Neal Norwitz1fe5f382007-08-31 04:32:55 +00001786 else if (PyLong_Check(left))
Tim Peters2a799bf2002-12-16 20:18:38 +00001787 result = multiply_int_timedelta(left,
1788 (PyDateTime_Delta *) right);
1789
1790 if (result == Py_NotImplemented)
1791 Py_INCREF(result);
1792 return result;
1793}
1794
1795static PyObject *
1796delta_divide(PyObject *left, PyObject *right)
1797{
1798 PyObject *result = Py_NotImplemented;
1799
1800 if (PyDelta_Check(left)) {
1801 /* delta * ??? */
Neal Norwitz1fe5f382007-08-31 04:32:55 +00001802 if (PyLong_Check(right))
Tim Peters2a799bf2002-12-16 20:18:38 +00001803 result = divide_timedelta_int(
1804 (PyDateTime_Delta *)left,
1805 right);
1806 }
1807
1808 if (result == Py_NotImplemented)
1809 Py_INCREF(result);
1810 return result;
1811}
1812
1813/* Fold in the value of the tag ("seconds", "weeks", etc) component of a
1814 * timedelta constructor. sofar is the # of microseconds accounted for
1815 * so far, and there are factor microseconds per current unit, the number
1816 * of which is given by num. num * factor is added to sofar in a
1817 * numerically careful way, and that's the result. Any fractional
1818 * microseconds left over (this can happen if num is a float type) are
1819 * added into *leftover.
1820 * Note that there are many ways this can give an error (NULL) return.
1821 */
1822static PyObject *
1823accum(const char* tag, PyObject *sofar, PyObject *num, PyObject *factor,
1824 double *leftover)
1825{
1826 PyObject *prod;
1827 PyObject *sum;
1828
1829 assert(num != NULL);
1830
Neal Norwitz1fe5f382007-08-31 04:32:55 +00001831 if (PyLong_Check(num)) {
Tim Peters2a799bf2002-12-16 20:18:38 +00001832 prod = PyNumber_Multiply(num, factor);
1833 if (prod == NULL)
1834 return NULL;
1835 sum = PyNumber_Add(sofar, prod);
1836 Py_DECREF(prod);
1837 return sum;
1838 }
1839
1840 if (PyFloat_Check(num)) {
1841 double dnum;
1842 double fracpart;
1843 double intpart;
1844 PyObject *x;
1845 PyObject *y;
1846
1847 /* The Plan: decompose num into an integer part and a
1848 * fractional part, num = intpart + fracpart.
1849 * Then num * factor ==
1850 * intpart * factor + fracpart * factor
1851 * and the LHS can be computed exactly in long arithmetic.
1852 * The RHS is again broken into an int part and frac part.
1853 * and the frac part is added into *leftover.
1854 */
1855 dnum = PyFloat_AsDouble(num);
1856 if (dnum == -1.0 && PyErr_Occurred())
1857 return NULL;
1858 fracpart = modf(dnum, &intpart);
1859 x = PyLong_FromDouble(intpart);
1860 if (x == NULL)
1861 return NULL;
1862
1863 prod = PyNumber_Multiply(x, factor);
1864 Py_DECREF(x);
1865 if (prod == NULL)
1866 return NULL;
1867
1868 sum = PyNumber_Add(sofar, prod);
1869 Py_DECREF(prod);
1870 if (sum == NULL)
1871 return NULL;
1872
1873 if (fracpart == 0.0)
1874 return sum;
1875 /* So far we've lost no information. Dealing with the
1876 * fractional part requires float arithmetic, and may
1877 * lose a little info.
1878 */
Neal Norwitz1fe5f382007-08-31 04:32:55 +00001879 assert(PyLong_Check(factor));
Guido van Rossumddefaf32007-01-14 03:31:43 +00001880 dnum = PyLong_AsDouble(factor);
Tim Peters2a799bf2002-12-16 20:18:38 +00001881
1882 dnum *= fracpart;
1883 fracpart = modf(dnum, &intpart);
1884 x = PyLong_FromDouble(intpart);
1885 if (x == NULL) {
1886 Py_DECREF(sum);
1887 return NULL;
1888 }
1889
1890 y = PyNumber_Add(sum, x);
1891 Py_DECREF(sum);
1892 Py_DECREF(x);
1893 *leftover += fracpart;
1894 return y;
1895 }
1896
1897 PyErr_Format(PyExc_TypeError,
1898 "unsupported type for timedelta %s component: %s",
Christian Heimes90aa7642007-12-19 02:45:37 +00001899 tag, Py_TYPE(num)->tp_name);
Tim Peters2a799bf2002-12-16 20:18:38 +00001900 return NULL;
1901}
1902
1903static PyObject *
1904delta_new(PyTypeObject *type, PyObject *args, PyObject *kw)
1905{
1906 PyObject *self = NULL;
1907
1908 /* Argument objects. */
1909 PyObject *day = NULL;
1910 PyObject *second = NULL;
1911 PyObject *us = NULL;
1912 PyObject *ms = NULL;
1913 PyObject *minute = NULL;
1914 PyObject *hour = NULL;
1915 PyObject *week = NULL;
1916
1917 PyObject *x = NULL; /* running sum of microseconds */
1918 PyObject *y = NULL; /* temp sum of microseconds */
1919 double leftover_us = 0.0;
1920
Martin v. Löwis02cbf4a2006-02-27 17:20:04 +00001921 static char *keywords[] = {
Tim Peters2a799bf2002-12-16 20:18:38 +00001922 "days", "seconds", "microseconds", "milliseconds",
1923 "minutes", "hours", "weeks", NULL
1924 };
1925
1926 if (PyArg_ParseTupleAndKeywords(args, kw, "|OOOOOOO:__new__",
1927 keywords,
1928 &day, &second, &us,
1929 &ms, &minute, &hour, &week) == 0)
1930 goto Done;
1931
Christian Heimes217cfd12007-12-02 14:31:20 +00001932 x = PyLong_FromLong(0);
Tim Peters2a799bf2002-12-16 20:18:38 +00001933 if (x == NULL)
1934 goto Done;
1935
1936#define CLEANUP \
1937 Py_DECREF(x); \
1938 x = y; \
1939 if (x == NULL) \
1940 goto Done
1941
1942 if (us) {
1943 y = accum("microseconds", x, us, us_per_us, &leftover_us);
1944 CLEANUP;
1945 }
1946 if (ms) {
1947 y = accum("milliseconds", x, ms, us_per_ms, &leftover_us);
1948 CLEANUP;
1949 }
1950 if (second) {
1951 y = accum("seconds", x, second, us_per_second, &leftover_us);
1952 CLEANUP;
1953 }
1954 if (minute) {
1955 y = accum("minutes", x, minute, us_per_minute, &leftover_us);
1956 CLEANUP;
1957 }
1958 if (hour) {
1959 y = accum("hours", x, hour, us_per_hour, &leftover_us);
1960 CLEANUP;
1961 }
1962 if (day) {
1963 y = accum("days", x, day, us_per_day, &leftover_us);
1964 CLEANUP;
1965 }
1966 if (week) {
1967 y = accum("weeks", x, week, us_per_week, &leftover_us);
1968 CLEANUP;
1969 }
1970 if (leftover_us) {
1971 /* Round to nearest whole # of us, and add into x. */
Tim Peters5d644dd2003-01-02 16:32:54 +00001972 PyObject *temp = PyLong_FromLong(round_to_long(leftover_us));
Tim Peters2a799bf2002-12-16 20:18:38 +00001973 if (temp == NULL) {
1974 Py_DECREF(x);
1975 goto Done;
1976 }
1977 y = PyNumber_Add(x, temp);
1978 Py_DECREF(temp);
1979 CLEANUP;
1980 }
1981
Tim Petersb0c854d2003-05-17 15:57:00 +00001982 self = microseconds_to_delta_ex(x, type);
Tim Peters2a799bf2002-12-16 20:18:38 +00001983 Py_DECREF(x);
1984Done:
1985 return self;
1986
1987#undef CLEANUP
1988}
1989
1990static int
Jack Diederich4dafcc42006-11-28 19:15:13 +00001991delta_bool(PyDateTime_Delta *self)
Tim Peters2a799bf2002-12-16 20:18:38 +00001992{
1993 return (GET_TD_DAYS(self) != 0
1994 || GET_TD_SECONDS(self) != 0
1995 || GET_TD_MICROSECONDS(self) != 0);
1996}
1997
1998static PyObject *
1999delta_repr(PyDateTime_Delta *self)
2000{
2001 if (GET_TD_MICROSECONDS(self) != 0)
Walter Dörwald1ab83302007-05-18 17:15:44 +00002002 return PyUnicode_FromFormat("%s(%d, %d, %d)",
Christian Heimes90aa7642007-12-19 02:45:37 +00002003 Py_TYPE(self)->tp_name,
Walter Dörwald7569dfe2007-05-19 21:49:49 +00002004 GET_TD_DAYS(self),
2005 GET_TD_SECONDS(self),
2006 GET_TD_MICROSECONDS(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00002007 if (GET_TD_SECONDS(self) != 0)
Walter Dörwald1ab83302007-05-18 17:15:44 +00002008 return PyUnicode_FromFormat("%s(%d, %d)",
Christian Heimes90aa7642007-12-19 02:45:37 +00002009 Py_TYPE(self)->tp_name,
Walter Dörwald7569dfe2007-05-19 21:49:49 +00002010 GET_TD_DAYS(self),
2011 GET_TD_SECONDS(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00002012
Walter Dörwald1ab83302007-05-18 17:15:44 +00002013 return PyUnicode_FromFormat("%s(%d)",
Christian Heimes90aa7642007-12-19 02:45:37 +00002014 Py_TYPE(self)->tp_name,
Walter Dörwald7569dfe2007-05-19 21:49:49 +00002015 GET_TD_DAYS(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00002016}
2017
2018static PyObject *
2019delta_str(PyDateTime_Delta *self)
2020{
Tim Peters2a799bf2002-12-16 20:18:38 +00002021 int us = GET_TD_MICROSECONDS(self);
Walter Dörwaldbaf853c2007-05-31 18:42:47 +00002022 int seconds = GET_TD_SECONDS(self);
2023 int minutes = divmod(seconds, 60, &seconds);
2024 int hours = divmod(minutes, 60, &minutes);
2025 int days = GET_TD_DAYS(self);
Tim Peters2a799bf2002-12-16 20:18:38 +00002026
2027 if (days) {
Walter Dörwaldbaf853c2007-05-31 18:42:47 +00002028 if (us)
2029 return PyUnicode_FromFormat("%d day%s, %d:%02d:%02d.%06d",
2030 days, (days == 1 || days == -1) ? "" : "s",
2031 hours, minutes, seconds, us);
2032 else
2033 return PyUnicode_FromFormat("%d day%s, %d:%02d:%02d",
2034 days, (days == 1 || days == -1) ? "" : "s",
2035 hours, minutes, seconds);
2036 } else {
2037 if (us)
2038 return PyUnicode_FromFormat("%d:%02d:%02d.%06d",
2039 hours, minutes, seconds, us);
2040 else
2041 return PyUnicode_FromFormat("%d:%02d:%02d",
2042 hours, minutes, seconds);
Tim Peters2a799bf2002-12-16 20:18:38 +00002043 }
2044
Tim Peters2a799bf2002-12-16 20:18:38 +00002045}
2046
Tim Peters371935f2003-02-01 01:52:50 +00002047/* Pickle support, a simple use of __reduce__. */
2048
Tim Petersb57f8f02003-02-01 02:54:15 +00002049/* __getstate__ isn't exposed */
Tim Peters2a799bf2002-12-16 20:18:38 +00002050static PyObject *
2051delta_getstate(PyDateTime_Delta *self)
2052{
2053 return Py_BuildValue("iii", GET_TD_DAYS(self),
2054 GET_TD_SECONDS(self),
2055 GET_TD_MICROSECONDS(self));
2056}
2057
Tim Peters2a799bf2002-12-16 20:18:38 +00002058static PyObject *
2059delta_reduce(PyDateTime_Delta* self)
2060{
Christian Heimes90aa7642007-12-19 02:45:37 +00002061 return Py_BuildValue("ON", Py_TYPE(self), delta_getstate(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00002062}
2063
2064#define OFFSET(field) offsetof(PyDateTime_Delta, field)
2065
2066static PyMemberDef delta_members[] = {
Guido van Rossum177e41a2003-01-30 22:06:23 +00002067
Neal Norwitzdfb80862002-12-19 02:30:56 +00002068 {"days", T_INT, OFFSET(days), READONLY,
Tim Peters2a799bf2002-12-16 20:18:38 +00002069 PyDoc_STR("Number of days.")},
2070
Neal Norwitzdfb80862002-12-19 02:30:56 +00002071 {"seconds", T_INT, OFFSET(seconds), READONLY,
Tim Peters2a799bf2002-12-16 20:18:38 +00002072 PyDoc_STR("Number of seconds (>= 0 and less than 1 day).")},
2073
Neal Norwitzdfb80862002-12-19 02:30:56 +00002074 {"microseconds", T_INT, OFFSET(microseconds), READONLY,
Tim Peters2a799bf2002-12-16 20:18:38 +00002075 PyDoc_STR("Number of microseconds (>= 0 and less than 1 second).")},
2076 {NULL}
2077};
2078
2079static PyMethodDef delta_methods[] = {
Guido van Rossum177e41a2003-01-30 22:06:23 +00002080 {"__reduce__", (PyCFunction)delta_reduce, METH_NOARGS,
2081 PyDoc_STR("__reduce__() -> (cls, state)")},
2082
Tim Peters2a799bf2002-12-16 20:18:38 +00002083 {NULL, NULL},
2084};
2085
2086static char delta_doc[] =
2087PyDoc_STR("Difference between two datetime values.");
2088
2089static PyNumberMethods delta_as_number = {
2090 delta_add, /* nb_add */
2091 delta_subtract, /* nb_subtract */
2092 delta_multiply, /* nb_multiply */
Tim Peters2a799bf2002-12-16 20:18:38 +00002093 0, /* nb_remainder */
2094 0, /* nb_divmod */
2095 0, /* nb_power */
2096 (unaryfunc)delta_negative, /* nb_negative */
2097 (unaryfunc)delta_positive, /* nb_positive */
2098 (unaryfunc)delta_abs, /* nb_absolute */
Jack Diederich4dafcc42006-11-28 19:15:13 +00002099 (inquiry)delta_bool, /* nb_bool */
Tim Peters2a799bf2002-12-16 20:18:38 +00002100 0, /*nb_invert*/
2101 0, /*nb_lshift*/
2102 0, /*nb_rshift*/
2103 0, /*nb_and*/
2104 0, /*nb_xor*/
2105 0, /*nb_or*/
Neil Schemenauer16c70752007-09-21 20:19:23 +00002106 0, /*nb_reserved*/
Tim Peters2a799bf2002-12-16 20:18:38 +00002107 0, /*nb_int*/
2108 0, /*nb_long*/
2109 0, /*nb_float*/
2110 0, /*nb_oct*/
2111 0, /*nb_hex*/
2112 0, /*nb_inplace_add*/
2113 0, /*nb_inplace_subtract*/
2114 0, /*nb_inplace_multiply*/
Tim Peters2a799bf2002-12-16 20:18:38 +00002115 0, /*nb_inplace_remainder*/
2116 0, /*nb_inplace_power*/
2117 0, /*nb_inplace_lshift*/
2118 0, /*nb_inplace_rshift*/
2119 0, /*nb_inplace_and*/
2120 0, /*nb_inplace_xor*/
2121 0, /*nb_inplace_or*/
2122 delta_divide, /* nb_floor_divide */
2123 0, /* nb_true_divide */
2124 0, /* nb_inplace_floor_divide */
2125 0, /* nb_inplace_true_divide */
2126};
2127
2128static PyTypeObject PyDateTime_DeltaType = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00002129 PyVarObject_HEAD_INIT(NULL, 0)
Tim Peters2a799bf2002-12-16 20:18:38 +00002130 "datetime.timedelta", /* tp_name */
2131 sizeof(PyDateTime_Delta), /* tp_basicsize */
2132 0, /* tp_itemsize */
2133 0, /* tp_dealloc */
2134 0, /* tp_print */
2135 0, /* tp_getattr */
2136 0, /* tp_setattr */
2137 0, /* tp_compare */
2138 (reprfunc)delta_repr, /* tp_repr */
2139 &delta_as_number, /* tp_as_number */
2140 0, /* tp_as_sequence */
2141 0, /* tp_as_mapping */
2142 (hashfunc)delta_hash, /* tp_hash */
2143 0, /* tp_call */
2144 (reprfunc)delta_str, /* tp_str */
2145 PyObject_GenericGetAttr, /* tp_getattro */
2146 0, /* tp_setattro */
2147 0, /* tp_as_buffer */
Guido van Rossum3cf5b1e2006-07-27 21:53:35 +00002148 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
Tim Peters2a799bf2002-12-16 20:18:38 +00002149 delta_doc, /* tp_doc */
2150 0, /* tp_traverse */
2151 0, /* tp_clear */
Guido van Rossum19960592006-08-24 17:29:38 +00002152 delta_richcompare, /* tp_richcompare */
Tim Peters2a799bf2002-12-16 20:18:38 +00002153 0, /* tp_weaklistoffset */
2154 0, /* tp_iter */
2155 0, /* tp_iternext */
2156 delta_methods, /* tp_methods */
2157 delta_members, /* tp_members */
2158 0, /* tp_getset */
2159 0, /* tp_base */
2160 0, /* tp_dict */
2161 0, /* tp_descr_get */
2162 0, /* tp_descr_set */
2163 0, /* tp_dictoffset */
2164 0, /* tp_init */
2165 0, /* tp_alloc */
2166 delta_new, /* tp_new */
Tim Peters4c530132003-05-16 22:44:06 +00002167 0, /* tp_free */
Tim Peters2a799bf2002-12-16 20:18:38 +00002168};
2169
2170/*
2171 * PyDateTime_Date implementation.
2172 */
2173
2174/* Accessor properties. */
2175
2176static PyObject *
2177date_year(PyDateTime_Date *self, void *unused)
2178{
Christian Heimes217cfd12007-12-02 14:31:20 +00002179 return PyLong_FromLong(GET_YEAR(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00002180}
2181
2182static PyObject *
2183date_month(PyDateTime_Date *self, void *unused)
2184{
Christian Heimes217cfd12007-12-02 14:31:20 +00002185 return PyLong_FromLong(GET_MONTH(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00002186}
2187
2188static PyObject *
2189date_day(PyDateTime_Date *self, void *unused)
2190{
Christian Heimes217cfd12007-12-02 14:31:20 +00002191 return PyLong_FromLong(GET_DAY(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00002192}
2193
2194static PyGetSetDef date_getset[] = {
2195 {"year", (getter)date_year},
2196 {"month", (getter)date_month},
2197 {"day", (getter)date_day},
2198 {NULL}
2199};
2200
2201/* Constructors. */
2202
Martin v. Löwis02cbf4a2006-02-27 17:20:04 +00002203static char *date_kws[] = {"year", "month", "day", NULL};
Tim Peters12bf3392002-12-24 05:41:27 +00002204
Tim Peters2a799bf2002-12-16 20:18:38 +00002205static PyObject *
2206date_new(PyTypeObject *type, PyObject *args, PyObject *kw)
2207{
2208 PyObject *self = NULL;
Tim Peters70533e22003-02-01 04:40:04 +00002209 PyObject *state;
Tim Peters2a799bf2002-12-16 20:18:38 +00002210 int year;
2211 int month;
2212 int day;
2213
Guido van Rossum177e41a2003-01-30 22:06:23 +00002214 /* Check for invocation from pickle with __getstate__ state */
2215 if (PyTuple_GET_SIZE(args) == 1 &&
Guido van Rossum254348e2007-11-21 19:29:53 +00002216 PyString_Check(state = PyTuple_GET_ITEM(args, 0)) &&
2217 PyString_GET_SIZE(state) == _PyDateTime_DATE_DATASIZE &&
2218 MONTH_IS_SANE(PyString_AS_STRING(state)[2]))
Guido van Rossum177e41a2003-01-30 22:06:23 +00002219 {
Tim Peters70533e22003-02-01 04:40:04 +00002220 PyDateTime_Date *me;
2221
Tim Peters604c0132004-06-07 23:04:33 +00002222 me = (PyDateTime_Date *) (type->tp_alloc(type, 0));
Tim Peters70533e22003-02-01 04:40:04 +00002223 if (me != NULL) {
Guido van Rossum254348e2007-11-21 19:29:53 +00002224 char *pdata = PyString_AS_STRING(state);
Tim Peters70533e22003-02-01 04:40:04 +00002225 memcpy(me->data, pdata, _PyDateTime_DATE_DATASIZE);
2226 me->hashcode = -1;
Guido van Rossum177e41a2003-01-30 22:06:23 +00002227 }
Tim Peters70533e22003-02-01 04:40:04 +00002228 return (PyObject *)me;
Guido van Rossum177e41a2003-01-30 22:06:23 +00002229 }
2230
Tim Peters12bf3392002-12-24 05:41:27 +00002231 if (PyArg_ParseTupleAndKeywords(args, kw, "iii", date_kws,
Tim Peters2a799bf2002-12-16 20:18:38 +00002232 &year, &month, &day)) {
2233 if (check_date_args(year, month, day) < 0)
2234 return NULL;
Guido van Rossum8b7a9a32003-04-14 22:01:58 +00002235 self = new_date_ex(year, month, day, type);
Tim Peters2a799bf2002-12-16 20:18:38 +00002236 }
2237 return self;
2238}
2239
2240/* Return new date from localtime(t). */
2241static PyObject *
Tim Peters1b6f7a92004-06-20 02:50:16 +00002242date_local_from_time_t(PyObject *cls, double ts)
Tim Peters2a799bf2002-12-16 20:18:38 +00002243{
2244 struct tm *tm;
Tim Peters1b6f7a92004-06-20 02:50:16 +00002245 time_t t;
Tim Peters2a799bf2002-12-16 20:18:38 +00002246 PyObject *result = NULL;
2247
Tim Peters1b6f7a92004-06-20 02:50:16 +00002248 t = _PyTime_DoubleToTimet(ts);
2249 if (t == (time_t)-1 && PyErr_Occurred())
2250 return NULL;
Tim Peters2a799bf2002-12-16 20:18:38 +00002251 tm = localtime(&t);
2252 if (tm)
2253 result = PyObject_CallFunction(cls, "iii",
2254 tm->tm_year + 1900,
2255 tm->tm_mon + 1,
2256 tm->tm_mday);
2257 else
2258 PyErr_SetString(PyExc_ValueError,
2259 "timestamp out of range for "
2260 "platform localtime() function");
2261 return result;
2262}
2263
2264/* Return new date from current time.
2265 * We say this is equivalent to fromtimestamp(time.time()), and the
2266 * only way to be sure of that is to *call* time.time(). That's not
2267 * generally the same as calling C's time.
2268 */
2269static PyObject *
2270date_today(PyObject *cls, PyObject *dummy)
2271{
2272 PyObject *time;
2273 PyObject *result;
2274
2275 time = time_time();
2276 if (time == NULL)
2277 return NULL;
2278
2279 /* Note well: today() is a class method, so this may not call
2280 * date.fromtimestamp. For example, it may call
2281 * datetime.fromtimestamp. That's why we need all the accuracy
2282 * time.time() delivers; if someone were gonzo about optimization,
2283 * date.today() could get away with plain C time().
2284 */
2285 result = PyObject_CallMethod(cls, "fromtimestamp", "O", time);
2286 Py_DECREF(time);
2287 return result;
2288}
2289
2290/* Return new date from given timestamp (Python timestamp -- a double). */
2291static PyObject *
2292date_fromtimestamp(PyObject *cls, PyObject *args)
2293{
2294 double timestamp;
2295 PyObject *result = NULL;
2296
2297 if (PyArg_ParseTuple(args, "d:fromtimestamp", &timestamp))
Tim Peters1b6f7a92004-06-20 02:50:16 +00002298 result = date_local_from_time_t(cls, timestamp);
Tim Peters2a799bf2002-12-16 20:18:38 +00002299 return result;
2300}
2301
2302/* Return new date from proleptic Gregorian ordinal. Raises ValueError if
2303 * the ordinal is out of range.
2304 */
2305static PyObject *
2306date_fromordinal(PyObject *cls, PyObject *args)
2307{
2308 PyObject *result = NULL;
2309 int ordinal;
2310
2311 if (PyArg_ParseTuple(args, "i:fromordinal", &ordinal)) {
2312 int year;
2313 int month;
2314 int day;
2315
2316 if (ordinal < 1)
2317 PyErr_SetString(PyExc_ValueError, "ordinal must be "
2318 ">= 1");
2319 else {
2320 ord_to_ymd(ordinal, &year, &month, &day);
2321 result = PyObject_CallFunction(cls, "iii",
2322 year, month, day);
2323 }
2324 }
2325 return result;
2326}
2327
2328/*
2329 * Date arithmetic.
2330 */
2331
2332/* date + timedelta -> date. If arg negate is true, subtract the timedelta
2333 * instead.
2334 */
2335static PyObject *
2336add_date_timedelta(PyDateTime_Date *date, PyDateTime_Delta *delta, int negate)
2337{
2338 PyObject *result = NULL;
2339 int year = GET_YEAR(date);
2340 int month = GET_MONTH(date);
2341 int deltadays = GET_TD_DAYS(delta);
2342 /* C-level overflow is impossible because |deltadays| < 1e9. */
2343 int day = GET_DAY(date) + (negate ? -deltadays : deltadays);
2344
2345 if (normalize_date(&year, &month, &day) >= 0)
2346 result = new_date(year, month, day);
2347 return result;
2348}
2349
2350static PyObject *
2351date_add(PyObject *left, PyObject *right)
2352{
2353 if (PyDateTime_Check(left) || PyDateTime_Check(right)) {
2354 Py_INCREF(Py_NotImplemented);
2355 return Py_NotImplemented;
2356 }
Tim Petersaa7d8492003-02-08 03:28:59 +00002357 if (PyDate_Check(left)) {
Tim Peters2a799bf2002-12-16 20:18:38 +00002358 /* date + ??? */
2359 if (PyDelta_Check(right))
2360 /* date + delta */
2361 return add_date_timedelta((PyDateTime_Date *) left,
2362 (PyDateTime_Delta *) right,
2363 0);
2364 }
2365 else {
2366 /* ??? + date
2367 * 'right' must be one of us, or we wouldn't have been called
2368 */
2369 if (PyDelta_Check(left))
2370 /* delta + date */
2371 return add_date_timedelta((PyDateTime_Date *) right,
2372 (PyDateTime_Delta *) left,
2373 0);
2374 }
2375 Py_INCREF(Py_NotImplemented);
2376 return Py_NotImplemented;
2377}
2378
2379static PyObject *
2380date_subtract(PyObject *left, PyObject *right)
2381{
2382 if (PyDateTime_Check(left) || PyDateTime_Check(right)) {
2383 Py_INCREF(Py_NotImplemented);
2384 return Py_NotImplemented;
2385 }
Tim Petersaa7d8492003-02-08 03:28:59 +00002386 if (PyDate_Check(left)) {
2387 if (PyDate_Check(right)) {
Tim Peters2a799bf2002-12-16 20:18:38 +00002388 /* date - date */
2389 int left_ord = ymd_to_ord(GET_YEAR(left),
2390 GET_MONTH(left),
2391 GET_DAY(left));
2392 int right_ord = ymd_to_ord(GET_YEAR(right),
2393 GET_MONTH(right),
2394 GET_DAY(right));
2395 return new_delta(left_ord - right_ord, 0, 0, 0);
2396 }
2397 if (PyDelta_Check(right)) {
2398 /* date - delta */
2399 return add_date_timedelta((PyDateTime_Date *) left,
2400 (PyDateTime_Delta *) right,
2401 1);
2402 }
2403 }
2404 Py_INCREF(Py_NotImplemented);
2405 return Py_NotImplemented;
2406}
2407
2408
2409/* Various ways to turn a date into a string. */
2410
2411static PyObject *
2412date_repr(PyDateTime_Date *self)
2413{
Walter Dörwald7569dfe2007-05-19 21:49:49 +00002414 return PyUnicode_FromFormat("%s(%d, %d, %d)",
Christian Heimes90aa7642007-12-19 02:45:37 +00002415 Py_TYPE(self)->tp_name,
Walter Dörwald7569dfe2007-05-19 21:49:49 +00002416 GET_YEAR(self), GET_MONTH(self), GET_DAY(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00002417}
2418
2419static PyObject *
2420date_isoformat(PyDateTime_Date *self)
2421{
Walter Dörwaldbafa1372007-05-31 17:50:48 +00002422 return PyUnicode_FromFormat("%04d-%02d-%02d",
2423 GET_YEAR(self), GET_MONTH(self), GET_DAY(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00002424}
2425
Tim Peterse2df5ff2003-05-02 18:39:55 +00002426/* str() calls the appropriate isoformat() method. */
Tim Peters2a799bf2002-12-16 20:18:38 +00002427static PyObject *
2428date_str(PyDateTime_Date *self)
2429{
2430 return PyObject_CallMethod((PyObject *)self, "isoformat", "()");
2431}
2432
2433
2434static PyObject *
2435date_ctime(PyDateTime_Date *self)
2436{
2437 return format_ctime(self, 0, 0, 0);
2438}
2439
2440static PyObject *
2441date_strftime(PyDateTime_Date *self, PyObject *args, PyObject *kw)
2442{
2443 /* This method can be inherited, and needs to call the
2444 * timetuple() method appropriate to self's class.
2445 */
2446 PyObject *result;
2447 PyObject *format;
2448 PyObject *tuple;
Martin v. Löwis02cbf4a2006-02-27 17:20:04 +00002449 static char *keywords[] = {"format", NULL};
Tim Peters2a799bf2002-12-16 20:18:38 +00002450
Guido van Rossumfd53fd62007-08-24 04:05:13 +00002451 if (! PyArg_ParseTupleAndKeywords(args, kw, "U:strftime", keywords,
Guido van Rossumbce56a62007-05-10 18:04:33 +00002452 &format))
Tim Peters2a799bf2002-12-16 20:18:38 +00002453 return NULL;
2454
2455 tuple = PyObject_CallMethod((PyObject *)self, "timetuple", "()");
2456 if (tuple == NULL)
2457 return NULL;
Tim Petersbad8ff02002-12-30 20:52:32 +00002458 result = wrap_strftime((PyObject *)self, format, tuple,
2459 (PyObject *)self);
Tim Peters2a799bf2002-12-16 20:18:38 +00002460 Py_DECREF(tuple);
2461 return result;
2462}
2463
Eric Smith1ba31142007-09-11 18:06:02 +00002464static PyObject *
2465date_format(PyDateTime_Date *self, PyObject *args)
2466{
2467 PyObject *format;
2468
2469 if (!PyArg_ParseTuple(args, "U:__format__", &format))
2470 return NULL;
2471
2472 /* if the format is zero length, return str(self) */
2473 if (PyUnicode_GetSize(format) == 0)
Thomas Heller519a0422007-11-15 20:48:54 +00002474 return PyObject_Str((PyObject *)self);
Eric Smith1ba31142007-09-11 18:06:02 +00002475
2476 return PyObject_CallMethod((PyObject *)self, "strftime", "O", format);
2477}
2478
Tim Peters2a799bf2002-12-16 20:18:38 +00002479/* ISO methods. */
2480
2481static PyObject *
2482date_isoweekday(PyDateTime_Date *self)
2483{
2484 int dow = weekday(GET_YEAR(self), GET_MONTH(self), GET_DAY(self));
2485
Christian Heimes217cfd12007-12-02 14:31:20 +00002486 return PyLong_FromLong(dow + 1);
Tim Peters2a799bf2002-12-16 20:18:38 +00002487}
2488
2489static PyObject *
2490date_isocalendar(PyDateTime_Date *self)
2491{
2492 int year = GET_YEAR(self);
2493 int week1_monday = iso_week1_monday(year);
2494 int today = ymd_to_ord(year, GET_MONTH(self), GET_DAY(self));
2495 int week;
2496 int day;
2497
2498 week = divmod(today - week1_monday, 7, &day);
2499 if (week < 0) {
2500 --year;
2501 week1_monday = iso_week1_monday(year);
2502 week = divmod(today - week1_monday, 7, &day);
2503 }
2504 else if (week >= 52 && today >= iso_week1_monday(year + 1)) {
2505 ++year;
2506 week = 0;
2507 }
2508 return Py_BuildValue("iii", year, week + 1, day + 1);
2509}
2510
2511/* Miscellaneous methods. */
2512
Tim Peters2a799bf2002-12-16 20:18:38 +00002513static PyObject *
Guido van Rossum19960592006-08-24 17:29:38 +00002514date_richcompare(PyObject *self, PyObject *other, int op)
Tim Peters2a799bf2002-12-16 20:18:38 +00002515{
Guido van Rossum19960592006-08-24 17:29:38 +00002516 if (PyDate_Check(other)) {
2517 int diff = memcmp(((PyDateTime_Date *)self)->data,
2518 ((PyDateTime_Date *)other)->data,
2519 _PyDateTime_DATE_DATASIZE);
2520 return diff_to_bool(diff, op);
2521 }
2522 else {
Tim Peters07534a62003-02-07 22:50:28 +00002523 Py_INCREF(Py_NotImplemented);
2524 return Py_NotImplemented;
Tim Peters2a799bf2002-12-16 20:18:38 +00002525 }
Tim Peters2a799bf2002-12-16 20:18:38 +00002526}
2527
2528static PyObject *
2529date_timetuple(PyDateTime_Date *self)
2530{
2531 return build_struct_time(GET_YEAR(self),
2532 GET_MONTH(self),
2533 GET_DAY(self),
2534 0, 0, 0, -1);
2535}
2536
Tim Peters12bf3392002-12-24 05:41:27 +00002537static PyObject *
2538date_replace(PyDateTime_Date *self, PyObject *args, PyObject *kw)
2539{
2540 PyObject *clone;
2541 PyObject *tuple;
2542 int year = GET_YEAR(self);
2543 int month = GET_MONTH(self);
2544 int day = GET_DAY(self);
2545
2546 if (! PyArg_ParseTupleAndKeywords(args, kw, "|iii:replace", date_kws,
2547 &year, &month, &day))
2548 return NULL;
2549 tuple = Py_BuildValue("iii", year, month, day);
2550 if (tuple == NULL)
2551 return NULL;
Christian Heimes90aa7642007-12-19 02:45:37 +00002552 clone = date_new(Py_TYPE(self), tuple, NULL);
Tim Peters12bf3392002-12-24 05:41:27 +00002553 Py_DECREF(tuple);
2554 return clone;
2555}
2556
Guido van Rossumfd53fd62007-08-24 04:05:13 +00002557/*
2558 Borrowed from stringobject.c, originally it was string_hash()
2559*/
2560static long
2561generic_hash(unsigned char *data, int len)
2562{
2563 register unsigned char *p;
2564 register long x;
2565
2566 p = (unsigned char *) data;
2567 x = *p << 7;
2568 while (--len >= 0)
2569 x = (1000003*x) ^ *p++;
2570 x ^= len;
2571 if (x == -1)
2572 x = -2;
2573
2574 return x;
2575}
2576
2577
2578static PyObject *date_getstate(PyDateTime_Date *self);
Tim Peters2a799bf2002-12-16 20:18:38 +00002579
2580static long
2581date_hash(PyDateTime_Date *self)
2582{
Guido van Rossumfd53fd62007-08-24 04:05:13 +00002583 if (self->hashcode == -1)
2584 self->hashcode = generic_hash(
2585 (unsigned char *)self->data, _PyDateTime_DATE_DATASIZE);
Guido van Rossum254348e2007-11-21 19:29:53 +00002586
Tim Peters2a799bf2002-12-16 20:18:38 +00002587 return self->hashcode;
2588}
2589
2590static PyObject *
2591date_toordinal(PyDateTime_Date *self)
2592{
Christian Heimes217cfd12007-12-02 14:31:20 +00002593 return PyLong_FromLong(ymd_to_ord(GET_YEAR(self), GET_MONTH(self),
Tim Peters2a799bf2002-12-16 20:18:38 +00002594 GET_DAY(self)));
2595}
2596
2597static PyObject *
2598date_weekday(PyDateTime_Date *self)
2599{
2600 int dow = weekday(GET_YEAR(self), GET_MONTH(self), GET_DAY(self));
2601
Christian Heimes217cfd12007-12-02 14:31:20 +00002602 return PyLong_FromLong(dow);
Tim Peters2a799bf2002-12-16 20:18:38 +00002603}
2604
Tim Peters371935f2003-02-01 01:52:50 +00002605/* Pickle support, a simple use of __reduce__. */
Tim Peters2a799bf2002-12-16 20:18:38 +00002606
Tim Petersb57f8f02003-02-01 02:54:15 +00002607/* __getstate__ isn't exposed */
Tim Peters2a799bf2002-12-16 20:18:38 +00002608static PyObject *
Guido van Rossumfd53fd62007-08-24 04:05:13 +00002609date_getstate(PyDateTime_Date *self)
Tim Peters2a799bf2002-12-16 20:18:38 +00002610{
Martin v. Löwis10a60b32007-07-18 02:28:27 +00002611 PyObject* field;
Guido van Rossum254348e2007-11-21 19:29:53 +00002612 field = PyString_FromStringAndSize((char*)self->data,
2613 _PyDateTime_DATE_DATASIZE);
Martin v. Löwis10a60b32007-07-18 02:28:27 +00002614 return Py_BuildValue("(N)", field);
Tim Peters2a799bf2002-12-16 20:18:38 +00002615}
2616
2617static PyObject *
Guido van Rossum177e41a2003-01-30 22:06:23 +00002618date_reduce(PyDateTime_Date *self, PyObject *arg)
Tim Peters2a799bf2002-12-16 20:18:38 +00002619{
Christian Heimes90aa7642007-12-19 02:45:37 +00002620 return Py_BuildValue("(ON)", Py_TYPE(self), date_getstate(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00002621}
2622
2623static PyMethodDef date_methods[] = {
Guido van Rossum177e41a2003-01-30 22:06:23 +00002624
Tim Peters2a799bf2002-12-16 20:18:38 +00002625 /* Class methods: */
Guido van Rossum177e41a2003-01-30 22:06:23 +00002626
Tim Peters2a799bf2002-12-16 20:18:38 +00002627 {"fromtimestamp", (PyCFunction)date_fromtimestamp, METH_VARARGS |
2628 METH_CLASS,
2629 PyDoc_STR("timestamp -> local date from a POSIX timestamp (like "
2630 "time.time()).")},
2631
2632 {"fromordinal", (PyCFunction)date_fromordinal, METH_VARARGS |
2633 METH_CLASS,
2634 PyDoc_STR("int -> date corresponding to a proleptic Gregorian "
2635 "ordinal.")},
2636
2637 {"today", (PyCFunction)date_today, METH_NOARGS | METH_CLASS,
2638 PyDoc_STR("Current date or datetime: same as "
2639 "self.__class__.fromtimestamp(time.time()).")},
2640
2641 /* Instance methods: */
2642
2643 {"ctime", (PyCFunction)date_ctime, METH_NOARGS,
2644 PyDoc_STR("Return ctime() style string.")},
2645
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002646 {"strftime", (PyCFunction)date_strftime, METH_VARARGS | METH_KEYWORDS,
Tim Peters2a799bf2002-12-16 20:18:38 +00002647 PyDoc_STR("format -> strftime() style string.")},
2648
Eric Smith1ba31142007-09-11 18:06:02 +00002649 {"__format__", (PyCFunction)date_format, METH_VARARGS,
2650 PyDoc_STR("Formats self with strftime.")},
2651
Tim Peters2a799bf2002-12-16 20:18:38 +00002652 {"timetuple", (PyCFunction)date_timetuple, METH_NOARGS,
2653 PyDoc_STR("Return time tuple, compatible with time.localtime().")},
2654
2655 {"isocalendar", (PyCFunction)date_isocalendar, METH_NOARGS,
2656 PyDoc_STR("Return a 3-tuple containing ISO year, week number, and "
2657 "weekday.")},
2658
2659 {"isoformat", (PyCFunction)date_isoformat, METH_NOARGS,
2660 PyDoc_STR("Return string in ISO 8601 format, YYYY-MM-DD.")},
2661
2662 {"isoweekday", (PyCFunction)date_isoweekday, METH_NOARGS,
2663 PyDoc_STR("Return the day of the week represented by the date.\n"
2664 "Monday == 1 ... Sunday == 7")},
2665
2666 {"toordinal", (PyCFunction)date_toordinal, METH_NOARGS,
2667 PyDoc_STR("Return proleptic Gregorian ordinal. January 1 of year "
2668 "1 is day 1.")},
2669
2670 {"weekday", (PyCFunction)date_weekday, METH_NOARGS,
2671 PyDoc_STR("Return the day of the week represented by the date.\n"
2672 "Monday == 0 ... Sunday == 6")},
2673
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002674 {"replace", (PyCFunction)date_replace, METH_VARARGS | METH_KEYWORDS,
Tim Peters12bf3392002-12-24 05:41:27 +00002675 PyDoc_STR("Return date with new specified fields.")},
2676
Guido van Rossum177e41a2003-01-30 22:06:23 +00002677 {"__reduce__", (PyCFunction)date_reduce, METH_NOARGS,
2678 PyDoc_STR("__reduce__() -> (cls, state)")},
2679
Tim Peters2a799bf2002-12-16 20:18:38 +00002680 {NULL, NULL}
2681};
2682
2683static char date_doc[] =
Raymond Hettinger3a4231d2004-12-19 20:13:24 +00002684PyDoc_STR("date(year, month, day) --> date object");
Tim Peters2a799bf2002-12-16 20:18:38 +00002685
2686static PyNumberMethods date_as_number = {
2687 date_add, /* nb_add */
2688 date_subtract, /* nb_subtract */
2689 0, /* nb_multiply */
Tim Peters2a799bf2002-12-16 20:18:38 +00002690 0, /* nb_remainder */
2691 0, /* nb_divmod */
2692 0, /* nb_power */
2693 0, /* nb_negative */
2694 0, /* nb_positive */
2695 0, /* nb_absolute */
Jack Diederich4dafcc42006-11-28 19:15:13 +00002696 0, /* nb_bool */
Tim Peters2a799bf2002-12-16 20:18:38 +00002697};
2698
2699static PyTypeObject PyDateTime_DateType = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00002700 PyVarObject_HEAD_INIT(NULL, 0)
Tim Peters2a799bf2002-12-16 20:18:38 +00002701 "datetime.date", /* tp_name */
2702 sizeof(PyDateTime_Date), /* tp_basicsize */
2703 0, /* tp_itemsize */
Guido van Rossum8b7a9a32003-04-14 22:01:58 +00002704 0, /* tp_dealloc */
Tim Peters2a799bf2002-12-16 20:18:38 +00002705 0, /* tp_print */
2706 0, /* tp_getattr */
2707 0, /* tp_setattr */
2708 0, /* tp_compare */
2709 (reprfunc)date_repr, /* tp_repr */
2710 &date_as_number, /* tp_as_number */
2711 0, /* tp_as_sequence */
2712 0, /* tp_as_mapping */
2713 (hashfunc)date_hash, /* tp_hash */
2714 0, /* tp_call */
2715 (reprfunc)date_str, /* tp_str */
2716 PyObject_GenericGetAttr, /* tp_getattro */
2717 0, /* tp_setattro */
2718 0, /* tp_as_buffer */
Guido van Rossum3cf5b1e2006-07-27 21:53:35 +00002719 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
Tim Peters2a799bf2002-12-16 20:18:38 +00002720 date_doc, /* tp_doc */
2721 0, /* tp_traverse */
2722 0, /* tp_clear */
Guido van Rossum19960592006-08-24 17:29:38 +00002723 date_richcompare, /* tp_richcompare */
Tim Peters2a799bf2002-12-16 20:18:38 +00002724 0, /* tp_weaklistoffset */
2725 0, /* tp_iter */
2726 0, /* tp_iternext */
2727 date_methods, /* tp_methods */
2728 0, /* tp_members */
2729 date_getset, /* tp_getset */
2730 0, /* tp_base */
2731 0, /* tp_dict */
2732 0, /* tp_descr_get */
2733 0, /* tp_descr_set */
2734 0, /* tp_dictoffset */
2735 0, /* tp_init */
2736 0, /* tp_alloc */
2737 date_new, /* tp_new */
Tim Peters4c530132003-05-16 22:44:06 +00002738 0, /* tp_free */
Tim Peters2a799bf2002-12-16 20:18:38 +00002739};
2740
2741/*
Tim Peters2a799bf2002-12-16 20:18:38 +00002742 * PyDateTime_TZInfo implementation.
2743 */
2744
2745/* This is a pure abstract base class, so doesn't do anything beyond
2746 * raising NotImplemented exceptions. Real tzinfo classes need
2747 * to derive from this. This is mostly for clarity, and for efficiency in
Tim Petersa9bc1682003-01-11 03:39:11 +00002748 * datetime and time constructors (their tzinfo arguments need to
Tim Peters2a799bf2002-12-16 20:18:38 +00002749 * be subclasses of this tzinfo class, which is easy and quick to check).
2750 *
2751 * Note: For reasons having to do with pickling of subclasses, we have
2752 * to allow tzinfo objects to be instantiated. This wasn't an issue
2753 * in the Python implementation (__init__() could raise NotImplementedError
2754 * there without ill effect), but doing so in the C implementation hit a
2755 * brick wall.
2756 */
2757
2758static PyObject *
2759tzinfo_nogo(const char* methodname)
2760{
2761 PyErr_Format(PyExc_NotImplementedError,
2762 "a tzinfo subclass must implement %s()",
2763 methodname);
2764 return NULL;
2765}
2766
2767/* Methods. A subclass must implement these. */
2768
Tim Peters52dcce22003-01-23 16:36:11 +00002769static PyObject *
Tim Peters2a799bf2002-12-16 20:18:38 +00002770tzinfo_tzname(PyDateTime_TZInfo *self, PyObject *dt)
2771{
2772 return tzinfo_nogo("tzname");
2773}
2774
Tim Peters52dcce22003-01-23 16:36:11 +00002775static PyObject *
Tim Peters2a799bf2002-12-16 20:18:38 +00002776tzinfo_utcoffset(PyDateTime_TZInfo *self, PyObject *dt)
2777{
2778 return tzinfo_nogo("utcoffset");
2779}
2780
Tim Peters52dcce22003-01-23 16:36:11 +00002781static PyObject *
Tim Peters2a799bf2002-12-16 20:18:38 +00002782tzinfo_dst(PyDateTime_TZInfo *self, PyObject *dt)
2783{
2784 return tzinfo_nogo("dst");
2785}
2786
Tim Peters52dcce22003-01-23 16:36:11 +00002787static PyObject *
2788tzinfo_fromutc(PyDateTime_TZInfo *self, PyDateTime_DateTime *dt)
2789{
2790 int y, m, d, hh, mm, ss, us;
2791
2792 PyObject *result;
2793 int off, dst;
2794 int none;
2795 int delta;
2796
2797 if (! PyDateTime_Check(dt)) {
2798 PyErr_SetString(PyExc_TypeError,
2799 "fromutc: argument must be a datetime");
2800 return NULL;
2801 }
2802 if (! HASTZINFO(dt) || dt->tzinfo != (PyObject *)self) {
2803 PyErr_SetString(PyExc_ValueError, "fromutc: dt.tzinfo "
2804 "is not self");
2805 return NULL;
2806 }
2807
2808 off = call_utcoffset(dt->tzinfo, (PyObject *)dt, &none);
2809 if (off == -1 && PyErr_Occurred())
2810 return NULL;
2811 if (none) {
2812 PyErr_SetString(PyExc_ValueError, "fromutc: non-None "
2813 "utcoffset() result required");
2814 return NULL;
2815 }
2816
2817 dst = call_dst(dt->tzinfo, (PyObject *)dt, &none);
2818 if (dst == -1 && PyErr_Occurred())
2819 return NULL;
2820 if (none) {
2821 PyErr_SetString(PyExc_ValueError, "fromutc: non-None "
2822 "dst() result required");
2823 return NULL;
2824 }
2825
2826 y = GET_YEAR(dt);
2827 m = GET_MONTH(dt);
2828 d = GET_DAY(dt);
2829 hh = DATE_GET_HOUR(dt);
2830 mm = DATE_GET_MINUTE(dt);
2831 ss = DATE_GET_SECOND(dt);
2832 us = DATE_GET_MICROSECOND(dt);
2833
2834 delta = off - dst;
2835 mm += delta;
2836 if ((mm < 0 || mm >= 60) &&
2837 normalize_datetime(&y, &m, &d, &hh, &mm, &ss, &us) < 0)
Tim Petersb1049e82003-01-23 17:20:36 +00002838 return NULL;
Tim Peters52dcce22003-01-23 16:36:11 +00002839 result = new_datetime(y, m, d, hh, mm, ss, us, dt->tzinfo);
2840 if (result == NULL)
2841 return result;
2842
2843 dst = call_dst(dt->tzinfo, result, &none);
2844 if (dst == -1 && PyErr_Occurred())
2845 goto Fail;
2846 if (none)
2847 goto Inconsistent;
2848 if (dst == 0)
2849 return result;
2850
2851 mm += dst;
2852 if ((mm < 0 || mm >= 60) &&
2853 normalize_datetime(&y, &m, &d, &hh, &mm, &ss, &us) < 0)
2854 goto Fail;
2855 Py_DECREF(result);
2856 result = new_datetime(y, m, d, hh, mm, ss, us, dt->tzinfo);
2857 return result;
2858
2859Inconsistent:
2860 PyErr_SetString(PyExc_ValueError, "fromutc: tz.dst() gave"
2861 "inconsistent results; cannot convert");
2862
2863 /* fall thru to failure */
2864Fail:
2865 Py_DECREF(result);
2866 return NULL;
2867}
2868
Tim Peters2a799bf2002-12-16 20:18:38 +00002869/*
2870 * Pickle support. This is solely so that tzinfo subclasses can use
Guido van Rossum177e41a2003-01-30 22:06:23 +00002871 * pickling -- tzinfo itself is supposed to be uninstantiable.
Tim Peters2a799bf2002-12-16 20:18:38 +00002872 */
2873
Guido van Rossum177e41a2003-01-30 22:06:23 +00002874static PyObject *
2875tzinfo_reduce(PyObject *self)
2876{
2877 PyObject *args, *state, *tmp;
2878 PyObject *getinitargs, *getstate;
Tim Peters2a799bf2002-12-16 20:18:38 +00002879
Guido van Rossum177e41a2003-01-30 22:06:23 +00002880 tmp = PyTuple_New(0);
2881 if (tmp == NULL)
2882 return NULL;
Tim Peters2a799bf2002-12-16 20:18:38 +00002883
Guido van Rossum177e41a2003-01-30 22:06:23 +00002884 getinitargs = PyObject_GetAttrString(self, "__getinitargs__");
2885 if (getinitargs != NULL) {
2886 args = PyObject_CallObject(getinitargs, tmp);
2887 Py_DECREF(getinitargs);
2888 if (args == NULL) {
2889 Py_DECREF(tmp);
2890 return NULL;
2891 }
2892 }
2893 else {
2894 PyErr_Clear();
2895 args = tmp;
2896 Py_INCREF(args);
2897 }
2898
2899 getstate = PyObject_GetAttrString(self, "__getstate__");
2900 if (getstate != NULL) {
2901 state = PyObject_CallObject(getstate, tmp);
2902 Py_DECREF(getstate);
2903 if (state == NULL) {
2904 Py_DECREF(args);
2905 Py_DECREF(tmp);
2906 return NULL;
2907 }
2908 }
2909 else {
2910 PyObject **dictptr;
2911 PyErr_Clear();
2912 state = Py_None;
2913 dictptr = _PyObject_GetDictPtr(self);
2914 if (dictptr && *dictptr && PyDict_Size(*dictptr))
2915 state = *dictptr;
2916 Py_INCREF(state);
2917 }
2918
2919 Py_DECREF(tmp);
2920
2921 if (state == Py_None) {
2922 Py_DECREF(state);
Christian Heimes90aa7642007-12-19 02:45:37 +00002923 return Py_BuildValue("(ON)", Py_TYPE(self), args);
Guido van Rossum177e41a2003-01-30 22:06:23 +00002924 }
2925 else
Christian Heimes90aa7642007-12-19 02:45:37 +00002926 return Py_BuildValue("(ONN)", Py_TYPE(self), args, state);
Guido van Rossum177e41a2003-01-30 22:06:23 +00002927}
Tim Peters2a799bf2002-12-16 20:18:38 +00002928
2929static PyMethodDef tzinfo_methods[] = {
Guido van Rossum177e41a2003-01-30 22:06:23 +00002930
Tim Peters2a799bf2002-12-16 20:18:38 +00002931 {"tzname", (PyCFunction)tzinfo_tzname, METH_O,
2932 PyDoc_STR("datetime -> string name of time zone.")},
2933
2934 {"utcoffset", (PyCFunction)tzinfo_utcoffset, METH_O,
2935 PyDoc_STR("datetime -> minutes east of UTC (negative for "
2936 "west of UTC).")},
2937
2938 {"dst", (PyCFunction)tzinfo_dst, METH_O,
2939 PyDoc_STR("datetime -> DST offset in minutes east of UTC.")},
2940
Tim Peters52dcce22003-01-23 16:36:11 +00002941 {"fromutc", (PyCFunction)tzinfo_fromutc, METH_O,
2942 PyDoc_STR("datetime in UTC -> datetime in local time.")},
2943
Guido van Rossum177e41a2003-01-30 22:06:23 +00002944 {"__reduce__", (PyCFunction)tzinfo_reduce, METH_NOARGS,
2945 PyDoc_STR("-> (cls, state)")},
2946
Tim Peters2a799bf2002-12-16 20:18:38 +00002947 {NULL, NULL}
2948};
2949
2950static char tzinfo_doc[] =
2951PyDoc_STR("Abstract base class for time zone info objects.");
2952
Neal Norwitz227b5332006-03-22 09:28:35 +00002953static PyTypeObject PyDateTime_TZInfoType = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00002954 PyVarObject_HEAD_INIT(NULL, 0)
Tim Peters2a799bf2002-12-16 20:18:38 +00002955 "datetime.tzinfo", /* tp_name */
2956 sizeof(PyDateTime_TZInfo), /* tp_basicsize */
2957 0, /* tp_itemsize */
2958 0, /* tp_dealloc */
2959 0, /* tp_print */
2960 0, /* tp_getattr */
2961 0, /* tp_setattr */
2962 0, /* tp_compare */
2963 0, /* tp_repr */
2964 0, /* tp_as_number */
2965 0, /* tp_as_sequence */
2966 0, /* tp_as_mapping */
2967 0, /* tp_hash */
2968 0, /* tp_call */
2969 0, /* tp_str */
2970 PyObject_GenericGetAttr, /* tp_getattro */
2971 0, /* tp_setattro */
2972 0, /* tp_as_buffer */
Guido van Rossum3cf5b1e2006-07-27 21:53:35 +00002973 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
Tim Peters2a799bf2002-12-16 20:18:38 +00002974 tzinfo_doc, /* tp_doc */
2975 0, /* tp_traverse */
2976 0, /* tp_clear */
2977 0, /* tp_richcompare */
2978 0, /* tp_weaklistoffset */
2979 0, /* tp_iter */
2980 0, /* tp_iternext */
2981 tzinfo_methods, /* tp_methods */
2982 0, /* tp_members */
2983 0, /* tp_getset */
2984 0, /* tp_base */
2985 0, /* tp_dict */
2986 0, /* tp_descr_get */
2987 0, /* tp_descr_set */
2988 0, /* tp_dictoffset */
2989 0, /* tp_init */
2990 0, /* tp_alloc */
2991 PyType_GenericNew, /* tp_new */
2992 0, /* tp_free */
2993};
2994
2995/*
Tim Peters37f39822003-01-10 03:49:02 +00002996 * PyDateTime_Time implementation.
Tim Peters2a799bf2002-12-16 20:18:38 +00002997 */
2998
Tim Peters37f39822003-01-10 03:49:02 +00002999/* Accessor properties.
Tim Peters2a799bf2002-12-16 20:18:38 +00003000 */
3001
3002static PyObject *
Tim Peters37f39822003-01-10 03:49:02 +00003003time_hour(PyDateTime_Time *self, void *unused)
Tim Peters2a799bf2002-12-16 20:18:38 +00003004{
Christian Heimes217cfd12007-12-02 14:31:20 +00003005 return PyLong_FromLong(TIME_GET_HOUR(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00003006}
3007
Tim Peters37f39822003-01-10 03:49:02 +00003008static PyObject *
3009time_minute(PyDateTime_Time *self, void *unused)
3010{
Christian Heimes217cfd12007-12-02 14:31:20 +00003011 return PyLong_FromLong(TIME_GET_MINUTE(self));
Tim Peters37f39822003-01-10 03:49:02 +00003012}
3013
3014/* The name time_second conflicted with some platform header file. */
3015static PyObject *
3016py_time_second(PyDateTime_Time *self, void *unused)
3017{
Christian Heimes217cfd12007-12-02 14:31:20 +00003018 return PyLong_FromLong(TIME_GET_SECOND(self));
Tim Peters37f39822003-01-10 03:49:02 +00003019}
3020
3021static PyObject *
3022time_microsecond(PyDateTime_Time *self, void *unused)
3023{
Christian Heimes217cfd12007-12-02 14:31:20 +00003024 return PyLong_FromLong(TIME_GET_MICROSECOND(self));
Tim Peters37f39822003-01-10 03:49:02 +00003025}
3026
3027static PyObject *
3028time_tzinfo(PyDateTime_Time *self, void *unused)
3029{
Tim Petersa032d2e2003-01-11 00:15:54 +00003030 PyObject *result = HASTZINFO(self) ? self->tzinfo : Py_None;
Tim Peters37f39822003-01-10 03:49:02 +00003031 Py_INCREF(result);
3032 return result;
3033}
3034
3035static PyGetSetDef time_getset[] = {
3036 {"hour", (getter)time_hour},
3037 {"minute", (getter)time_minute},
3038 {"second", (getter)py_time_second},
3039 {"microsecond", (getter)time_microsecond},
3040 {"tzinfo", (getter)time_tzinfo},
Tim Peters2a799bf2002-12-16 20:18:38 +00003041 {NULL}
3042};
3043
3044/*
3045 * Constructors.
3046 */
3047
Martin v. Löwis02cbf4a2006-02-27 17:20:04 +00003048static char *time_kws[] = {"hour", "minute", "second", "microsecond",
Tim Peters37f39822003-01-10 03:49:02 +00003049 "tzinfo", NULL};
Tim Peters12bf3392002-12-24 05:41:27 +00003050
Tim Peters2a799bf2002-12-16 20:18:38 +00003051static PyObject *
Tim Peters37f39822003-01-10 03:49:02 +00003052time_new(PyTypeObject *type, PyObject *args, PyObject *kw)
Tim Peters2a799bf2002-12-16 20:18:38 +00003053{
3054 PyObject *self = NULL;
Tim Peters70533e22003-02-01 04:40:04 +00003055 PyObject *state;
Tim Peters2a799bf2002-12-16 20:18:38 +00003056 int hour = 0;
3057 int minute = 0;
3058 int second = 0;
3059 int usecond = 0;
3060 PyObject *tzinfo = Py_None;
3061
Guido van Rossum177e41a2003-01-30 22:06:23 +00003062 /* Check for invocation from pickle with __getstate__ state */
3063 if (PyTuple_GET_SIZE(args) >= 1 &&
3064 PyTuple_GET_SIZE(args) <= 2 &&
Guido van Rossum254348e2007-11-21 19:29:53 +00003065 PyString_Check(state = PyTuple_GET_ITEM(args, 0)) &&
3066 PyString_GET_SIZE(state) == _PyDateTime_TIME_DATASIZE &&
3067 ((unsigned char) (PyString_AS_STRING(state)[0])) < 24)
Guido van Rossum177e41a2003-01-30 22:06:23 +00003068 {
Tim Peters70533e22003-02-01 04:40:04 +00003069 PyDateTime_Time *me;
3070 char aware;
3071
3072 if (PyTuple_GET_SIZE(args) == 2) {
Guido van Rossum177e41a2003-01-30 22:06:23 +00003073 tzinfo = PyTuple_GET_ITEM(args, 1);
Tim Peters70533e22003-02-01 04:40:04 +00003074 if (check_tzinfo_subclass(tzinfo) < 0) {
3075 PyErr_SetString(PyExc_TypeError, "bad "
3076 "tzinfo state arg");
3077 return NULL;
Guido van Rossum177e41a2003-01-30 22:06:23 +00003078 }
3079 }
Tim Peters70533e22003-02-01 04:40:04 +00003080 aware = (char)(tzinfo != Py_None);
Tim Peters604c0132004-06-07 23:04:33 +00003081 me = (PyDateTime_Time *) (type->tp_alloc(type, aware));
Tim Peters70533e22003-02-01 04:40:04 +00003082 if (me != NULL) {
Guido van Rossum254348e2007-11-21 19:29:53 +00003083 char *pdata = PyString_AS_STRING(state);
Tim Peters70533e22003-02-01 04:40:04 +00003084
3085 memcpy(me->data, pdata, _PyDateTime_TIME_DATASIZE);
3086 me->hashcode = -1;
3087 me->hastzinfo = aware;
3088 if (aware) {
3089 Py_INCREF(tzinfo);
3090 me->tzinfo = tzinfo;
3091 }
3092 }
3093 return (PyObject *)me;
Guido van Rossum177e41a2003-01-30 22:06:23 +00003094 }
3095
Tim Peters37f39822003-01-10 03:49:02 +00003096 if (PyArg_ParseTupleAndKeywords(args, kw, "|iiiiO", time_kws,
Tim Peters2a799bf2002-12-16 20:18:38 +00003097 &hour, &minute, &second, &usecond,
3098 &tzinfo)) {
3099 if (check_time_args(hour, minute, second, usecond) < 0)
3100 return NULL;
3101 if (check_tzinfo_subclass(tzinfo) < 0)
3102 return NULL;
Tim Petersa98924a2003-05-17 05:55:19 +00003103 self = new_time_ex(hour, minute, second, usecond, tzinfo,
3104 type);
Tim Peters2a799bf2002-12-16 20:18:38 +00003105 }
3106 return self;
3107}
3108
3109/*
3110 * Destructor.
3111 */
3112
3113static void
Tim Peters37f39822003-01-10 03:49:02 +00003114time_dealloc(PyDateTime_Time *self)
Tim Peters2a799bf2002-12-16 20:18:38 +00003115{
Tim Petersa032d2e2003-01-11 00:15:54 +00003116 if (HASTZINFO(self)) {
Tim Peters37f39822003-01-10 03:49:02 +00003117 Py_XDECREF(self->tzinfo);
Neal Norwitz8e914d92003-01-10 15:29:16 +00003118 }
Christian Heimes90aa7642007-12-19 02:45:37 +00003119 Py_TYPE(self)->tp_free((PyObject *)self);
Tim Peters2a799bf2002-12-16 20:18:38 +00003120}
3121
3122/*
Tim Peters855fe882002-12-22 03:43:39 +00003123 * Indirect access to tzinfo methods.
Tim Peters2a799bf2002-12-16 20:18:38 +00003124 */
3125
Tim Peters2a799bf2002-12-16 20:18:38 +00003126/* These are all METH_NOARGS, so don't need to check the arglist. */
3127static PyObject *
Tim Peters37f39822003-01-10 03:49:02 +00003128time_utcoffset(PyDateTime_Time *self, PyObject *unused) {
Tim Petersa032d2e2003-01-11 00:15:54 +00003129 return offset_as_timedelta(HASTZINFO(self) ? self->tzinfo : Py_None,
Tim Peters37f39822003-01-10 03:49:02 +00003130 "utcoffset", Py_None);
Tim Peters2a799bf2002-12-16 20:18:38 +00003131}
3132
3133static PyObject *
Tim Peters37f39822003-01-10 03:49:02 +00003134time_dst(PyDateTime_Time *self, PyObject *unused) {
Tim Petersa032d2e2003-01-11 00:15:54 +00003135 return offset_as_timedelta(HASTZINFO(self) ? self->tzinfo : Py_None,
Tim Peters37f39822003-01-10 03:49:02 +00003136 "dst", Py_None);
Tim Peters855fe882002-12-22 03:43:39 +00003137}
3138
3139static PyObject *
Tim Peters37f39822003-01-10 03:49:02 +00003140time_tzname(PyDateTime_Time *self, PyObject *unused) {
Tim Petersa032d2e2003-01-11 00:15:54 +00003141 return call_tzname(HASTZINFO(self) ? self->tzinfo : Py_None,
Tim Peters37f39822003-01-10 03:49:02 +00003142 Py_None);
Tim Peters2a799bf2002-12-16 20:18:38 +00003143}
3144
3145/*
Tim Peters37f39822003-01-10 03:49:02 +00003146 * Various ways to turn a time into a string.
Tim Peters2a799bf2002-12-16 20:18:38 +00003147 */
3148
3149static PyObject *
Tim Peters37f39822003-01-10 03:49:02 +00003150time_repr(PyDateTime_Time *self)
Tim Peters2a799bf2002-12-16 20:18:38 +00003151{
Christian Heimes90aa7642007-12-19 02:45:37 +00003152 const char *type_name = Py_TYPE(self)->tp_name;
Tim Peters37f39822003-01-10 03:49:02 +00003153 int h = TIME_GET_HOUR(self);
3154 int m = TIME_GET_MINUTE(self);
3155 int s = TIME_GET_SECOND(self);
3156 int us = TIME_GET_MICROSECOND(self);
3157 PyObject *result = NULL;
Tim Peters2a799bf2002-12-16 20:18:38 +00003158
Tim Peters37f39822003-01-10 03:49:02 +00003159 if (us)
Walter Dörwald7569dfe2007-05-19 21:49:49 +00003160 result = PyUnicode_FromFormat("%s(%d, %d, %d, %d)",
3161 type_name, h, m, s, us);
Tim Peters37f39822003-01-10 03:49:02 +00003162 else if (s)
Walter Dörwald7569dfe2007-05-19 21:49:49 +00003163 result = PyUnicode_FromFormat("%s(%d, %d, %d)",
3164 type_name, h, m, s);
Tim Peters37f39822003-01-10 03:49:02 +00003165 else
Walter Dörwald7569dfe2007-05-19 21:49:49 +00003166 result = PyUnicode_FromFormat("%s(%d, %d)", type_name, h, m);
Tim Petersa032d2e2003-01-11 00:15:54 +00003167 if (result != NULL && HASTZINFO(self))
Tim Peters37f39822003-01-10 03:49:02 +00003168 result = append_keyword_tzinfo(result, self->tzinfo);
3169 return result;
Tim Peters2a799bf2002-12-16 20:18:38 +00003170}
3171
Tim Peters37f39822003-01-10 03:49:02 +00003172static PyObject *
3173time_str(PyDateTime_Time *self)
3174{
3175 return PyObject_CallMethod((PyObject *)self, "isoformat", "()");
3176}
Tim Peters2a799bf2002-12-16 20:18:38 +00003177
3178static PyObject *
Thomas Wouterscf297e42007-02-23 15:07:44 +00003179time_isoformat(PyDateTime_Time *self, PyObject *unused)
Tim Peters2a799bf2002-12-16 20:18:38 +00003180{
3181 char buf[100];
Tim Peters37f39822003-01-10 03:49:02 +00003182 PyObject *result;
Walter Dörwaldbafa1372007-05-31 17:50:48 +00003183 int us = TIME_GET_MICROSECOND(self);;
Tim Peters2a799bf2002-12-16 20:18:38 +00003184
Walter Dörwaldbafa1372007-05-31 17:50:48 +00003185 if (us)
3186 result = PyUnicode_FromFormat("%02d:%02d:%02d.%06d",
3187 TIME_GET_HOUR(self),
3188 TIME_GET_MINUTE(self),
3189 TIME_GET_SECOND(self),
3190 us);
3191 else
3192 result = PyUnicode_FromFormat("%02d:%02d:%02d",
3193 TIME_GET_HOUR(self),
3194 TIME_GET_MINUTE(self),
3195 TIME_GET_SECOND(self));
Tim Peters37f39822003-01-10 03:49:02 +00003196
Tim Petersa032d2e2003-01-11 00:15:54 +00003197 if (result == NULL || ! HASTZINFO(self) || self->tzinfo == Py_None)
Tim Peters2a799bf2002-12-16 20:18:38 +00003198 return result;
3199
3200 /* We need to append the UTC offset. */
3201 if (format_utcoffset(buf, sizeof(buf), ":", self->tzinfo,
Tim Petersbad8ff02002-12-30 20:52:32 +00003202 Py_None) < 0) {
Tim Peters2a799bf2002-12-16 20:18:38 +00003203 Py_DECREF(result);
3204 return NULL;
3205 }
Walter Dörwaldbafa1372007-05-31 17:50:48 +00003206 PyUnicode_AppendAndDel(&result, PyUnicode_FromString(buf));
Tim Peters2a799bf2002-12-16 20:18:38 +00003207 return result;
3208}
3209
Tim Peters37f39822003-01-10 03:49:02 +00003210static PyObject *
3211time_strftime(PyDateTime_Time *self, PyObject *args, PyObject *kw)
3212{
3213 PyObject *result;
3214 PyObject *format;
3215 PyObject *tuple;
Martin v. Löwis02cbf4a2006-02-27 17:20:04 +00003216 static char *keywords[] = {"format", NULL};
Tim Peters37f39822003-01-10 03:49:02 +00003217
Guido van Rossum98297ee2007-11-06 21:34:58 +00003218 if (! PyArg_ParseTupleAndKeywords(args, kw, "U:strftime", keywords,
Guido van Rossumbce56a62007-05-10 18:04:33 +00003219 &format))
Tim Peters37f39822003-01-10 03:49:02 +00003220 return NULL;
3221
3222 /* Python's strftime does insane things with the year part of the
3223 * timetuple. The year is forced to (the otherwise nonsensical)
3224 * 1900 to worm around that.
3225 */
3226 tuple = Py_BuildValue("iiiiiiiii",
Brett Cannond1080a32004-03-02 04:38:10 +00003227 1900, 1, 1, /* year, month, day */
Tim Peters37f39822003-01-10 03:49:02 +00003228 TIME_GET_HOUR(self),
3229 TIME_GET_MINUTE(self),
3230 TIME_GET_SECOND(self),
Brett Cannond1080a32004-03-02 04:38:10 +00003231 0, 1, -1); /* weekday, daynum, dst */
Tim Peters37f39822003-01-10 03:49:02 +00003232 if (tuple == NULL)
3233 return NULL;
3234 assert(PyTuple_Size(tuple) == 9);
3235 result = wrap_strftime((PyObject *)self, format, tuple, Py_None);
3236 Py_DECREF(tuple);
3237 return result;
3238}
Tim Peters2a799bf2002-12-16 20:18:38 +00003239
3240/*
3241 * Miscellaneous methods.
3242 */
3243
Tim Peters37f39822003-01-10 03:49:02 +00003244static PyObject *
Guido van Rossum19960592006-08-24 17:29:38 +00003245time_richcompare(PyObject *self, PyObject *other, int op)
Tim Peters37f39822003-01-10 03:49:02 +00003246{
3247 int diff;
3248 naivety n1, n2;
3249 int offset1, offset2;
3250
3251 if (! PyTime_Check(other)) {
Guido van Rossum19960592006-08-24 17:29:38 +00003252 Py_INCREF(Py_NotImplemented);
3253 return Py_NotImplemented;
Tim Peters37f39822003-01-10 03:49:02 +00003254 }
Guido van Rossum19960592006-08-24 17:29:38 +00003255 if (classify_two_utcoffsets(self, &offset1, &n1, Py_None,
3256 other, &offset2, &n2, Py_None) < 0)
Tim Peters37f39822003-01-10 03:49:02 +00003257 return NULL;
3258 assert(n1 != OFFSET_UNKNOWN && n2 != OFFSET_UNKNOWN);
3259 /* If they're both naive, or both aware and have the same offsets,
3260 * we get off cheap. Note that if they're both naive, offset1 ==
3261 * offset2 == 0 at this point.
3262 */
3263 if (n1 == n2 && offset1 == offset2) {
Guido van Rossum19960592006-08-24 17:29:38 +00003264 diff = memcmp(((PyDateTime_Time *)self)->data,
3265 ((PyDateTime_Time *)other)->data,
Tim Peters37f39822003-01-10 03:49:02 +00003266 _PyDateTime_TIME_DATASIZE);
3267 return diff_to_bool(diff, op);
3268 }
3269
3270 if (n1 == OFFSET_AWARE && n2 == OFFSET_AWARE) {
3271 assert(offset1 != offset2); /* else last "if" handled it */
3272 /* Convert everything except microseconds to seconds. These
3273 * can't overflow (no more than the # of seconds in 2 days).
3274 */
3275 offset1 = TIME_GET_HOUR(self) * 3600 +
3276 (TIME_GET_MINUTE(self) - offset1) * 60 +
3277 TIME_GET_SECOND(self);
3278 offset2 = TIME_GET_HOUR(other) * 3600 +
3279 (TIME_GET_MINUTE(other) - offset2) * 60 +
3280 TIME_GET_SECOND(other);
3281 diff = offset1 - offset2;
3282 if (diff == 0)
3283 diff = TIME_GET_MICROSECOND(self) -
3284 TIME_GET_MICROSECOND(other);
3285 return diff_to_bool(diff, op);
3286 }
3287
3288 assert(n1 != n2);
3289 PyErr_SetString(PyExc_TypeError,
3290 "can't compare offset-naive and "
3291 "offset-aware times");
3292 return NULL;
3293}
3294
3295static long
3296time_hash(PyDateTime_Time *self)
3297{
3298 if (self->hashcode == -1) {
3299 naivety n;
3300 int offset;
3301 PyObject *temp;
3302
3303 n = classify_utcoffset((PyObject *)self, Py_None, &offset);
3304 assert(n != OFFSET_UNKNOWN);
3305 if (n == OFFSET_ERROR)
3306 return -1;
3307
3308 /* Reduce this to a hash of another object. */
Guido van Rossumfd53fd62007-08-24 04:05:13 +00003309 if (offset == 0) {
3310 self->hashcode = generic_hash(
3311 (unsigned char *)self->data, _PyDateTime_TIME_DATASIZE);
3312 return self->hashcode;
3313 }
Tim Peters37f39822003-01-10 03:49:02 +00003314 else {
3315 int hour;
3316 int minute;
3317
3318 assert(n == OFFSET_AWARE);
Tim Petersa032d2e2003-01-11 00:15:54 +00003319 assert(HASTZINFO(self));
Tim Peters37f39822003-01-10 03:49:02 +00003320 hour = divmod(TIME_GET_HOUR(self) * 60 +
3321 TIME_GET_MINUTE(self) - offset,
3322 60,
3323 &minute);
3324 if (0 <= hour && hour < 24)
3325 temp = new_time(hour, minute,
3326 TIME_GET_SECOND(self),
3327 TIME_GET_MICROSECOND(self),
3328 Py_None);
3329 else
3330 temp = Py_BuildValue("iiii",
3331 hour, minute,
3332 TIME_GET_SECOND(self),
3333 TIME_GET_MICROSECOND(self));
3334 }
3335 if (temp != NULL) {
3336 self->hashcode = PyObject_Hash(temp);
3337 Py_DECREF(temp);
3338 }
3339 }
3340 return self->hashcode;
3341}
Tim Peters2a799bf2002-12-16 20:18:38 +00003342
Tim Peters12bf3392002-12-24 05:41:27 +00003343static PyObject *
Tim Peters37f39822003-01-10 03:49:02 +00003344time_replace(PyDateTime_Time *self, PyObject *args, PyObject *kw)
Tim Peters12bf3392002-12-24 05:41:27 +00003345{
3346 PyObject *clone;
3347 PyObject *tuple;
3348 int hh = TIME_GET_HOUR(self);
3349 int mm = TIME_GET_MINUTE(self);
3350 int ss = TIME_GET_SECOND(self);
3351 int us = TIME_GET_MICROSECOND(self);
Tim Petersa032d2e2003-01-11 00:15:54 +00003352 PyObject *tzinfo = HASTZINFO(self) ? self->tzinfo : Py_None;
Tim Peters12bf3392002-12-24 05:41:27 +00003353
3354 if (! PyArg_ParseTupleAndKeywords(args, kw, "|iiiiO:replace",
Tim Peters37f39822003-01-10 03:49:02 +00003355 time_kws,
Tim Peters12bf3392002-12-24 05:41:27 +00003356 &hh, &mm, &ss, &us, &tzinfo))
3357 return NULL;
3358 tuple = Py_BuildValue("iiiiO", hh, mm, ss, us, tzinfo);
3359 if (tuple == NULL)
3360 return NULL;
Christian Heimes90aa7642007-12-19 02:45:37 +00003361 clone = time_new(Py_TYPE(self), tuple, NULL);
Tim Peters12bf3392002-12-24 05:41:27 +00003362 Py_DECREF(tuple);
3363 return clone;
3364}
3365
Tim Peters2a799bf2002-12-16 20:18:38 +00003366static int
Jack Diederich4dafcc42006-11-28 19:15:13 +00003367time_bool(PyDateTime_Time *self)
Tim Peters2a799bf2002-12-16 20:18:38 +00003368{
3369 int offset;
3370 int none;
3371
3372 if (TIME_GET_SECOND(self) || TIME_GET_MICROSECOND(self)) {
3373 /* Since utcoffset is in whole minutes, nothing can
3374 * alter the conclusion that this is nonzero.
3375 */
3376 return 1;
3377 }
3378 offset = 0;
Tim Petersa032d2e2003-01-11 00:15:54 +00003379 if (HASTZINFO(self) && self->tzinfo != Py_None) {
Tim Petersbad8ff02002-12-30 20:52:32 +00003380 offset = call_utcoffset(self->tzinfo, Py_None, &none);
Tim Peters2a799bf2002-12-16 20:18:38 +00003381 if (offset == -1 && PyErr_Occurred())
3382 return -1;
3383 }
3384 return (TIME_GET_MINUTE(self) - offset + TIME_GET_HOUR(self)*60) != 0;
3385}
3386
Tim Peters371935f2003-02-01 01:52:50 +00003387/* Pickle support, a simple use of __reduce__. */
Tim Peters2a799bf2002-12-16 20:18:38 +00003388
Tim Peters33e0f382003-01-10 02:05:14 +00003389/* Let basestate be the non-tzinfo data string.
Tim Peters2a799bf2002-12-16 20:18:38 +00003390 * If tzinfo is None, this returns (basestate,), else (basestate, tzinfo).
3391 * So it's a tuple in any (non-error) case.
Tim Petersb57f8f02003-02-01 02:54:15 +00003392 * __getstate__ isn't exposed.
Tim Peters2a799bf2002-12-16 20:18:38 +00003393 */
3394static PyObject *
Tim Peters37f39822003-01-10 03:49:02 +00003395time_getstate(PyDateTime_Time *self)
Tim Peters2a799bf2002-12-16 20:18:38 +00003396{
3397 PyObject *basestate;
3398 PyObject *result = NULL;
3399
Guido van Rossum254348e2007-11-21 19:29:53 +00003400 basestate = PyString_FromStringAndSize((char *)self->data,
Tim Peters33e0f382003-01-10 02:05:14 +00003401 _PyDateTime_TIME_DATASIZE);
Tim Peters2a799bf2002-12-16 20:18:38 +00003402 if (basestate != NULL) {
Tim Petersa032d2e2003-01-11 00:15:54 +00003403 if (! HASTZINFO(self) || self->tzinfo == Py_None)
Raymond Hettinger8ae46892003-10-12 19:09:37 +00003404 result = PyTuple_Pack(1, basestate);
Tim Peters2a799bf2002-12-16 20:18:38 +00003405 else
Raymond Hettinger8ae46892003-10-12 19:09:37 +00003406 result = PyTuple_Pack(2, basestate, self->tzinfo);
Tim Peters2a799bf2002-12-16 20:18:38 +00003407 Py_DECREF(basestate);
3408 }
3409 return result;
3410}
3411
3412static PyObject *
Guido van Rossum177e41a2003-01-30 22:06:23 +00003413time_reduce(PyDateTime_Time *self, PyObject *arg)
Tim Peters2a799bf2002-12-16 20:18:38 +00003414{
Christian Heimes90aa7642007-12-19 02:45:37 +00003415 return Py_BuildValue("(ON)", Py_TYPE(self), time_getstate(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00003416}
3417
Tim Peters37f39822003-01-10 03:49:02 +00003418static PyMethodDef time_methods[] = {
Guido van Rossum177e41a2003-01-30 22:06:23 +00003419
Thomas Wouterscf297e42007-02-23 15:07:44 +00003420 {"isoformat", (PyCFunction)time_isoformat, METH_NOARGS,
Tim Peters2a799bf2002-12-16 20:18:38 +00003421 PyDoc_STR("Return string in ISO 8601 format, HH:MM:SS[.mmmmmm]"
3422 "[+HH:MM].")},
3423
Guido van Rossumd59da4b2007-05-22 18:11:13 +00003424 {"strftime", (PyCFunction)time_strftime, METH_VARARGS | METH_KEYWORDS,
Tim Peters37f39822003-01-10 03:49:02 +00003425 PyDoc_STR("format -> strftime() style string.")},
3426
Eric Smith8fd3eba2008-02-17 19:48:00 +00003427 {"__format__", (PyCFunction)date_format, METH_VARARGS,
Eric Smith1ba31142007-09-11 18:06:02 +00003428 PyDoc_STR("Formats self with strftime.")},
3429
Tim Peters37f39822003-01-10 03:49:02 +00003430 {"utcoffset", (PyCFunction)time_utcoffset, METH_NOARGS,
Tim Peters2a799bf2002-12-16 20:18:38 +00003431 PyDoc_STR("Return self.tzinfo.utcoffset(self).")},
3432
Tim Peters37f39822003-01-10 03:49:02 +00003433 {"tzname", (PyCFunction)time_tzname, METH_NOARGS,
Tim Peters2a799bf2002-12-16 20:18:38 +00003434 PyDoc_STR("Return self.tzinfo.tzname(self).")},
3435
Tim Peters37f39822003-01-10 03:49:02 +00003436 {"dst", (PyCFunction)time_dst, METH_NOARGS,
Tim Peters2a799bf2002-12-16 20:18:38 +00003437 PyDoc_STR("Return self.tzinfo.dst(self).")},
3438
Guido van Rossumd59da4b2007-05-22 18:11:13 +00003439 {"replace", (PyCFunction)time_replace, METH_VARARGS | METH_KEYWORDS,
Tim Peters37f39822003-01-10 03:49:02 +00003440 PyDoc_STR("Return time with new specified fields.")},
Tim Peters12bf3392002-12-24 05:41:27 +00003441
Guido van Rossum177e41a2003-01-30 22:06:23 +00003442 {"__reduce__", (PyCFunction)time_reduce, METH_NOARGS,
3443 PyDoc_STR("__reduce__() -> (cls, state)")},
3444
Tim Peters2a799bf2002-12-16 20:18:38 +00003445 {NULL, NULL}
Tim Peters2a799bf2002-12-16 20:18:38 +00003446};
3447
Tim Peters37f39822003-01-10 03:49:02 +00003448static char time_doc[] =
Raymond Hettinger3a4231d2004-12-19 20:13:24 +00003449PyDoc_STR("time([hour[, minute[, second[, microsecond[, tzinfo]]]]]) --> a time object\n\
3450\n\
3451All arguments are optional. tzinfo may be None, or an instance of\n\
3452a tzinfo subclass. The remaining arguments may be ints or longs.\n");
Tim Peters2a799bf2002-12-16 20:18:38 +00003453
Tim Peters37f39822003-01-10 03:49:02 +00003454static PyNumberMethods time_as_number = {
Tim Peters2a799bf2002-12-16 20:18:38 +00003455 0, /* nb_add */
3456 0, /* nb_subtract */
3457 0, /* nb_multiply */
Tim Peters2a799bf2002-12-16 20:18:38 +00003458 0, /* nb_remainder */
3459 0, /* nb_divmod */
3460 0, /* nb_power */
3461 0, /* nb_negative */
3462 0, /* nb_positive */
3463 0, /* nb_absolute */
Jack Diederich4dafcc42006-11-28 19:15:13 +00003464 (inquiry)time_bool, /* nb_bool */
Tim Peters2a799bf2002-12-16 20:18:38 +00003465};
3466
Neal Norwitz227b5332006-03-22 09:28:35 +00003467static PyTypeObject PyDateTime_TimeType = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00003468 PyVarObject_HEAD_INIT(NULL, 0)
Tim Peters0bf60bd2003-01-08 20:40:01 +00003469 "datetime.time", /* tp_name */
Tim Peters37f39822003-01-10 03:49:02 +00003470 sizeof(PyDateTime_Time), /* tp_basicsize */
Tim Peters2a799bf2002-12-16 20:18:38 +00003471 0, /* tp_itemsize */
Tim Peters37f39822003-01-10 03:49:02 +00003472 (destructor)time_dealloc, /* tp_dealloc */
Tim Peters2a799bf2002-12-16 20:18:38 +00003473 0, /* tp_print */
3474 0, /* tp_getattr */
3475 0, /* tp_setattr */
3476 0, /* tp_compare */
Tim Peters37f39822003-01-10 03:49:02 +00003477 (reprfunc)time_repr, /* tp_repr */
3478 &time_as_number, /* tp_as_number */
Tim Peters2a799bf2002-12-16 20:18:38 +00003479 0, /* tp_as_sequence */
3480 0, /* tp_as_mapping */
Tim Peters37f39822003-01-10 03:49:02 +00003481 (hashfunc)time_hash, /* tp_hash */
Tim Peters2a799bf2002-12-16 20:18:38 +00003482 0, /* tp_call */
Tim Peters37f39822003-01-10 03:49:02 +00003483 (reprfunc)time_str, /* tp_str */
Tim Peters2a799bf2002-12-16 20:18:38 +00003484 PyObject_GenericGetAttr, /* tp_getattro */
3485 0, /* tp_setattro */
3486 0, /* tp_as_buffer */
Guido van Rossum3cf5b1e2006-07-27 21:53:35 +00003487 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
Tim Peters37f39822003-01-10 03:49:02 +00003488 time_doc, /* tp_doc */
Tim Peters2a799bf2002-12-16 20:18:38 +00003489 0, /* tp_traverse */
3490 0, /* tp_clear */
Guido van Rossum19960592006-08-24 17:29:38 +00003491 time_richcompare, /* tp_richcompare */
Tim Peters2a799bf2002-12-16 20:18:38 +00003492 0, /* tp_weaklistoffset */
3493 0, /* tp_iter */
3494 0, /* tp_iternext */
Tim Peters37f39822003-01-10 03:49:02 +00003495 time_methods, /* tp_methods */
Tim Peters2a799bf2002-12-16 20:18:38 +00003496 0, /* tp_members */
Tim Peters37f39822003-01-10 03:49:02 +00003497 time_getset, /* tp_getset */
3498 0, /* tp_base */
Tim Peters2a799bf2002-12-16 20:18:38 +00003499 0, /* tp_dict */
3500 0, /* tp_descr_get */
3501 0, /* tp_descr_set */
3502 0, /* tp_dictoffset */
3503 0, /* tp_init */
Tim Petersa98924a2003-05-17 05:55:19 +00003504 time_alloc, /* tp_alloc */
Tim Peters37f39822003-01-10 03:49:02 +00003505 time_new, /* tp_new */
Tim Peters4c530132003-05-16 22:44:06 +00003506 0, /* tp_free */
Tim Peters2a799bf2002-12-16 20:18:38 +00003507};
3508
3509/*
Tim Petersa9bc1682003-01-11 03:39:11 +00003510 * PyDateTime_DateTime implementation.
Tim Peters2a799bf2002-12-16 20:18:38 +00003511 */
3512
Tim Petersa9bc1682003-01-11 03:39:11 +00003513/* Accessor properties. Properties for day, month, and year are inherited
3514 * from date.
Tim Peters2a799bf2002-12-16 20:18:38 +00003515 */
3516
3517static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00003518datetime_hour(PyDateTime_DateTime *self, void *unused)
Tim Peters2a799bf2002-12-16 20:18:38 +00003519{
Christian Heimes217cfd12007-12-02 14:31:20 +00003520 return PyLong_FromLong(DATE_GET_HOUR(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00003521}
3522
Tim Petersa9bc1682003-01-11 03:39:11 +00003523static PyObject *
3524datetime_minute(PyDateTime_DateTime *self, void *unused)
3525{
Christian Heimes217cfd12007-12-02 14:31:20 +00003526 return PyLong_FromLong(DATE_GET_MINUTE(self));
Tim Petersa9bc1682003-01-11 03:39:11 +00003527}
3528
3529static PyObject *
3530datetime_second(PyDateTime_DateTime *self, void *unused)
3531{
Christian Heimes217cfd12007-12-02 14:31:20 +00003532 return PyLong_FromLong(DATE_GET_SECOND(self));
Tim Petersa9bc1682003-01-11 03:39:11 +00003533}
3534
3535static PyObject *
3536datetime_microsecond(PyDateTime_DateTime *self, void *unused)
3537{
Christian Heimes217cfd12007-12-02 14:31:20 +00003538 return PyLong_FromLong(DATE_GET_MICROSECOND(self));
Tim Petersa9bc1682003-01-11 03:39:11 +00003539}
3540
3541static PyObject *
3542datetime_tzinfo(PyDateTime_DateTime *self, void *unused)
3543{
3544 PyObject *result = HASTZINFO(self) ? self->tzinfo : Py_None;
3545 Py_INCREF(result);
3546 return result;
3547}
3548
3549static PyGetSetDef datetime_getset[] = {
3550 {"hour", (getter)datetime_hour},
3551 {"minute", (getter)datetime_minute},
3552 {"second", (getter)datetime_second},
3553 {"microsecond", (getter)datetime_microsecond},
3554 {"tzinfo", (getter)datetime_tzinfo},
Tim Peters2a799bf2002-12-16 20:18:38 +00003555 {NULL}
3556};
3557
3558/*
3559 * Constructors.
Tim Peters2a799bf2002-12-16 20:18:38 +00003560 */
3561
Martin v. Löwis02cbf4a2006-02-27 17:20:04 +00003562static char *datetime_kws[] = {
Tim Peters12bf3392002-12-24 05:41:27 +00003563 "year", "month", "day", "hour", "minute", "second",
3564 "microsecond", "tzinfo", NULL
3565};
3566
Tim Peters2a799bf2002-12-16 20:18:38 +00003567static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00003568datetime_new(PyTypeObject *type, PyObject *args, PyObject *kw)
Tim Peters2a799bf2002-12-16 20:18:38 +00003569{
3570 PyObject *self = NULL;
Tim Peters70533e22003-02-01 04:40:04 +00003571 PyObject *state;
Tim Peters2a799bf2002-12-16 20:18:38 +00003572 int year;
3573 int month;
3574 int day;
3575 int hour = 0;
3576 int minute = 0;
3577 int second = 0;
3578 int usecond = 0;
3579 PyObject *tzinfo = Py_None;
3580
Guido van Rossum177e41a2003-01-30 22:06:23 +00003581 /* Check for invocation from pickle with __getstate__ state */
3582 if (PyTuple_GET_SIZE(args) >= 1 &&
3583 PyTuple_GET_SIZE(args) <= 2 &&
Guido van Rossum254348e2007-11-21 19:29:53 +00003584 PyString_Check(state = PyTuple_GET_ITEM(args, 0)) &&
3585 PyString_GET_SIZE(state) == _PyDateTime_DATETIME_DATASIZE &&
3586 MONTH_IS_SANE(PyString_AS_STRING(state)[2]))
Guido van Rossum177e41a2003-01-30 22:06:23 +00003587 {
Tim Peters70533e22003-02-01 04:40:04 +00003588 PyDateTime_DateTime *me;
3589 char aware;
3590
3591 if (PyTuple_GET_SIZE(args) == 2) {
Guido van Rossum177e41a2003-01-30 22:06:23 +00003592 tzinfo = PyTuple_GET_ITEM(args, 1);
Tim Peters70533e22003-02-01 04:40:04 +00003593 if (check_tzinfo_subclass(tzinfo) < 0) {
3594 PyErr_SetString(PyExc_TypeError, "bad "
3595 "tzinfo state arg");
3596 return NULL;
Guido van Rossum177e41a2003-01-30 22:06:23 +00003597 }
3598 }
Tim Peters70533e22003-02-01 04:40:04 +00003599 aware = (char)(tzinfo != Py_None);
Tim Peters604c0132004-06-07 23:04:33 +00003600 me = (PyDateTime_DateTime *) (type->tp_alloc(type , aware));
Tim Peters70533e22003-02-01 04:40:04 +00003601 if (me != NULL) {
Guido van Rossum254348e2007-11-21 19:29:53 +00003602 char *pdata = PyString_AS_STRING(state);
Tim Peters70533e22003-02-01 04:40:04 +00003603
3604 memcpy(me->data, pdata, _PyDateTime_DATETIME_DATASIZE);
3605 me->hashcode = -1;
3606 me->hastzinfo = aware;
3607 if (aware) {
3608 Py_INCREF(tzinfo);
3609 me->tzinfo = tzinfo;
3610 }
3611 }
3612 return (PyObject *)me;
Guido van Rossum177e41a2003-01-30 22:06:23 +00003613 }
3614
Tim Petersa9bc1682003-01-11 03:39:11 +00003615 if (PyArg_ParseTupleAndKeywords(args, kw, "iii|iiiiO", datetime_kws,
Tim Peters2a799bf2002-12-16 20:18:38 +00003616 &year, &month, &day, &hour, &minute,
3617 &second, &usecond, &tzinfo)) {
3618 if (check_date_args(year, month, day) < 0)
3619 return NULL;
3620 if (check_time_args(hour, minute, second, usecond) < 0)
3621 return NULL;
3622 if (check_tzinfo_subclass(tzinfo) < 0)
3623 return NULL;
Tim Petersa98924a2003-05-17 05:55:19 +00003624 self = new_datetime_ex(year, month, day,
3625 hour, minute, second, usecond,
3626 tzinfo, type);
Tim Peters2a799bf2002-12-16 20:18:38 +00003627 }
3628 return self;
3629}
3630
Tim Petersa9bc1682003-01-11 03:39:11 +00003631/* TM_FUNC is the shared type of localtime() and gmtime(). */
3632typedef struct tm *(*TM_FUNC)(const time_t *timer);
3633
3634/* Internal helper.
3635 * Build datetime from a time_t and a distinct count of microseconds.
3636 * Pass localtime or gmtime for f, to control the interpretation of timet.
3637 */
3638static PyObject *
3639datetime_from_timet_and_us(PyObject *cls, TM_FUNC f, time_t timet, int us,
3640 PyObject *tzinfo)
3641{
3642 struct tm *tm;
3643 PyObject *result = NULL;
3644
3645 tm = f(&timet);
3646 if (tm) {
3647 /* The platform localtime/gmtime may insert leap seconds,
3648 * indicated by tm->tm_sec > 59. We don't care about them,
3649 * except to the extent that passing them on to the datetime
3650 * constructor would raise ValueError for a reason that
3651 * made no sense to the user.
3652 */
3653 if (tm->tm_sec > 59)
3654 tm->tm_sec = 59;
3655 result = PyObject_CallFunction(cls, "iiiiiiiO",
3656 tm->tm_year + 1900,
3657 tm->tm_mon + 1,
3658 tm->tm_mday,
3659 tm->tm_hour,
3660 tm->tm_min,
3661 tm->tm_sec,
3662 us,
3663 tzinfo);
3664 }
3665 else
3666 PyErr_SetString(PyExc_ValueError,
3667 "timestamp out of range for "
3668 "platform localtime()/gmtime() function");
3669 return result;
3670}
3671
3672/* Internal helper.
3673 * Build datetime from a Python timestamp. Pass localtime or gmtime for f,
3674 * to control the interpretation of the timestamp. Since a double doesn't
3675 * have enough bits to cover a datetime's full range of precision, it's
3676 * better to call datetime_from_timet_and_us provided you have a way
3677 * to get that much precision (e.g., C time() isn't good enough).
3678 */
3679static PyObject *
3680datetime_from_timestamp(PyObject *cls, TM_FUNC f, double timestamp,
3681 PyObject *tzinfo)
3682{
Tim Peters1b6f7a92004-06-20 02:50:16 +00003683 time_t timet;
3684 double fraction;
3685 int us;
Tim Petersa9bc1682003-01-11 03:39:11 +00003686
Tim Peters1b6f7a92004-06-20 02:50:16 +00003687 timet = _PyTime_DoubleToTimet(timestamp);
3688 if (timet == (time_t)-1 && PyErr_Occurred())
3689 return NULL;
3690 fraction = timestamp - (double)timet;
3691 us = (int)round_to_long(fraction * 1e6);
Guido van Rossumd8faa362007-04-27 19:54:29 +00003692 if (us < 0) {
3693 /* Truncation towards zero is not what we wanted
3694 for negative numbers (Python's mod semantics) */
3695 timet -= 1;
3696 us += 1000000;
3697 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00003698 /* If timestamp is less than one microsecond smaller than a
3699 * full second, round up. Otherwise, ValueErrors are raised
3700 * for some floats. */
3701 if (us == 1000000) {
3702 timet += 1;
3703 us = 0;
3704 }
Tim Petersa9bc1682003-01-11 03:39:11 +00003705 return datetime_from_timet_and_us(cls, f, timet, us, tzinfo);
3706}
3707
3708/* Internal helper.
3709 * Build most accurate possible datetime for current time. Pass localtime or
3710 * gmtime for f as appropriate.
3711 */
3712static PyObject *
3713datetime_best_possible(PyObject *cls, TM_FUNC f, PyObject *tzinfo)
3714{
3715#ifdef HAVE_GETTIMEOFDAY
3716 struct timeval t;
3717
3718#ifdef GETTIMEOFDAY_NO_TZ
3719 gettimeofday(&t);
3720#else
3721 gettimeofday(&t, (struct timezone *)NULL);
3722#endif
3723 return datetime_from_timet_and_us(cls, f, t.tv_sec, (int)t.tv_usec,
3724 tzinfo);
3725
3726#else /* ! HAVE_GETTIMEOFDAY */
3727 /* No flavor of gettimeofday exists on this platform. Python's
3728 * time.time() does a lot of other platform tricks to get the
3729 * best time it can on the platform, and we're not going to do
3730 * better than that (if we could, the better code would belong
3731 * in time.time()!) We're limited by the precision of a double,
3732 * though.
3733 */
3734 PyObject *time;
3735 double dtime;
3736
3737 time = time_time();
3738 if (time == NULL)
3739 return NULL;
3740 dtime = PyFloat_AsDouble(time);
3741 Py_DECREF(time);
3742 if (dtime == -1.0 && PyErr_Occurred())
3743 return NULL;
3744 return datetime_from_timestamp(cls, f, dtime, tzinfo);
3745#endif /* ! HAVE_GETTIMEOFDAY */
3746}
3747
Tim Peters2a799bf2002-12-16 20:18:38 +00003748/* Return best possible local time -- this isn't constrained by the
3749 * precision of a timestamp.
3750 */
3751static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00003752datetime_now(PyObject *cls, PyObject *args, PyObject *kw)
Tim Peters2a799bf2002-12-16 20:18:38 +00003753{
Tim Peters10cadce2003-01-23 19:58:02 +00003754 PyObject *self;
Tim Peters2a799bf2002-12-16 20:18:38 +00003755 PyObject *tzinfo = Py_None;
Martin v. Löwis02cbf4a2006-02-27 17:20:04 +00003756 static char *keywords[] = {"tz", NULL};
Tim Peters2a799bf2002-12-16 20:18:38 +00003757
Tim Peters10cadce2003-01-23 19:58:02 +00003758 if (! PyArg_ParseTupleAndKeywords(args, kw, "|O:now", keywords,
3759 &tzinfo))
3760 return NULL;
3761 if (check_tzinfo_subclass(tzinfo) < 0)
3762 return NULL;
3763
3764 self = datetime_best_possible(cls,
3765 tzinfo == Py_None ? localtime : gmtime,
3766 tzinfo);
3767 if (self != NULL && tzinfo != Py_None) {
3768 /* Convert UTC to tzinfo's zone. */
3769 PyObject *temp = self;
Tim Peters2a44a8d2003-01-23 20:53:10 +00003770 self = PyObject_CallMethod(tzinfo, "fromutc", "O", self);
Tim Peters10cadce2003-01-23 19:58:02 +00003771 Py_DECREF(temp);
Tim Peters2a799bf2002-12-16 20:18:38 +00003772 }
3773 return self;
3774}
3775
Tim Petersa9bc1682003-01-11 03:39:11 +00003776/* Return best possible UTC time -- this isn't constrained by the
3777 * precision of a timestamp.
3778 */
3779static PyObject *
3780datetime_utcnow(PyObject *cls, PyObject *dummy)
3781{
3782 return datetime_best_possible(cls, gmtime, Py_None);
3783}
3784
Tim Peters2a799bf2002-12-16 20:18:38 +00003785/* Return new local datetime from timestamp (Python timestamp -- a double). */
3786static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00003787datetime_fromtimestamp(PyObject *cls, PyObject *args, PyObject *kw)
Tim Peters2a799bf2002-12-16 20:18:38 +00003788{
Tim Peters2a44a8d2003-01-23 20:53:10 +00003789 PyObject *self;
Tim Peters2a799bf2002-12-16 20:18:38 +00003790 double timestamp;
3791 PyObject *tzinfo = Py_None;
Martin v. Löwis02cbf4a2006-02-27 17:20:04 +00003792 static char *keywords[] = {"timestamp", "tz", NULL};
Tim Peters2a799bf2002-12-16 20:18:38 +00003793
Tim Peters2a44a8d2003-01-23 20:53:10 +00003794 if (! PyArg_ParseTupleAndKeywords(args, kw, "d|O:fromtimestamp",
3795 keywords, &timestamp, &tzinfo))
3796 return NULL;
3797 if (check_tzinfo_subclass(tzinfo) < 0)
3798 return NULL;
3799
3800 self = datetime_from_timestamp(cls,
3801 tzinfo == Py_None ? localtime : gmtime,
3802 timestamp,
3803 tzinfo);
3804 if (self != NULL && tzinfo != Py_None) {
3805 /* Convert UTC to tzinfo's zone. */
3806 PyObject *temp = self;
3807 self = PyObject_CallMethod(tzinfo, "fromutc", "O", self);
3808 Py_DECREF(temp);
Tim Peters2a799bf2002-12-16 20:18:38 +00003809 }
3810 return self;
3811}
3812
Tim Petersa9bc1682003-01-11 03:39:11 +00003813/* Return new UTC datetime from timestamp (Python timestamp -- a double). */
3814static PyObject *
3815datetime_utcfromtimestamp(PyObject *cls, PyObject *args)
3816{
3817 double timestamp;
3818 PyObject *result = NULL;
Tim Peters2a799bf2002-12-16 20:18:38 +00003819
Tim Petersa9bc1682003-01-11 03:39:11 +00003820 if (PyArg_ParseTuple(args, "d:utcfromtimestamp", &timestamp))
3821 result = datetime_from_timestamp(cls, gmtime, timestamp,
3822 Py_None);
3823 return result;
3824}
3825
Skip Montanaro0af3ade2005-01-13 04:12:31 +00003826/* Return new datetime from time.strptime(). */
3827static PyObject *
3828datetime_strptime(PyObject *cls, PyObject *args)
3829{
Christian Heimesdd15f6c2008-03-16 00:07:10 +00003830 static PyObject *module = NULL;
3831 PyObject *result = NULL, *obj, *st = NULL, *frac = NULL;
Guido van Rossume8a17aa2007-08-29 17:28:42 +00003832 const Py_UNICODE *string, *format;
Skip Montanaro0af3ade2005-01-13 04:12:31 +00003833
Guido van Rossumfd53fd62007-08-24 04:05:13 +00003834 if (!PyArg_ParseTuple(args, "uu:strptime", &string, &format))
Skip Montanaro0af3ade2005-01-13 04:12:31 +00003835 return NULL;
3836
Christian Heimesdd15f6c2008-03-16 00:07:10 +00003837 if (module == NULL &&
3838 (module = PyImport_ImportModuleNoBlock("_strptime")) == NULL)
Skip Montanaro0af3ade2005-01-13 04:12:31 +00003839 return NULL;
Skip Montanaro0af3ade2005-01-13 04:12:31 +00003840
Christian Heimesdd15f6c2008-03-16 00:07:10 +00003841 /* _strptime._strptime returns a two-element tuple. The first
3842 element is a time.struct_time object. The second is the
3843 microseconds (which are not defined for time.struct_time). */
Mark Dickinsonfc689dd2008-03-16 03:45:34 +00003844 obj = PyObject_CallMethod(module, "_strptime", "uu", string, format);
Skip Montanaro0af3ade2005-01-13 04:12:31 +00003845 if (obj != NULL) {
Christian Heimesdd15f6c2008-03-16 00:07:10 +00003846 int i, good_timetuple = 1;
3847 long int ia[7];
3848 if (PySequence_Check(obj) && PySequence_Size(obj) == 2) {
3849 st = PySequence_GetItem(obj, 0);
3850 frac = PySequence_GetItem(obj, 1);
3851 if (st == NULL || frac == NULL)
3852 good_timetuple = 0;
3853 /* copy y/m/d/h/m/s values out of the
3854 time.struct_time */
3855 if (good_timetuple &&
3856 PySequence_Check(st) &&
3857 PySequence_Size(st) >= 6) {
3858 for (i=0; i < 6; i++) {
3859 PyObject *p = PySequence_GetItem(st, i);
3860 if (p == NULL) {
3861 good_timetuple = 0;
3862 break;
3863 }
3864 if (PyLong_Check(p))
3865 ia[i] = PyLong_AsLong(p);
3866 else
3867 good_timetuple = 0;
3868 Py_DECREF(p);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003869 }
Christian Heimesdd15f6c2008-03-16 00:07:10 +00003870/* if (PyLong_CheckExact(p)) {
Martin v. Löwisd1a1d1e2007-12-04 22:10:37 +00003871 ia[i] = PyLong_AsLongAndOverflow(p, &overflow);
3872 if (overflow)
3873 good_timetuple = 0;
3874 }
Skip Montanaro0af3ade2005-01-13 04:12:31 +00003875 else
3876 good_timetuple = 0;
3877 Py_DECREF(p);
Christian Heimesdd15f6c2008-03-16 00:07:10 +00003878*/ }
3879 else
3880 good_timetuple = 0;
3881 /* follow that up with a little dose of microseconds */
3882 if (PyLong_Check(frac))
3883 ia[6] = PyLong_AsLong(frac);
3884 else
3885 good_timetuple = 0;
3886 }
Skip Montanaro0af3ade2005-01-13 04:12:31 +00003887 else
3888 good_timetuple = 0;
3889 if (good_timetuple)
Christian Heimesdd15f6c2008-03-16 00:07:10 +00003890 result = PyObject_CallFunction(cls, "iiiiiii",
3891 ia[0], ia[1], ia[2],
3892 ia[3], ia[4], ia[5],
3893 ia[6]);
Skip Montanaro0af3ade2005-01-13 04:12:31 +00003894 else
3895 PyErr_SetString(PyExc_ValueError,
Christian Heimesdd15f6c2008-03-16 00:07:10 +00003896 "unexpected value from _strptime._strptime");
Skip Montanaro0af3ade2005-01-13 04:12:31 +00003897 }
Christian Heimesdd15f6c2008-03-16 00:07:10 +00003898 Py_XDECREF(obj);
3899 Py_XDECREF(st);
3900 Py_XDECREF(frac);
Skip Montanaro0af3ade2005-01-13 04:12:31 +00003901 return result;
3902}
3903
Tim Petersa9bc1682003-01-11 03:39:11 +00003904/* Return new datetime from date/datetime and time arguments. */
3905static PyObject *
3906datetime_combine(PyObject *cls, PyObject *args, PyObject *kw)
3907{
Martin v. Löwis02cbf4a2006-02-27 17:20:04 +00003908 static char *keywords[] = {"date", "time", NULL};
Tim Petersa9bc1682003-01-11 03:39:11 +00003909 PyObject *date;
3910 PyObject *time;
3911 PyObject *result = NULL;
3912
3913 if (PyArg_ParseTupleAndKeywords(args, kw, "O!O!:combine", keywords,
3914 &PyDateTime_DateType, &date,
3915 &PyDateTime_TimeType, &time)) {
3916 PyObject *tzinfo = Py_None;
3917
3918 if (HASTZINFO(time))
3919 tzinfo = ((PyDateTime_Time *)time)->tzinfo;
3920 result = PyObject_CallFunction(cls, "iiiiiiiO",
3921 GET_YEAR(date),
3922 GET_MONTH(date),
3923 GET_DAY(date),
3924 TIME_GET_HOUR(time),
3925 TIME_GET_MINUTE(time),
3926 TIME_GET_SECOND(time),
3927 TIME_GET_MICROSECOND(time),
3928 tzinfo);
3929 }
3930 return result;
3931}
Tim Peters2a799bf2002-12-16 20:18:38 +00003932
3933/*
3934 * Destructor.
3935 */
3936
3937static void
Tim Petersa9bc1682003-01-11 03:39:11 +00003938datetime_dealloc(PyDateTime_DateTime *self)
Tim Peters2a799bf2002-12-16 20:18:38 +00003939{
Tim Petersa9bc1682003-01-11 03:39:11 +00003940 if (HASTZINFO(self)) {
3941 Py_XDECREF(self->tzinfo);
3942 }
Christian Heimes90aa7642007-12-19 02:45:37 +00003943 Py_TYPE(self)->tp_free((PyObject *)self);
Tim Peters2a799bf2002-12-16 20:18:38 +00003944}
3945
3946/*
3947 * Indirect access to tzinfo methods.
3948 */
3949
Tim Peters2a799bf2002-12-16 20:18:38 +00003950/* These are all METH_NOARGS, so don't need to check the arglist. */
3951static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00003952datetime_utcoffset(PyDateTime_DateTime *self, PyObject *unused) {
3953 return offset_as_timedelta(HASTZINFO(self) ? self->tzinfo : Py_None,
3954 "utcoffset", (PyObject *)self);
Tim Peters2a799bf2002-12-16 20:18:38 +00003955}
3956
3957static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00003958datetime_dst(PyDateTime_DateTime *self, PyObject *unused) {
3959 return offset_as_timedelta(HASTZINFO(self) ? self->tzinfo : Py_None,
3960 "dst", (PyObject *)self);
Tim Peters855fe882002-12-22 03:43:39 +00003961}
3962
3963static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00003964datetime_tzname(PyDateTime_DateTime *self, PyObject *unused) {
3965 return call_tzname(HASTZINFO(self) ? self->tzinfo : Py_None,
3966 (PyObject *)self);
Tim Peters2a799bf2002-12-16 20:18:38 +00003967}
3968
3969/*
Tim Petersa9bc1682003-01-11 03:39:11 +00003970 * datetime arithmetic.
Tim Peters2a799bf2002-12-16 20:18:38 +00003971 */
3972
Tim Petersa9bc1682003-01-11 03:39:11 +00003973/* factor must be 1 (to add) or -1 (to subtract). The result inherits
3974 * the tzinfo state of date.
Tim Peters2a799bf2002-12-16 20:18:38 +00003975 */
3976static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00003977add_datetime_timedelta(PyDateTime_DateTime *date, PyDateTime_Delta *delta,
3978 int factor)
Tim Peters2a799bf2002-12-16 20:18:38 +00003979{
Tim Petersa9bc1682003-01-11 03:39:11 +00003980 /* Note that the C-level additions can't overflow, because of
3981 * invariant bounds on the member values.
3982 */
3983 int year = GET_YEAR(date);
3984 int month = GET_MONTH(date);
3985 int day = GET_DAY(date) + GET_TD_DAYS(delta) * factor;
3986 int hour = DATE_GET_HOUR(date);
3987 int minute = DATE_GET_MINUTE(date);
3988 int second = DATE_GET_SECOND(date) + GET_TD_SECONDS(delta) * factor;
3989 int microsecond = DATE_GET_MICROSECOND(date) +
3990 GET_TD_MICROSECONDS(delta) * factor;
Tim Peters2a799bf2002-12-16 20:18:38 +00003991
Tim Petersa9bc1682003-01-11 03:39:11 +00003992 assert(factor == 1 || factor == -1);
3993 if (normalize_datetime(&year, &month, &day,
3994 &hour, &minute, &second, &microsecond) < 0)
3995 return NULL;
3996 else
3997 return new_datetime(year, month, day,
3998 hour, minute, second, microsecond,
3999 HASTZINFO(date) ? date->tzinfo : Py_None);
Tim Peters2a799bf2002-12-16 20:18:38 +00004000}
4001
4002static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00004003datetime_add(PyObject *left, PyObject *right)
Tim Peters2a799bf2002-12-16 20:18:38 +00004004{
Tim Petersa9bc1682003-01-11 03:39:11 +00004005 if (PyDateTime_Check(left)) {
4006 /* datetime + ??? */
4007 if (PyDelta_Check(right))
4008 /* datetime + delta */
4009 return add_datetime_timedelta(
4010 (PyDateTime_DateTime *)left,
4011 (PyDateTime_Delta *)right,
4012 1);
4013 }
4014 else if (PyDelta_Check(left)) {
4015 /* delta + datetime */
4016 return add_datetime_timedelta((PyDateTime_DateTime *) right,
4017 (PyDateTime_Delta *) left,
4018 1);
4019 }
4020 Py_INCREF(Py_NotImplemented);
4021 return Py_NotImplemented;
Tim Peters2a799bf2002-12-16 20:18:38 +00004022}
4023
4024static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00004025datetime_subtract(PyObject *left, PyObject *right)
Tim Peters2a799bf2002-12-16 20:18:38 +00004026{
4027 PyObject *result = Py_NotImplemented;
4028
4029 if (PyDateTime_Check(left)) {
4030 /* datetime - ??? */
4031 if (PyDateTime_Check(right)) {
4032 /* datetime - datetime */
4033 naivety n1, n2;
4034 int offset1, offset2;
Tim Petersa9bc1682003-01-11 03:39:11 +00004035 int delta_d, delta_s, delta_us;
Tim Peters2a799bf2002-12-16 20:18:38 +00004036
Tim Peterse39a80c2002-12-30 21:28:52 +00004037 if (classify_two_utcoffsets(left, &offset1, &n1, left,
4038 right, &offset2, &n2,
4039 right) < 0)
Tim Peters00237032002-12-27 02:21:51 +00004040 return NULL;
Tim Peters8702d5f2002-12-27 02:26:16 +00004041 assert(n1 != OFFSET_UNKNOWN && n2 != OFFSET_UNKNOWN);
Tim Peters2a799bf2002-12-16 20:18:38 +00004042 if (n1 != n2) {
4043 PyErr_SetString(PyExc_TypeError,
4044 "can't subtract offset-naive and "
4045 "offset-aware datetimes");
4046 return NULL;
4047 }
Tim Petersa9bc1682003-01-11 03:39:11 +00004048 delta_d = ymd_to_ord(GET_YEAR(left),
4049 GET_MONTH(left),
4050 GET_DAY(left)) -
4051 ymd_to_ord(GET_YEAR(right),
4052 GET_MONTH(right),
4053 GET_DAY(right));
4054 /* These can't overflow, since the values are
4055 * normalized. At most this gives the number of
4056 * seconds in one day.
4057 */
4058 delta_s = (DATE_GET_HOUR(left) -
4059 DATE_GET_HOUR(right)) * 3600 +
4060 (DATE_GET_MINUTE(left) -
4061 DATE_GET_MINUTE(right)) * 60 +
4062 (DATE_GET_SECOND(left) -
4063 DATE_GET_SECOND(right));
4064 delta_us = DATE_GET_MICROSECOND(left) -
4065 DATE_GET_MICROSECOND(right);
Tim Peters2a799bf2002-12-16 20:18:38 +00004066 /* (left - offset1) - (right - offset2) =
4067 * (left - right) + (offset2 - offset1)
4068 */
Tim Petersa9bc1682003-01-11 03:39:11 +00004069 delta_s += (offset2 - offset1) * 60;
4070 result = new_delta(delta_d, delta_s, delta_us, 1);
Tim Peters2a799bf2002-12-16 20:18:38 +00004071 }
4072 else if (PyDelta_Check(right)) {
Tim Petersa9bc1682003-01-11 03:39:11 +00004073 /* datetime - delta */
4074 result = add_datetime_timedelta(
Tim Peters2a799bf2002-12-16 20:18:38 +00004075 (PyDateTime_DateTime *)left,
Tim Petersa9bc1682003-01-11 03:39:11 +00004076 (PyDateTime_Delta *)right,
4077 -1);
Tim Peters2a799bf2002-12-16 20:18:38 +00004078 }
4079 }
4080
4081 if (result == Py_NotImplemented)
4082 Py_INCREF(result);
4083 return result;
4084}
4085
4086/* Various ways to turn a datetime into a string. */
4087
4088static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00004089datetime_repr(PyDateTime_DateTime *self)
Tim Peters2a799bf2002-12-16 20:18:38 +00004090{
Christian Heimes90aa7642007-12-19 02:45:37 +00004091 const char *type_name = Py_TYPE(self)->tp_name;
Tim Petersa9bc1682003-01-11 03:39:11 +00004092 PyObject *baserepr;
Tim Peters2a799bf2002-12-16 20:18:38 +00004093
Tim Petersa9bc1682003-01-11 03:39:11 +00004094 if (DATE_GET_MICROSECOND(self)) {
Walter Dörwald7569dfe2007-05-19 21:49:49 +00004095 baserepr = PyUnicode_FromFormat(
Tim Petersa9bc1682003-01-11 03:39:11 +00004096 "%s(%d, %d, %d, %d, %d, %d, %d)",
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004097 type_name,
Tim Petersa9bc1682003-01-11 03:39:11 +00004098 GET_YEAR(self), GET_MONTH(self), GET_DAY(self),
4099 DATE_GET_HOUR(self), DATE_GET_MINUTE(self),
4100 DATE_GET_SECOND(self),
4101 DATE_GET_MICROSECOND(self));
4102 }
4103 else if (DATE_GET_SECOND(self)) {
Walter Dörwald7569dfe2007-05-19 21:49:49 +00004104 baserepr = PyUnicode_FromFormat(
Tim Petersa9bc1682003-01-11 03:39:11 +00004105 "%s(%d, %d, %d, %d, %d, %d)",
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004106 type_name,
Tim Petersa9bc1682003-01-11 03:39:11 +00004107 GET_YEAR(self), GET_MONTH(self), GET_DAY(self),
4108 DATE_GET_HOUR(self), DATE_GET_MINUTE(self),
4109 DATE_GET_SECOND(self));
4110 }
4111 else {
Walter Dörwald7569dfe2007-05-19 21:49:49 +00004112 baserepr = PyUnicode_FromFormat(
Tim Petersa9bc1682003-01-11 03:39:11 +00004113 "%s(%d, %d, %d, %d, %d)",
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004114 type_name,
Tim Petersa9bc1682003-01-11 03:39:11 +00004115 GET_YEAR(self), GET_MONTH(self), GET_DAY(self),
4116 DATE_GET_HOUR(self), DATE_GET_MINUTE(self));
4117 }
Tim Petersa9bc1682003-01-11 03:39:11 +00004118 if (baserepr == NULL || ! HASTZINFO(self))
4119 return baserepr;
Tim Peters2a799bf2002-12-16 20:18:38 +00004120 return append_keyword_tzinfo(baserepr, self->tzinfo);
4121}
4122
Tim Petersa9bc1682003-01-11 03:39:11 +00004123static PyObject *
4124datetime_str(PyDateTime_DateTime *self)
4125{
4126 return PyObject_CallMethod((PyObject *)self, "isoformat", "(s)", " ");
4127}
Tim Peters2a799bf2002-12-16 20:18:38 +00004128
4129static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00004130datetime_isoformat(PyDateTime_DateTime *self, PyObject *args, PyObject *kw)
Tim Peters2a799bf2002-12-16 20:18:38 +00004131{
Walter Dörwaldbc1f8862007-06-20 11:02:38 +00004132 int sep = 'T';
Martin v. Löwis02cbf4a2006-02-27 17:20:04 +00004133 static char *keywords[] = {"sep", NULL};
Tim Petersa9bc1682003-01-11 03:39:11 +00004134 char buffer[100];
Tim Petersa9bc1682003-01-11 03:39:11 +00004135 PyObject *result;
Walter Dörwaldbafa1372007-05-31 17:50:48 +00004136 int us = DATE_GET_MICROSECOND(self);
Tim Peters2a799bf2002-12-16 20:18:38 +00004137
Walter Dörwaldd0941302007-07-01 21:58:22 +00004138 if (!PyArg_ParseTupleAndKeywords(args, kw, "|C:isoformat", keywords, &sep))
Tim Petersa9bc1682003-01-11 03:39:11 +00004139 return NULL;
Walter Dörwaldbafa1372007-05-31 17:50:48 +00004140 if (us)
4141 result = PyUnicode_FromFormat("%04d-%02d-%02d%c%02d:%02d:%02d.%06d",
4142 GET_YEAR(self), GET_MONTH(self),
4143 GET_DAY(self), (int)sep,
4144 DATE_GET_HOUR(self), DATE_GET_MINUTE(self),
4145 DATE_GET_SECOND(self), us);
4146 else
4147 result = PyUnicode_FromFormat("%04d-%02d-%02d%c%02d:%02d:%02d",
4148 GET_YEAR(self), GET_MONTH(self),
4149 GET_DAY(self), (int)sep,
4150 DATE_GET_HOUR(self), DATE_GET_MINUTE(self),
4151 DATE_GET_SECOND(self));
4152
4153 if (!result || !HASTZINFO(self))
Tim Peters2a799bf2002-12-16 20:18:38 +00004154 return result;
4155
4156 /* We need to append the UTC offset. */
Tim Petersa9bc1682003-01-11 03:39:11 +00004157 if (format_utcoffset(buffer, sizeof(buffer), ":", self->tzinfo,
Tim Peters2a799bf2002-12-16 20:18:38 +00004158 (PyObject *)self) < 0) {
4159 Py_DECREF(result);
4160 return NULL;
4161 }
Walter Dörwaldbafa1372007-05-31 17:50:48 +00004162 PyUnicode_AppendAndDel(&result, PyUnicode_FromString(buffer));
Tim Peters2a799bf2002-12-16 20:18:38 +00004163 return result;
4164}
4165
Tim Petersa9bc1682003-01-11 03:39:11 +00004166static PyObject *
4167datetime_ctime(PyDateTime_DateTime *self)
4168{
4169 return format_ctime((PyDateTime_Date *)self,
4170 DATE_GET_HOUR(self),
4171 DATE_GET_MINUTE(self),
4172 DATE_GET_SECOND(self));
4173}
4174
Tim Peters2a799bf2002-12-16 20:18:38 +00004175/* Miscellaneous methods. */
4176
Tim Petersa9bc1682003-01-11 03:39:11 +00004177static PyObject *
Guido van Rossum19960592006-08-24 17:29:38 +00004178datetime_richcompare(PyObject *self, PyObject *other, int op)
Tim Petersa9bc1682003-01-11 03:39:11 +00004179{
4180 int diff;
4181 naivety n1, n2;
4182 int offset1, offset2;
4183
4184 if (! PyDateTime_Check(other)) {
Guido van Rossum19960592006-08-24 17:29:38 +00004185 if (PyDate_Check(other)) {
4186 /* Prevent invocation of date_richcompare. We want to
4187 return NotImplemented here to give the other object
4188 a chance. But since DateTime is a subclass of
4189 Date, if the other object is a Date, it would
4190 compute an ordering based on the date part alone,
4191 and we don't want that. So force unequal or
4192 uncomparable here in that case. */
4193 if (op == Py_EQ)
4194 Py_RETURN_FALSE;
4195 if (op == Py_NE)
4196 Py_RETURN_TRUE;
4197 return cmperror(self, other);
Tim Peters8d81a012003-01-24 22:36:34 +00004198 }
Guido van Rossum19960592006-08-24 17:29:38 +00004199 Py_INCREF(Py_NotImplemented);
4200 return Py_NotImplemented;
Tim Petersa9bc1682003-01-11 03:39:11 +00004201 }
4202
Guido van Rossum19960592006-08-24 17:29:38 +00004203 if (classify_two_utcoffsets(self, &offset1, &n1, self,
4204 other, &offset2, &n2, other) < 0)
Tim Petersa9bc1682003-01-11 03:39:11 +00004205 return NULL;
4206 assert(n1 != OFFSET_UNKNOWN && n2 != OFFSET_UNKNOWN);
4207 /* If they're both naive, or both aware and have the same offsets,
4208 * we get off cheap. Note that if they're both naive, offset1 ==
4209 * offset2 == 0 at this point.
4210 */
4211 if (n1 == n2 && offset1 == offset2) {
Guido van Rossum19960592006-08-24 17:29:38 +00004212 diff = memcmp(((PyDateTime_DateTime *)self)->data,
4213 ((PyDateTime_DateTime *)other)->data,
Tim Petersa9bc1682003-01-11 03:39:11 +00004214 _PyDateTime_DATETIME_DATASIZE);
4215 return diff_to_bool(diff, op);
4216 }
4217
4218 if (n1 == OFFSET_AWARE && n2 == OFFSET_AWARE) {
4219 PyDateTime_Delta *delta;
4220
4221 assert(offset1 != offset2); /* else last "if" handled it */
4222 delta = (PyDateTime_Delta *)datetime_subtract((PyObject *)self,
4223 other);
4224 if (delta == NULL)
4225 return NULL;
4226 diff = GET_TD_DAYS(delta);
4227 if (diff == 0)
4228 diff = GET_TD_SECONDS(delta) |
4229 GET_TD_MICROSECONDS(delta);
4230 Py_DECREF(delta);
4231 return diff_to_bool(diff, op);
4232 }
4233
4234 assert(n1 != n2);
4235 PyErr_SetString(PyExc_TypeError,
4236 "can't compare offset-naive and "
4237 "offset-aware datetimes");
4238 return NULL;
4239}
4240
4241static long
4242datetime_hash(PyDateTime_DateTime *self)
4243{
4244 if (self->hashcode == -1) {
4245 naivety n;
4246 int offset;
4247 PyObject *temp;
4248
4249 n = classify_utcoffset((PyObject *)self, (PyObject *)self,
4250 &offset);
4251 assert(n != OFFSET_UNKNOWN);
4252 if (n == OFFSET_ERROR)
4253 return -1;
4254
4255 /* Reduce this to a hash of another object. */
Guido van Rossumfd53fd62007-08-24 04:05:13 +00004256 if (n == OFFSET_NAIVE) {
4257 self->hashcode = generic_hash(
4258 (unsigned char *)self->data, _PyDateTime_DATETIME_DATASIZE);
4259 return self->hashcode;
4260 }
Tim Petersa9bc1682003-01-11 03:39:11 +00004261 else {
4262 int days;
4263 int seconds;
4264
4265 assert(n == OFFSET_AWARE);
4266 assert(HASTZINFO(self));
4267 days = ymd_to_ord(GET_YEAR(self),
4268 GET_MONTH(self),
4269 GET_DAY(self));
4270 seconds = DATE_GET_HOUR(self) * 3600 +
4271 (DATE_GET_MINUTE(self) - offset) * 60 +
4272 DATE_GET_SECOND(self);
4273 temp = new_delta(days,
4274 seconds,
4275 DATE_GET_MICROSECOND(self),
4276 1);
4277 }
4278 if (temp != NULL) {
4279 self->hashcode = PyObject_Hash(temp);
4280 Py_DECREF(temp);
4281 }
4282 }
4283 return self->hashcode;
4284}
Tim Peters2a799bf2002-12-16 20:18:38 +00004285
4286static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00004287datetime_replace(PyDateTime_DateTime *self, PyObject *args, PyObject *kw)
Tim Peters12bf3392002-12-24 05:41:27 +00004288{
4289 PyObject *clone;
4290 PyObject *tuple;
4291 int y = GET_YEAR(self);
4292 int m = GET_MONTH(self);
4293 int d = GET_DAY(self);
4294 int hh = DATE_GET_HOUR(self);
4295 int mm = DATE_GET_MINUTE(self);
4296 int ss = DATE_GET_SECOND(self);
4297 int us = DATE_GET_MICROSECOND(self);
Tim Petersa9bc1682003-01-11 03:39:11 +00004298 PyObject *tzinfo = HASTZINFO(self) ? self->tzinfo : Py_None;
Tim Peters12bf3392002-12-24 05:41:27 +00004299
4300 if (! PyArg_ParseTupleAndKeywords(args, kw, "|iiiiiiiO:replace",
Tim Petersa9bc1682003-01-11 03:39:11 +00004301 datetime_kws,
Tim Peters12bf3392002-12-24 05:41:27 +00004302 &y, &m, &d, &hh, &mm, &ss, &us,
4303 &tzinfo))
4304 return NULL;
4305 tuple = Py_BuildValue("iiiiiiiO", y, m, d, hh, mm, ss, us, tzinfo);
4306 if (tuple == NULL)
4307 return NULL;
Christian Heimes90aa7642007-12-19 02:45:37 +00004308 clone = datetime_new(Py_TYPE(self), tuple, NULL);
Tim Peters12bf3392002-12-24 05:41:27 +00004309 Py_DECREF(tuple);
4310 return clone;
4311}
4312
4313static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00004314datetime_astimezone(PyDateTime_DateTime *self, PyObject *args, PyObject *kw)
Tim Peters80475bb2002-12-25 07:40:55 +00004315{
Tim Peters52dcce22003-01-23 16:36:11 +00004316 int y, m, d, hh, mm, ss, us;
Tim Peters521fc152002-12-31 17:36:56 +00004317 PyObject *result;
Tim Peters52dcce22003-01-23 16:36:11 +00004318 int offset, none;
Tim Peters521fc152002-12-31 17:36:56 +00004319
Tim Peters80475bb2002-12-25 07:40:55 +00004320 PyObject *tzinfo;
Martin v. Löwis02cbf4a2006-02-27 17:20:04 +00004321 static char *keywords[] = {"tz", NULL};
Tim Peters80475bb2002-12-25 07:40:55 +00004322
Tim Peters52dcce22003-01-23 16:36:11 +00004323 if (! PyArg_ParseTupleAndKeywords(args, kw, "O!:astimezone", keywords,
4324 &PyDateTime_TZInfoType, &tzinfo))
Tim Peters80475bb2002-12-25 07:40:55 +00004325 return NULL;
4326
Tim Peters52dcce22003-01-23 16:36:11 +00004327 if (!HASTZINFO(self) || self->tzinfo == Py_None)
4328 goto NeedAware;
Tim Peters521fc152002-12-31 17:36:56 +00004329
Tim Peters52dcce22003-01-23 16:36:11 +00004330 /* Conversion to self's own time zone is a NOP. */
4331 if (self->tzinfo == tzinfo) {
4332 Py_INCREF(self);
4333 return (PyObject *)self;
Tim Peters710fb152003-01-02 19:35:54 +00004334 }
Tim Peters521fc152002-12-31 17:36:56 +00004335
Tim Peters52dcce22003-01-23 16:36:11 +00004336 /* Convert self to UTC. */
4337 offset = call_utcoffset(self->tzinfo, (PyObject *)self, &none);
4338 if (offset == -1 && PyErr_Occurred())
4339 return NULL;
4340 if (none)
4341 goto NeedAware;
Tim Petersf3615152003-01-01 21:51:37 +00004342
Tim Peters52dcce22003-01-23 16:36:11 +00004343 y = GET_YEAR(self);
4344 m = GET_MONTH(self);
4345 d = GET_DAY(self);
4346 hh = DATE_GET_HOUR(self);
4347 mm = DATE_GET_MINUTE(self);
4348 ss = DATE_GET_SECOND(self);
4349 us = DATE_GET_MICROSECOND(self);
4350
4351 mm -= offset;
Tim Petersf3615152003-01-01 21:51:37 +00004352 if ((mm < 0 || mm >= 60) &&
4353 normalize_datetime(&y, &m, &d, &hh, &mm, &ss, &us) < 0)
Tim Peters52dcce22003-01-23 16:36:11 +00004354 return NULL;
4355
4356 /* Attach new tzinfo and let fromutc() do the rest. */
4357 result = new_datetime(y, m, d, hh, mm, ss, us, tzinfo);
4358 if (result != NULL) {
4359 PyObject *temp = result;
4360
4361 result = PyObject_CallMethod(tzinfo, "fromutc", "O", temp);
4362 Py_DECREF(temp);
4363 }
Tim Petersadf64202003-01-04 06:03:15 +00004364 return result;
Tim Peters521fc152002-12-31 17:36:56 +00004365
Tim Peters52dcce22003-01-23 16:36:11 +00004366NeedAware:
4367 PyErr_SetString(PyExc_ValueError, "astimezone() cannot be applied to "
4368 "a naive datetime");
Tim Peters521fc152002-12-31 17:36:56 +00004369 return NULL;
Tim Peters80475bb2002-12-25 07:40:55 +00004370}
4371
4372static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00004373datetime_timetuple(PyDateTime_DateTime *self)
Tim Peters2a799bf2002-12-16 20:18:38 +00004374{
4375 int dstflag = -1;
4376
Tim Petersa9bc1682003-01-11 03:39:11 +00004377 if (HASTZINFO(self) && self->tzinfo != Py_None) {
Tim Peters2a799bf2002-12-16 20:18:38 +00004378 int none;
4379
4380 dstflag = call_dst(self->tzinfo, (PyObject *)self, &none);
4381 if (dstflag == -1 && PyErr_Occurred())
4382 return NULL;
4383
4384 if (none)
4385 dstflag = -1;
4386 else if (dstflag != 0)
4387 dstflag = 1;
4388
4389 }
4390 return build_struct_time(GET_YEAR(self),
4391 GET_MONTH(self),
4392 GET_DAY(self),
4393 DATE_GET_HOUR(self),
4394 DATE_GET_MINUTE(self),
4395 DATE_GET_SECOND(self),
4396 dstflag);
4397}
4398
4399static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00004400datetime_getdate(PyDateTime_DateTime *self)
4401{
4402 return new_date(GET_YEAR(self),
4403 GET_MONTH(self),
4404 GET_DAY(self));
4405}
4406
4407static PyObject *
4408datetime_gettime(PyDateTime_DateTime *self)
4409{
4410 return new_time(DATE_GET_HOUR(self),
4411 DATE_GET_MINUTE(self),
4412 DATE_GET_SECOND(self),
4413 DATE_GET_MICROSECOND(self),
4414 Py_None);
4415}
4416
4417static PyObject *
4418datetime_gettimetz(PyDateTime_DateTime *self)
4419{
4420 return new_time(DATE_GET_HOUR(self),
4421 DATE_GET_MINUTE(self),
4422 DATE_GET_SECOND(self),
4423 DATE_GET_MICROSECOND(self),
4424 HASTZINFO(self) ? self->tzinfo : Py_None);
4425}
4426
4427static PyObject *
4428datetime_utctimetuple(PyDateTime_DateTime *self)
Tim Peters2a799bf2002-12-16 20:18:38 +00004429{
4430 int y = GET_YEAR(self);
4431 int m = GET_MONTH(self);
4432 int d = GET_DAY(self);
4433 int hh = DATE_GET_HOUR(self);
4434 int mm = DATE_GET_MINUTE(self);
4435 int ss = DATE_GET_SECOND(self);
4436 int us = 0; /* microseconds are ignored in a timetuple */
4437 int offset = 0;
4438
Tim Petersa9bc1682003-01-11 03:39:11 +00004439 if (HASTZINFO(self) && self->tzinfo != Py_None) {
Tim Peters2a799bf2002-12-16 20:18:38 +00004440 int none;
4441
4442 offset = call_utcoffset(self->tzinfo, (PyObject *)self, &none);
4443 if (offset == -1 && PyErr_Occurred())
4444 return NULL;
4445 }
4446 /* Even if offset is 0, don't call timetuple() -- tm_isdst should be
4447 * 0 in a UTC timetuple regardless of what dst() says.
4448 */
4449 if (offset) {
4450 /* Subtract offset minutes & normalize. */
4451 int stat;
4452
4453 mm -= offset;
4454 stat = normalize_datetime(&y, &m, &d, &hh, &mm, &ss, &us);
4455 if (stat < 0) {
4456 /* At the edges, it's possible we overflowed
4457 * beyond MINYEAR or MAXYEAR.
4458 */
4459 if (PyErr_ExceptionMatches(PyExc_OverflowError))
4460 PyErr_Clear();
4461 else
4462 return NULL;
4463 }
4464 }
4465 return build_struct_time(y, m, d, hh, mm, ss, 0);
4466}
4467
Tim Peters371935f2003-02-01 01:52:50 +00004468/* Pickle support, a simple use of __reduce__. */
Tim Peters33e0f382003-01-10 02:05:14 +00004469
Tim Petersa9bc1682003-01-11 03:39:11 +00004470/* Let basestate be the non-tzinfo data string.
Tim Peters2a799bf2002-12-16 20:18:38 +00004471 * If tzinfo is None, this returns (basestate,), else (basestate, tzinfo).
4472 * So it's a tuple in any (non-error) case.
Tim Petersb57f8f02003-02-01 02:54:15 +00004473 * __getstate__ isn't exposed.
Tim Peters2a799bf2002-12-16 20:18:38 +00004474 */
4475static PyObject *
Tim Petersa9bc1682003-01-11 03:39:11 +00004476datetime_getstate(PyDateTime_DateTime *self)
Tim Peters2a799bf2002-12-16 20:18:38 +00004477{
4478 PyObject *basestate;
4479 PyObject *result = NULL;
4480
Guido van Rossum254348e2007-11-21 19:29:53 +00004481 basestate = PyString_FromStringAndSize((char *)self->data,
4482 _PyDateTime_DATETIME_DATASIZE);
Tim Peters2a799bf2002-12-16 20:18:38 +00004483 if (basestate != NULL) {
Tim Petersa9bc1682003-01-11 03:39:11 +00004484 if (! HASTZINFO(self) || self->tzinfo == Py_None)
Raymond Hettinger8ae46892003-10-12 19:09:37 +00004485 result = PyTuple_Pack(1, basestate);
Tim Peters2a799bf2002-12-16 20:18:38 +00004486 else
Raymond Hettinger8ae46892003-10-12 19:09:37 +00004487 result = PyTuple_Pack(2, basestate, self->tzinfo);
Tim Peters2a799bf2002-12-16 20:18:38 +00004488 Py_DECREF(basestate);
4489 }
4490 return result;
4491}
4492
4493static PyObject *
Guido van Rossum177e41a2003-01-30 22:06:23 +00004494datetime_reduce(PyDateTime_DateTime *self, PyObject *arg)
Tim Peters2a799bf2002-12-16 20:18:38 +00004495{
Christian Heimes90aa7642007-12-19 02:45:37 +00004496 return Py_BuildValue("(ON)", Py_TYPE(self), datetime_getstate(self));
Tim Peters2a799bf2002-12-16 20:18:38 +00004497}
4498
Tim Petersa9bc1682003-01-11 03:39:11 +00004499static PyMethodDef datetime_methods[] = {
Guido van Rossum177e41a2003-01-30 22:06:23 +00004500
Tim Peters2a799bf2002-12-16 20:18:38 +00004501 /* Class methods: */
Tim Peters2a799bf2002-12-16 20:18:38 +00004502
Tim Petersa9bc1682003-01-11 03:39:11 +00004503 {"now", (PyCFunction)datetime_now,
Guido van Rossumd59da4b2007-05-22 18:11:13 +00004504 METH_VARARGS | METH_KEYWORDS | METH_CLASS,
Neal Norwitz2fbe5372003-01-23 21:09:05 +00004505 PyDoc_STR("[tz] -> new datetime with tz's local day and time.")},
Tim Peters2a799bf2002-12-16 20:18:38 +00004506
Tim Petersa9bc1682003-01-11 03:39:11 +00004507 {"utcnow", (PyCFunction)datetime_utcnow,
4508 METH_NOARGS | METH_CLASS,
4509 PyDoc_STR("Return a new datetime representing UTC day and time.")},
4510
4511 {"fromtimestamp", (PyCFunction)datetime_fromtimestamp,
Guido van Rossumd59da4b2007-05-22 18:11:13 +00004512 METH_VARARGS | METH_KEYWORDS | METH_CLASS,
Tim Peters2a44a8d2003-01-23 20:53:10 +00004513 PyDoc_STR("timestamp[, tz] -> tz's local time from POSIX timestamp.")},
Tim Peters2a799bf2002-12-16 20:18:38 +00004514
Tim Petersa9bc1682003-01-11 03:39:11 +00004515 {"utcfromtimestamp", (PyCFunction)datetime_utcfromtimestamp,
4516 METH_VARARGS | METH_CLASS,
4517 PyDoc_STR("timestamp -> UTC datetime from a POSIX timestamp "
4518 "(like time.time()).")},
4519
Skip Montanaro0af3ade2005-01-13 04:12:31 +00004520 {"strptime", (PyCFunction)datetime_strptime,
4521 METH_VARARGS | METH_CLASS,
4522 PyDoc_STR("string, format -> new datetime parsed from a string "
4523 "(like time.strptime()).")},
4524
Tim Petersa9bc1682003-01-11 03:39:11 +00004525 {"combine", (PyCFunction)datetime_combine,
4526 METH_VARARGS | METH_KEYWORDS | METH_CLASS,
4527 PyDoc_STR("date, time -> datetime with same date and time fields")},
4528
Tim Peters2a799bf2002-12-16 20:18:38 +00004529 /* Instance methods: */
Guido van Rossum177e41a2003-01-30 22:06:23 +00004530
Tim Petersa9bc1682003-01-11 03:39:11 +00004531 {"date", (PyCFunction)datetime_getdate, METH_NOARGS,
4532 PyDoc_STR("Return date object with same year, month and day.")},
4533
4534 {"time", (PyCFunction)datetime_gettime, METH_NOARGS,
4535 PyDoc_STR("Return time object with same time but with tzinfo=None.")},
4536
4537 {"timetz", (PyCFunction)datetime_gettimetz, METH_NOARGS,
4538 PyDoc_STR("Return time object with same time and tzinfo.")},
4539
4540 {"ctime", (PyCFunction)datetime_ctime, METH_NOARGS,
4541 PyDoc_STR("Return ctime() style string.")},
4542
4543 {"timetuple", (PyCFunction)datetime_timetuple, METH_NOARGS,
Tim Peters2a799bf2002-12-16 20:18:38 +00004544 PyDoc_STR("Return time tuple, compatible with time.localtime().")},
4545
Tim Petersa9bc1682003-01-11 03:39:11 +00004546 {"utctimetuple", (PyCFunction)datetime_utctimetuple, METH_NOARGS,
Tim Peters2a799bf2002-12-16 20:18:38 +00004547 PyDoc_STR("Return UTC time tuple, compatible with time.localtime().")},
4548
Guido van Rossumd59da4b2007-05-22 18:11:13 +00004549 {"isoformat", (PyCFunction)datetime_isoformat, METH_VARARGS | METH_KEYWORDS,
Tim Peters2a799bf2002-12-16 20:18:38 +00004550 PyDoc_STR("[sep] -> string in ISO 8601 format, "
4551 "YYYY-MM-DDTHH:MM:SS[.mmmmmm][+HH:MM].\n\n"
4552 "sep is used to separate the year from the time, and "
4553 "defaults to 'T'.")},
4554
Tim Petersa9bc1682003-01-11 03:39:11 +00004555 {"utcoffset", (PyCFunction)datetime_utcoffset, METH_NOARGS,
Tim Peters2a799bf2002-12-16 20:18:38 +00004556 PyDoc_STR("Return self.tzinfo.utcoffset(self).")},
4557
Tim Petersa9bc1682003-01-11 03:39:11 +00004558 {"tzname", (PyCFunction)datetime_tzname, METH_NOARGS,
Tim Peters2a799bf2002-12-16 20:18:38 +00004559 PyDoc_STR("Return self.tzinfo.tzname(self).")},
4560
Tim Petersa9bc1682003-01-11 03:39:11 +00004561 {"dst", (PyCFunction)datetime_dst, METH_NOARGS,
Tim Peters2a799bf2002-12-16 20:18:38 +00004562 PyDoc_STR("Return self.tzinfo.dst(self).")},
4563
Guido van Rossumd59da4b2007-05-22 18:11:13 +00004564 {"replace", (PyCFunction)datetime_replace, METH_VARARGS | METH_KEYWORDS,
Tim Petersa9bc1682003-01-11 03:39:11 +00004565 PyDoc_STR("Return datetime with new specified fields.")},
Tim Peters12bf3392002-12-24 05:41:27 +00004566
Guido van Rossumd59da4b2007-05-22 18:11:13 +00004567 {"astimezone", (PyCFunction)datetime_astimezone, METH_VARARGS | METH_KEYWORDS,
Tim Peters80475bb2002-12-25 07:40:55 +00004568 PyDoc_STR("tz -> convert to local time in new timezone tz\n")},
4569
Guido van Rossum177e41a2003-01-30 22:06:23 +00004570 {"__reduce__", (PyCFunction)datetime_reduce, METH_NOARGS,
4571 PyDoc_STR("__reduce__() -> (cls, state)")},
4572
Tim Peters2a799bf2002-12-16 20:18:38 +00004573 {NULL, NULL}
4574};
4575
Tim Petersa9bc1682003-01-11 03:39:11 +00004576static char datetime_doc[] =
Raymond Hettinger3a4231d2004-12-19 20:13:24 +00004577PyDoc_STR("datetime(year, month, day[, hour[, minute[, second[, microsecond[,tzinfo]]]]])\n\
4578\n\
4579The year, month and day arguments are required. tzinfo may be None, or an\n\
4580instance of a tzinfo subclass. The remaining arguments may be ints or longs.\n");
Tim Peters2a799bf2002-12-16 20:18:38 +00004581
Tim Petersa9bc1682003-01-11 03:39:11 +00004582static PyNumberMethods datetime_as_number = {
4583 datetime_add, /* nb_add */
4584 datetime_subtract, /* nb_subtract */
Tim Peters2a799bf2002-12-16 20:18:38 +00004585 0, /* nb_multiply */
Tim Peters2a799bf2002-12-16 20:18:38 +00004586 0, /* nb_remainder */
4587 0, /* nb_divmod */
4588 0, /* nb_power */
4589 0, /* nb_negative */
4590 0, /* nb_positive */
4591 0, /* nb_absolute */
Jack Diederich4dafcc42006-11-28 19:15:13 +00004592 0, /* nb_bool */
Tim Peters2a799bf2002-12-16 20:18:38 +00004593};
4594
Neal Norwitz227b5332006-03-22 09:28:35 +00004595static PyTypeObject PyDateTime_DateTimeType = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00004596 PyVarObject_HEAD_INIT(NULL, 0)
Tim Peters0bf60bd2003-01-08 20:40:01 +00004597 "datetime.datetime", /* tp_name */
Tim Petersa9bc1682003-01-11 03:39:11 +00004598 sizeof(PyDateTime_DateTime), /* tp_basicsize */
Tim Peters2a799bf2002-12-16 20:18:38 +00004599 0, /* tp_itemsize */
Tim Petersa9bc1682003-01-11 03:39:11 +00004600 (destructor)datetime_dealloc, /* tp_dealloc */
Tim Peters2a799bf2002-12-16 20:18:38 +00004601 0, /* tp_print */
4602 0, /* tp_getattr */
4603 0, /* tp_setattr */
4604 0, /* tp_compare */
Tim Petersa9bc1682003-01-11 03:39:11 +00004605 (reprfunc)datetime_repr, /* tp_repr */
4606 &datetime_as_number, /* tp_as_number */
Tim Peters2a799bf2002-12-16 20:18:38 +00004607 0, /* tp_as_sequence */
4608 0, /* tp_as_mapping */
Tim Petersa9bc1682003-01-11 03:39:11 +00004609 (hashfunc)datetime_hash, /* tp_hash */
Tim Peters2a799bf2002-12-16 20:18:38 +00004610 0, /* tp_call */
Tim Petersa9bc1682003-01-11 03:39:11 +00004611 (reprfunc)datetime_str, /* tp_str */
Tim Peters2a799bf2002-12-16 20:18:38 +00004612 PyObject_GenericGetAttr, /* tp_getattro */
4613 0, /* tp_setattro */
4614 0, /* tp_as_buffer */
Guido van Rossum3cf5b1e2006-07-27 21:53:35 +00004615 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
Tim Petersa9bc1682003-01-11 03:39:11 +00004616 datetime_doc, /* tp_doc */
Tim Peters2a799bf2002-12-16 20:18:38 +00004617 0, /* tp_traverse */
4618 0, /* tp_clear */
Guido van Rossum19960592006-08-24 17:29:38 +00004619 datetime_richcompare, /* tp_richcompare */
Tim Peters2a799bf2002-12-16 20:18:38 +00004620 0, /* tp_weaklistoffset */
4621 0, /* tp_iter */
4622 0, /* tp_iternext */
Tim Petersa9bc1682003-01-11 03:39:11 +00004623 datetime_methods, /* tp_methods */
Tim Peters2a799bf2002-12-16 20:18:38 +00004624 0, /* tp_members */
Tim Petersa9bc1682003-01-11 03:39:11 +00004625 datetime_getset, /* tp_getset */
4626 &PyDateTime_DateType, /* tp_base */
Tim Peters2a799bf2002-12-16 20:18:38 +00004627 0, /* tp_dict */
4628 0, /* tp_descr_get */
4629 0, /* tp_descr_set */
4630 0, /* tp_dictoffset */
4631 0, /* tp_init */
Tim Petersa98924a2003-05-17 05:55:19 +00004632 datetime_alloc, /* tp_alloc */
Tim Petersa9bc1682003-01-11 03:39:11 +00004633 datetime_new, /* tp_new */
Tim Peters4c530132003-05-16 22:44:06 +00004634 0, /* tp_free */
Tim Peters2a799bf2002-12-16 20:18:38 +00004635};
4636
4637/* ---------------------------------------------------------------------------
4638 * Module methods and initialization.
4639 */
4640
4641static PyMethodDef module_methods[] = {
Tim Peters2a799bf2002-12-16 20:18:38 +00004642 {NULL, NULL}
4643};
4644
Tim Peters9ddf40b2004-06-20 22:41:32 +00004645/* C API. Clients get at this via PyDateTime_IMPORT, defined in
4646 * datetime.h.
4647 */
4648static PyDateTime_CAPI CAPI = {
4649 &PyDateTime_DateType,
4650 &PyDateTime_DateTimeType,
4651 &PyDateTime_TimeType,
4652 &PyDateTime_DeltaType,
4653 &PyDateTime_TZInfoType,
4654 new_date_ex,
4655 new_datetime_ex,
4656 new_time_ex,
4657 new_delta_ex,
4658 datetime_fromtimestamp,
4659 date_fromtimestamp
4660};
4661
4662
Tim Peters2a799bf2002-12-16 20:18:38 +00004663PyMODINIT_FUNC
4664initdatetime(void)
4665{
4666 PyObject *m; /* a module object */
4667 PyObject *d; /* its dict */
4668 PyObject *x;
4669
Tim Peters2a799bf2002-12-16 20:18:38 +00004670 m = Py_InitModule3("datetime", module_methods,
4671 "Fast implementation of the datetime type.");
Neal Norwitz1ac754f2006-01-19 06:09:39 +00004672 if (m == NULL)
4673 return;
Tim Peters2a799bf2002-12-16 20:18:38 +00004674
4675 if (PyType_Ready(&PyDateTime_DateType) < 0)
4676 return;
4677 if (PyType_Ready(&PyDateTime_DateTimeType) < 0)
4678 return;
4679 if (PyType_Ready(&PyDateTime_DeltaType) < 0)
4680 return;
4681 if (PyType_Ready(&PyDateTime_TimeType) < 0)
4682 return;
4683 if (PyType_Ready(&PyDateTime_TZInfoType) < 0)
4684 return;
Tim Peters2a799bf2002-12-16 20:18:38 +00004685
Tim Peters2a799bf2002-12-16 20:18:38 +00004686 /* timedelta values */
4687 d = PyDateTime_DeltaType.tp_dict;
4688
Tim Peters2a799bf2002-12-16 20:18:38 +00004689 x = new_delta(0, 0, 1, 0);
4690 if (x == NULL || PyDict_SetItemString(d, "resolution", x) < 0)
4691 return;
4692 Py_DECREF(x);
4693
4694 x = new_delta(-MAX_DELTA_DAYS, 0, 0, 0);
4695 if (x == NULL || PyDict_SetItemString(d, "min", x) < 0)
4696 return;
4697 Py_DECREF(x);
4698
4699 x = new_delta(MAX_DELTA_DAYS, 24*3600-1, 1000000-1, 0);
4700 if (x == NULL || PyDict_SetItemString(d, "max", x) < 0)
4701 return;
4702 Py_DECREF(x);
4703
4704 /* date values */
4705 d = PyDateTime_DateType.tp_dict;
4706
4707 x = new_date(1, 1, 1);
4708 if (x == NULL || PyDict_SetItemString(d, "min", x) < 0)
4709 return;
4710 Py_DECREF(x);
4711
4712 x = new_date(MAXYEAR, 12, 31);
4713 if (x == NULL || PyDict_SetItemString(d, "max", x) < 0)
4714 return;
4715 Py_DECREF(x);
4716
4717 x = new_delta(1, 0, 0, 0);
4718 if (x == NULL || PyDict_SetItemString(d, "resolution", x) < 0)
4719 return;
4720 Py_DECREF(x);
4721
Tim Peters37f39822003-01-10 03:49:02 +00004722 /* time values */
4723 d = PyDateTime_TimeType.tp_dict;
Tim Peters2a799bf2002-12-16 20:18:38 +00004724
Tim Peters37f39822003-01-10 03:49:02 +00004725 x = new_time(0, 0, 0, 0, Py_None);
Tim Peters2a799bf2002-12-16 20:18:38 +00004726 if (x == NULL || PyDict_SetItemString(d, "min", x) < 0)
4727 return;
4728 Py_DECREF(x);
4729
Tim Peters37f39822003-01-10 03:49:02 +00004730 x = new_time(23, 59, 59, 999999, Py_None);
Tim Peters2a799bf2002-12-16 20:18:38 +00004731 if (x == NULL || PyDict_SetItemString(d, "max", x) < 0)
4732 return;
4733 Py_DECREF(x);
4734
4735 x = new_delta(0, 0, 1, 0);
4736 if (x == NULL || PyDict_SetItemString(d, "resolution", x) < 0)
4737 return;
4738 Py_DECREF(x);
4739
Tim Petersa9bc1682003-01-11 03:39:11 +00004740 /* datetime values */
4741 d = PyDateTime_DateTimeType.tp_dict;
Tim Peters2a799bf2002-12-16 20:18:38 +00004742
Tim Petersa9bc1682003-01-11 03:39:11 +00004743 x = new_datetime(1, 1, 1, 0, 0, 0, 0, Py_None);
Tim Peters2a799bf2002-12-16 20:18:38 +00004744 if (x == NULL || PyDict_SetItemString(d, "min", x) < 0)
4745 return;
4746 Py_DECREF(x);
4747
Tim Petersa9bc1682003-01-11 03:39:11 +00004748 x = new_datetime(MAXYEAR, 12, 31, 23, 59, 59, 999999, Py_None);
Tim Peters2a799bf2002-12-16 20:18:38 +00004749 if (x == NULL || PyDict_SetItemString(d, "max", x) < 0)
4750 return;
4751 Py_DECREF(x);
4752
4753 x = new_delta(0, 0, 1, 0);
4754 if (x == NULL || PyDict_SetItemString(d, "resolution", x) < 0)
4755 return;
4756 Py_DECREF(x);
4757
Tim Peters2a799bf2002-12-16 20:18:38 +00004758 /* module initialization */
4759 PyModule_AddIntConstant(m, "MINYEAR", MINYEAR);
4760 PyModule_AddIntConstant(m, "MAXYEAR", MAXYEAR);
4761
4762 Py_INCREF(&PyDateTime_DateType);
4763 PyModule_AddObject(m, "date", (PyObject *) &PyDateTime_DateType);
4764
Tim Petersa9bc1682003-01-11 03:39:11 +00004765 Py_INCREF(&PyDateTime_DateTimeType);
4766 PyModule_AddObject(m, "datetime",
4767 (PyObject *)&PyDateTime_DateTimeType);
4768
4769 Py_INCREF(&PyDateTime_TimeType);
4770 PyModule_AddObject(m, "time", (PyObject *) &PyDateTime_TimeType);
4771
Tim Peters2a799bf2002-12-16 20:18:38 +00004772 Py_INCREF(&PyDateTime_DeltaType);
4773 PyModule_AddObject(m, "timedelta", (PyObject *) &PyDateTime_DeltaType);
4774
Tim Peters2a799bf2002-12-16 20:18:38 +00004775 Py_INCREF(&PyDateTime_TZInfoType);
4776 PyModule_AddObject(m, "tzinfo", (PyObject *) &PyDateTime_TZInfoType);
4777
Tim Peters9ddf40b2004-06-20 22:41:32 +00004778 x = PyCObject_FromVoidPtrAndDesc(&CAPI, (void*) DATETIME_API_MAGIC,
4779 NULL);
4780 if (x == NULL)
4781 return;
4782 PyModule_AddObject(m, "datetime_CAPI", x);
4783
Tim Peters2a799bf2002-12-16 20:18:38 +00004784 /* A 4-year cycle has an extra leap day over what we'd get from
4785 * pasting together 4 single years.
4786 */
4787 assert(DI4Y == 4 * 365 + 1);
4788 assert(DI4Y == days_before_year(4+1));
4789
4790 /* Similarly, a 400-year cycle has an extra leap day over what we'd
4791 * get from pasting together 4 100-year cycles.
4792 */
4793 assert(DI400Y == 4 * DI100Y + 1);
4794 assert(DI400Y == days_before_year(400+1));
4795
4796 /* OTOH, a 100-year cycle has one fewer leap day than we'd get from
4797 * pasting together 25 4-year cycles.
4798 */
4799 assert(DI100Y == 25 * DI4Y - 1);
4800 assert(DI100Y == days_before_year(100+1));
4801
Christian Heimes217cfd12007-12-02 14:31:20 +00004802 us_per_us = PyLong_FromLong(1);
4803 us_per_ms = PyLong_FromLong(1000);
4804 us_per_second = PyLong_FromLong(1000000);
4805 us_per_minute = PyLong_FromLong(60000000);
4806 seconds_per_day = PyLong_FromLong(24 * 3600);
Tim Peters2a799bf2002-12-16 20:18:38 +00004807 if (us_per_us == NULL || us_per_ms == NULL || us_per_second == NULL ||
4808 us_per_minute == NULL || seconds_per_day == NULL)
4809 return;
4810
4811 /* The rest are too big for 32-bit ints, but even
4812 * us_per_week fits in 40 bits, so doubles should be exact.
4813 */
4814 us_per_hour = PyLong_FromDouble(3600000000.0);
4815 us_per_day = PyLong_FromDouble(86400000000.0);
4816 us_per_week = PyLong_FromDouble(604800000000.0);
4817 if (us_per_hour == NULL || us_per_day == NULL || us_per_week == NULL)
4818 return;
4819}
Tim Petersf3615152003-01-01 21:51:37 +00004820
4821/* ---------------------------------------------------------------------------
Tim Petersa9bc1682003-01-11 03:39:11 +00004822Some time zone algebra. For a datetime x, let
Tim Petersf3615152003-01-01 21:51:37 +00004823 x.n = x stripped of its timezone -- its naive time.
4824 x.o = x.utcoffset(), and assuming that doesn't raise an exception or
4825 return None
4826 x.d = x.dst(), and assuming that doesn't raise an exception or
4827 return None
4828 x.s = x's standard offset, x.o - x.d
4829
4830Now some derived rules, where k is a duration (timedelta).
4831
48321. x.o = x.s + x.d
4833 This follows from the definition of x.s.
4834
Tim Petersc5dc4da2003-01-02 17:55:03 +000048352. If x and y have the same tzinfo member, x.s = y.s.
Tim Petersf3615152003-01-01 21:51:37 +00004836 This is actually a requirement, an assumption we need to make about
4837 sane tzinfo classes.
4838
48393. The naive UTC time corresponding to x is x.n - x.o.
4840 This is again a requirement for a sane tzinfo class.
4841
48424. (x+k).s = x.s
Tim Peters8bb5ad22003-01-24 02:44:45 +00004843 This follows from #2, and that datimetimetz+timedelta preserves tzinfo.
Tim Petersf3615152003-01-01 21:51:37 +00004844
Tim Petersc5dc4da2003-01-02 17:55:03 +000048455. (x+k).n = x.n + k
Tim Petersf3615152003-01-01 21:51:37 +00004846 Again follows from how arithmetic is defined.
4847
Tim Peters8bb5ad22003-01-24 02:44:45 +00004848Now we can explain tz.fromutc(x). Let's assume it's an interesting case
Tim Petersf3615152003-01-01 21:51:37 +00004849(meaning that the various tzinfo methods exist, and don't blow up or return
4850None when called).
4851
Tim Petersa9bc1682003-01-11 03:39:11 +00004852The function wants to return a datetime y with timezone tz, equivalent to x.
Tim Peters8bb5ad22003-01-24 02:44:45 +00004853x is already in UTC.
Tim Petersf3615152003-01-01 21:51:37 +00004854
4855By #3, we want
4856
Tim Peters8bb5ad22003-01-24 02:44:45 +00004857 y.n - y.o = x.n [1]
Tim Petersf3615152003-01-01 21:51:37 +00004858
4859The algorithm starts by attaching tz to x.n, and calling that y. So
4860x.n = y.n at the start. Then it wants to add a duration k to y, so that [1]
4861becomes true; in effect, we want to solve [2] for k:
4862
Tim Peters8bb5ad22003-01-24 02:44:45 +00004863 (y+k).n - (y+k).o = x.n [2]
Tim Petersf3615152003-01-01 21:51:37 +00004864
4865By #1, this is the same as
4866
Tim Peters8bb5ad22003-01-24 02:44:45 +00004867 (y+k).n - ((y+k).s + (y+k).d) = x.n [3]
Tim Petersf3615152003-01-01 21:51:37 +00004868
4869By #5, (y+k).n = y.n + k, which equals x.n + k because x.n=y.n at the start.
4870Substituting that into [3],
4871
Tim Peters8bb5ad22003-01-24 02:44:45 +00004872 x.n + k - (y+k).s - (y+k).d = x.n; the x.n terms cancel, leaving
4873 k - (y+k).s - (y+k).d = 0; rearranging,
4874 k = (y+k).s - (y+k).d; by #4, (y+k).s == y.s, so
4875 k = y.s - (y+k).d
Tim Petersf3615152003-01-01 21:51:37 +00004876
Tim Peters8bb5ad22003-01-24 02:44:45 +00004877On the RHS, (y+k).d can't be computed directly, but y.s can be, and we
4878approximate k by ignoring the (y+k).d term at first. Note that k can't be
4879very large, since all offset-returning methods return a duration of magnitude
4880less than 24 hours. For that reason, if y is firmly in std time, (y+k).d must
4881be 0, so ignoring it has no consequence then.
Tim Petersf3615152003-01-01 21:51:37 +00004882
4883In any case, the new value is
4884
Tim Peters8bb5ad22003-01-24 02:44:45 +00004885 z = y + y.s [4]
Tim Petersf3615152003-01-01 21:51:37 +00004886
Tim Peters8bb5ad22003-01-24 02:44:45 +00004887It's helpful to step back at look at [4] from a higher level: it's simply
4888mapping from UTC to tz's standard time.
Tim Petersc5dc4da2003-01-02 17:55:03 +00004889
4890At this point, if
4891
Tim Peters8bb5ad22003-01-24 02:44:45 +00004892 z.n - z.o = x.n [5]
Tim Petersc5dc4da2003-01-02 17:55:03 +00004893
4894we have an equivalent time, and are almost done. The insecurity here is
Tim Petersf3615152003-01-01 21:51:37 +00004895at the start of daylight time. Picture US Eastern for concreteness. The wall
4896time jumps from 1:59 to 3:00, and wall hours of the form 2:MM don't make good
Tim Peters8bb5ad22003-01-24 02:44:45 +00004897sense then. The docs ask that an Eastern tzinfo class consider such a time to
4898be EDT (because it's "after 2"), which is a redundant spelling of 1:MM EST
4899on the day DST starts. We want to return the 1:MM EST spelling because that's
Tim Petersf3615152003-01-01 21:51:37 +00004900the only spelling that makes sense on the local wall clock.
4901
Tim Petersc5dc4da2003-01-02 17:55:03 +00004902In fact, if [5] holds at this point, we do have the standard-time spelling,
4903but that takes a bit of proof. We first prove a stronger result. What's the
4904difference between the LHS and RHS of [5]? Let
Tim Petersf3615152003-01-01 21:51:37 +00004905
Tim Peters8bb5ad22003-01-24 02:44:45 +00004906 diff = x.n - (z.n - z.o) [6]
Tim Petersf3615152003-01-01 21:51:37 +00004907
Tim Petersc5dc4da2003-01-02 17:55:03 +00004908Now
4909 z.n = by [4]
Tim Peters8bb5ad22003-01-24 02:44:45 +00004910 (y + y.s).n = by #5
4911 y.n + y.s = since y.n = x.n
4912 x.n + y.s = since z and y are have the same tzinfo member,
4913 y.s = z.s by #2
4914 x.n + z.s
Tim Petersf3615152003-01-01 21:51:37 +00004915
Tim Petersc5dc4da2003-01-02 17:55:03 +00004916Plugging that back into [6] gives
Tim Petersf3615152003-01-01 21:51:37 +00004917
Tim Petersc5dc4da2003-01-02 17:55:03 +00004918 diff =
Tim Peters8bb5ad22003-01-24 02:44:45 +00004919 x.n - ((x.n + z.s) - z.o) = expanding
4920 x.n - x.n - z.s + z.o = cancelling
4921 - z.s + z.o = by #2
Tim Petersc5dc4da2003-01-02 17:55:03 +00004922 z.d
Tim Petersf3615152003-01-01 21:51:37 +00004923
Tim Petersc5dc4da2003-01-02 17:55:03 +00004924So diff = z.d.
Tim Petersf3615152003-01-01 21:51:37 +00004925
Tim Petersc5dc4da2003-01-02 17:55:03 +00004926If [5] is true now, diff = 0, so z.d = 0 too, and we have the standard-time
Tim Peters8bb5ad22003-01-24 02:44:45 +00004927spelling we wanted in the endcase described above. We're done. Contrarily,
4928if z.d = 0, then we have a UTC equivalent, and are also done.
Tim Petersf3615152003-01-01 21:51:37 +00004929
Tim Petersc5dc4da2003-01-02 17:55:03 +00004930If [5] is not true now, diff = z.d != 0, and z.d is the offset we need to
4931add to z (in effect, z is in tz's standard time, and we need to shift the
Tim Peters8bb5ad22003-01-24 02:44:45 +00004932local clock into tz's daylight time).
Tim Petersf3615152003-01-01 21:51:37 +00004933
Tim Petersc5dc4da2003-01-02 17:55:03 +00004934Let
Tim Petersf3615152003-01-01 21:51:37 +00004935
Tim Peters4fede1a2003-01-04 00:26:59 +00004936 z' = z + z.d = z + diff [7]
Tim Petersc3bb26a2003-01-02 03:14:59 +00004937
Tim Peters4fede1a2003-01-04 00:26:59 +00004938and we can again ask whether
Tim Petersc3bb26a2003-01-02 03:14:59 +00004939
Tim Peters8bb5ad22003-01-24 02:44:45 +00004940 z'.n - z'.o = x.n [8]
Tim Petersc3bb26a2003-01-02 03:14:59 +00004941
Tim Peters8bb5ad22003-01-24 02:44:45 +00004942If so, we're done. If not, the tzinfo class is insane, according to the
4943assumptions we've made. This also requires a bit of proof. As before, let's
4944compute the difference between the LHS and RHS of [8] (and skipping some of
4945the justifications for the kinds of substitutions we've done several times
4946already):
Tim Peters4fede1a2003-01-04 00:26:59 +00004947
Tim Peters8bb5ad22003-01-24 02:44:45 +00004948 diff' = x.n - (z'.n - z'.o) = replacing z'.n via [7]
4949 x.n - (z.n + diff - z'.o) = replacing diff via [6]
4950 x.n - (z.n + x.n - (z.n - z.o) - z'.o) =
4951 x.n - z.n - x.n + z.n - z.o + z'.o = cancel x.n
4952 - z.n + z.n - z.o + z'.o = cancel z.n
Tim Peters4fede1a2003-01-04 00:26:59 +00004953 - z.o + z'.o = #1 twice
4954 -z.s - z.d + z'.s + z'.d = z and z' have same tzinfo
4955 z'.d - z.d
4956
4957So z' is UTC-equivalent to x iff z'.d = z.d at this point. If they are equal,
Tim Peters8bb5ad22003-01-24 02:44:45 +00004958we've found the UTC-equivalent so are done. In fact, we stop with [7] and
4959return z', not bothering to compute z'.d.
Tim Peters4fede1a2003-01-04 00:26:59 +00004960
Tim Peters8bb5ad22003-01-24 02:44:45 +00004961How could z.d and z'd differ? z' = z + z.d [7], so merely moving z' by
4962a dst() offset, and starting *from* a time already in DST (we know z.d != 0),
4963would have to change the result dst() returns: we start in DST, and moving
4964a little further into it takes us out of DST.
Tim Peters4fede1a2003-01-04 00:26:59 +00004965
Tim Peters8bb5ad22003-01-24 02:44:45 +00004966There isn't a sane case where this can happen. The closest it gets is at
4967the end of DST, where there's an hour in UTC with no spelling in a hybrid
4968tzinfo class. In US Eastern, that's 5:MM UTC = 0:MM EST = 1:MM EDT. During
4969that hour, on an Eastern clock 1:MM is taken as being in standard time (6:MM
4970UTC) because the docs insist on that, but 0:MM is taken as being in daylight
4971time (4:MM UTC). There is no local time mapping to 5:MM UTC. The local
4972clock jumps from 1:59 back to 1:00 again, and repeats the 1:MM hour in
4973standard time. Since that's what the local clock *does*, we want to map both
4974UTC hours 5:MM and 6:MM to 1:MM Eastern. The result is ambiguous
Tim Peters4fede1a2003-01-04 00:26:59 +00004975in local time, but so it goes -- it's the way the local clock works.
4976
Tim Peters8bb5ad22003-01-24 02:44:45 +00004977When x = 5:MM UTC is the input to this algorithm, x.o=0, y.o=-5 and y.d=0,
4978so z=0:MM. z.d=60 (minutes) then, so [5] doesn't hold and we keep going.
4979z' = z + z.d = 1:MM then, and z'.d=0, and z'.d - z.d = -60 != 0 so [8]
Tim Peters4fede1a2003-01-04 00:26:59 +00004980(correctly) concludes that z' is not UTC-equivalent to x.
4981
4982Because we know z.d said z was in daylight time (else [5] would have held and
4983we would have stopped then), and we know z.d != z'.d (else [8] would have held
Walter Dörwaldf0dfc7a2003-10-20 14:01:56 +00004984and we would have stopped then), and there are only 2 possible values dst() can
Tim Peters4fede1a2003-01-04 00:26:59 +00004985return in Eastern, it follows that z'.d must be 0 (which it is in the example,
4986but the reasoning doesn't depend on the example -- it depends on there being
4987two possible dst() outcomes, one zero and the other non-zero). Therefore
Tim Peters8bb5ad22003-01-24 02:44:45 +00004988z' must be in standard time, and is the spelling we want in this case.
4989
4990Note again that z' is not UTC-equivalent as far as the hybrid tzinfo class is
4991concerned (because it takes z' as being in standard time rather than the
4992daylight time we intend here), but returning it gives the real-life "local
4993clock repeats an hour" behavior when mapping the "unspellable" UTC hour into
4994tz.
4995
4996When the input is 6:MM, z=1:MM and z.d=0, and we stop at once, again with
4997the 1:MM standard time spelling we want.
4998
4999So how can this break? One of the assumptions must be violated. Two
5000possibilities:
5001
50021) [2] effectively says that y.s is invariant across all y belong to a given
5003 time zone. This isn't true if, for political reasons or continental drift,
5004 a region decides to change its base offset from UTC.
5005
50062) There may be versions of "double daylight" time where the tail end of
5007 the analysis gives up a step too early. I haven't thought about that
5008 enough to say.
5009
5010In any case, it's clear that the default fromutc() is strong enough to handle
5011"almost all" time zones: so long as the standard offset is invariant, it
5012doesn't matter if daylight time transition points change from year to year, or
5013if daylight time is skipped in some years; it doesn't matter how large or
5014small dst() may get within its bounds; and it doesn't even matter if some
5015perverse time zone returns a negative dst()). So a breaking case must be
5016pretty bizarre, and a tzinfo subclass can override fromutc() if it is.
Tim Petersf3615152003-01-01 21:51:37 +00005017--------------------------------------------------------------------------- */