blob: 197d2c0b8dd357eaf60e6f90c6c638d83199691d [file] [log] [blame]
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001/* Time module */
2
Barry Warsaw9a2a8a81996-12-06 23:32:14 +00003#include "Python.h"
Guido van Rossum3f5da241990-12-20 15:06:42 +00004
Guido van Rossum87ce7bb1998-06-09 16:30:31 +00005#include <ctype.h>
6
Victor Stinnerec895392012-04-29 02:41:27 +02007#ifdef HAVE_SYS_TIMES_H
8#include <sys/times.h>
9#endif
10
Thomas Wouters0e3f5912006-08-11 14:57:12 +000011#ifdef HAVE_SYS_TYPES_H
Guido van Rossumb6775db1994-08-01 11:34:53 +000012#include <sys/types.h>
Victor Stinnerec895392012-04-29 02:41:27 +020013#endif
14
15#if defined(HAVE_SYS_RESOURCE_H)
16#include <sys/resource.h>
17#endif
Guido van Rossum6d946f91992-08-14 13:49:30 +000018
Guido van Rossumb6775db1994-08-01 11:34:53 +000019#ifdef QUICKWIN
20#include <io.h>
21#endif
22
Guido van Rossum7bf22de1997-12-02 20:34:19 +000023#if defined(__WATCOMC__) && !defined(__QNX__)
Guido van Rossumbceeac81996-05-23 22:53:47 +000024#include <i86.h>
25#else
Guido van Rossumcac6c721996-09-06 13:34:02 +000026#ifdef MS_WINDOWS
Mark Hammond975e3922002-07-16 01:29:19 +000027#define WIN32_LEAN_AND_MEAN
Guido van Rossum258ccd42001-03-02 06:53:29 +000028#include <windows.h>
Mark Hammond975e3922002-07-16 01:29:19 +000029#include "pythread.h"
Guido van Rossumcac6c721996-09-06 13:34:02 +000030#endif /* MS_WINDOWS */
Guido van Rossum7bf22de1997-12-02 20:34:19 +000031#endif /* !__WATCOMC__ || __QNX__ */
Guido van Rossum234f9421993-06-17 12:35:49 +000032
33/* Forward declarations */
Victor Stinnercb29f012015-03-27 13:31:18 +010034static int pysleep(_PyTime_t);
Victor Stinnerec895392012-04-29 02:41:27 +020035static PyObject* floattime(_Py_clock_info_t *info);
36
Barry Warsaw9a2a8a81996-12-06 23:32:14 +000037static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +010038time_time(PyObject *self, PyObject *unused)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000039{
Victor Stinnerec895392012-04-29 02:41:27 +020040 return floattime(NULL);
Guido van Rossumb6775db1994-08-01 11:34:53 +000041}
42
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000043PyDoc_STRVAR(time_doc,
Victor Stinner4195b5c2012-02-08 23:03:19 +010044"time() -> floating point number\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +000045\n\
46Return the current time in seconds since the Epoch.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000047Fractions of a second may be present if the system clock provides them.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +000048
Victor Stinner85fdfa82012-01-27 00:38:48 +010049#if defined(HAVE_CLOCK)
50
51#ifndef CLOCKS_PER_SEC
52#ifdef CLK_TCK
53#define CLOCKS_PER_SEC CLK_TCK
54#else
55#define CLOCKS_PER_SEC 1000000
56#endif
57#endif
58
Victor Stinner4195b5c2012-02-08 23:03:19 +010059static PyObject *
Victor Stinnerec895392012-04-29 02:41:27 +020060floatclock(_Py_clock_info_t *info)
Victor Stinner85fdfa82012-01-27 00:38:48 +010061{
Victor Stinner4195b5c2012-02-08 23:03:19 +010062 clock_t value;
63 value = clock();
64 if (value == (clock_t)-1) {
Victor Stinner85fdfa82012-01-27 00:38:48 +010065 PyErr_SetString(PyExc_RuntimeError,
66 "the processor time used is not available "
67 "or its value cannot be represented");
Victor Stinner4195b5c2012-02-08 23:03:19 +010068 return NULL;
Victor Stinner85fdfa82012-01-27 00:38:48 +010069 }
Victor Stinnerec895392012-04-29 02:41:27 +020070 if (info) {
71 info->implementation = "clock()";
72 info->resolution = 1.0 / (double)CLOCKS_PER_SEC;
Benjamin Peterson49a69e42012-05-01 09:38:34 -040073 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +020074 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +020075 }
Victor Stinner4195b5c2012-02-08 23:03:19 +010076 return PyFloat_FromDouble((double)value / CLOCKS_PER_SEC);
Victor Stinner85fdfa82012-01-27 00:38:48 +010077}
78#endif /* HAVE_CLOCK */
79
Victor Stinnerf427a142014-10-22 12:33:23 +020080#ifdef MS_WINDOWS
Victor Stinnerec895392012-04-29 02:41:27 +020081#define WIN32_PERF_COUNTER
Victor Stinner9122fdd2011-07-04 13:55:40 +020082/* Win32 has better clock replacement; we have our own version, due to Mark
83 Hammond and Tim Peters */
Victor Stinner54884492014-08-29 16:51:33 +020084static PyObject*
85win_perf_counter(_Py_clock_info_t *info)
Guido van Rossum3917c221997-04-02 05:35:28 +000086{
Victor Stinner8b302012012-02-07 23:29:46 +010087 static LONGLONG cpu_frequency = 0;
Victor Stinner4195b5c2012-02-08 23:03:19 +010088 static LONGLONG ctrStart;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000089 LARGE_INTEGER now;
Victor Stinner4195b5c2012-02-08 23:03:19 +010090 double diff;
Guido van Rossum3917c221997-04-02 05:35:28 +000091
Victor Stinner8b302012012-02-07 23:29:46 +010092 if (cpu_frequency == 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000093 LARGE_INTEGER freq;
Victor Stinner8b302012012-02-07 23:29:46 +010094 QueryPerformanceCounter(&now);
Victor Stinner4195b5c2012-02-08 23:03:19 +010095 ctrStart = now.QuadPart;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000096 if (!QueryPerformanceFrequency(&freq) || freq.QuadPart == 0) {
Victor Stinner54884492014-08-29 16:51:33 +020097 PyErr_SetFromWindowsErr(0);
98 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000099 }
Victor Stinner8b302012012-02-07 23:29:46 +0100100 cpu_frequency = freq.QuadPart;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000101 }
102 QueryPerformanceCounter(&now);
Victor Stinner4195b5c2012-02-08 23:03:19 +0100103 diff = (double)(now.QuadPart - ctrStart);
Victor Stinnerec895392012-04-29 02:41:27 +0200104 if (info) {
105 info->implementation = "QueryPerformanceCounter()";
106 info->resolution = 1.0 / (double)cpu_frequency;
Benjamin Peterson49a69e42012-05-01 09:38:34 -0400107 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +0200108 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +0200109 }
Victor Stinner54884492014-08-29 16:51:33 +0200110 return PyFloat_FromDouble(diff / (double)cpu_frequency);
Guido van Rossum3917c221997-04-02 05:35:28 +0000111}
Victor Stinnerf427a142014-10-22 12:33:23 +0200112#endif /* MS_WINDOWS */
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000113
Victor Stinnerec895392012-04-29 02:41:27 +0200114#if defined(WIN32_PERF_COUNTER) || defined(HAVE_CLOCK)
115#define PYCLOCK
116static PyObject*
117pyclock(_Py_clock_info_t *info)
118{
119#ifdef WIN32_PERF_COUNTER
Victor Stinner54884492014-08-29 16:51:33 +0200120 return win_perf_counter(info);
121#else
Victor Stinnerec895392012-04-29 02:41:27 +0200122 return floatclock(info);
Victor Stinner54884492014-08-29 16:51:33 +0200123#endif
Victor Stinnerec895392012-04-29 02:41:27 +0200124}
125
Victor Stinner9122fdd2011-07-04 13:55:40 +0200126static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +0100127time_clock(PyObject *self, PyObject *unused)
Victor Stinner9122fdd2011-07-04 13:55:40 +0200128{
Victor Stinnerec895392012-04-29 02:41:27 +0200129 return pyclock(NULL);
Victor Stinner9122fdd2011-07-04 13:55:40 +0200130}
Victor Stinner9122fdd2011-07-04 13:55:40 +0200131
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000132PyDoc_STRVAR(clock_doc,
Victor Stinner4195b5c2012-02-08 23:03:19 +0100133"clock() -> floating point number\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000134\n\
135Return the CPU time or real time since the start of the process or since\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000136the first call to clock(). This has as much precision as the system\n\
137records.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000138#endif
139
Victor Stinnere0be4232011-10-25 13:06:09 +0200140#ifdef HAVE_CLOCK_GETTIME
141static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +0100142time_clock_gettime(PyObject *self, PyObject *args)
Victor Stinnere0be4232011-10-25 13:06:09 +0200143{
144 int ret;
Victor Stinnerb8d01692012-04-13 23:44:05 +0200145 int clk_id;
Victor Stinnere0be4232011-10-25 13:06:09 +0200146 struct timespec tp;
147
Victor Stinner4195b5c2012-02-08 23:03:19 +0100148 if (!PyArg_ParseTuple(args, "i:clock_gettime", &clk_id))
Victor Stinnere0be4232011-10-25 13:06:09 +0200149 return NULL;
150
151 ret = clock_gettime((clockid_t)clk_id, &tp);
Antoine Pitrou2c085602012-01-18 01:41:44 +0100152 if (ret != 0) {
Victor Stinnera734af32014-07-31 13:07:17 +0200153 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrou2c085602012-01-18 01:41:44 +0100154 return NULL;
155 }
Victor Stinner4195b5c2012-02-08 23:03:19 +0100156 return PyFloat_FromDouble(tp.tv_sec + tp.tv_nsec * 1e-9);
Victor Stinnere0be4232011-10-25 13:06:09 +0200157}
158
159PyDoc_STRVAR(clock_gettime_doc,
Victor Stinner4195b5c2012-02-08 23:03:19 +0100160"clock_gettime(clk_id) -> floating point number\n\
Victor Stinnere0be4232011-10-25 13:06:09 +0200161\n\
162Return the time of the specified clock clk_id.");
Victor Stinner30d79472012-04-03 00:45:07 +0200163
164static PyObject *
165time_clock_settime(PyObject *self, PyObject *args)
166{
Victor Stinnerb8d01692012-04-13 23:44:05 +0200167 int clk_id;
Victor Stinner30d79472012-04-03 00:45:07 +0200168 PyObject *obj;
Victor Stinnerb3b45442015-03-28 04:09:41 +0100169 _PyTime_t t;
Victor Stinner30d79472012-04-03 00:45:07 +0200170 struct timespec tp;
171 int ret;
172
173 if (!PyArg_ParseTuple(args, "iO:clock_settime", &clk_id, &obj))
174 return NULL;
175
Victor Stinner02937aa2015-03-28 05:02:39 +0100176 if (_PyTime_FromSecondsObject(&t, obj, _PyTime_ROUND_FLOOR) < 0)
Victor Stinner30d79472012-04-03 00:45:07 +0200177 return NULL;
Victor Stinnerb3b45442015-03-28 04:09:41 +0100178
179 if (_PyTime_AsTimespec(t, &tp) == -1)
180 return NULL;
Victor Stinner30d79472012-04-03 00:45:07 +0200181
182 ret = clock_settime((clockid_t)clk_id, &tp);
183 if (ret != 0) {
Victor Stinnera734af32014-07-31 13:07:17 +0200184 PyErr_SetFromErrno(PyExc_OSError);
Victor Stinner30d79472012-04-03 00:45:07 +0200185 return NULL;
186 }
187 Py_RETURN_NONE;
188}
189
190PyDoc_STRVAR(clock_settime_doc,
191"clock_settime(clk_id, time)\n\
192\n\
193Set the time of the specified clock clk_id.");
Victor Stinnere0be4232011-10-25 13:06:09 +0200194
Victor Stinnere0be4232011-10-25 13:06:09 +0200195static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +0100196time_clock_getres(PyObject *self, PyObject *args)
Victor Stinnere0be4232011-10-25 13:06:09 +0200197{
198 int ret;
Victor Stinnerb8d01692012-04-13 23:44:05 +0200199 int clk_id;
Victor Stinnere0be4232011-10-25 13:06:09 +0200200 struct timespec tp;
201
Victor Stinner4195b5c2012-02-08 23:03:19 +0100202 if (!PyArg_ParseTuple(args, "i:clock_getres", &clk_id))
Victor Stinnere0be4232011-10-25 13:06:09 +0200203 return NULL;
204
205 ret = clock_getres((clockid_t)clk_id, &tp);
Antoine Pitrou2c085602012-01-18 01:41:44 +0100206 if (ret != 0) {
Victor Stinnera734af32014-07-31 13:07:17 +0200207 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrou2c085602012-01-18 01:41:44 +0100208 return NULL;
209 }
Victor Stinner4195b5c2012-02-08 23:03:19 +0100210
211 return PyFloat_FromDouble(tp.tv_sec + tp.tv_nsec * 1e-9);
Victor Stinnere0be4232011-10-25 13:06:09 +0200212}
213
214PyDoc_STRVAR(clock_getres_doc,
Victor Stinner4195b5c2012-02-08 23:03:19 +0100215"clock_getres(clk_id) -> floating point number\n\
Victor Stinnere0be4232011-10-25 13:06:09 +0200216\n\
217Return the resolution (precision) of the specified clock clk_id.");
Victor Stinnerb8d01692012-04-13 23:44:05 +0200218#endif /* HAVE_CLOCK_GETTIME */
Victor Stinnere0be4232011-10-25 13:06:09 +0200219
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000220static PyObject *
Victor Stinnercb29f012015-03-27 13:31:18 +0100221time_sleep(PyObject *self, PyObject *obj)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000222{
Victor Stinnercb29f012015-03-27 13:31:18 +0100223 _PyTime_t secs;
Victor Stinner869e1772015-03-30 03:49:14 +0200224 if (_PyTime_FromSecondsObject(&secs, obj, _PyTime_ROUND_CEILING))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000225 return NULL;
Victor Stinner7f53a502011-07-05 22:00:25 +0200226 if (secs < 0) {
227 PyErr_SetString(PyExc_ValueError,
228 "sleep length must be non-negative");
229 return NULL;
230 }
Victor Stinnercb29f012015-03-27 13:31:18 +0100231 if (pysleep(secs) != 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000232 return NULL;
233 Py_INCREF(Py_None);
234 return Py_None;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000235}
236
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000237PyDoc_STRVAR(sleep_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000238"sleep(seconds)\n\
239\n\
240Delay execution for a given number of seconds. The argument may be\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000241a floating point number for subsecond precision.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000242
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000243static PyStructSequence_Field struct_time_type_fields[] = {
Alexander Belopolsky69f3fd02010-06-05 15:04:51 +0000244 {"tm_year", "year, for example, 1993"},
245 {"tm_mon", "month of year, range [1, 12]"},
246 {"tm_mday", "day of month, range [1, 31]"},
247 {"tm_hour", "hours, range [0, 23]"},
248 {"tm_min", "minutes, range [0, 59]"},
249 {"tm_sec", "seconds, range [0, 61])"},
250 {"tm_wday", "day of week, range [0, 6], Monday is 0"},
251 {"tm_yday", "day of year, range [1, 366]"},
252 {"tm_isdst", "1 if summer time is in effect, 0 if not, and -1 if unknown"},
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400253#ifdef HAVE_STRUCT_TM_TM_ZONE
254 {"tm_zone", "abbreviation of timezone name"},
255 {"tm_gmtoff", "offset from UTC in seconds"},
256#endif /* HAVE_STRUCT_TM_TM_ZONE */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000257 {0}
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000258};
259
260static PyStructSequence_Desc struct_time_type_desc = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000261 "time.struct_time",
Alexander Belopolsky69f3fd02010-06-05 15:04:51 +0000262 "The time value as returned by gmtime(), localtime(), and strptime(), and\n"
263 " accepted by asctime(), mktime() and strftime(). May be considered as a\n"
264 " sequence of 9 integers.\n\n"
265 " Note that several fields' values are not the same as those defined by\n"
266 " the C language standard for struct tm. For example, the value of the\n"
267 " field tm_year is the actual year, not year - 1900. See individual\n"
268 " fields' descriptions for details.",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000269 struct_time_type_fields,
270 9,
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000271};
Tim Peters9ad4b682002-02-13 05:14:18 +0000272
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000273static int initialized;
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000274static PyTypeObject StructTimeType;
275
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400276
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000277static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000278tmtotuple(struct tm *p)
Guido van Rossum87ce7bb1998-06-09 16:30:31 +0000279{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000280 PyObject *v = PyStructSequence_New(&StructTimeType);
281 if (v == NULL)
282 return NULL;
Tim Peters9ad4b682002-02-13 05:14:18 +0000283
Christian Heimes217cfd12007-12-02 14:31:20 +0000284#define SET(i,val) PyStructSequence_SET_ITEM(v, i, PyLong_FromLong((long) val))
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000285
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000286 SET(0, p->tm_year + 1900);
287 SET(1, p->tm_mon + 1); /* Want January == 1 */
288 SET(2, p->tm_mday);
289 SET(3, p->tm_hour);
290 SET(4, p->tm_min);
291 SET(5, p->tm_sec);
292 SET(6, (p->tm_wday + 6) % 7); /* Want Monday == 0 */
293 SET(7, p->tm_yday + 1); /* Want January, 1 == 1 */
294 SET(8, p->tm_isdst);
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400295#ifdef HAVE_STRUCT_TM_TM_ZONE
296 PyStructSequence_SET_ITEM(v, 9,
297 PyUnicode_DecodeLocale(p->tm_zone, "surrogateescape"));
298 SET(10, p->tm_gmtoff);
299#endif /* HAVE_STRUCT_TM_TM_ZONE */
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000300#undef SET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000301 if (PyErr_Occurred()) {
302 Py_XDECREF(v);
303 return NULL;
304 }
Guido van Rossum98bf58f2001-10-18 20:34:25 +0000305
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000306 return v;
Guido van Rossum87ce7bb1998-06-09 16:30:31 +0000307}
308
Fred Drakef901abd2004-08-03 17:58:55 +0000309/* Parse arg tuple that can contain an optional float-or-None value;
310 format needs to be "|O:name".
311 Returns non-zero on success (parallels PyArg_ParseTuple).
312*/
313static int
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100314parse_time_t_args(PyObject *args, char *format, time_t *pwhen)
Fred Drakef901abd2004-08-03 17:58:55 +0000315{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000316 PyObject *ot = NULL;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100317 time_t whent;
Fred Drakef901abd2004-08-03 17:58:55 +0000318
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000319 if (!PyArg_ParseTuple(args, format, &ot))
320 return 0;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100321 if (ot == NULL || ot == Py_None) {
322 whent = time(NULL);
323 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000324 else {
Victor Stinner02937aa2015-03-28 05:02:39 +0100325 if (_PyTime_ObjectToTime_t(ot, &whent, _PyTime_ROUND_FLOOR) == -1)
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100326 return 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000327 }
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100328 *pwhen = whent;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000329 return 1;
Fred Drakef901abd2004-08-03 17:58:55 +0000330}
331
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000332static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000333time_gmtime(PyObject *self, PyObject *args)
Guido van Rossum234f9421993-06-17 12:35:49 +0000334{
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100335 time_t when;
336 struct tm buf, *local;
337
338 if (!parse_time_t_args(args, "|O:gmtime", &when))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000339 return NULL;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100340
341 errno = 0;
342 local = gmtime(&when);
343 if (local == NULL) {
344#ifdef EINVAL
345 if (errno == 0)
346 errno = EINVAL;
347#endif
348 return PyErr_SetFromErrno(PyExc_OSError);
349 }
350 buf = *local;
351 return tmtotuple(&buf);
Guido van Rossum234f9421993-06-17 12:35:49 +0000352}
353
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000354PyDoc_STRVAR(gmtime_doc,
Christian Heimes9a371592007-12-28 14:08:13 +0000355"gmtime([seconds]) -> (tm_year, tm_mon, tm_mday, tm_hour, tm_min,\n\
Fred Drake193a3f62002-03-12 21:38:49 +0000356 tm_sec, tm_wday, tm_yday, tm_isdst)\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000357\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000358Convert seconds since the Epoch to a time tuple expressing UTC (a.k.a.\n\
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400359GMT). When 'seconds' is not passed in, convert the current time instead.\n\
360\n\
361If the platform supports the tm_gmtoff and tm_zone, they are available as\n\
362attributes only.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000363
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100364static int
365pylocaltime(time_t *timep, struct tm *result)
366{
367 struct tm *local;
368
369 assert (timep != NULL);
370 local = localtime(timep);
371 if (local == NULL) {
372 /* unconvertible time */
373#ifdef EINVAL
374 if (errno == 0)
375 errno = EINVAL;
376#endif
377 PyErr_SetFromErrno(PyExc_OSError);
378 return -1;
379 }
380 *result = *local;
381 return 0;
382}
383
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000384static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000385time_localtime(PyObject *self, PyObject *args)
Guido van Rossum234f9421993-06-17 12:35:49 +0000386{
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100387 time_t when;
388 struct tm buf;
389
390 if (!parse_time_t_args(args, "|O:localtime", &when))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000391 return NULL;
Alexander Belopolskyd9738242012-06-12 16:14:17 -0400392 if (pylocaltime(&when, &buf) == -1)
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100393 return NULL;
394 return tmtotuple(&buf);
Guido van Rossum234f9421993-06-17 12:35:49 +0000395}
396
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000397PyDoc_STRVAR(localtime_doc,
Christian Heimes9a371592007-12-28 14:08:13 +0000398"localtime([seconds]) -> (tm_year,tm_mon,tm_mday,tm_hour,tm_min,\n\
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000399 tm_sec,tm_wday,tm_yday,tm_isdst)\n\
Martin v. Löwisb3cfc1d2001-12-02 12:27:43 +0000400\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000401Convert seconds since the Epoch to a time tuple expressing local time.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000402When 'seconds' is not passed in, convert the current time instead.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000403
Alexander Belopolsky38e29962010-10-01 14:18:49 +0000404/* Convert 9-item tuple to tm structure. Return 1 on success, set
405 * an exception and return 0 on error.
406 */
Guido van Rossum9e90a671993-06-24 11:10:19 +0000407static int
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000408gettmarg(PyObject *args, struct tm *p)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000409{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000410 int y;
Guido van Rossumcfbaecc1998-08-25 14:51:12 +0000411
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000412 memset((void *) p, '\0', sizeof(struct tm));
Guido van Rossumb9081262007-08-25 03:14:09 +0000413
Alexander Belopolsky610e5442011-01-06 21:57:06 +0000414 if (!PyTuple_Check(args)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000415 PyErr_SetString(PyExc_TypeError,
416 "Tuple or struct_time argument required");
417 return 0;
418 }
Skip Montanaro41cfce92007-08-24 21:11:00 +0000419
Alexander Belopolsky610e5442011-01-06 21:57:06 +0000420 if (!PyArg_ParseTuple(args, "iiiiiiiii",
421 &y, &p->tm_mon, &p->tm_mday,
422 &p->tm_hour, &p->tm_min, &p->tm_sec,
423 &p->tm_wday, &p->tm_yday, &p->tm_isdst))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000424 return 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000425 p->tm_year = y - 1900;
426 p->tm_mon--;
427 p->tm_wday = (p->tm_wday + 1) % 7;
428 p->tm_yday--;
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400429#ifdef HAVE_STRUCT_TM_TM_ZONE
430 if (Py_TYPE(args) == &StructTimeType) {
431 PyObject *item;
432 item = PyTuple_GET_ITEM(args, 9);
433 p->tm_zone = item == Py_None ? NULL : _PyUnicode_AsString(item);
434 item = PyTuple_GET_ITEM(args, 10);
435 p->tm_gmtoff = item == Py_None ? 0 : PyLong_AsLong(item);
436 if (PyErr_Occurred())
437 return 0;
438 }
439#endif /* HAVE_STRUCT_TM_TM_ZONE */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000440 return 1;
Guido van Rossum9e90a671993-06-24 11:10:19 +0000441}
442
Alexander Belopolsky38e29962010-10-01 14:18:49 +0000443/* Check values of the struct tm fields before it is passed to strftime() and
444 * asctime(). Return 1 if all values are valid, otherwise set an exception
445 * and returns 0.
446 */
Victor Stinneref128102010-10-07 01:00:52 +0000447static int
448checktm(struct tm* buf)
Alexander Belopolsky38e29962010-10-01 14:18:49 +0000449{
Victor Stinneref128102010-10-07 01:00:52 +0000450 /* Checks added to make sure strftime() and asctime() does not crash Python by
451 indexing blindly into some array for a textual representation
452 by some bad index (fixes bug #897625 and #6608).
453
454 Also support values of zero from Python code for arguments in which
455 that is out of range by forcing that value to the lowest value that
456 is valid (fixed bug #1520914).
457
458 Valid ranges based on what is allowed in struct tm:
459
460 - tm_year: [0, max(int)] (1)
461 - tm_mon: [0, 11] (2)
462 - tm_mday: [1, 31]
463 - tm_hour: [0, 23]
464 - tm_min: [0, 59]
465 - tm_sec: [0, 60]
466 - tm_wday: [0, 6] (1)
467 - tm_yday: [0, 365] (2)
468 - tm_isdst: [-max(int), max(int)]
469
470 (1) gettmarg() handles bounds-checking.
471 (2) Python's acceptable range is one greater than the range in C,
Alexander Belopolsky38e29962010-10-01 14:18:49 +0000472 thus need to check against automatic decrement by gettmarg().
473 */
474 if (buf->tm_mon == -1)
475 buf->tm_mon = 0;
476 else if (buf->tm_mon < 0 || buf->tm_mon > 11) {
477 PyErr_SetString(PyExc_ValueError, "month out of range");
478 return 0;
479 }
480 if (buf->tm_mday == 0)
481 buf->tm_mday = 1;
482 else if (buf->tm_mday < 0 || buf->tm_mday > 31) {
483 PyErr_SetString(PyExc_ValueError, "day of month out of range");
484 return 0;
485 }
486 if (buf->tm_hour < 0 || buf->tm_hour > 23) {
487 PyErr_SetString(PyExc_ValueError, "hour out of range");
488 return 0;
489 }
490 if (buf->tm_min < 0 || buf->tm_min > 59) {
491 PyErr_SetString(PyExc_ValueError, "minute out of range");
492 return 0;
493 }
494 if (buf->tm_sec < 0 || buf->tm_sec > 61) {
495 PyErr_SetString(PyExc_ValueError, "seconds out of range");
496 return 0;
497 }
498 /* tm_wday does not need checking of its upper-bound since taking
499 ``% 7`` in gettmarg() automatically restricts the range. */
500 if (buf->tm_wday < 0) {
501 PyErr_SetString(PyExc_ValueError, "day of week out of range");
502 return 0;
503 }
504 if (buf->tm_yday == -1)
505 buf->tm_yday = 0;
506 else if (buf->tm_yday < 0 || buf->tm_yday > 365) {
507 PyErr_SetString(PyExc_ValueError, "day of year out of range");
508 return 0;
509 }
510 return 1;
511}
512
Victor Stinnerc1f32ca2011-10-14 02:36:13 +0200513#ifdef MS_WINDOWS
514 /* wcsftime() doesn't format correctly time zones, see issue #10653 */
515# undef HAVE_WCSFTIME
516#endif
Alexander Belopolskycf774542012-10-02 18:39:16 -0400517#define STRFTIME_FORMAT_CODES \
518"Commonly used format codes:\n\
519\n\
520%Y Year with century as a decimal number.\n\
521%m Month as a decimal number [01,12].\n\
522%d Day of the month as a decimal number [01,31].\n\
523%H Hour (24-hour clock) as a decimal number [00,23].\n\
524%M Minute as a decimal number [00,59].\n\
525%S Second as a decimal number [00,61].\n\
526%z Time zone offset from UTC.\n\
527%a Locale's abbreviated weekday name.\n\
528%A Locale's full weekday name.\n\
529%b Locale's abbreviated month name.\n\
530%B Locale's full month name.\n\
531%c Locale's appropriate date and time representation.\n\
532%I Hour (12-hour clock) as a decimal number [01,12].\n\
533%p Locale's equivalent of either AM or PM.\n\
534\n\
535Other codes may be available on your platform. See documentation for\n\
536the C library strftime function.\n"
Victor Stinnerc1f32ca2011-10-14 02:36:13 +0200537
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000538#ifdef HAVE_STRFTIME
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000539#ifdef HAVE_WCSFTIME
540#define time_char wchar_t
541#define format_time wcsftime
542#define time_strlen wcslen
543#else
544#define time_char char
545#define format_time strftime
546#define time_strlen strlen
547#endif
548
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000549static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000550time_strftime(PyObject *self, PyObject *args)
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000551{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000552 PyObject *tup = NULL;
553 struct tm buf;
554 const time_char *fmt;
Victor Stinnerb2904782010-09-29 10:34:19 +0000555#ifdef HAVE_WCSFTIME
556 wchar_t *format;
557#else
558 PyObject *format;
559#endif
Victor Stinneref128102010-10-07 01:00:52 +0000560 PyObject *format_arg;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000561 size_t fmtlen, buflen;
Victor Stinnerb2904782010-09-29 10:34:19 +0000562 time_char *outbuf = NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000563 size_t i;
Victor Stinnerb2904782010-09-29 10:34:19 +0000564 PyObject *ret = NULL;
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000565
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000566 memset((void *) &buf, '\0', sizeof(buf));
Guido van Rossum1f41f841998-04-27 19:04:26 +0000567
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000568 /* Will always expect a unicode string to be passed as format.
569 Given that there's no str type anymore in py3k this seems safe.
570 */
Victor Stinneref128102010-10-07 01:00:52 +0000571 if (!PyArg_ParseTuple(args, "U|O:strftime", &format_arg, &tup))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000572 return NULL;
Thomas Woutersfe385252001-01-19 23:16:56 +0000573
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000574 if (tup == NULL) {
575 time_t tt = time(NULL);
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100576 if (pylocaltime(&tt, &buf) == -1)
577 return NULL;
Alexander Belopolsky38e29962010-10-01 14:18:49 +0000578 }
579 else if (!gettmarg(tup, &buf) || !checktm(&buf))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000580 return NULL;
Guido van Rossum10b164a2001-09-25 13:59:01 +0000581
Victor Stinner36b82d82013-06-25 02:33:53 +0200582#if defined(_MSC_VER) || defined(sun) || defined(_AIX)
Victor Stinner73ea29c2011-01-08 01:56:31 +0000583 if (buf.tm_year + 1900 < 1 || 9999 < buf.tm_year + 1900) {
Victor Stinner6f0e4f92011-03-21 02:14:53 +0100584 PyErr_SetString(PyExc_ValueError,
585 "strftime() requires year in [1; 9999]");
Alexander Belopolsky0dd06f42011-01-08 01:23:02 +0000586 return NULL;
Alexander Belopolskyc64708a2011-01-07 19:59:19 +0000587 }
Victor Stinner73ea29c2011-01-08 01:56:31 +0000588#endif
Alexander Belopolskyc64708a2011-01-07 19:59:19 +0000589
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000590 /* Normalize tm_isdst just in case someone foolishly implements %Z
591 based on the assumption that tm_isdst falls within the range of
592 [-1, 1] */
593 if (buf.tm_isdst < -1)
594 buf.tm_isdst = -1;
595 else if (buf.tm_isdst > 1)
596 buf.tm_isdst = 1;
Brett Cannond1080a32004-03-02 04:38:10 +0000597
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000598#ifdef HAVE_WCSFTIME
Victor Stinnerbeb4135b2010-10-07 01:02:42 +0000599 format = PyUnicode_AsWideCharString(format_arg, NULL);
Victor Stinnerb2904782010-09-29 10:34:19 +0000600 if (format == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000601 return NULL;
Victor Stinnerb2904782010-09-29 10:34:19 +0000602 fmt = format;
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000603#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000604 /* Convert the unicode string to an ascii one */
Victor Stinner1b579672011-12-17 05:47:23 +0100605 format = PyUnicode_EncodeLocale(format_arg, "surrogateescape");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000606 if (format == NULL)
607 return NULL;
608 fmt = PyBytes_AS_STRING(format);
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000609#endif
Amaury Forgeot d'Arcb5be6d42009-03-02 23:52:57 +0000610
Stefan Krah4aea7d32012-02-27 16:30:26 +0100611#if defined(MS_WINDOWS) && !defined(HAVE_WCSFTIME)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000612 /* check that the format string contains only valid directives */
Victor Stinner5a3ff792011-10-16 19:08:23 +0200613 for(outbuf = strchr(fmt, '%');
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000614 outbuf != NULL;
Victor Stinner5a3ff792011-10-16 19:08:23 +0200615 outbuf = strchr(outbuf+2, '%'))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000616 {
617 if (outbuf[1]=='#')
618 ++outbuf; /* not documented by python, */
Tim Golden6e51b8f2013-11-12 12:36:54 +0000619 if ((outbuf[1] == 'y') && buf.tm_year < 0)
620 {
621 PyErr_SetString(PyExc_ValueError,
622 "format %y requires year >= 1900 on Windows");
623 Py_DECREF(format);
624 return NULL;
625 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000626 }
Victor Stinner93965f72013-11-23 14:59:33 +0100627#elif (defined(_AIX) || defined(sun)) && defined(HAVE_WCSFTIME)
Victor Stinner55329f82013-11-17 23:39:21 +0100628 for(outbuf = wcschr(fmt, '%');
629 outbuf != NULL;
630 outbuf = wcschr(outbuf+2, '%'))
631 {
632 /* Issue #19634: On AIX, wcsftime("y", (1899, 1, 1, 0, 0, 0, 0, 0, 0))
633 returns "0/" instead of "99" */
634 if (outbuf[1] == L'y' && buf.tm_year < 0) {
635 PyErr_SetString(PyExc_ValueError,
636 "format %y requires year >= 1900 on AIX");
Victor Stinner55329f82013-11-17 23:39:21 +0100637 return NULL;
638 }
639 }
Amaury Forgeot d'Arcb5be6d42009-03-02 23:52:57 +0000640#endif
641
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000642 fmtlen = time_strlen(fmt);
Guido van Rossumc222ec21999-02-23 00:00:10 +0000643
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000644 /* I hate these functions that presume you know how big the output
645 * will be ahead of time...
646 */
647 for (i = 1024; ; i += i) {
Victor Stinner136ea492011-12-17 22:37:18 +0100648#if defined _MSC_VER && _MSC_VER >= 1400 && defined(__STDC_SECURE_LIB__)
Antoine Pitrouc345ce12011-12-16 12:28:32 +0100649 int err;
Victor Stinner136ea492011-12-17 22:37:18 +0100650#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000651 outbuf = (time_char *)PyMem_Malloc(i*sizeof(time_char));
652 if (outbuf == NULL) {
Victor Stinnerb2904782010-09-29 10:34:19 +0000653 PyErr_NoMemory();
654 break;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000655 }
Steve Dowerd9ef74e2015-05-22 15:08:34 -0700656 _Py_BEGIN_SUPPRESS_IPH
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000657 buflen = format_time(outbuf, i, fmt, &buf);
Steve Dowerd9ef74e2015-05-22 15:08:34 -0700658 _Py_END_SUPPRESS_IPH
Victor Stinner136ea492011-12-17 22:37:18 +0100659#if defined _MSC_VER && _MSC_VER >= 1400 && defined(__STDC_SECURE_LIB__)
Antoine Pitrouc345ce12011-12-16 12:28:32 +0100660 err = errno;
Victor Stinner136ea492011-12-17 22:37:18 +0100661#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000662 if (buflen > 0 || i >= 256 * fmtlen) {
663 /* If the buffer is 256 times as long as the format,
664 it's probably not failing for lack of room!
665 More likely, the format yields an empty result,
666 e.g. an empty format, or %Z when the timezone
667 is unknown. */
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000668#ifdef HAVE_WCSFTIME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000669 ret = PyUnicode_FromWideChar(outbuf, buflen);
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000670#else
Victor Stinner1b579672011-12-17 05:47:23 +0100671 ret = PyUnicode_DecodeLocaleAndSize(outbuf, buflen,
672 "surrogateescape");
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000673#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000674 PyMem_Free(outbuf);
Victor Stinnerb2904782010-09-29 10:34:19 +0000675 break;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000676 }
677 PyMem_Free(outbuf);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000678#if defined _MSC_VER && _MSC_VER >= 1400 && defined(__STDC_SECURE_LIB__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000679 /* VisualStudio .NET 2005 does this properly */
Antoine Pitrouc345ce12011-12-16 12:28:32 +0100680 if (buflen == 0 && err == EINVAL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000681 PyErr_SetString(PyExc_ValueError, "Invalid format string");
Victor Stinnerb2904782010-09-29 10:34:19 +0000682 break;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000683 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000684#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000685 }
Victor Stinnerb2904782010-09-29 10:34:19 +0000686#ifdef HAVE_WCSFTIME
687 PyMem_Free(format);
688#else
689 Py_DECREF(format);
690#endif
691 return ret;
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000692}
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000693
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000694#undef time_char
695#undef format_time
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000696PyDoc_STRVAR(strftime_doc,
Thomas Woutersfe385252001-01-19 23:16:56 +0000697"strftime(format[, tuple]) -> string\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000698\n\
699Convert a time tuple to a string according to a format specification.\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000700See the library reference manual for formatting codes. When the time tuple\n\
Alexander Belopolskycf774542012-10-02 18:39:16 -0400701is not present, current time as returned by localtime() is used.\n\
702\n" STRFTIME_FORMAT_CODES);
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000703#endif /* HAVE_STRFTIME */
704
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000705static PyObject *
706time_strptime(PyObject *self, PyObject *args)
707{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000708 PyObject *strptime_module = PyImport_ImportModuleNoBlock("_strptime");
709 PyObject *strptime_result;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200710 _Py_IDENTIFIER(_strptime_time);
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000711
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000712 if (!strptime_module)
713 return NULL;
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200714 strptime_result = _PyObject_CallMethodId(strptime_module,
715 &PyId__strptime_time, "O", args);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000716 Py_DECREF(strptime_module);
717 return strptime_result;
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000718}
719
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000720
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000721PyDoc_STRVAR(strptime_doc,
Brett Cannon20def8b2003-07-01 05:16:08 +0000722"strptime(string, format) -> struct_time\n\
Martin v. Löwisb3cfc1d2001-12-02 12:27:43 +0000723\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000724Parse a string to a time tuple according to a format specification.\n\
Alexander Belopolskycf774542012-10-02 18:39:16 -0400725See the library reference manual for formatting codes (same as\n\
726strftime()).\n\
727\n" STRFTIME_FORMAT_CODES);
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000728
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000729static PyObject *
730_asctime(struct tm *timeptr)
731{
732 /* Inspired by Open Group reference implementation available at
733 * http://pubs.opengroup.org/onlinepubs/009695399/functions/asctime.html */
Victor Stinner499dfcf2011-03-21 13:26:24 +0100734 static char wday_name[7][4] = {
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000735 "Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"
736 };
Victor Stinner499dfcf2011-03-21 13:26:24 +0100737 static char mon_name[12][4] = {
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000738 "Jan", "Feb", "Mar", "Apr", "May", "Jun",
739 "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
740 };
Victor Stinner499dfcf2011-03-21 13:26:24 +0100741 return PyUnicode_FromFormat(
742 "%s %s%3d %.2d:%.2d:%.2d %d",
743 wday_name[timeptr->tm_wday],
744 mon_name[timeptr->tm_mon],
745 timeptr->tm_mday, timeptr->tm_hour,
746 timeptr->tm_min, timeptr->tm_sec,
747 1900 + timeptr->tm_year);
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000748}
Guido van Rossum87ce7bb1998-06-09 16:30:31 +0000749
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000750static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000751time_asctime(PyObject *self, PyObject *args)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000752{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000753 PyObject *tup = NULL;
754 struct tm buf;
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000755
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000756 if (!PyArg_UnpackTuple(args, "asctime", 0, 1, &tup))
757 return NULL;
758 if (tup == NULL) {
759 time_t tt = time(NULL);
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100760 if (pylocaltime(&tt, &buf) == -1)
761 return NULL;
762
Alexander Belopolsky38e29962010-10-01 14:18:49 +0000763 } else if (!gettmarg(tup, &buf) || !checktm(&buf))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000764 return NULL;
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000765 return _asctime(&buf);
Guido van Rossum9e90a671993-06-24 11:10:19 +0000766}
767
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000768PyDoc_STRVAR(asctime_doc,
Thomas Woutersfe385252001-01-19 23:16:56 +0000769"asctime([tuple]) -> string\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000770\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000771Convert a time tuple to a string, e.g. 'Sat Jun 06 16:26:11 1998'.\n\
772When the time tuple is not present, current time as returned by localtime()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000773is used.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000774
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000775static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000776time_ctime(PyObject *self, PyObject *args)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000777{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000778 time_t tt;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100779 struct tm buf;
780 if (!parse_time_t_args(args, "|O:ctime", &tt))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000781 return NULL;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100782 if (pylocaltime(&tt, &buf) == -1)
Alexander Belopolsky5da468f2011-01-04 17:15:52 +0000783 return NULL;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100784 return _asctime(&buf);
Guido van Rossum9e90a671993-06-24 11:10:19 +0000785}
786
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000787PyDoc_STRVAR(ctime_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000788"ctime(seconds) -> string\n\
789\n\
790Convert a time in seconds since the Epoch to a string in local time.\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000791This is equivalent to asctime(localtime(seconds)). When the time tuple is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000792not present, current time as returned by localtime() is used.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000793
Guido van Rossum60cd8131998-03-06 17:16:21 +0000794#ifdef HAVE_MKTIME
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000795static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +0100796time_mktime(PyObject *self, PyObject *tup)
Guido van Rossum234f9421993-06-17 12:35:49 +0000797{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000798 struct tm buf;
799 time_t tt;
800 if (!gettmarg(tup, &buf))
801 return NULL;
Victor Stinner1ac42612014-02-21 09:27:17 +0100802#ifdef _AIX
803 /* year < 1902 or year > 2037 */
804 if (buf.tm_year < 2 || buf.tm_year > 137) {
805 /* Issue #19748: On AIX, mktime() doesn't report overflow error for
806 * timestamp < -2^31 or timestamp > 2**31-1. */
807 PyErr_SetString(PyExc_OverflowError,
808 "mktime argument out of range");
809 return NULL;
810 }
811#else
Alexander Belopolskyb7d40d12011-01-11 01:21:25 +0000812 buf.tm_wday = -1; /* sentinel; original value ignored */
Victor Stinner1ac42612014-02-21 09:27:17 +0100813#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000814 tt = mktime(&buf);
Alexander Belopolskyb7d40d12011-01-11 01:21:25 +0000815 /* Return value of -1 does not necessarily mean an error, but tm_wday
Ezio Melotti13925002011-03-16 11:05:33 +0200816 * cannot remain set to -1 if mktime succeeded. */
Victor Stinner93037492013-06-25 22:54:35 +0200817 if (tt == (time_t)(-1)
818#ifndef _AIX
819 /* Return value of -1 does not necessarily mean an error, but
820 * tm_wday cannot remain set to -1 if mktime succeeded. */
821 && buf.tm_wday == -1
822#else
823 /* on AIX, tm_wday is always sets, even on error */
824#endif
825 )
826 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000827 PyErr_SetString(PyExc_OverflowError,
828 "mktime argument out of range");
829 return NULL;
830 }
Victor Stinner4195b5c2012-02-08 23:03:19 +0100831 return PyFloat_FromDouble((double)tt);
Guido van Rossum234f9421993-06-17 12:35:49 +0000832}
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000833
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000834PyDoc_STRVAR(mktime_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000835"mktime(tuple) -> floating point number\n\
836\n\
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400837Convert a time tuple in local time to seconds since the Epoch.\n\
838Note that mktime(gmtime(0)) will not generally return zero for most\n\
839time zones; instead the returned value will either be equal to that\n\
840of the timezone or altzone attributes on the time module.");
Guido van Rossum60cd8131998-03-06 17:16:21 +0000841#endif /* HAVE_MKTIME */
Guido van Rossum234f9421993-06-17 12:35:49 +0000842
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000843#ifdef HAVE_WORKING_TZSET
Martin v. Löwis1a214512008-06-11 05:26:20 +0000844static void PyInit_timezone(PyObject *module);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000845
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000846static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000847time_tzset(PyObject *self, PyObject *unused)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000848{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000849 PyObject* m;
Fred Drake9bb74322002-04-01 14:49:59 +0000850
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000851 m = PyImport_ImportModuleNoBlock("time");
852 if (m == NULL) {
853 return NULL;
854 }
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000855
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000856 tzset();
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000857
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000858 /* Reset timezone, altzone, daylight and tzname */
859 PyInit_timezone(m);
860 Py_DECREF(m);
Victor Stinner2ff51b82013-07-17 21:42:45 +0200861 if (PyErr_Occurred())
862 return NULL;
Tim Peters1b6f7a92004-06-20 02:50:16 +0000863
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000864 Py_INCREF(Py_None);
865 return Py_None;
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000866}
867
868PyDoc_STRVAR(tzset_doc,
R. David Murray4d55bf92010-12-14 00:55:46 +0000869"tzset()\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000870\n\
871Initialize, or reinitialize, the local timezone to the value stored in\n\
872os.environ['TZ']. The TZ environment variable should be specified in\n\
Neal Norwitzdc8e1942004-07-20 22:34:37 +0000873standard Unix timezone format as documented in the tzset man page\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000874(eg. 'US/Eastern', 'Europe/Amsterdam'). Unknown timezones will silently\n\
875fall back to UTC. If the TZ environment variable is not set, the local\n\
876timezone is set to the systems best guess of wallclock time.\n\
877Changing the TZ environment variable without calling tzset *may* change\n\
878the local timezone used by methods such as localtime, but this behaviour\n\
879should not be relied on.");
880#endif /* HAVE_WORKING_TZSET */
881
Victor Stinnerae586492014-09-02 23:18:25 +0200882static PyObject *
Victor Stinnerec895392012-04-29 02:41:27 +0200883pymonotonic(_Py_clock_info_t *info)
Victor Stinnerb94b2662012-01-18 01:50:21 +0100884{
Victor Stinner4bfb4602015-03-27 22:27:24 +0100885 _PyTime_t t;
886 double d;
887 if (_PyTime_GetMonotonicClockWithInfo(&t, info) < 0) {
Victor Stinnerae586492014-09-02 23:18:25 +0200888 assert(info != NULL);
Victor Stinner071eca32012-03-15 01:17:09 +0100889 return NULL;
890 }
Victor Stinner4bfb4602015-03-27 22:27:24 +0100891 d = _PyTime_AsSecondsDouble(t);
892 return PyFloat_FromDouble(d);
Victor Stinner8b302012012-02-07 23:29:46 +0100893}
894
Victor Stinner071eca32012-03-15 01:17:09 +0100895static PyObject *
Victor Stinnerec895392012-04-29 02:41:27 +0200896time_monotonic(PyObject *self, PyObject *unused)
Victor Stinner071eca32012-03-15 01:17:09 +0100897{
Victor Stinnerec895392012-04-29 02:41:27 +0200898 return pymonotonic(NULL);
Victor Stinner071eca32012-03-15 01:17:09 +0100899}
900
Victor Stinnerec895392012-04-29 02:41:27 +0200901PyDoc_STRVAR(monotonic_doc,
902"monotonic() -> float\n\
Victor Stinner8b302012012-02-07 23:29:46 +0100903\n\
Victor Stinnerec895392012-04-29 02:41:27 +0200904Monotonic clock, cannot go backward.");
Victor Stinnerec919cc2012-03-15 00:58:32 +0100905
Victor Stinnerec895392012-04-29 02:41:27 +0200906static PyObject*
907perf_counter(_Py_clock_info_t *info)
908{
Victor Stinner54884492014-08-29 16:51:33 +0200909#ifdef WIN32_PERF_COUNTER
910 return win_perf_counter(info);
911#else
Victor Stinnerae586492014-09-02 23:18:25 +0200912 return pymonotonic(info);
Victor Stinner54884492014-08-29 16:51:33 +0200913#endif
Victor Stinnerec895392012-04-29 02:41:27 +0200914}
915
916static PyObject *
917time_perf_counter(PyObject *self, PyObject *unused)
918{
919 return perf_counter(NULL);
920}
921
922PyDoc_STRVAR(perf_counter_doc,
923"perf_counter() -> float\n\
924\n\
925Performance counter for benchmarking.");
926
927static PyObject*
928py_process_time(_Py_clock_info_t *info)
929{
930#if defined(MS_WINDOWS)
931 HANDLE process;
932 FILETIME creation_time, exit_time, kernel_time, user_time;
933 ULARGE_INTEGER large;
934 double total;
935 BOOL ok;
936
937 process = GetCurrentProcess();
938 ok = GetProcessTimes(process, &creation_time, &exit_time, &kernel_time, &user_time);
939 if (!ok)
940 return PyErr_SetFromWindowsErr(0);
941
942 large.u.LowPart = kernel_time.dwLowDateTime;
943 large.u.HighPart = kernel_time.dwHighDateTime;
944 total = (double)large.QuadPart;
945 large.u.LowPart = user_time.dwLowDateTime;
946 large.u.HighPart = user_time.dwHighDateTime;
947 total += (double)large.QuadPart;
948 if (info) {
949 info->implementation = "GetProcessTimes()";
950 info->resolution = 1e-7;
Benjamin Peterson49a69e42012-05-01 09:38:34 -0400951 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +0200952 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +0200953 }
954 return PyFloat_FromDouble(total * 1e-7);
955#else
956
957#if defined(HAVE_SYS_RESOURCE_H)
958 struct rusage ru;
959#endif
960#ifdef HAVE_TIMES
961 struct tms t;
962 static long ticks_per_second = -1;
963#endif
964
965#if defined(HAVE_CLOCK_GETTIME) \
966 && (defined(CLOCK_PROCESS_CPUTIME_ID) || defined(CLOCK_PROF))
967 struct timespec tp;
968#ifdef CLOCK_PROF
969 const clockid_t clk_id = CLOCK_PROF;
970 const char *function = "clock_gettime(CLOCK_PROF)";
971#else
972 const clockid_t clk_id = CLOCK_PROCESS_CPUTIME_ID;
973 const char *function = "clock_gettime(CLOCK_PROCESS_CPUTIME_ID)";
974#endif
975
976 if (clock_gettime(clk_id, &tp) == 0) {
977 if (info) {
978 struct timespec res;
979 info->implementation = function;
Benjamin Peterson49a69e42012-05-01 09:38:34 -0400980 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +0200981 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +0200982 if (clock_getres(clk_id, &res) == 0)
983 info->resolution = res.tv_sec + res.tv_nsec * 1e-9;
984 else
985 info->resolution = 1e-9;
986 }
987 return PyFloat_FromDouble(tp.tv_sec + tp.tv_nsec * 1e-9);
988 }
989#endif
990
991#if defined(HAVE_SYS_RESOURCE_H)
992 if (getrusage(RUSAGE_SELF, &ru) == 0) {
993 double total;
994 total = ru.ru_utime.tv_sec + ru.ru_utime.tv_usec * 1e-6;
995 total += ru.ru_stime.tv_sec + ru.ru_stime.tv_usec * 1e-6;
996 if (info) {
997 info->implementation = "getrusage(RUSAGE_SELF)";
Benjamin Peterson49a69e42012-05-01 09:38:34 -0400998 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +0200999 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001000 info->resolution = 1e-6;
1001 }
1002 return PyFloat_FromDouble(total);
1003 }
1004#endif
1005
1006#ifdef HAVE_TIMES
1007 if (times(&t) != (clock_t)-1) {
1008 double total;
1009
1010 if (ticks_per_second == -1) {
1011#if defined(HAVE_SYSCONF) && defined(_SC_CLK_TCK)
1012 ticks_per_second = sysconf(_SC_CLK_TCK);
1013 if (ticks_per_second < 1)
1014 ticks_per_second = -1;
1015#elif defined(HZ)
1016 ticks_per_second = HZ;
1017#else
1018 ticks_per_second = 60; /* magic fallback value; may be bogus */
1019#endif
1020 }
1021
1022 if (ticks_per_second != -1) {
1023 total = (double)t.tms_utime / ticks_per_second;
1024 total += (double)t.tms_stime / ticks_per_second;
1025 if (info) {
1026 info->implementation = "times()";
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001027 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001028 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001029 info->resolution = 1.0 / ticks_per_second;
1030 }
1031 return PyFloat_FromDouble(total);
1032 }
1033 }
1034#endif
1035
1036 return floatclock(info);
1037#endif
1038}
1039
1040static PyObject *
1041time_process_time(PyObject *self, PyObject *unused)
1042{
1043 return py_process_time(NULL);
1044}
1045
1046PyDoc_STRVAR(process_time_doc,
1047"process_time() -> float\n\
1048\n\
1049Process time for profiling: sum of the kernel and user-space CPU time.");
1050
1051
Victor Stinnerec895392012-04-29 02:41:27 +02001052static PyObject *
1053time_get_clock_info(PyObject *self, PyObject *args)
1054{
1055 char *name;
Victor Stinnerec895392012-04-29 02:41:27 +02001056 _Py_clock_info_t info;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001057 PyObject *obj = NULL, *dict, *ns;
Victor Stinnerec895392012-04-29 02:41:27 +02001058
1059 if (!PyArg_ParseTuple(args, "s:get_clock_info", &name))
1060 return NULL;
1061
1062#ifdef Py_DEBUG
1063 info.implementation = NULL;
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001064 info.monotonic = -1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001065 info.adjustable = -1;
Victor Stinnerec895392012-04-29 02:41:27 +02001066 info.resolution = -1.0;
1067#else
1068 info.implementation = "";
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001069 info.monotonic = 0;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001070 info.adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001071 info.resolution = 1.0;
1072#endif
1073
1074 if (strcmp(name, "time") == 0)
1075 obj = floattime(&info);
1076#ifdef PYCLOCK
1077 else if (strcmp(name, "clock") == 0)
1078 obj = pyclock(&info);
1079#endif
Victor Stinnerec895392012-04-29 02:41:27 +02001080 else if (strcmp(name, "monotonic") == 0)
1081 obj = pymonotonic(&info);
Victor Stinnerec895392012-04-29 02:41:27 +02001082 else if (strcmp(name, "perf_counter") == 0)
1083 obj = perf_counter(&info);
1084 else if (strcmp(name, "process_time") == 0)
1085 obj = py_process_time(&info);
1086 else {
1087 PyErr_SetString(PyExc_ValueError, "unknown clock");
1088 return NULL;
1089 }
1090 if (obj == NULL)
1091 return NULL;
1092 Py_DECREF(obj);
1093
Victor Stinnerbda4b882012-06-12 22:11:44 +02001094 dict = PyDict_New();
1095 if (dict == NULL)
Victor Stinnerec895392012-04-29 02:41:27 +02001096 return NULL;
1097
1098 assert(info.implementation != NULL);
1099 obj = PyUnicode_FromString(info.implementation);
1100 if (obj == NULL)
1101 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001102 if (PyDict_SetItemString(dict, "implementation", obj) == -1)
1103 goto error;
1104 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001105
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001106 assert(info.monotonic != -1);
1107 obj = PyBool_FromLong(info.monotonic);
Victor Stinnerec895392012-04-29 02:41:27 +02001108 if (obj == NULL)
1109 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001110 if (PyDict_SetItemString(dict, "monotonic", obj) == -1)
1111 goto error;
1112 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001113
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001114 assert(info.adjustable != -1);
1115 obj = PyBool_FromLong(info.adjustable);
Victor Stinnerec895392012-04-29 02:41:27 +02001116 if (obj == NULL)
1117 goto error;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001118 if (PyDict_SetItemString(dict, "adjustable", obj) == -1)
Victor Stinnerbda4b882012-06-12 22:11:44 +02001119 goto error;
1120 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001121
1122 assert(info.resolution > 0.0);
1123 assert(info.resolution <= 1.0);
1124 obj = PyFloat_FromDouble(info.resolution);
1125 if (obj == NULL)
1126 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001127 if (PyDict_SetItemString(dict, "resolution", obj) == -1)
1128 goto error;
1129 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001130
Victor Stinnerbda4b882012-06-12 22:11:44 +02001131 ns = _PyNamespace_New(dict);
1132 Py_DECREF(dict);
1133 return ns;
Victor Stinnerec895392012-04-29 02:41:27 +02001134
1135error:
Victor Stinnerbda4b882012-06-12 22:11:44 +02001136 Py_DECREF(dict);
1137 Py_XDECREF(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001138 return NULL;
1139}
1140
1141PyDoc_STRVAR(get_clock_info_doc,
1142"get_clock_info(name: str) -> dict\n\
1143\n\
1144Get information of the specified clock.");
Victor Stinner8b302012012-02-07 23:29:46 +01001145
Martin v. Löwisd218dc12008-04-07 03:17:54 +00001146static void
Martin v. Löwis1a214512008-06-11 05:26:20 +00001147PyInit_timezone(PyObject *m) {
1148 /* This code moved from PyInit_time wholesale to allow calling it from
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001149 time_tzset. In the future, some parts of it can be moved back
1150 (for platforms that don't HAVE_WORKING_TZSET, when we know what they
1151 are), and the extraneous calls to tzset(3) should be removed.
1152 I haven't done this yet, as I don't want to change this code as
1153 little as possible when introducing the time.tzset and time.tzsetwall
1154 methods. This should simply be a method of doing the following once,
1155 at the top of this function and removing the call to tzset() from
1156 time_tzset():
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001157
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001158 #ifdef HAVE_TZSET
1159 tzset()
1160 #endif
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001161
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001162 And I'm lazy and hate C so nyer.
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001163 */
Guido van Rossum10b164a2001-09-25 13:59:01 +00001164#if defined(HAVE_TZNAME) && !defined(__GLIBC__) && !defined(__CYGWIN__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001165 PyObject *otz0, *otz1;
1166 tzset();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001167 PyModule_AddIntConstant(m, "timezone", timezone);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001168#ifdef HAVE_ALTZONE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001169 PyModule_AddIntConstant(m, "altzone", altzone);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001170#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001171 PyModule_AddIntConstant(m, "altzone", timezone-3600);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001172#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001173 PyModule_AddIntConstant(m, "daylight", daylight);
Victor Stinner1b579672011-12-17 05:47:23 +01001174 otz0 = PyUnicode_DecodeLocale(tzname[0], "surrogateescape");
1175 otz1 = PyUnicode_DecodeLocale(tzname[1], "surrogateescape");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001176 PyModule_AddObject(m, "tzname", Py_BuildValue("(NN)", otz0, otz1));
Guido van Rossum10b164a2001-09-25 13:59:01 +00001177#else /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001178#ifdef HAVE_STRUCT_TM_TM_ZONE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001179 {
Guido van Rossum234f9421993-06-17 12:35:49 +00001180#define YEAR ((time_t)((365 * 24 + 6) * 3600))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001181 time_t t;
1182 struct tm *p;
1183 long janzone, julyzone;
1184 char janname[10], julyname[10];
1185 t = (time((time_t *)0) / YEAR) * YEAR;
1186 p = localtime(&t);
1187 janzone = -p->tm_gmtoff;
1188 strncpy(janname, p->tm_zone ? p->tm_zone : " ", 9);
1189 janname[9] = '\0';
1190 t += YEAR/2;
1191 p = localtime(&t);
1192 julyzone = -p->tm_gmtoff;
1193 strncpy(julyname, p->tm_zone ? p->tm_zone : " ", 9);
1194 julyname[9] = '\0';
Guido van Rossum10b164a2001-09-25 13:59:01 +00001195
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001196 if( janzone < julyzone ) {
1197 /* DST is reversed in the southern hemisphere */
1198 PyModule_AddIntConstant(m, "timezone", julyzone);
1199 PyModule_AddIntConstant(m, "altzone", janzone);
1200 PyModule_AddIntConstant(m, "daylight",
1201 janzone != julyzone);
1202 PyModule_AddObject(m, "tzname",
1203 Py_BuildValue("(zz)",
1204 julyname, janname));
1205 } else {
1206 PyModule_AddIntConstant(m, "timezone", janzone);
1207 PyModule_AddIntConstant(m, "altzone", julyzone);
1208 PyModule_AddIntConstant(m, "daylight",
1209 janzone != julyzone);
1210 PyModule_AddObject(m, "tzname",
1211 Py_BuildValue("(zz)",
1212 janname, julyname));
1213 }
1214 }
Guido van Rossume6a4b7b1997-10-08 15:27:56 +00001215#else
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001216#endif /* HAVE_STRUCT_TM_TM_ZONE */
Tim Peters26ae7cd2001-03-20 03:26:49 +00001217#ifdef __CYGWIN__
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001218 tzset();
1219 PyModule_AddIntConstant(m, "timezone", _timezone);
1220 PyModule_AddIntConstant(m, "altzone", _timezone-3600);
1221 PyModule_AddIntConstant(m, "daylight", _daylight);
1222 PyModule_AddObject(m, "tzname",
1223 Py_BuildValue("(zz)", _tzname[0], _tzname[1]));
Tim Peters26ae7cd2001-03-20 03:26:49 +00001224#endif /* __CYGWIN__ */
Guido van Rossum10b164a2001-09-25 13:59:01 +00001225#endif /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
Victor Stinnere0be4232011-10-25 13:06:09 +02001226
Victor Stinnerec895392012-04-29 02:41:27 +02001227#if defined(HAVE_CLOCK_GETTIME)
Victor Stinnere0be4232011-10-25 13:06:09 +02001228 PyModule_AddIntMacro(m, CLOCK_REALTIME);
Victor Stinnere0be4232011-10-25 13:06:09 +02001229#ifdef CLOCK_MONOTONIC
1230 PyModule_AddIntMacro(m, CLOCK_MONOTONIC);
1231#endif
1232#ifdef CLOCK_MONOTONIC_RAW
1233 PyModule_AddIntMacro(m, CLOCK_MONOTONIC_RAW);
1234#endif
Victor Stinner1470f352012-04-03 00:31:17 +02001235#ifdef CLOCK_HIGHRES
1236 PyModule_AddIntMacro(m, CLOCK_HIGHRES);
1237#endif
Victor Stinnere0be4232011-10-25 13:06:09 +02001238#ifdef CLOCK_PROCESS_CPUTIME_ID
1239 PyModule_AddIntMacro(m, CLOCK_PROCESS_CPUTIME_ID);
1240#endif
1241#ifdef CLOCK_THREAD_CPUTIME_ID
1242 PyModule_AddIntMacro(m, CLOCK_THREAD_CPUTIME_ID);
1243#endif
1244#endif /* HAVE_CLOCK_GETTIME */
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001245}
1246
1247
1248static PyMethodDef time_methods[] = {
Victor Stinner4195b5c2012-02-08 23:03:19 +01001249 {"time", time_time, METH_NOARGS, time_doc},
Victor Stinnerec895392012-04-29 02:41:27 +02001250#ifdef PYCLOCK
Victor Stinner4195b5c2012-02-08 23:03:19 +01001251 {"clock", time_clock, METH_NOARGS, clock_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001252#endif
Victor Stinnere0be4232011-10-25 13:06:09 +02001253#ifdef HAVE_CLOCK_GETTIME
Victor Stinner4195b5c2012-02-08 23:03:19 +01001254 {"clock_gettime", time_clock_gettime, METH_VARARGS, clock_gettime_doc},
Victor Stinner30d79472012-04-03 00:45:07 +02001255 {"clock_settime", time_clock_settime, METH_VARARGS, clock_settime_doc},
Victor Stinner4195b5c2012-02-08 23:03:19 +01001256 {"clock_getres", time_clock_getres, METH_VARARGS, clock_getres_doc},
Victor Stinnere0be4232011-10-25 13:06:09 +02001257#endif
Victor Stinnercb29f012015-03-27 13:31:18 +01001258 {"sleep", time_sleep, METH_O, sleep_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001259 {"gmtime", time_gmtime, METH_VARARGS, gmtime_doc},
1260 {"localtime", time_localtime, METH_VARARGS, localtime_doc},
1261 {"asctime", time_asctime, METH_VARARGS, asctime_doc},
1262 {"ctime", time_ctime, METH_VARARGS, ctime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001263#ifdef HAVE_MKTIME
Victor Stinner4195b5c2012-02-08 23:03:19 +01001264 {"mktime", time_mktime, METH_O, mktime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001265#endif
1266#ifdef HAVE_STRFTIME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001267 {"strftime", time_strftime, METH_VARARGS, strftime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001268#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001269 {"strptime", time_strptime, METH_VARARGS, strptime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001270#ifdef HAVE_WORKING_TZSET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001271 {"tzset", time_tzset, METH_NOARGS, tzset_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001272#endif
Victor Stinnerec895392012-04-29 02:41:27 +02001273 {"monotonic", time_monotonic, METH_NOARGS, monotonic_doc},
Victor Stinnerec895392012-04-29 02:41:27 +02001274 {"process_time", time_process_time, METH_NOARGS, process_time_doc},
1275 {"perf_counter", time_perf_counter, METH_NOARGS, perf_counter_doc},
1276 {"get_clock_info", time_get_clock_info, METH_VARARGS, get_clock_info_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001277 {NULL, NULL} /* sentinel */
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001278};
1279
1280
1281PyDoc_STRVAR(module_doc,
1282"This module provides various functions to manipulate time values.\n\
1283\n\
1284There are two standard representations of time. One is the number\n\
1285of seconds since the Epoch, in UTC (a.k.a. GMT). It may be an integer\n\
1286or a floating point number (to represent fractions of seconds).\n\
1287The Epoch is system-defined; on Unix, it is generally January 1st, 1970.\n\
1288The actual value can be retrieved by calling gmtime(0).\n\
1289\n\
1290The other representation is a tuple of 9 integers giving local time.\n\
1291The tuple items are:\n\
Alexander Belopolsky03163ac2011-05-02 12:20:52 -04001292 year (including century, e.g. 1998)\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001293 month (1-12)\n\
1294 day (1-31)\n\
1295 hours (0-23)\n\
1296 minutes (0-59)\n\
1297 seconds (0-59)\n\
1298 weekday (0-6, Monday is 0)\n\
1299 Julian day (day in the year, 1-366)\n\
1300 DST (Daylight Savings Time) flag (-1, 0 or 1)\n\
1301If the DST flag is 0, the time is given in the regular time zone;\n\
1302if it is 1, the time is given in the DST time zone;\n\
1303if it is -1, mktime() should guess based on the date and time.\n\
1304\n\
1305Variables:\n\
1306\n\
1307timezone -- difference in seconds between UTC and local standard time\n\
1308altzone -- difference in seconds between UTC and local DST time\n\
1309daylight -- whether local time should reflect DST\n\
1310tzname -- tuple of (standard time zone name, DST time zone name)\n\
1311\n\
1312Functions:\n\
1313\n\
1314time() -- return current time in seconds since the Epoch as a float\n\
1315clock() -- return CPU time since process start as a float\n\
1316sleep() -- delay for a number of seconds given as a float\n\
1317gmtime() -- convert seconds since Epoch to UTC tuple\n\
1318localtime() -- convert seconds since Epoch to local time tuple\n\
1319asctime() -- convert time tuple to string\n\
1320ctime() -- convert time in seconds to string\n\
1321mktime() -- convert local time tuple to seconds since Epoch\n\
1322strftime() -- convert time tuple to string according to format specification\n\
1323strptime() -- parse string to time tuple according to format specification\n\
1324tzset() -- change the local timezone");
1325
1326
Martin v. Löwis1a214512008-06-11 05:26:20 +00001327
1328static struct PyModuleDef timemodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001329 PyModuleDef_HEAD_INIT,
1330 "time",
1331 module_doc,
1332 -1,
1333 time_methods,
1334 NULL,
1335 NULL,
1336 NULL,
1337 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00001338};
1339
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001340PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00001341PyInit_time(void)
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001342{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001343 PyObject *m;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001344 m = PyModule_Create(&timemodule);
1345 if (m == NULL)
1346 return NULL;
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001347
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001348 /* Set, or reset, module variables like time.timezone */
1349 PyInit_timezone(m);
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001350
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001351 if (!initialized) {
Victor Stinner1c8f0592013-07-22 22:24:54 +02001352 if (PyStructSequence_InitType2(&StructTimeType,
1353 &struct_time_type_desc) < 0)
1354 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001355 }
1356 Py_INCREF(&StructTimeType);
Alexander Belopolskyc142bba2012-06-13 22:15:26 -04001357#ifdef HAVE_STRUCT_TM_TM_ZONE
1358 PyModule_AddIntConstant(m, "_STRUCT_TM_ITEMS", 11);
1359#else
1360 PyModule_AddIntConstant(m, "_STRUCT_TM_ITEMS", 9);
1361#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001362 PyModule_AddObject(m, "struct_time", (PyObject*) &StructTimeType);
1363 initialized = 1;
1364 return m;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001365}
1366
Victor Stinner071eca32012-03-15 01:17:09 +01001367static PyObject*
Victor Stinnerec895392012-04-29 02:41:27 +02001368floattime(_Py_clock_info_t *info)
Victor Stinner4195b5c2012-02-08 23:03:19 +01001369{
Victor Stinnera47b8812015-03-27 18:16:17 +01001370 _PyTime_t t;
1371 double d;
1372 if (_PyTime_GetSystemClockWithInfo(&t, info) < 0) {
Victor Stinner00111242014-08-29 16:31:59 +02001373 assert(info != NULL);
1374 return NULL;
1375 }
Victor Stinnera47b8812015-03-27 18:16:17 +01001376 d = _PyTime_AsSecondsDouble(t);
1377 return PyFloat_FromDouble(d);
Victor Stinner4195b5c2012-02-08 23:03:19 +01001378}
1379
1380
Victor Stinnercb29f012015-03-27 13:31:18 +01001381/* Implement pysleep() for various platforms.
Guido van Rossumb6775db1994-08-01 11:34:53 +00001382 When interrupted (or when another error occurs), return -1 and
1383 set an exception; else return 0. */
1384
1385static int
Victor Stinnercb29f012015-03-27 13:31:18 +01001386pysleep(_PyTime_t secs)
Guido van Rossum426035c1991-02-19 12:27:35 +00001387{
Victor Stinnercb29f012015-03-27 13:31:18 +01001388 _PyTime_t deadline, monotonic;
Victor Stinner79d68f92015-03-19 21:54:09 +01001389#ifndef MS_WINDOWS
1390 struct timeval timeout;
Victor Stinner79d68f92015-03-19 21:54:09 +01001391 int err = 0;
1392#else
Victor Stinnercb29f012015-03-27 13:31:18 +01001393 _PyTime_t millisecs;
Victor Stinner79d68f92015-03-19 21:54:09 +01001394 unsigned long ul_millis;
1395 DWORD rc;
1396 HANDLE hInterruptEvent;
Victor Stinner0c2fd892015-03-17 10:49:17 +01001397#endif
Victor Stinner79d68f92015-03-19 21:54:09 +01001398
Victor Stinnercb29f012015-03-27 13:31:18 +01001399 deadline = _PyTime_GetMonotonicClock() + secs;
Victor Stinner79d68f92015-03-19 21:54:09 +01001400
1401 do {
1402#ifndef MS_WINDOWS
Victor Stinner869e1772015-03-30 03:49:14 +02001403 if (_PyTime_AsTimeval(secs, &timeout, _PyTime_ROUND_CEILING) < 0)
Victor Stinnercb29f012015-03-27 13:31:18 +01001404 return -1;
Victor Stinner79d68f92015-03-19 21:54:09 +01001405
1406 Py_BEGIN_ALLOW_THREADS
1407 err = select(0, (fd_set *)0, (fd_set *)0, (fd_set *)0, &timeout);
1408 Py_END_ALLOW_THREADS
1409
1410 if (err == 0)
1411 break;
1412
1413 if (errno != EINTR) {
Victor Stinner0c2fd892015-03-17 10:49:17 +01001414 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001415 return -1;
1416 }
Victor Stinner79d68f92015-03-19 21:54:09 +01001417#else
Victor Stinner869e1772015-03-30 03:49:14 +02001418 millisecs = _PyTime_AsMilliseconds(secs, _PyTime_ROUND_CEILING);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001419 if (millisecs > (double)ULONG_MAX) {
1420 PyErr_SetString(PyExc_OverflowError,
1421 "sleep length is too large");
1422 return -1;
1423 }
Victor Stinner79d68f92015-03-19 21:54:09 +01001424
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001425 /* Allow sleep(0) to maintain win32 semantics, and as decreed
1426 * by Guido, only the main thread can be interrupted.
1427 */
1428 ul_millis = (unsigned long)millisecs;
Victor Stinner79d68f92015-03-19 21:54:09 +01001429 if (ul_millis == 0 || !_PyOS_IsMainThread()) {
1430 Py_BEGIN_ALLOW_THREADS
Victor Stinner0eac1302015-03-20 03:06:12 +01001431 Sleep(ul_millis);
Victor Stinner79d68f92015-03-19 21:54:09 +01001432 Py_END_ALLOW_THREADS
1433 break;
Victor Stinner945c82e2015-03-12 16:19:01 +01001434 }
Victor Stinner79d68f92015-03-19 21:54:09 +01001435
1436 hInterruptEvent = _PyOS_SigintEvent();
1437 ResetEvent(hInterruptEvent);
1438
1439 Py_BEGIN_ALLOW_THREADS
1440 rc = WaitForSingleObjectEx(hInterruptEvent, ul_millis, FALSE);
Victor Stinner0c2fd892015-03-17 10:49:17 +01001441 Py_END_ALLOW_THREADS
Victor Stinner79d68f92015-03-19 21:54:09 +01001442
1443 if (rc != WAIT_OBJECT_0)
1444 break;
Victor Stinner945c82e2015-03-12 16:19:01 +01001445#endif
Victor Stinner0c2fd892015-03-17 10:49:17 +01001446
Victor Stinner79d68f92015-03-19 21:54:09 +01001447 /* sleep was interrupted by SIGINT */
1448 if (PyErr_CheckSignals())
1449 return -1;
1450
Victor Stinnercb29f012015-03-27 13:31:18 +01001451 monotonic = _PyTime_GetMonotonicClock();
1452 secs = deadline - monotonic;
Victor Stinner6aa446c2015-03-30 21:33:51 +02001453 if (secs < 0)
Victor Stinner79d68f92015-03-19 21:54:09 +01001454 break;
1455 /* retry with the recomputed delay */
1456 } while (1);
1457
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001458 return 0;
Guido van Rossum80c9d881991-04-16 08:47:51 +00001459}