blob: 78dec7c94a6a7975ceeda82626c34c05c22aed61 [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002/* Time module */
3
Barry Warsaw9a2a8a81996-12-06 23:32:14 +00004#include "Python.h"
Guido van Rossum98bf58f2001-10-18 20:34:25 +00005#include "structseq.h"
Tim Peters1b6f7a92004-06-20 02:50:16 +00006#include "timefuncs.h"
Guido van Rossum3f5da241990-12-20 15:06:42 +00007
Ronald Oussorend06b6f22006-04-23 11:59:25 +00008#ifdef __APPLE__
9#if defined(HAVE_GETTIMEOFDAY) && defined(HAVE_FTIME)
10 /*
11 * floattime falls back to ftime when getttimeofday fails because the latter
12 * might fail on some platforms. This fallback is unwanted on MacOSX because
13 * that makes it impossible to use a binary build on OSX 10.4 on earlier
14 * releases of the OS. Therefore claim we don't support ftime.
15 */
16# undef HAVE_FTIME
17#endif
18#endif
19
Guido van Rossum87ce7bb1998-06-09 16:30:31 +000020#include <ctype.h>
21
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +000022#ifdef HAVE_SYS_TYPES_H
Guido van Rossumb6775db1994-08-01 11:34:53 +000023#include <sys/types.h>
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +000024#endif /* HAVE_SYS_TYPES_H */
Guido van Rossum6d946f91992-08-14 13:49:30 +000025
Guido van Rossumb6775db1994-08-01 11:34:53 +000026#ifdef QUICKWIN
27#include <io.h>
28#endif
29
Guido van Rossumb6775db1994-08-01 11:34:53 +000030#ifdef HAVE_FTIME
31#include <sys/timeb.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000032#if !defined(MS_WINDOWS) && !defined(PYOS_OS2)
Thomas Woutersbd4bc4e2000-07-22 23:57:55 +000033extern int ftime(struct timeb *);
Guido van Rossum52174571996-12-09 18:38:52 +000034#endif /* MS_WINDOWS */
35#endif /* HAVE_FTIME */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000036
Guido van Rossum7bf22de1997-12-02 20:34:19 +000037#if defined(__WATCOMC__) && !defined(__QNX__)
Guido van Rossumbceeac81996-05-23 22:53:47 +000038#include <i86.h>
39#else
Guido van Rossumcac6c721996-09-06 13:34:02 +000040#ifdef MS_WINDOWS
Mark Hammond975e3922002-07-16 01:29:19 +000041#define WIN32_LEAN_AND_MEAN
Guido van Rossum258ccd42001-03-02 06:53:29 +000042#include <windows.h>
Mark Hammond975e3922002-07-16 01:29:19 +000043#include "pythread.h"
44
45/* helper to allow us to interrupt sleep() on Windows*/
46static HANDLE hInterruptEvent = NULL;
47static BOOL WINAPI PyCtrlHandler(DWORD dwCtrlType)
48{
Antoine Pitrouc83ea132010-05-09 14:46:46 +000049 SetEvent(hInterruptEvent);
50 /* allow other default handlers to be called.
51 Default Python handler will setup the
52 KeyboardInterrupt exception.
53 */
54 return FALSE;
Mark Hammond975e3922002-07-16 01:29:19 +000055}
56static long main_thread;
57
58
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000059#if defined(__BORLANDC__)
Guido van Rossumb2fb3641996-09-07 00:47:35 +000060/* These overrides not needed for Win32 */
Guido van Rossumb6775db1994-08-01 11:34:53 +000061#define timezone _timezone
Guido van Rossumcc081121995-03-14 15:05:41 +000062#define tzname _tzname
63#define daylight _daylight
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000064#endif /* __BORLANDC__ */
Guido van Rossumcac6c721996-09-06 13:34:02 +000065#endif /* MS_WINDOWS */
Guido van Rossum7bf22de1997-12-02 20:34:19 +000066#endif /* !__WATCOMC__ || __QNX__ */
Guido van Rossum234f9421993-06-17 12:35:49 +000067
Tim Peters7a822da2006-05-25 21:50:17 +000068#if defined(MS_WINDOWS) && !defined(__BORLANDC__)
Tim Petersc285e622006-05-25 22:25:25 +000069/* Win32 has better clock replacement; we have our own version below. */
70#undef HAVE_CLOCK
Tim Peters7a822da2006-05-25 21:50:17 +000071#endif /* MS_WINDOWS && !defined(__BORLANDC__) */
Guido van Rossum3917c221997-04-02 05:35:28 +000072
Andrew MacIntyre7bf68332002-03-03 02:59:16 +000073#if defined(PYOS_OS2)
74#define INCL_DOS
75#define INCL_ERRORS
76#include <os2.h>
77#endif
78
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000079#if defined(PYCC_VACPP)
Guido van Rossum26452411998-09-28 22:07:11 +000080#include <sys/time.h>
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +000081#endif
82
Guido van Rossumbcc20741998-08-04 22:53:56 +000083#ifdef __BEOS__
Fred Drake56221a72000-08-15 18:52:33 +000084#include <time.h>
Guido van Rossumbcc20741998-08-04 22:53:56 +000085/* For bigtime_t, snooze(). - [cjh] */
86#include <support/SupportDefs.h>
87#include <kernel/OS.h>
88#endif
89
Martin v. Löwisa94568a2003-05-10 07:36:56 +000090#ifdef RISCOS
91extern int riscos_sleep(double);
92#endif
93
Guido van Rossum234f9421993-06-17 12:35:49 +000094/* Forward declarations */
Tim Petersdbd9ba62000-07-09 03:09:57 +000095static int floatsleep(double);
Thomas Woutersed77bac2000-07-24 15:26:39 +000096static double floattime(void);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000097
Guido van Rossumcfbaecc1998-08-25 14:51:12 +000098/* For Y2K check */
99static PyObject *moddict;
100
Tim Peters1b6f7a92004-06-20 02:50:16 +0000101/* Exposed in timefuncs.h. */
102time_t
Brett Cannon298c3802004-06-19 20:48:43 +0000103_PyTime_DoubleToTimet(double x)
104{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000105 time_t result;
106 double diff;
Brett Cannon298c3802004-06-19 20:48:43 +0000107
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000108 result = (time_t)x;
109 /* How much info did we lose? time_t may be an integral or
110 * floating type, and we don't know which. If it's integral,
111 * we don't know whether C truncates, rounds, returns the floor,
112 * etc. If we lost a second or more, the C rounding is
113 * unreasonable, or the input just doesn't fit in a time_t;
114 * call it an error regardless. Note that the original cast to
115 * time_t can cause a C error too, but nothing we can do to
116 * worm around that.
117 */
118 diff = x - (double)result;
119 if (diff <= -1.0 || diff >= 1.0) {
120 PyErr_SetString(PyExc_ValueError,
121 "timestamp out of range for platform time_t");
122 result = (time_t)-1;
123 }
124 return result;
Brett Cannon298c3802004-06-19 20:48:43 +0000125}
126
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000127static PyObject *
Georg Brandl96a8c392006-05-29 21:04:52 +0000128time_time(PyObject *self, PyObject *unused)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000129{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000130 double secs;
131 secs = floattime();
132 if (secs == 0.0) {
133 PyErr_SetFromErrno(PyExc_IOError);
134 return NULL;
135 }
136 return PyFloat_FromDouble(secs);
Guido van Rossumb6775db1994-08-01 11:34:53 +0000137}
138
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000139PyDoc_STRVAR(time_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000140"time() -> floating point number\n\
141\n\
142Return the current time in seconds since the Epoch.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000143Fractions of a second may be present if the system clock provides them.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000144
Guido van Rossumb6775db1994-08-01 11:34:53 +0000145#ifdef HAVE_CLOCK
146
147#ifndef CLOCKS_PER_SEC
Guido van Rossum1b66a4f1996-02-25 04:50:33 +0000148#ifdef CLK_TCK
149#define CLOCKS_PER_SEC CLK_TCK
150#else
Guido van Rossumb6775db1994-08-01 11:34:53 +0000151#define CLOCKS_PER_SEC 1000000
152#endif
Guido van Rossum1b66a4f1996-02-25 04:50:33 +0000153#endif
Guido van Rossumb6775db1994-08-01 11:34:53 +0000154
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000155static PyObject *
Georg Brandl96a8c392006-05-29 21:04:52 +0000156time_clock(PyObject *self, PyObject *unused)
Guido van Rossumb6775db1994-08-01 11:34:53 +0000157{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000158 return PyFloat_FromDouble(((double)clock()) / CLOCKS_PER_SEC);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000159}
Guido van Rossumb6775db1994-08-01 11:34:53 +0000160#endif /* HAVE_CLOCK */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000161
Tim Petersc285e622006-05-25 22:25:25 +0000162#if defined(MS_WINDOWS) && !defined(__BORLANDC__)
Mark Hammond7ba5e812002-02-12 04:02:33 +0000163/* Due to Mark Hammond and Tim Peters */
Guido van Rossum3917c221997-04-02 05:35:28 +0000164static PyObject *
Georg Brandl96a8c392006-05-29 21:04:52 +0000165time_clock(PyObject *self, PyObject *unused)
Guido van Rossum3917c221997-04-02 05:35:28 +0000166{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000167 static LARGE_INTEGER ctrStart;
168 static double divisor = 0.0;
169 LARGE_INTEGER now;
170 double diff;
Guido van Rossum3917c221997-04-02 05:35:28 +0000171
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000172 if (divisor == 0.0) {
173 LARGE_INTEGER freq;
174 QueryPerformanceCounter(&ctrStart);
175 if (!QueryPerformanceFrequency(&freq) || freq.QuadPart == 0) {
176 /* Unlikely to happen - this works on all intel
177 machines at least! Revert to clock() */
178 return PyFloat_FromDouble(((double)clock()) /
179 CLOCKS_PER_SEC);
180 }
181 divisor = (double)freq.QuadPart;
182 }
183 QueryPerformanceCounter(&now);
184 diff = (double)(now.QuadPart - ctrStart.QuadPart);
185 return PyFloat_FromDouble(diff / divisor);
Guido van Rossum3917c221997-04-02 05:35:28 +0000186}
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000187
Guido van Rossum3917c221997-04-02 05:35:28 +0000188#define HAVE_CLOCK /* So it gets included in the methods */
Tim Petersc285e622006-05-25 22:25:25 +0000189#endif /* MS_WINDOWS && !defined(__BORLANDC__) */
Guido van Rossum3917c221997-04-02 05:35:28 +0000190
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000191#ifdef HAVE_CLOCK
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000192PyDoc_STRVAR(clock_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000193"clock() -> floating point number\n\
194\n\
195Return the CPU time or real time since the start of the process or since\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000196the first call to clock(). This has as much precision as the system\n\
197records.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000198#endif
199
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000200static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000201time_sleep(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000202{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000203 double secs;
204 if (!PyArg_ParseTuple(args, "d:sleep", &secs))
205 return NULL;
206 if (floatsleep(secs) != 0)
207 return NULL;
208 Py_INCREF(Py_None);
209 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000210}
211
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000212PyDoc_STRVAR(sleep_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000213"sleep(seconds)\n\
214\n\
215Delay execution for a given number of seconds. The argument may be\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000216a floating point number for subsecond precision.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000217
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000218static PyStructSequence_Field struct_time_type_fields[] = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000219 {"tm_year", NULL},
220 {"tm_mon", NULL},
221 {"tm_mday", NULL},
222 {"tm_hour", NULL},
223 {"tm_min", NULL},
224 {"tm_sec", NULL},
225 {"tm_wday", NULL},
226 {"tm_yday", NULL},
227 {"tm_isdst", NULL},
228 {0}
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000229};
230
231static PyStructSequence_Desc struct_time_type_desc = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000232 "time.struct_time",
233 NULL,
234 struct_time_type_fields,
235 9,
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000236};
Tim Peters9ad4b682002-02-13 05:14:18 +0000237
Martin v. Löwis19ab6c92006-04-16 18:55:50 +0000238static int initialized;
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000239static PyTypeObject StructTimeType;
240
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000241static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000242tmtotuple(struct tm *p)
Guido van Rossum87ce7bb1998-06-09 16:30:31 +0000243{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000244 PyObject *v = PyStructSequence_New(&StructTimeType);
245 if (v == NULL)
246 return NULL;
Tim Peters9ad4b682002-02-13 05:14:18 +0000247
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000248#define SET(i,val) PyStructSequence_SET_ITEM(v, i, PyInt_FromLong((long) val))
249
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000250 SET(0, p->tm_year + 1900);
251 SET(1, p->tm_mon + 1); /* Want January == 1 */
252 SET(2, p->tm_mday);
253 SET(3, p->tm_hour);
254 SET(4, p->tm_min);
255 SET(5, p->tm_sec);
256 SET(6, (p->tm_wday + 6) % 7); /* Want Monday == 0 */
257 SET(7, p->tm_yday + 1); /* Want January, 1 == 1 */
258 SET(8, p->tm_isdst);
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000259#undef SET
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000260 if (PyErr_Occurred()) {
261 Py_XDECREF(v);
262 return NULL;
263 }
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000264
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000265 return v;
Guido van Rossum87ce7bb1998-06-09 16:30:31 +0000266}
267
268static PyObject *
Brett Cannon298c3802004-06-19 20:48:43 +0000269time_convert(double when, struct tm * (*function)(const time_t *))
Guido van Rossum234f9421993-06-17 12:35:49 +0000270{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000271 struct tm *p;
272 time_t whent = _PyTime_DoubleToTimet(when);
Brett Cannon298c3802004-06-19 20:48:43 +0000273
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000274 if (whent == (time_t)-1 && PyErr_Occurred())
275 return NULL;
276 errno = 0;
277 p = function(&whent);
278 if (p == NULL) {
Guido van Rossum6e8583d1996-10-08 14:19:52 +0000279#ifdef EINVAL
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000280 if (errno == 0)
281 errno = EINVAL;
Guido van Rossum6e8583d1996-10-08 14:19:52 +0000282#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000283 return PyErr_SetFromErrno(PyExc_ValueError);
284 }
285 return tmtotuple(p);
Guido van Rossum234f9421993-06-17 12:35:49 +0000286}
287
Fred Drakef901abd2004-08-03 17:58:55 +0000288/* Parse arg tuple that can contain an optional float-or-None value;
289 format needs to be "|O:name".
290 Returns non-zero on success (parallels PyArg_ParseTuple).
291*/
292static int
293parse_time_double_args(PyObject *args, char *format, double *pwhen)
294{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000295 PyObject *ot = NULL;
Fred Drakef901abd2004-08-03 17:58:55 +0000296
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000297 if (!PyArg_ParseTuple(args, format, &ot))
298 return 0;
299 if (ot == NULL || ot == Py_None)
300 *pwhen = floattime();
301 else {
302 double when = PyFloat_AsDouble(ot);
303 if (PyErr_Occurred())
304 return 0;
305 *pwhen = when;
306 }
307 return 1;
Fred Drakef901abd2004-08-03 17:58:55 +0000308}
309
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000310static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000311time_gmtime(PyObject *self, PyObject *args)
Guido van Rossum234f9421993-06-17 12:35:49 +0000312{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000313 double when;
314 if (!parse_time_double_args(args, "|O:gmtime", &when))
315 return NULL;
316 return time_convert(when, gmtime);
Guido van Rossum234f9421993-06-17 12:35:49 +0000317}
318
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000319PyDoc_STRVAR(gmtime_doc,
Brett Cannon8d993aa2007-12-24 19:58:25 +0000320"gmtime([seconds]) -> (tm_year, tm_mon, tm_mday, tm_hour, tm_min,\n\
Fred Drake193a3f62002-03-12 21:38:49 +0000321 tm_sec, tm_wday, tm_yday, tm_isdst)\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000322\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000323Convert seconds since the Epoch to a time tuple expressing UTC (a.k.a.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000324GMT). When 'seconds' is not passed in, convert the current time instead.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000325
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000326static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000327time_localtime(PyObject *self, PyObject *args)
Guido van Rossum234f9421993-06-17 12:35:49 +0000328{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000329 double when;
330 if (!parse_time_double_args(args, "|O:localtime", &when))
331 return NULL;
332 return time_convert(when, localtime);
Guido van Rossum234f9421993-06-17 12:35:49 +0000333}
334
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000335PyDoc_STRVAR(localtime_doc,
Brett Cannon8d993aa2007-12-24 19:58:25 +0000336"localtime([seconds]) -> (tm_year,tm_mon,tm_mday,tm_hour,tm_min,\n\
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000337 tm_sec,tm_wday,tm_yday,tm_isdst)\n\
Martin v. Löwisb3cfc1d2001-12-02 12:27:43 +0000338\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000339Convert seconds since the Epoch to a time tuple expressing local time.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000340When 'seconds' is not passed in, convert the current time instead.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000341
Guido van Rossum9e90a671993-06-24 11:10:19 +0000342static int
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000343gettmarg(PyObject *args, struct tm *p)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000344{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000345 int y;
346 memset((void *) p, '\0', sizeof(struct tm));
Guido van Rossumcfbaecc1998-08-25 14:51:12 +0000347
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000348 if (!PyArg_Parse(args, "(iiiiiiiii)",
349 &y,
350 &p->tm_mon,
351 &p->tm_mday,
352 &p->tm_hour,
353 &p->tm_min,
354 &p->tm_sec,
355 &p->tm_wday,
356 &p->tm_yday,
357 &p->tm_isdst))
358 return 0;
359 if (y < 1900) {
360 PyObject *accept = PyDict_GetItemString(moddict,
361 "accept2dyear");
362 if (accept == NULL || !PyInt_Check(accept) ||
363 PyInt_AsLong(accept) == 0) {
364 PyErr_SetString(PyExc_ValueError,
365 "year >= 1900 required");
366 return 0;
367 }
368 if (69 <= y && y <= 99)
369 y += 1900;
370 else if (0 <= y && y <= 68)
371 y += 2000;
372 else {
373 PyErr_SetString(PyExc_ValueError,
374 "year out of range");
375 return 0;
376 }
377 }
378 p->tm_year = y - 1900;
379 p->tm_mon--;
380 p->tm_wday = (p->tm_wday + 1) % 7;
381 p->tm_yday--;
382 return 1;
Guido van Rossum9e90a671993-06-24 11:10:19 +0000383}
384
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000385#ifdef HAVE_STRFTIME
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000386static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000387time_strftime(PyObject *self, PyObject *args)
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000388{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000389 PyObject *tup = NULL;
390 struct tm buf;
391 const char *fmt;
392 size_t fmtlen, buflen;
393 char *outbuf = 0;
394 size_t i;
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000395
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000396 memset((void *) &buf, '\0', sizeof(buf));
Guido van Rossum1f41f841998-04-27 19:04:26 +0000397
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000398 if (!PyArg_ParseTuple(args, "s|O:strftime", &fmt, &tup))
399 return NULL;
Thomas Woutersfe385252001-01-19 23:16:56 +0000400
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000401 if (tup == NULL) {
402 time_t tt = time(NULL);
403 buf = *localtime(&tt);
404 } else if (!gettmarg(tup, &buf))
405 return NULL;
Guido van Rossum10b164a2001-09-25 13:59:01 +0000406
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000407 /* Checks added to make sure strftime() does not crash Python by
408 indexing blindly into some array for a textual representation
409 by some bad index (fixes bug #897625).
Tim Peters1b6f7a92004-06-20 02:50:16 +0000410
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000411 Also support values of zero from Python code for arguments in which
412 that is out of range by forcing that value to the lowest value that
413 is valid (fixed bug #1520914).
Brett Cannoncaebe222006-07-18 04:41:36 +0000414
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000415 Valid ranges based on what is allowed in struct tm:
Brett Cannoncaebe222006-07-18 04:41:36 +0000416
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000417 - tm_year: [0, max(int)] (1)
418 - tm_mon: [0, 11] (2)
419 - tm_mday: [1, 31]
420 - tm_hour: [0, 23]
421 - tm_min: [0, 59]
422 - tm_sec: [0, 60]
423 - tm_wday: [0, 6] (1)
424 - tm_yday: [0, 365] (2)
425 - tm_isdst: [-max(int), max(int)]
Brett Cannoncaebe222006-07-18 04:41:36 +0000426
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000427 (1) gettmarg() handles bounds-checking.
428 (2) Python's acceptable range is one greater than the range in C,
429 thus need to check against automatic decrement by gettmarg().
430 */
431 if (buf.tm_mon == -1)
432 buf.tm_mon = 0;
433 else if (buf.tm_mon < 0 || buf.tm_mon > 11) {
434 PyErr_SetString(PyExc_ValueError, "month out of range");
435 return NULL;
436 }
437 if (buf.tm_mday == 0)
438 buf.tm_mday = 1;
439 else if (buf.tm_mday < 0 || buf.tm_mday > 31) {
440 PyErr_SetString(PyExc_ValueError, "day of month out of range");
441 return NULL;
442 }
443 if (buf.tm_hour < 0 || buf.tm_hour > 23) {
444 PyErr_SetString(PyExc_ValueError, "hour out of range");
445 return NULL;
446 }
447 if (buf.tm_min < 0 || buf.tm_min > 59) {
448 PyErr_SetString(PyExc_ValueError, "minute out of range");
449 return NULL;
450 }
451 if (buf.tm_sec < 0 || buf.tm_sec > 61) {
452 PyErr_SetString(PyExc_ValueError, "seconds out of range");
453 return NULL;
454 }
455 /* tm_wday does not need checking of its upper-bound since taking
456 ``% 7`` in gettmarg() automatically restricts the range. */
457 if (buf.tm_wday < 0) {
458 PyErr_SetString(PyExc_ValueError, "day of week out of range");
459 return NULL;
460 }
461 if (buf.tm_yday == -1)
462 buf.tm_yday = 0;
463 else if (buf.tm_yday < 0 || buf.tm_yday > 365) {
464 PyErr_SetString(PyExc_ValueError, "day of year out of range");
465 return NULL;
466 }
467 /* Normalize tm_isdst just in case someone foolishly implements %Z
468 based on the assumption that tm_isdst falls within the range of
469 [-1, 1] */
470 if (buf.tm_isdst < -1)
471 buf.tm_isdst = -1;
472 else if (buf.tm_isdst > 1)
473 buf.tm_isdst = 1;
Brett Cannond1080a32004-03-02 04:38:10 +0000474
Kristján Valur Jónssonfd4c8722009-02-04 10:05:25 +0000475#ifdef MS_WINDOWS
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000476 /* check that the format string contains only valid directives */
477 for(outbuf = strchr(fmt, '%');
478 outbuf != NULL;
479 outbuf = strchr(outbuf+2, '%'))
480 {
481 if (outbuf[1]=='#')
482 ++outbuf; /* not documented by python, */
483 if (outbuf[1]=='\0' ||
484 !strchr("aAbBcdfHIjmMpSUwWxXyYzZ%", outbuf[1]))
485 {
486 PyErr_SetString(PyExc_ValueError, "Invalid format string");
487 return 0;
488 }
489 }
Kristján Valur Jónssonfd4c8722009-02-04 10:05:25 +0000490#endif
491
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000492 fmtlen = strlen(fmt);
Guido van Rossumc222ec21999-02-23 00:00:10 +0000493
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000494 /* I hate these functions that presume you know how big the output
495 * will be ahead of time...
496 */
497 for (i = 1024; ; i += i) {
498 outbuf = (char *)malloc(i);
499 if (outbuf == NULL) {
500 return PyErr_NoMemory();
501 }
502 buflen = strftime(outbuf, i, fmt, &buf);
503 if (buflen > 0 || i >= 256 * fmtlen) {
504 /* If the buffer is 256 times as long as the format,
505 it's probably not failing for lack of room!
506 More likely, the format yields an empty result,
507 e.g. an empty format, or %Z when the timezone
508 is unknown. */
509 PyObject *ret;
510 ret = PyString_FromStringAndSize(outbuf, buflen);
511 free(outbuf);
512 return ret;
513 }
514 free(outbuf);
Kristján Valur Jónsson74c3ea02006-07-03 14:59:05 +0000515#if defined _MSC_VER && _MSC_VER >= 1400 && defined(__STDC_SECURE_LIB__)
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000516 /* VisualStudio .NET 2005 does this properly */
517 if (buflen == 0 && errno == EINVAL) {
518 PyErr_SetString(PyExc_ValueError, "Invalid format string");
519 return 0;
520 }
Kristján Valur Jónssonf6083172006-06-12 15:45:12 +0000521#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000522
523 }
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000524}
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000525
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000526PyDoc_STRVAR(strftime_doc,
Thomas Woutersfe385252001-01-19 23:16:56 +0000527"strftime(format[, tuple]) -> string\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000528\n\
529Convert a time tuple to a string according to a format specification.\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000530See the library reference manual for formatting codes. When the time tuple\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000531is not present, current time as returned by localtime() is used.");
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000532#endif /* HAVE_STRFTIME */
533
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000534static PyObject *
535time_strptime(PyObject *self, PyObject *args)
536{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000537 PyObject *strptime_module = PyImport_ImportModuleNoBlock("_strptime");
538 PyObject *strptime_result;
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000539
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000540 if (!strptime_module)
541 return NULL;
542 strptime_result = PyObject_CallMethod(strptime_module,
543 "_strptime_time", "O", args);
544 Py_DECREF(strptime_module);
545 return strptime_result;
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000546}
547
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000548PyDoc_STRVAR(strptime_doc,
Brett Cannon20def8b2003-07-01 05:16:08 +0000549"strptime(string, format) -> struct_time\n\
Martin v. Löwisb3cfc1d2001-12-02 12:27:43 +0000550\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000551Parse a string to a time tuple according to a format specification.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000552See the library reference manual for formatting codes (same as strftime()).");
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000553
Guido van Rossum87ce7bb1998-06-09 16:30:31 +0000554
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000555static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000556time_asctime(PyObject *self, PyObject *args)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000557{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000558 PyObject *tup = NULL;
559 struct tm buf;
560 char *p;
561 if (!PyArg_UnpackTuple(args, "asctime", 0, 1, &tup))
562 return NULL;
563 if (tup == NULL) {
564 time_t tt = time(NULL);
565 buf = *localtime(&tt);
566 } else if (!gettmarg(tup, &buf))
567 return NULL;
568 p = asctime(&buf);
569 if (p[24] == '\n')
570 p[24] = '\0';
571 return PyString_FromString(p);
Guido van Rossum9e90a671993-06-24 11:10:19 +0000572}
573
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000574PyDoc_STRVAR(asctime_doc,
Thomas Woutersfe385252001-01-19 23:16:56 +0000575"asctime([tuple]) -> string\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000576\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000577Convert a time tuple to a string, e.g. 'Sat Jun 06 16:26:11 1998'.\n\
578When the time tuple is not present, current time as returned by localtime()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000579is used.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000580
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000581static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000582time_ctime(PyObject *self, PyObject *args)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000583{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000584 PyObject *ot = NULL;
585 time_t tt;
586 char *p;
Guido van Rossum10b164a2001-09-25 13:59:01 +0000587
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000588 if (!PyArg_UnpackTuple(args, "ctime", 0, 1, &ot))
589 return NULL;
590 if (ot == NULL || ot == Py_None)
591 tt = time(NULL);
592 else {
593 double dt = PyFloat_AsDouble(ot);
594 if (PyErr_Occurred())
595 return NULL;
596 tt = _PyTime_DoubleToTimet(dt);
597 if (tt == (time_t)-1 && PyErr_Occurred())
598 return NULL;
599 }
600 p = ctime(&tt);
601 if (p == NULL) {
602 PyErr_SetString(PyExc_ValueError, "unconvertible time");
603 return NULL;
604 }
605 if (p[24] == '\n')
606 p[24] = '\0';
607 return PyString_FromString(p);
Guido van Rossum9e90a671993-06-24 11:10:19 +0000608}
609
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000610PyDoc_STRVAR(ctime_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000611"ctime(seconds) -> string\n\
612\n\
613Convert a time in seconds since the Epoch to a string in local time.\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000614This is equivalent to asctime(localtime(seconds)). When the time tuple is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000615not present, current time as returned by localtime() is used.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000616
Guido van Rossum60cd8131998-03-06 17:16:21 +0000617#ifdef HAVE_MKTIME
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000618static PyObject *
Georg Brandl96a8c392006-05-29 21:04:52 +0000619time_mktime(PyObject *self, PyObject *tup)
Guido van Rossum234f9421993-06-17 12:35:49 +0000620{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000621 struct tm buf;
622 time_t tt;
623 if (!gettmarg(tup, &buf))
624 return NULL;
625 tt = mktime(&buf);
626 if (tt == (time_t)(-1)) {
627 PyErr_SetString(PyExc_OverflowError,
628 "mktime argument out of range");
629 return NULL;
630 }
631 return PyFloat_FromDouble((double)tt);
Guido van Rossum234f9421993-06-17 12:35:49 +0000632}
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000633
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000634PyDoc_STRVAR(mktime_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000635"mktime(tuple) -> floating point number\n\
636\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000637Convert a time tuple in local time to seconds since the Epoch.");
Guido van Rossum60cd8131998-03-06 17:16:21 +0000638#endif /* HAVE_MKTIME */
Guido van Rossum234f9421993-06-17 12:35:49 +0000639
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000640#ifdef HAVE_WORKING_TZSET
Skip Montanaro9e0fa7a2008-04-05 19:47:47 +0000641static void inittimezone(PyObject *module);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000642
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000643static PyObject *
Georg Brandl96a8c392006-05-29 21:04:52 +0000644time_tzset(PyObject *self, PyObject *unused)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000645{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000646 PyObject* m;
Fred Drake9bb74322002-04-01 14:49:59 +0000647
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000648 m = PyImport_ImportModuleNoBlock("time");
649 if (m == NULL) {
650 return NULL;
651 }
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000652
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000653 tzset();
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000654
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000655 /* Reset timezone, altzone, daylight and tzname */
656 inittimezone(m);
657 Py_DECREF(m);
Tim Peters1b6f7a92004-06-20 02:50:16 +0000658
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000659 Py_INCREF(Py_None);
660 return Py_None;
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000661}
662
663PyDoc_STRVAR(tzset_doc,
664"tzset(zone)\n\
665\n\
666Initialize, or reinitialize, the local timezone to the value stored in\n\
667os.environ['TZ']. The TZ environment variable should be specified in\n\
Neal Norwitzdc8e1942004-07-20 22:34:37 +0000668standard Unix timezone format as documented in the tzset man page\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000669(eg. 'US/Eastern', 'Europe/Amsterdam'). Unknown timezones will silently\n\
670fall back to UTC. If the TZ environment variable is not set, the local\n\
671timezone is set to the systems best guess of wallclock time.\n\
672Changing the TZ environment variable without calling tzset *may* change\n\
673the local timezone used by methods such as localtime, but this behaviour\n\
674should not be relied on.");
675#endif /* HAVE_WORKING_TZSET */
676
Skip Montanaro9e0fa7a2008-04-05 19:47:47 +0000677static void
678inittimezone(PyObject *m) {
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000679 /* This code moved from inittime wholesale to allow calling it from
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000680 time_tzset. In the future, some parts of it can be moved back
681 (for platforms that don't HAVE_WORKING_TZSET, when we know what they
682 are), and the extraneous calls to tzset(3) should be removed.
683 I haven't done this yet, as I don't want to change this code as
684 little as possible when introducing the time.tzset and time.tzsetwall
685 methods. This should simply be a method of doing the following once,
686 at the top of this function and removing the call to tzset() from
687 time_tzset():
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000688
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000689 #ifdef HAVE_TZSET
690 tzset()
691 #endif
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000692
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000693 And I'm lazy and hate C so nyer.
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000694 */
Guido van Rossum10b164a2001-09-25 13:59:01 +0000695#if defined(HAVE_TZNAME) && !defined(__GLIBC__) && !defined(__CYGWIN__)
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000696 tzset();
Guido van Rossum26452411998-09-28 22:07:11 +0000697#ifdef PYOS_OS2
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000698 PyModule_AddIntConstant(m, "timezone", _timezone);
Guido van Rossum26452411998-09-28 22:07:11 +0000699#else /* !PYOS_OS2 */
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000700 PyModule_AddIntConstant(m, "timezone", timezone);
Guido van Rossum26452411998-09-28 22:07:11 +0000701#endif /* PYOS_OS2 */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000702#ifdef HAVE_ALTZONE
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000703 PyModule_AddIntConstant(m, "altzone", altzone);
Guido van Rossumb6775db1994-08-01 11:34:53 +0000704#else
Guido van Rossum26452411998-09-28 22:07:11 +0000705#ifdef PYOS_OS2
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000706 PyModule_AddIntConstant(m, "altzone", _timezone-3600);
Guido van Rossum26452411998-09-28 22:07:11 +0000707#else /* !PYOS_OS2 */
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000708 PyModule_AddIntConstant(m, "altzone", timezone-3600);
Guido van Rossum26452411998-09-28 22:07:11 +0000709#endif /* PYOS_OS2 */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000710#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000711 PyModule_AddIntConstant(m, "daylight", daylight);
712 PyModule_AddObject(m, "tzname",
713 Py_BuildValue("(zz)", tzname[0], tzname[1]));
Guido van Rossum10b164a2001-09-25 13:59:01 +0000714#else /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
Martin v. Löwis60a5d722002-10-16 20:28:25 +0000715#ifdef HAVE_STRUCT_TM_TM_ZONE
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000716 {
Guido van Rossum234f9421993-06-17 12:35:49 +0000717#define YEAR ((time_t)((365 * 24 + 6) * 3600))
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000718 time_t t;
719 struct tm *p;
720 long janzone, julyzone;
721 char janname[10], julyname[10];
722 t = (time((time_t *)0) / YEAR) * YEAR;
723 p = localtime(&t);
724 janzone = -p->tm_gmtoff;
725 strncpy(janname, p->tm_zone ? p->tm_zone : " ", 9);
726 janname[9] = '\0';
727 t += YEAR/2;
728 p = localtime(&t);
729 julyzone = -p->tm_gmtoff;
730 strncpy(julyname, p->tm_zone ? p->tm_zone : " ", 9);
731 julyname[9] = '\0';
Guido van Rossum10b164a2001-09-25 13:59:01 +0000732
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000733 if( janzone < julyzone ) {
734 /* DST is reversed in the southern hemisphere */
735 PyModule_AddIntConstant(m, "timezone", julyzone);
736 PyModule_AddIntConstant(m, "altzone", janzone);
737 PyModule_AddIntConstant(m, "daylight",
738 janzone != julyzone);
739 PyModule_AddObject(m, "tzname",
740 Py_BuildValue("(zz)",
741 julyname, janname));
742 } else {
743 PyModule_AddIntConstant(m, "timezone", janzone);
744 PyModule_AddIntConstant(m, "altzone", julyzone);
745 PyModule_AddIntConstant(m, "daylight",
746 janzone != julyzone);
747 PyModule_AddObject(m, "tzname",
748 Py_BuildValue("(zz)",
749 janname, julyname));
750 }
751 }
Guido van Rossume6a4b7b1997-10-08 15:27:56 +0000752#else
Martin v. Löwis60a5d722002-10-16 20:28:25 +0000753#endif /* HAVE_STRUCT_TM_TM_ZONE */
Tim Peters26ae7cd2001-03-20 03:26:49 +0000754#ifdef __CYGWIN__
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000755 tzset();
756 PyModule_AddIntConstant(m, "timezone", _timezone);
757 PyModule_AddIntConstant(m, "altzone", _timezone-3600);
758 PyModule_AddIntConstant(m, "daylight", _daylight);
759 PyModule_AddObject(m, "tzname",
760 Py_BuildValue("(zz)", _tzname[0], _tzname[1]));
Tim Peters26ae7cd2001-03-20 03:26:49 +0000761#endif /* __CYGWIN__ */
Guido van Rossum10b164a2001-09-25 13:59:01 +0000762#endif /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000763}
764
765
766static PyMethodDef time_methods[] = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000767 {"time", time_time, METH_NOARGS, time_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000768#ifdef HAVE_CLOCK
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000769 {"clock", time_clock, METH_NOARGS, clock_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000770#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000771 {"sleep", time_sleep, METH_VARARGS, sleep_doc},
772 {"gmtime", time_gmtime, METH_VARARGS, gmtime_doc},
773 {"localtime", time_localtime, METH_VARARGS, localtime_doc},
774 {"asctime", time_asctime, METH_VARARGS, asctime_doc},
775 {"ctime", time_ctime, METH_VARARGS, ctime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000776#ifdef HAVE_MKTIME
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000777 {"mktime", time_mktime, METH_O, mktime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000778#endif
779#ifdef HAVE_STRFTIME
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000780 {"strftime", time_strftime, METH_VARARGS, strftime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000781#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000782 {"strptime", time_strptime, METH_VARARGS, strptime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000783#ifdef HAVE_WORKING_TZSET
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000784 {"tzset", time_tzset, METH_NOARGS, tzset_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000785#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000786 {NULL, NULL} /* sentinel */
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000787};
788
789
790PyDoc_STRVAR(module_doc,
791"This module provides various functions to manipulate time values.\n\
792\n\
793There are two standard representations of time. One is the number\n\
794of seconds since the Epoch, in UTC (a.k.a. GMT). It may be an integer\n\
795or a floating point number (to represent fractions of seconds).\n\
796The Epoch is system-defined; on Unix, it is generally January 1st, 1970.\n\
797The actual value can be retrieved by calling gmtime(0).\n\
798\n\
799The other representation is a tuple of 9 integers giving local time.\n\
800The tuple items are:\n\
801 year (four digits, e.g. 1998)\n\
802 month (1-12)\n\
803 day (1-31)\n\
804 hours (0-23)\n\
805 minutes (0-59)\n\
806 seconds (0-59)\n\
807 weekday (0-6, Monday is 0)\n\
808 Julian day (day in the year, 1-366)\n\
809 DST (Daylight Savings Time) flag (-1, 0 or 1)\n\
810If the DST flag is 0, the time is given in the regular time zone;\n\
811if it is 1, the time is given in the DST time zone;\n\
812if it is -1, mktime() should guess based on the date and time.\n\
813\n\
814Variables:\n\
815\n\
816timezone -- difference in seconds between UTC and local standard time\n\
817altzone -- difference in seconds between UTC and local DST time\n\
818daylight -- whether local time should reflect DST\n\
819tzname -- tuple of (standard time zone name, DST time zone name)\n\
820\n\
821Functions:\n\
822\n\
823time() -- return current time in seconds since the Epoch as a float\n\
824clock() -- return CPU time since process start as a float\n\
825sleep() -- delay for a number of seconds given as a float\n\
826gmtime() -- convert seconds since Epoch to UTC tuple\n\
827localtime() -- convert seconds since Epoch to local time tuple\n\
828asctime() -- convert time tuple to string\n\
829ctime() -- convert time in seconds to string\n\
830mktime() -- convert local time tuple to seconds since Epoch\n\
831strftime() -- convert time tuple to string according to format specification\n\
832strptime() -- parse string to time tuple according to format specification\n\
833tzset() -- change the local timezone");
834
835
836PyMODINIT_FUNC
837inittime(void)
838{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000839 PyObject *m;
840 char *p;
841 m = Py_InitModule3("time", time_methods, module_doc);
842 if (m == NULL)
843 return;
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000844
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000845 /* Accept 2-digit dates unless PYTHONY2K is set and non-empty */
846 p = Py_GETENV("PYTHONY2K");
847 PyModule_AddIntConstant(m, "accept2dyear", (long) (!p || !*p));
848 /* Squirrel away the module's dictionary for the y2k check */
849 moddict = PyModule_GetDict(m);
850 Py_INCREF(moddict);
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000851
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000852 /* Set, or reset, module variables like time.timezone */
853 inittimezone(m);
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000854
Mark Hammond975e3922002-07-16 01:29:19 +0000855#ifdef MS_WINDOWS
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000856 /* Helper to allow interrupts for Windows.
857 If Ctrl+C event delivered while not sleeping
858 it will be ignored.
859 */
860 main_thread = PyThread_get_thread_ident();
861 hInterruptEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
862 SetConsoleCtrlHandler( PyCtrlHandler, TRUE);
Mark Hammond975e3922002-07-16 01:29:19 +0000863#endif /* MS_WINDOWS */
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000864 if (!initialized) {
865 PyStructSequence_InitType(&StructTimeType,
866 &struct_time_type_desc);
867 }
868 Py_INCREF(&StructTimeType);
869 PyModule_AddObject(m, "struct_time", (PyObject*) &StructTimeType);
870 initialized = 1;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000871}
872
873
Guido van Rossumb6775db1994-08-01 11:34:53 +0000874/* Implement floattime() for various platforms */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000875
Guido van Rossumb6775db1994-08-01 11:34:53 +0000876static double
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000877floattime(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000878{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000879 /* There are three ways to get the time:
880 (1) gettimeofday() -- resolution in microseconds
881 (2) ftime() -- resolution in milliseconds
882 (3) time() -- resolution in seconds
883 In all cases the return value is a float in seconds.
884 Since on some systems (e.g. SCO ODT 3.0) gettimeofday() may
885 fail, so we fall back on ftime() or time().
886 Note: clock resolution does not imply clock accuracy! */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000887#ifdef HAVE_GETTIMEOFDAY
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000888 {
889 struct timeval t;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000890#ifdef GETTIMEOFDAY_NO_TZ
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000891 if (gettimeofday(&t) == 0)
892 return (double)t.tv_sec + t.tv_usec*0.000001;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000893#else /* !GETTIMEOFDAY_NO_TZ */
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000894 if (gettimeofday(&t, (struct timezone *)NULL) == 0)
895 return (double)t.tv_sec + t.tv_usec*0.000001;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000896#endif /* !GETTIMEOFDAY_NO_TZ */
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000897 }
Ronald Oussorend06b6f22006-04-23 11:59:25 +0000898
Guido van Rossumb6775db1994-08-01 11:34:53 +0000899#endif /* !HAVE_GETTIMEOFDAY */
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000900 {
Guido van Rossumd3eb5771999-03-09 16:07:23 +0000901#if defined(HAVE_FTIME)
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000902 struct timeb t;
903 ftime(&t);
904 return (double)t.time + (double)t.millitm * (double)0.001;
Guido van Rossumb6775db1994-08-01 11:34:53 +0000905#else /* !HAVE_FTIME */
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000906 time_t secs;
907 time(&secs);
908 return (double)secs;
Guido van Rossumb6775db1994-08-01 11:34:53 +0000909#endif /* !HAVE_FTIME */
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000910 }
Guido van Rossum426035c1991-02-19 12:27:35 +0000911}
912
Guido van Rossumb6775db1994-08-01 11:34:53 +0000913
914/* Implement floatsleep() for various platforms.
915 When interrupted (or when another error occurs), return -1 and
916 set an exception; else return 0. */
917
918static int
Guido van Rossuma320fd31995-03-09 12:14:15 +0000919floatsleep(double secs)
Guido van Rossum426035c1991-02-19 12:27:35 +0000920{
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000921/* XXX Should test for MS_WINDOWS first! */
Andrew MacIntyre7bf68332002-03-03 02:59:16 +0000922#if defined(HAVE_SELECT) && !defined(__BEOS__) && !defined(__EMX__)
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000923 struct timeval t;
924 double frac;
925 frac = fmod(secs, 1.0);
926 secs = floor(secs);
927 t.tv_sec = (long)secs;
928 t.tv_usec = (long)(frac*1000000.0);
929 Py_BEGIN_ALLOW_THREADS
930 if (select(0, (fd_set *)0, (fd_set *)0, (fd_set *)0, &t) != 0) {
Guido van Rossum09cbb011999-11-08 15:32:27 +0000931#ifdef EINTR
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000932 if (errno != EINTR) {
Guido van Rossum09cbb011999-11-08 15:32:27 +0000933#else
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000934 if (1) {
Guido van Rossum09cbb011999-11-08 15:32:27 +0000935#endif
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000936 Py_BLOCK_THREADS
937 PyErr_SetFromErrno(PyExc_IOError);
938 return -1;
939 }
940 }
941 Py_END_ALLOW_THREADS
Martin v. Löwis02af9642002-01-16 11:04:06 +0000942#elif defined(__WATCOMC__) && !defined(__QNX__)
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000943 /* XXX Can't interrupt this sleep */
944 Py_BEGIN_ALLOW_THREADS
945 delay((int)(secs * 1000 + 0.5)); /* delay() uses milliseconds */
946 Py_END_ALLOW_THREADS
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000947#elif defined(MS_WINDOWS)
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000948 {
949 double millisecs = secs * 1000.0;
950 unsigned long ul_millis;
Tim Peters513a1cd2003-01-19 04:54:58 +0000951
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000952 if (millisecs > (double)ULONG_MAX) {
953 PyErr_SetString(PyExc_OverflowError,
954 "sleep length is too large");
955 return -1;
956 }
957 Py_BEGIN_ALLOW_THREADS
958 /* Allow sleep(0) to maintain win32 semantics, and as decreed
959 * by Guido, only the main thread can be interrupted.
960 */
961 ul_millis = (unsigned long)millisecs;
962 if (ul_millis == 0 ||
963 main_thread != PyThread_get_thread_ident())
964 Sleep(ul_millis);
965 else {
966 DWORD rc;
967 ResetEvent(hInterruptEvent);
968 rc = WaitForSingleObject(hInterruptEvent, ul_millis);
969 if (rc == WAIT_OBJECT_0) {
970 /* Yield to make sure real Python signal
971 * handler called.
972 */
973 Sleep(1);
974 Py_BLOCK_THREADS
975 errno = EINTR;
976 PyErr_SetFromErrno(PyExc_IOError);
977 return -1;
978 }
979 }
980 Py_END_ALLOW_THREADS
981 }
Martin v. Löwis02af9642002-01-16 11:04:06 +0000982#elif defined(PYOS_OS2)
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000983 /* This Sleep *IS* Interruptable by Exceptions */
984 Py_BEGIN_ALLOW_THREADS
985 if (DosSleep(secs * 1000) != NO_ERROR) {
986 Py_BLOCK_THREADS
987 PyErr_SetFromErrno(PyExc_IOError);
988 return -1;
989 }
990 Py_END_ALLOW_THREADS
Martin v. Löwis02af9642002-01-16 11:04:06 +0000991#elif defined(__BEOS__)
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000992 /* This sleep *CAN BE* interrupted. */
993 {
994 if( secs <= 0.0 ) {
995 return;
996 }
Guido van Rossum10b164a2001-09-25 13:59:01 +0000997
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000998 Py_BEGIN_ALLOW_THREADS
999 /* BeOS snooze() is in microseconds... */
1000 if( snooze( (bigtime_t)( secs * 1000.0 * 1000.0 ) ) == B_INTERRUPTED ) {
1001 Py_BLOCK_THREADS
1002 PyErr_SetFromErrno( PyExc_IOError );
1003 return -1;
1004 }
1005 Py_END_ALLOW_THREADS
1006 }
Martin v. Löwis02af9642002-01-16 11:04:06 +00001007#elif defined(RISCOS)
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001008 if (secs <= 0.0)
1009 return 0;
1010 Py_BEGIN_ALLOW_THREADS
1011 /* This sleep *CAN BE* interrupted. */
1012 if ( riscos_sleep(secs) )
1013 return -1;
1014 Py_END_ALLOW_THREADS
Martin v. Löwis02af9642002-01-16 11:04:06 +00001015#elif defined(PLAN9)
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001016 {
1017 double millisecs = secs * 1000.0;
1018 if (millisecs > (double)LONG_MAX) {
1019 PyErr_SetString(PyExc_OverflowError, "sleep length is too large");
1020 return -1;
1021 }
1022 /* This sleep *CAN BE* interrupted. */
1023 Py_BEGIN_ALLOW_THREADS
1024 if(sleep((long)millisecs) < 0){
1025 Py_BLOCK_THREADS
1026 PyErr_SetFromErrno(PyExc_IOError);
1027 return -1;
1028 }
1029 Py_END_ALLOW_THREADS
1030 }
Martin v. Löwis02af9642002-01-16 11:04:06 +00001031#else
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001032 /* XXX Can't interrupt this sleep */
1033 Py_BEGIN_ALLOW_THREADS
1034 sleep((int)secs);
1035 Py_END_ALLOW_THREADS
Martin v. Löwis02af9642002-01-16 11:04:06 +00001036#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001037
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001038 return 0;
Guido van Rossum80c9d881991-04-16 08:47:51 +00001039}
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001040
1041