blob: 2f4092d64e184e65349d3f948ee390cd2aa65a41 [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{
49 SetEvent(hInterruptEvent);
50 /* allow other default handlers to be called.
51 Default Python handler will setup the
52 KeyboardInterrupt exception.
53 */
54 return FALSE;
55}
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{
105 time_t result;
106 double diff;
107
108 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;
125}
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{
Guido van Rossumb6775db1994-08-01 11:34:53 +0000130 double secs;
Guido van Rossumb6775db1994-08-01 11:34:53 +0000131 secs = floattime();
132 if (secs == 0.0) {
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000133 PyErr_SetFromErrno(PyExc_IOError);
Guido van Rossuma2b7f401993-01-04 09:09:59 +0000134 return NULL;
135 }
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000136 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{
Barry Warsaw9a2a8a81996-12-06 23:32:14 +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{
Tim Peters9ad4b682002-02-13 05:14:18 +0000167 static LARGE_INTEGER ctrStart;
Mark Hammond7ba5e812002-02-12 04:02:33 +0000168 static double divisor = 0.0;
Tim Peters9ad4b682002-02-13 05:14:18 +0000169 LARGE_INTEGER now;
Mark Hammond7ba5e812002-02-12 04:02:33 +0000170 double diff;
Guido van Rossum3917c221997-04-02 05:35:28 +0000171
Mark Hammond7ba5e812002-02-12 04:02:33 +0000172 if (divisor == 0.0) {
Tim Peters9ad4b682002-02-13 05:14:18 +0000173 LARGE_INTEGER freq;
174 QueryPerformanceCounter(&ctrStart);
175 if (!QueryPerformanceFrequency(&freq) || freq.QuadPart == 0) {
Mark Hammond7ba5e812002-02-12 04:02:33 +0000176 /* Unlikely to happen - this works on all intel
177 machines at least! Revert to clock() */
Georg Brandl30712ab2007-03-29 12:42:07 +0000178 return PyFloat_FromDouble(((double)clock()) /
179 CLOCKS_PER_SEC);
Guido van Rossum3917c221997-04-02 05:35:28 +0000180 }
Tim Peters9ad4b682002-02-13 05:14:18 +0000181 divisor = (double)freq.QuadPart;
Guido van Rossum3917c221997-04-02 05:35:28 +0000182 }
Tim Peters9ad4b682002-02-13 05:14:18 +0000183 QueryPerformanceCounter(&now);
184 diff = (double)(now.QuadPart - ctrStart.QuadPart);
Mark Hammond7ba5e812002-02-12 04:02:33 +0000185 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{
Guido van Rossum775f4da1993-01-09 17:18:52 +0000203 double secs;
Thomas Woutersfe385252001-01-19 23:16:56 +0000204 if (!PyArg_ParseTuple(args, "d:sleep", &secs))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000205 return NULL;
Guido van Rossum8607ae21997-11-03 22:04:46 +0000206 if (floatsleep(secs) != 0)
207 return NULL;
208 Py_INCREF(Py_None);
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000209 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[] = {
219 {"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}
229};
230
231static PyStructSequence_Desc struct_time_type_desc = {
Guido van Rossum14648392001-12-08 18:02:58 +0000232 "time.struct_time",
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000233 NULL,
234 struct_time_type_fields,
235 9,
236};
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{
Guido van Rossum98bf58f2001-10-18 20:34:25 +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
250 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);
259#undef SET
260 if (PyErr_Occurred()) {
261 Py_XDECREF(v);
262 return NULL;
263 }
264
265 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{
Guido van Rossum6e8583d1996-10-08 14:19:52 +0000271 struct tm *p;
Brett Cannon298c3802004-06-19 20:48:43 +0000272 time_t whent = _PyTime_DoubleToTimet(when);
273
274 if (whent == (time_t)-1 && PyErr_Occurred())
275 return NULL;
Guido van Rossum6e8583d1996-10-08 14:19:52 +0000276 errno = 0;
Brett Cannon298c3802004-06-19 20:48:43 +0000277 p = function(&whent);
Guido van Rossum6e8583d1996-10-08 14:19:52 +0000278 if (p == NULL) {
279#ifdef EINVAL
Guido van Rossum0b1ff661996-11-02 17:31:22 +0000280 if (errno == 0)
Guido van Rossum6e8583d1996-10-08 14:19:52 +0000281 errno = EINVAL;
282#endif
Tim Peters8b19a932003-01-17 20:08:54 +0000283 return PyErr_SetFromErrno(PyExc_ValueError);
Guido van Rossum6e8583d1996-10-08 14:19:52 +0000284 }
Guido van Rossum87ce7bb1998-06-09 16:30:31 +0000285 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{
295 PyObject *ot = NULL;
296
297 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;
308}
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{
313 double when;
Fred Drakef901abd2004-08-03 17:58:55 +0000314 if (!parse_time_double_args(args, "|O:gmtime", &when))
Guido van Rossum234f9421993-06-17 12:35:49 +0000315 return NULL;
Brett Cannon298c3802004-06-19 20:48:43 +0000316 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{
329 double when;
Fred Drakef901abd2004-08-03 17:58:55 +0000330 if (!parse_time_double_args(args, "|O:localtime", &when))
Guido van Rossum234f9421993-06-17 12:35:49 +0000331 return NULL;
Brett Cannon298c3802004-06-19 20:48:43 +0000332 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\
337 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{
Guido van Rossumcfbaecc1998-08-25 14:51:12 +0000345 int y;
Thomas Wouters334fb892000-07-25 12:56:38 +0000346 memset((void *) p, '\0', sizeof(struct tm));
Guido van Rossumcfbaecc1998-08-25 14:51:12 +0000347
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000348 if (!PyArg_Parse(args, "(iiiiiiiii)",
Guido van Rossumcfbaecc1998-08-25 14:51:12 +0000349 &y,
Barry Warsaw4a6cf411997-01-13 22:44:55 +0000350 &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))
Guido van Rossum9e90a671993-06-24 11:10:19 +0000358 return 0;
Guido van Rossumcfbaecc1998-08-25 14:51:12 +0000359 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,
Skip Montanaro1a10aac2001-08-22 12:39:16 +0000374 "year out of range");
Guido van Rossumcfbaecc1998-08-25 14:51:12 +0000375 return 0;
376 }
377 }
378 p->tm_year = y - 1900;
Guido van Rossum9e90a671993-06-24 11:10:19 +0000379 p->tm_mon--;
380 p->tm_wday = (p->tm_wday + 1) % 7;
381 p->tm_yday--;
382 return 1;
383}
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{
Thomas Woutersfe385252001-01-19 23:16:56 +0000389 PyObject *tup = NULL;
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000390 struct tm buf;
391 const char *fmt;
Guido van Rossumfa481162000-06-28 21:33:59 +0000392 size_t fmtlen, buflen;
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000393 char *outbuf = 0;
Guido van Rossumfa481162000-06-28 21:33:59 +0000394 size_t i;
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000395
Thomas Wouters334fb892000-07-25 12:56:38 +0000396 memset((void *) &buf, '\0', sizeof(buf));
Guido van Rossum1f41f841998-04-27 19:04:26 +0000397
Thomas Woutersfe385252001-01-19 23:16:56 +0000398 if (!PyArg_ParseTuple(args, "s|O:strftime", &fmt, &tup))
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000399 return NULL;
Thomas Woutersfe385252001-01-19 23:16:56 +0000400
401 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
Brett Cannond1080a32004-03-02 04:38:10 +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
Brett Cannoncaebe222006-07-18 04:41:36 +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
Brett Cannon468e45e2006-07-21 22:44:07 +0000413 is valid (fixed bug #1520914).
Brett Cannoncaebe222006-07-18 04:41:36 +0000414
415 Valid ranges based on what is allowed in struct tm:
416
417 - 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)]
426
427 (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().
Brett Cannond1080a32004-03-02 04:38:10 +0000430 */
Brett Cannoncaebe222006-07-18 04:41:36 +0000431 if (buf.tm_mon == -1)
432 buf.tm_mon = 0;
433 else if (buf.tm_mon < 0 || buf.tm_mon > 11) {
Brett Cannond1080a32004-03-02 04:38:10 +0000434 PyErr_SetString(PyExc_ValueError, "month out of range");
435 return NULL;
436 }
Brett Cannoncaebe222006-07-18 04:41:36 +0000437 if (buf.tm_mday == 0)
438 buf.tm_mday = 1;
439 else if (buf.tm_mday < 0 || buf.tm_mday > 31) {
Brett Cannond1080a32004-03-02 04:38:10 +0000440 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 }
Brett Cannoncaebe222006-07-18 04:41:36 +0000461 if (buf.tm_yday == -1)
462 buf.tm_yday = 0;
463 else if (buf.tm_yday < 0 || buf.tm_yday > 365) {
Brett Cannond1080a32004-03-02 04:38:10 +0000464 PyErr_SetString(PyExc_ValueError, "day of year out of range");
465 return NULL;
466 }
467 if (buf.tm_isdst < -1 || buf.tm_isdst > 1) {
468 PyErr_SetString(PyExc_ValueError,
469 "daylight savings flag out of range");
470 return NULL;
471 }
472
Kristján Valur Jónssonfd4c8722009-02-04 10:05:25 +0000473#ifdef MS_WINDOWS
474 /* check that the format string contains only valid directives */
475 for(outbuf = strchr(fmt, '%');
476 outbuf != NULL;
477 outbuf = strchr(outbuf+2, '%'))
478 {
479 if (outbuf[1]=='#')
480 ++outbuf; /* not documented by python, */
481 if (outbuf[1]=='\0' ||
482 !strchr("aAbBcdfHIjmMpSUwWxXyYzZ%", outbuf[1]))
483 {
484 PyErr_SetString(PyExc_ValueError, "Invalid format string");
485 return 0;
486 }
487 }
488#endif
489
Guido van Rossumc222ec21999-02-23 00:00:10 +0000490 fmtlen = strlen(fmt);
491
Barry Warsaw4a6cf411997-01-13 22:44:55 +0000492 /* I hate these functions that presume you know how big the output
493 * will be ahead of time...
494 */
Guido van Rossumc222ec21999-02-23 00:00:10 +0000495 for (i = 1024; ; i += i) {
Anthony Baxter7cbc0f52006-04-13 07:19:01 +0000496 outbuf = (char *)malloc(i);
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000497 if (outbuf == NULL) {
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000498 return PyErr_NoMemory();
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000499 }
Guido van Rossumc222ec21999-02-23 00:00:10 +0000500 buflen = strftime(outbuf, i, fmt, &buf);
501 if (buflen > 0 || i >= 256 * fmtlen) {
502 /* If the buffer is 256 times as long as the format,
503 it's probably not failing for lack of room!
504 More likely, the format yields an empty result,
505 e.g. an empty format, or %Z when the timezone
506 is unknown. */
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000507 PyObject *ret;
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000508 ret = PyString_FromStringAndSize(outbuf, buflen);
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000509 free(outbuf);
510 return ret;
511 }
512 free(outbuf);
Kristján Valur Jónsson74c3ea02006-07-03 14:59:05 +0000513#if defined _MSC_VER && _MSC_VER >= 1400 && defined(__STDC_SECURE_LIB__)
Kristján Valur Jónssonf6083172006-06-12 15:45:12 +0000514 /* VisualStudio .NET 2005 does this properly */
515 if (buflen == 0 && errno == EINVAL) {
516 PyErr_SetString(PyExc_ValueError, "Invalid format string");
517 return 0;
518 }
519#endif
520
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000521 }
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000522}
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000523
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000524PyDoc_STRVAR(strftime_doc,
Thomas Woutersfe385252001-01-19 23:16:56 +0000525"strftime(format[, tuple]) -> string\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000526\n\
527Convert a time tuple to a string according to a format specification.\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000528See the library reference manual for formatting codes. When the time tuple\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000529is not present, current time as returned by localtime() is used.");
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000530#endif /* HAVE_STRFTIME */
531
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000532static PyObject *
533time_strptime(PyObject *self, PyObject *args)
534{
Christian Heimes000a0742008-01-03 22:16:32 +0000535 PyObject *strptime_module = PyImport_ImportModuleNoBlock("_strptime");
Raymond Hettinger502168a2003-04-10 16:03:22 +0000536 PyObject *strptime_result;
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000537
Tim Peters513a1cd2003-01-19 04:54:58 +0000538 if (!strptime_module)
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000539 return NULL;
Skip Montanarofc070d22008-03-15 16:04:45 +0000540 strptime_result = PyObject_CallMethod(strptime_module, "_strptime_time", "O", args);
Raymond Hettinger502168a2003-04-10 16:03:22 +0000541 Py_DECREF(strptime_module);
542 return strptime_result;
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000543}
544
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000545PyDoc_STRVAR(strptime_doc,
Brett Cannon20def8b2003-07-01 05:16:08 +0000546"strptime(string, format) -> struct_time\n\
Martin v. Löwisb3cfc1d2001-12-02 12:27:43 +0000547\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000548Parse a string to a time tuple according to a format specification.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000549See the library reference manual for formatting codes (same as strftime()).");
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000550
Guido van Rossum87ce7bb1998-06-09 16:30:31 +0000551
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000552static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000553time_asctime(PyObject *self, PyObject *args)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000554{
Thomas Woutersfe385252001-01-19 23:16:56 +0000555 PyObject *tup = NULL;
Guido van Rossum9e90a671993-06-24 11:10:19 +0000556 struct tm buf;
557 char *p;
Georg Brandl96a8c392006-05-29 21:04:52 +0000558 if (!PyArg_UnpackTuple(args, "asctime", 0, 1, &tup))
Guido van Rossumb2b42dd2000-01-12 16:38:20 +0000559 return NULL;
Thomas Woutersfe385252001-01-19 23:16:56 +0000560 if (tup == NULL) {
561 time_t tt = time(NULL);
562 buf = *localtime(&tt);
563 } else if (!gettmarg(tup, &buf))
Guido van Rossum9e90a671993-06-24 11:10:19 +0000564 return NULL;
565 p = asctime(&buf);
566 if (p[24] == '\n')
567 p[24] = '\0';
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000568 return PyString_FromString(p);
Guido van Rossum9e90a671993-06-24 11:10:19 +0000569}
570
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000571PyDoc_STRVAR(asctime_doc,
Thomas Woutersfe385252001-01-19 23:16:56 +0000572"asctime([tuple]) -> string\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000573\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000574Convert a time tuple to a string, e.g. 'Sat Jun 06 16:26:11 1998'.\n\
575When the time tuple is not present, current time as returned by localtime()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000576is used.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000577
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000578static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000579time_ctime(PyObject *self, PyObject *args)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000580{
Fred Drakef901abd2004-08-03 17:58:55 +0000581 PyObject *ot = NULL;
Guido van Rossum9e90a671993-06-24 11:10:19 +0000582 time_t tt;
583 char *p;
Guido van Rossum10b164a2001-09-25 13:59:01 +0000584
Georg Brandl96a8c392006-05-29 21:04:52 +0000585 if (!PyArg_UnpackTuple(args, "ctime", 0, 1, &ot))
Fred Drakef901abd2004-08-03 17:58:55 +0000586 return NULL;
587 if (ot == NULL || ot == Py_None)
Thomas Woutersfe385252001-01-19 23:16:56 +0000588 tt = time(NULL);
589 else {
Fred Drakef901abd2004-08-03 17:58:55 +0000590 double dt = PyFloat_AsDouble(ot);
591 if (PyErr_Occurred())
Thomas Woutersfe385252001-01-19 23:16:56 +0000592 return NULL;
Brett Cannon298c3802004-06-19 20:48:43 +0000593 tt = _PyTime_DoubleToTimet(dt);
594 if (tt == (time_t)-1 && PyErr_Occurred())
595 return NULL;
Thomas Woutersfe385252001-01-19 23:16:56 +0000596 }
Guido van Rossum9e90a671993-06-24 11:10:19 +0000597 p = ctime(&tt);
Guido van Rossum78535701998-03-03 22:19:10 +0000598 if (p == NULL) {
599 PyErr_SetString(PyExc_ValueError, "unconvertible time");
600 return NULL;
601 }
Guido van Rossum9e90a671993-06-24 11:10:19 +0000602 if (p[24] == '\n')
603 p[24] = '\0';
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000604 return PyString_FromString(p);
Guido van Rossum9e90a671993-06-24 11:10:19 +0000605}
606
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000607PyDoc_STRVAR(ctime_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000608"ctime(seconds) -> string\n\
609\n\
610Convert a time in seconds since the Epoch to a string in local time.\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000611This is equivalent to asctime(localtime(seconds)). When the time tuple is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000612not present, current time as returned by localtime() is used.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000613
Guido van Rossum60cd8131998-03-06 17:16:21 +0000614#ifdef HAVE_MKTIME
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000615static PyObject *
Georg Brandl96a8c392006-05-29 21:04:52 +0000616time_mktime(PyObject *self, PyObject *tup)
Guido van Rossum234f9421993-06-17 12:35:49 +0000617{
618 struct tm buf;
Guido van Rossumbceeac81996-05-23 22:53:47 +0000619 time_t tt;
Guido van Rossumb2b42dd2000-01-12 16:38:20 +0000620 if (!gettmarg(tup, &buf))
Guido van Rossum234f9421993-06-17 12:35:49 +0000621 return NULL;
Guido van Rossumbceeac81996-05-23 22:53:47 +0000622 tt = mktime(&buf);
623 if (tt == (time_t)(-1)) {
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000624 PyErr_SetString(PyExc_OverflowError,
Guido van Rossum10b164a2001-09-25 13:59:01 +0000625 "mktime argument out of range");
Guido van Rossumbceeac81996-05-23 22:53:47 +0000626 return NULL;
627 }
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000628 return PyFloat_FromDouble((double)tt);
Guido van Rossum234f9421993-06-17 12:35:49 +0000629}
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000630
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000631PyDoc_STRVAR(mktime_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000632"mktime(tuple) -> floating point number\n\
633\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000634Convert a time tuple in local time to seconds since the Epoch.");
Guido van Rossum60cd8131998-03-06 17:16:21 +0000635#endif /* HAVE_MKTIME */
Guido van Rossum234f9421993-06-17 12:35:49 +0000636
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000637#ifdef HAVE_WORKING_TZSET
Skip Montanaro9e0fa7a2008-04-05 19:47:47 +0000638static void inittimezone(PyObject *module);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000639
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000640static PyObject *
Georg Brandl96a8c392006-05-29 21:04:52 +0000641time_tzset(PyObject *self, PyObject *unused)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000642{
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000643 PyObject* m;
Fred Drake9bb74322002-04-01 14:49:59 +0000644
Christian Heimes000a0742008-01-03 22:16:32 +0000645 m = PyImport_ImportModuleNoBlock("time");
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000646 if (m == NULL) {
647 return NULL;
648 }
649
650 tzset();
651
652 /* Reset timezone, altzone, daylight and tzname */
653 inittimezone(m);
654 Py_DECREF(m);
Tim Peters1b6f7a92004-06-20 02:50:16 +0000655
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000656 Py_INCREF(Py_None);
657 return Py_None;
658}
659
660PyDoc_STRVAR(tzset_doc,
661"tzset(zone)\n\
662\n\
663Initialize, or reinitialize, the local timezone to the value stored in\n\
664os.environ['TZ']. The TZ environment variable should be specified in\n\
Neal Norwitzdc8e1942004-07-20 22:34:37 +0000665standard Unix timezone format as documented in the tzset man page\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000666(eg. 'US/Eastern', 'Europe/Amsterdam'). Unknown timezones will silently\n\
667fall back to UTC. If the TZ environment variable is not set, the local\n\
668timezone is set to the systems best guess of wallclock time.\n\
669Changing the TZ environment variable without calling tzset *may* change\n\
670the local timezone used by methods such as localtime, but this behaviour\n\
671should not be relied on.");
672#endif /* HAVE_WORKING_TZSET */
673
Skip Montanaro9e0fa7a2008-04-05 19:47:47 +0000674static void
675inittimezone(PyObject *m) {
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000676 /* This code moved from inittime wholesale to allow calling it from
677 time_tzset. In the future, some parts of it can be moved back
678 (for platforms that don't HAVE_WORKING_TZSET, when we know what they
Skip Montanaro28112d02008-04-06 17:05:57 +0000679 are), and the extraneous calls to tzset(3) should be removed.
Walter Dörwaldfd7e1622006-12-21 18:06:30 +0000680 I haven't done this yet, as I don't want to change this code as
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000681 little as possible when introducing the time.tzset and time.tzsetwall
682 methods. This should simply be a method of doing the following once,
683 at the top of this function and removing the call to tzset() from
684 time_tzset():
685
686 #ifdef HAVE_TZSET
687 tzset()
688 #endif
689
690 And I'm lazy and hate C so nyer.
691 */
Guido van Rossum10b164a2001-09-25 13:59:01 +0000692#if defined(HAVE_TZNAME) && !defined(__GLIBC__) && !defined(__CYGWIN__)
Guido van Rossum234f9421993-06-17 12:35:49 +0000693 tzset();
Guido van Rossum26452411998-09-28 22:07:11 +0000694#ifdef PYOS_OS2
Fred Drake9bb74322002-04-01 14:49:59 +0000695 PyModule_AddIntConstant(m, "timezone", _timezone);
Guido van Rossum26452411998-09-28 22:07:11 +0000696#else /* !PYOS_OS2 */
Fred Drake9bb74322002-04-01 14:49:59 +0000697 PyModule_AddIntConstant(m, "timezone", timezone);
Guido van Rossum26452411998-09-28 22:07:11 +0000698#endif /* PYOS_OS2 */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000699#ifdef HAVE_ALTZONE
Fred Drake9bb74322002-04-01 14:49:59 +0000700 PyModule_AddIntConstant(m, "altzone", altzone);
Guido van Rossumb6775db1994-08-01 11:34:53 +0000701#else
Guido van Rossum26452411998-09-28 22:07:11 +0000702#ifdef PYOS_OS2
Fred Drake9bb74322002-04-01 14:49:59 +0000703 PyModule_AddIntConstant(m, "altzone", _timezone-3600);
Guido van Rossum26452411998-09-28 22:07:11 +0000704#else /* !PYOS_OS2 */
Fred Drake9bb74322002-04-01 14:49:59 +0000705 PyModule_AddIntConstant(m, "altzone", timezone-3600);
Guido van Rossum26452411998-09-28 22:07:11 +0000706#endif /* PYOS_OS2 */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000707#endif
Fred Drake9bb74322002-04-01 14:49:59 +0000708 PyModule_AddIntConstant(m, "daylight", daylight);
709 PyModule_AddObject(m, "tzname",
710 Py_BuildValue("(zz)", tzname[0], tzname[1]));
Guido van Rossum10b164a2001-09-25 13:59:01 +0000711#else /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
Martin v. Löwis60a5d722002-10-16 20:28:25 +0000712#ifdef HAVE_STRUCT_TM_TM_ZONE
Guido van Rossum234f9421993-06-17 12:35:49 +0000713 {
714#define YEAR ((time_t)((365 * 24 + 6) * 3600))
715 time_t t;
716 struct tm *p;
Guido van Rossum57731601999-03-29 19:12:04 +0000717 long janzone, julyzone;
718 char janname[10], julyname[10];
Guido van Rossum234f9421993-06-17 12:35:49 +0000719 t = (time((time_t *)0) / YEAR) * YEAR;
720 p = localtime(&t);
Guido van Rossum57731601999-03-29 19:12:04 +0000721 janzone = -p->tm_gmtoff;
722 strncpy(janname, p->tm_zone ? p->tm_zone : " ", 9);
723 janname[9] = '\0';
Guido van Rossum234f9421993-06-17 12:35:49 +0000724 t += YEAR/2;
725 p = localtime(&t);
Guido van Rossum57731601999-03-29 19:12:04 +0000726 julyzone = -p->tm_gmtoff;
727 strncpy(julyname, p->tm_zone ? p->tm_zone : " ", 9);
728 julyname[9] = '\0';
Guido van Rossum10b164a2001-09-25 13:59:01 +0000729
Guido van Rossum57731601999-03-29 19:12:04 +0000730 if( janzone < julyzone ) {
731 /* DST is reversed in the southern hemisphere */
Fred Drake9bb74322002-04-01 14:49:59 +0000732 PyModule_AddIntConstant(m, "timezone", julyzone);
733 PyModule_AddIntConstant(m, "altzone", janzone);
734 PyModule_AddIntConstant(m, "daylight",
735 janzone != julyzone);
736 PyModule_AddObject(m, "tzname",
737 Py_BuildValue("(zz)",
738 julyname, janname));
Guido van Rossum57731601999-03-29 19:12:04 +0000739 } else {
Fred Drake9bb74322002-04-01 14:49:59 +0000740 PyModule_AddIntConstant(m, "timezone", janzone);
741 PyModule_AddIntConstant(m, "altzone", julyzone);
742 PyModule_AddIntConstant(m, "daylight",
743 janzone != julyzone);
744 PyModule_AddObject(m, "tzname",
745 Py_BuildValue("(zz)",
746 janname, julyname));
Guido van Rossum57731601999-03-29 19:12:04 +0000747 }
Guido van Rossum234f9421993-06-17 12:35:49 +0000748 }
Guido van Rossume6a4b7b1997-10-08 15:27:56 +0000749#else
Martin v. Löwis60a5d722002-10-16 20:28:25 +0000750#endif /* HAVE_STRUCT_TM_TM_ZONE */
Tim Peters26ae7cd2001-03-20 03:26:49 +0000751#ifdef __CYGWIN__
752 tzset();
Fred Drake9bb74322002-04-01 14:49:59 +0000753 PyModule_AddIntConstant(m, "timezone", _timezone);
Georg Brandl378d5922006-05-17 14:26:50 +0000754 PyModule_AddIntConstant(m, "altzone", _timezone-3600);
Fred Drake9bb74322002-04-01 14:49:59 +0000755 PyModule_AddIntConstant(m, "daylight", _daylight);
756 PyModule_AddObject(m, "tzname",
757 Py_BuildValue("(zz)", _tzname[0], _tzname[1]));
Tim Peters26ae7cd2001-03-20 03:26:49 +0000758#endif /* __CYGWIN__ */
Guido van Rossum10b164a2001-09-25 13:59:01 +0000759#endif /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000760}
761
762
763static PyMethodDef time_methods[] = {
Georg Brandl96a8c392006-05-29 21:04:52 +0000764 {"time", time_time, METH_NOARGS, time_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000765#ifdef HAVE_CLOCK
Georg Brandl96a8c392006-05-29 21:04:52 +0000766 {"clock", time_clock, METH_NOARGS, clock_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000767#endif
768 {"sleep", time_sleep, METH_VARARGS, sleep_doc},
769 {"gmtime", time_gmtime, METH_VARARGS, gmtime_doc},
770 {"localtime", time_localtime, METH_VARARGS, localtime_doc},
771 {"asctime", time_asctime, METH_VARARGS, asctime_doc},
772 {"ctime", time_ctime, METH_VARARGS, ctime_doc},
773#ifdef HAVE_MKTIME
Georg Brandl96a8c392006-05-29 21:04:52 +0000774 {"mktime", time_mktime, METH_O, mktime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000775#endif
776#ifdef HAVE_STRFTIME
777 {"strftime", time_strftime, METH_VARARGS, strftime_doc},
778#endif
779 {"strptime", time_strptime, METH_VARARGS, strptime_doc},
780#ifdef HAVE_WORKING_TZSET
Georg Brandl96a8c392006-05-29 21:04:52 +0000781 {"tzset", time_tzset, METH_NOARGS, tzset_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000782#endif
783 {NULL, NULL} /* sentinel */
784};
785
786
787PyDoc_STRVAR(module_doc,
788"This module provides various functions to manipulate time values.\n\
789\n\
790There are two standard representations of time. One is the number\n\
791of seconds since the Epoch, in UTC (a.k.a. GMT). It may be an integer\n\
792or a floating point number (to represent fractions of seconds).\n\
793The Epoch is system-defined; on Unix, it is generally January 1st, 1970.\n\
794The actual value can be retrieved by calling gmtime(0).\n\
795\n\
796The other representation is a tuple of 9 integers giving local time.\n\
797The tuple items are:\n\
798 year (four digits, e.g. 1998)\n\
799 month (1-12)\n\
800 day (1-31)\n\
801 hours (0-23)\n\
802 minutes (0-59)\n\
803 seconds (0-59)\n\
804 weekday (0-6, Monday is 0)\n\
805 Julian day (day in the year, 1-366)\n\
806 DST (Daylight Savings Time) flag (-1, 0 or 1)\n\
807If the DST flag is 0, the time is given in the regular time zone;\n\
808if it is 1, the time is given in the DST time zone;\n\
809if it is -1, mktime() should guess based on the date and time.\n\
810\n\
811Variables:\n\
812\n\
813timezone -- difference in seconds between UTC and local standard time\n\
814altzone -- difference in seconds between UTC and local DST time\n\
815daylight -- whether local time should reflect DST\n\
816tzname -- tuple of (standard time zone name, DST time zone name)\n\
817\n\
818Functions:\n\
819\n\
820time() -- return current time in seconds since the Epoch as a float\n\
821clock() -- return CPU time since process start as a float\n\
822sleep() -- delay for a number of seconds given as a float\n\
823gmtime() -- convert seconds since Epoch to UTC tuple\n\
824localtime() -- convert seconds since Epoch to local time tuple\n\
825asctime() -- convert time tuple to string\n\
826ctime() -- convert time in seconds to string\n\
827mktime() -- convert local time tuple to seconds since Epoch\n\
828strftime() -- convert time tuple to string according to format specification\n\
829strptime() -- parse string to time tuple according to format specification\n\
830tzset() -- change the local timezone");
831
832
833PyMODINIT_FUNC
834inittime(void)
835{
836 PyObject *m;
837 char *p;
838 m = Py_InitModule3("time", time_methods, module_doc);
Neal Norwitz1ac754f2006-01-19 06:09:39 +0000839 if (m == NULL)
840 return;
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000841
842 /* Accept 2-digit dates unless PYTHONY2K is set and non-empty */
843 p = Py_GETENV("PYTHONY2K");
844 PyModule_AddIntConstant(m, "accept2dyear", (long) (!p || !*p));
845 /* Squirrel away the module's dictionary for the y2k check */
846 moddict = PyModule_GetDict(m);
847 Py_INCREF(moddict);
848
849 /* Set, or reset, module variables like time.timezone */
850 inittimezone(m);
851
Mark Hammond975e3922002-07-16 01:29:19 +0000852#ifdef MS_WINDOWS
853 /* Helper to allow interrupts for Windows.
854 If Ctrl+C event delivered while not sleeping
855 it will be ignored.
856 */
857 main_thread = PyThread_get_thread_ident();
858 hInterruptEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
859 SetConsoleCtrlHandler( PyCtrlHandler, TRUE);
860#endif /* MS_WINDOWS */
Martin v. Löwis19ab6c92006-04-16 18:55:50 +0000861 if (!initialized) {
Tim Peters7a822da2006-05-25 21:50:17 +0000862 PyStructSequence_InitType(&StructTimeType,
Martin v. Löwis19ab6c92006-04-16 18:55:50 +0000863 &struct_time_type_desc);
864 }
Fred Drake9bb74322002-04-01 14:49:59 +0000865 Py_INCREF(&StructTimeType);
866 PyModule_AddObject(m, "struct_time", (PyObject*) &StructTimeType);
Martin v. Löwis19ab6c92006-04-16 18:55:50 +0000867 initialized = 1;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000868}
869
870
Guido van Rossumb6775db1994-08-01 11:34:53 +0000871/* Implement floattime() for various platforms */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000872
Guido van Rossumb6775db1994-08-01 11:34:53 +0000873static double
Thomas Woutersf3f33dc2000-07-21 06:00:07 +0000874floattime(void)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000875{
Guido van Rossumb6775db1994-08-01 11:34:53 +0000876 /* There are three ways to get the time:
Barry Warsaw4a6cf411997-01-13 22:44:55 +0000877 (1) gettimeofday() -- resolution in microseconds
878 (2) ftime() -- resolution in milliseconds
879 (3) time() -- resolution in seconds
880 In all cases the return value is a float in seconds.
881 Since on some systems (e.g. SCO ODT 3.0) gettimeofday() may
882 fail, so we fall back on ftime() or time().
883 Note: clock resolution does not imply clock accuracy! */
Guido van Rossumb6775db1994-08-01 11:34:53 +0000884#ifdef HAVE_GETTIMEOFDAY
Barry Warsaw4a6cf411997-01-13 22:44:55 +0000885 {
886 struct timeval t;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000887#ifdef GETTIMEOFDAY_NO_TZ
Barry Warsaw4a6cf411997-01-13 22:44:55 +0000888 if (gettimeofday(&t) == 0)
889 return (double)t.tv_sec + t.tv_usec*0.000001;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000890#else /* !GETTIMEOFDAY_NO_TZ */
Barry Warsaw4a6cf411997-01-13 22:44:55 +0000891 if (gettimeofday(&t, (struct timezone *)NULL) == 0)
892 return (double)t.tv_sec + t.tv_usec*0.000001;
Guido van Rossum3bbc62e1995-01-02 19:30:30 +0000893#endif /* !GETTIMEOFDAY_NO_TZ */
Barry Warsaw4a6cf411997-01-13 22:44:55 +0000894 }
Ronald Oussorend06b6f22006-04-23 11:59:25 +0000895
Guido van Rossumb6775db1994-08-01 11:34:53 +0000896#endif /* !HAVE_GETTIMEOFDAY */
Barry Warsaw4a6cf411997-01-13 22:44:55 +0000897 {
Guido van Rossumd3eb5771999-03-09 16:07:23 +0000898#if defined(HAVE_FTIME)
Barry Warsaw4a6cf411997-01-13 22:44:55 +0000899 struct timeb t;
900 ftime(&t);
901 return (double)t.time + (double)t.millitm * (double)0.001;
Guido van Rossumb6775db1994-08-01 11:34:53 +0000902#else /* !HAVE_FTIME */
Barry Warsaw4a6cf411997-01-13 22:44:55 +0000903 time_t secs;
904 time(&secs);
905 return (double)secs;
Guido van Rossumb6775db1994-08-01 11:34:53 +0000906#endif /* !HAVE_FTIME */
Barry Warsaw4a6cf411997-01-13 22:44:55 +0000907 }
Guido van Rossum426035c1991-02-19 12:27:35 +0000908}
909
Guido van Rossumb6775db1994-08-01 11:34:53 +0000910
911/* Implement floatsleep() for various platforms.
912 When interrupted (or when another error occurs), return -1 and
913 set an exception; else return 0. */
914
915static int
Guido van Rossuma320fd31995-03-09 12:14:15 +0000916floatsleep(double secs)
Guido van Rossum426035c1991-02-19 12:27:35 +0000917{
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000918/* XXX Should test for MS_WINDOWS first! */
Andrew MacIntyre7bf68332002-03-03 02:59:16 +0000919#if defined(HAVE_SELECT) && !defined(__BEOS__) && !defined(__EMX__)
Guido van Rossum426035c1991-02-19 12:27:35 +0000920 struct timeval t;
Guido van Rossum775f4da1993-01-09 17:18:52 +0000921 double frac;
Guido van Rossum775f4da1993-01-09 17:18:52 +0000922 frac = fmod(secs, 1.0);
923 secs = floor(secs);
924 t.tv_sec = (long)secs;
925 t.tv_usec = (long)(frac*1000000.0);
Guido van Rossum8607ae21997-11-03 22:04:46 +0000926 Py_BEGIN_ALLOW_THREADS
Guido van Rossumb6775db1994-08-01 11:34:53 +0000927 if (select(0, (fd_set *)0, (fd_set *)0, (fd_set *)0, &t) != 0) {
Guido van Rossum09cbb011999-11-08 15:32:27 +0000928#ifdef EINTR
Guido van Rossuma5456d51999-08-19 14:40:27 +0000929 if (errno != EINTR) {
Guido van Rossum09cbb011999-11-08 15:32:27 +0000930#else
931 if (1) {
932#endif
Andrew M. Kuchlingc24ca4b2000-03-24 20:35:20 +0000933 Py_BLOCK_THREADS
Guido van Rossuma5456d51999-08-19 14:40:27 +0000934 PyErr_SetFromErrno(PyExc_IOError);
935 return -1;
936 }
Guido van Rossumb6775db1994-08-01 11:34:53 +0000937 }
Guido van Rossum8607ae21997-11-03 22:04:46 +0000938 Py_END_ALLOW_THREADS
Martin v. Löwis02af9642002-01-16 11:04:06 +0000939#elif defined(__WATCOMC__) && !defined(__QNX__)
Guido van Rossumbceeac81996-05-23 22:53:47 +0000940 /* XXX Can't interrupt this sleep */
Guido van Rossum8607ae21997-11-03 22:04:46 +0000941 Py_BEGIN_ALLOW_THREADS
Guido van Rossumbceeac81996-05-23 22:53:47 +0000942 delay((int)(secs * 1000 + 0.5)); /* delay() uses milliseconds */
Guido van Rossum8607ae21997-11-03 22:04:46 +0000943 Py_END_ALLOW_THREADS
Martin v. Löwis6238d2b2002-06-30 15:26:10 +0000944#elif defined(MS_WINDOWS)
Fred Drake0e123952000-06-29 21:31:02 +0000945 {
946 double millisecs = secs * 1000.0;
Tim Peters513a1cd2003-01-19 04:54:58 +0000947 unsigned long ul_millis;
948
Fred Drake0e123952000-06-29 21:31:02 +0000949 if (millisecs > (double)ULONG_MAX) {
Tim Peters513a1cd2003-01-19 04:54:58 +0000950 PyErr_SetString(PyExc_OverflowError,
951 "sleep length is too large");
Fred Drake0e123952000-06-29 21:31:02 +0000952 return -1;
953 }
Fred Drake0e123952000-06-29 21:31:02 +0000954 Py_BEGIN_ALLOW_THREADS
Tim Peters513a1cd2003-01-19 04:54:58 +0000955 /* Allow sleep(0) to maintain win32 semantics, and as decreed
956 * by Guido, only the main thread can be interrupted.
957 */
958 ul_millis = (unsigned long)millisecs;
959 if (ul_millis == 0 ||
960 main_thread != PyThread_get_thread_ident())
961 Sleep(ul_millis);
Mark Hammond975e3922002-07-16 01:29:19 +0000962 else {
963 DWORD rc;
964 ResetEvent(hInterruptEvent);
Tim Peters513a1cd2003-01-19 04:54:58 +0000965 rc = WaitForSingleObject(hInterruptEvent, ul_millis);
966 if (rc == WAIT_OBJECT_0) {
967 /* Yield to make sure real Python signal
968 * handler called.
969 */
Mark Hammond975e3922002-07-16 01:29:19 +0000970 Sleep(1);
971 Py_BLOCK_THREADS
Mark Hammond975e3922002-07-16 01:29:19 +0000972 errno = EINTR;
973 PyErr_SetFromErrno(PyExc_IOError);
974 return -1;
975 }
976 }
Fred Drake0e123952000-06-29 21:31:02 +0000977 Py_END_ALLOW_THREADS
978 }
Martin v. Löwis02af9642002-01-16 11:04:06 +0000979#elif defined(PYOS_OS2)
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000980 /* This Sleep *IS* Interruptable by Exceptions */
Guido van Rossum1d0d7e41997-12-29 20:03:10 +0000981 Py_BEGIN_ALLOW_THREADS
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000982 if (DosSleep(secs * 1000) != NO_ERROR) {
Guido van Rossum1d0d7e41997-12-29 20:03:10 +0000983 Py_BLOCK_THREADS
Guido van Rossum8e9ebfd1997-11-22 21:53:48 +0000984 PyErr_SetFromErrno(PyExc_IOError);
985 return -1;
986 }
Guido van Rossum1d0d7e41997-12-29 20:03:10 +0000987 Py_END_ALLOW_THREADS
Martin v. Löwis02af9642002-01-16 11:04:06 +0000988#elif defined(__BEOS__)
Guido van Rossumbcc20741998-08-04 22:53:56 +0000989 /* This sleep *CAN BE* interrupted. */
990 {
Guido van Rossumbcc20741998-08-04 22:53:56 +0000991 if( secs <= 0.0 ) {
992 return;
993 }
Guido van Rossum10b164a2001-09-25 13:59:01 +0000994
Guido van Rossumbcc20741998-08-04 22:53:56 +0000995 Py_BEGIN_ALLOW_THREADS
Guido van Rossumd3eb5771999-03-09 16:07:23 +0000996 /* BeOS snooze() is in microseconds... */
997 if( snooze( (bigtime_t)( secs * 1000.0 * 1000.0 ) ) == B_INTERRUPTED ) {
Guido van Rossumbcc20741998-08-04 22:53:56 +0000998 Py_BLOCK_THREADS
999 PyErr_SetFromErrno( PyExc_IOError );
1000 return -1;
1001 }
1002 Py_END_ALLOW_THREADS
1003 }
Martin v. Löwis02af9642002-01-16 11:04:06 +00001004#elif defined(RISCOS)
Guido van Rossumbceccf52001-04-10 22:07:43 +00001005 if (secs <= 0.0)
1006 return 0;
1007 Py_BEGIN_ALLOW_THREADS
1008 /* This sleep *CAN BE* interrupted. */
Martin v. Löwisa94568a2003-05-10 07:36:56 +00001009 if ( riscos_sleep(secs) )
Guido van Rossumbceccf52001-04-10 22:07:43 +00001010 return -1;
1011 Py_END_ALLOW_THREADS
Martin v. Löwis02af9642002-01-16 11:04:06 +00001012#elif defined(PLAN9)
1013 {
1014 double millisecs = secs * 1000.0;
1015 if (millisecs > (double)LONG_MAX) {
1016 PyErr_SetString(PyExc_OverflowError, "sleep length is too large");
1017 return -1;
1018 }
1019 /* This sleep *CAN BE* interrupted. */
1020 Py_BEGIN_ALLOW_THREADS
1021 if(sleep((long)millisecs) < 0){
1022 Py_BLOCK_THREADS
1023 PyErr_SetFromErrno(PyExc_IOError);
1024 return -1;
1025 }
1026 Py_END_ALLOW_THREADS
1027 }
1028#else
Guido van Rossumb6775db1994-08-01 11:34:53 +00001029 /* XXX Can't interrupt this sleep */
Guido van Rossum8607ae21997-11-03 22:04:46 +00001030 Py_BEGIN_ALLOW_THREADS
Guido van Rossumb6775db1994-08-01 11:34:53 +00001031 sleep((int)secs);
Guido van Rossum8607ae21997-11-03 22:04:46 +00001032 Py_END_ALLOW_THREADS
Martin v. Löwis02af9642002-01-16 11:04:06 +00001033#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001034
Guido van Rossumb6775db1994-08-01 11:34:53 +00001035 return 0;
Guido van Rossum80c9d881991-04-16 08:47:51 +00001036}
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001037
1038