blob: 7f5f3149afdb746aceb0d412c061a9f99ce3ec6c [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 */
Tim Petersdbd9ba62000-07-09 03:09:57 +000034static int floatsleep(double);
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;
Antoine Pitroucf8a1e52013-04-17 22:06:44 +0200169 time_t tv_sec;
170 long tv_nsec;
Victor Stinner30d79472012-04-03 00:45:07 +0200171 struct timespec tp;
172 int ret;
173
174 if (!PyArg_ParseTuple(args, "iO:clock_settime", &clk_id, &obj))
175 return NULL;
176
Victor Stinner3c1b3792014-02-17 00:02:43 +0100177 if (_PyTime_ObjectToTimespec(obj, &tv_sec, &tv_nsec, _PyTime_ROUND_DOWN) == -1)
Victor Stinner30d79472012-04-03 00:45:07 +0200178 return NULL;
Antoine Pitroucf8a1e52013-04-17 22:06:44 +0200179 tp.tv_sec = tv_sec;
180 tp.tv_nsec = tv_nsec;
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 *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000221time_sleep(PyObject *self, PyObject *args)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000222{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000223 double secs;
224 if (!PyArg_ParseTuple(args, "d:sleep", &secs))
225 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 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000231 if (floatsleep(secs) != 0)
232 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 Stinner3c1b3792014-02-17 00:02:43 +0100325 if (_PyTime_ObjectToTime_t(ot, &whent, _PyTime_ROUND_DOWN) == -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, */
619 if (outbuf[1]=='\0' ||
Victor Stinner5a3ff792011-10-16 19:08:23 +0200620 !strchr("aAbBcdHIjmMpSUwWxXyYzZ%", outbuf[1]))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000621 {
622 PyErr_SetString(PyExc_ValueError, "Invalid format string");
Stefan Krah4aea7d32012-02-27 16:30:26 +0100623 Py_DECREF(format);
624 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000625 }
Tim Golden6e51b8f2013-11-12 12:36:54 +0000626 if ((outbuf[1] == 'y') && buf.tm_year < 0)
627 {
628 PyErr_SetString(PyExc_ValueError,
629 "format %y requires year >= 1900 on Windows");
630 Py_DECREF(format);
631 return NULL;
632 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000633 }
Victor Stinner93965f72013-11-23 14:59:33 +0100634#elif (defined(_AIX) || defined(sun)) && defined(HAVE_WCSFTIME)
Victor Stinner55329f82013-11-17 23:39:21 +0100635 for(outbuf = wcschr(fmt, '%');
636 outbuf != NULL;
637 outbuf = wcschr(outbuf+2, '%'))
638 {
639 /* Issue #19634: On AIX, wcsftime("y", (1899, 1, 1, 0, 0, 0, 0, 0, 0))
640 returns "0/" instead of "99" */
641 if (outbuf[1] == L'y' && buf.tm_year < 0) {
642 PyErr_SetString(PyExc_ValueError,
643 "format %y requires year >= 1900 on AIX");
Victor Stinner55329f82013-11-17 23:39:21 +0100644 return NULL;
645 }
646 }
Amaury Forgeot d'Arcb5be6d42009-03-02 23:52:57 +0000647#endif
648
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000649 fmtlen = time_strlen(fmt);
Guido van Rossumc222ec21999-02-23 00:00:10 +0000650
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000651 /* I hate these functions that presume you know how big the output
652 * will be ahead of time...
653 */
654 for (i = 1024; ; i += i) {
Victor Stinner136ea492011-12-17 22:37:18 +0100655#if defined _MSC_VER && _MSC_VER >= 1400 && defined(__STDC_SECURE_LIB__)
Antoine Pitrouc345ce12011-12-16 12:28:32 +0100656 int err;
Victor Stinner136ea492011-12-17 22:37:18 +0100657#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000658 outbuf = (time_char *)PyMem_Malloc(i*sizeof(time_char));
659 if (outbuf == NULL) {
Victor Stinnerb2904782010-09-29 10:34:19 +0000660 PyErr_NoMemory();
661 break;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000662 }
663 buflen = format_time(outbuf, i, fmt, &buf);
Victor Stinner136ea492011-12-17 22:37:18 +0100664#if defined _MSC_VER && _MSC_VER >= 1400 && defined(__STDC_SECURE_LIB__)
Antoine Pitrouc345ce12011-12-16 12:28:32 +0100665 err = errno;
Victor Stinner136ea492011-12-17 22:37:18 +0100666#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000667 if (buflen > 0 || i >= 256 * fmtlen) {
668 /* If the buffer is 256 times as long as the format,
669 it's probably not failing for lack of room!
670 More likely, the format yields an empty result,
671 e.g. an empty format, or %Z when the timezone
672 is unknown. */
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000673#ifdef HAVE_WCSFTIME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000674 ret = PyUnicode_FromWideChar(outbuf, buflen);
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000675#else
Victor Stinner1b579672011-12-17 05:47:23 +0100676 ret = PyUnicode_DecodeLocaleAndSize(outbuf, buflen,
677 "surrogateescape");
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000678#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000679 PyMem_Free(outbuf);
Victor Stinnerb2904782010-09-29 10:34:19 +0000680 break;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000681 }
682 PyMem_Free(outbuf);
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000683#if defined _MSC_VER && _MSC_VER >= 1400 && defined(__STDC_SECURE_LIB__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000684 /* VisualStudio .NET 2005 does this properly */
Antoine Pitrouc345ce12011-12-16 12:28:32 +0100685 if (buflen == 0 && err == EINVAL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000686 PyErr_SetString(PyExc_ValueError, "Invalid format string");
Victor Stinnerb2904782010-09-29 10:34:19 +0000687 break;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000688 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000689#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000690 }
Victor Stinnerb2904782010-09-29 10:34:19 +0000691#ifdef HAVE_WCSFTIME
692 PyMem_Free(format);
693#else
694 Py_DECREF(format);
695#endif
696 return ret;
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000697}
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000698
Martin v. Löwis1b01ccd2009-05-30 06:13:40 +0000699#undef time_char
700#undef format_time
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000701PyDoc_STRVAR(strftime_doc,
Thomas Woutersfe385252001-01-19 23:16:56 +0000702"strftime(format[, tuple]) -> string\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000703\n\
704Convert a time tuple to a string according to a format specification.\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000705See the library reference manual for formatting codes. When the time tuple\n\
Alexander Belopolskycf774542012-10-02 18:39:16 -0400706is not present, current time as returned by localtime() is used.\n\
707\n" STRFTIME_FORMAT_CODES);
Guido van Rossum8d8c1ee1995-09-13 17:38:35 +0000708#endif /* HAVE_STRFTIME */
709
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000710static PyObject *
711time_strptime(PyObject *self, PyObject *args)
712{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000713 PyObject *strptime_module = PyImport_ImportModuleNoBlock("_strptime");
714 PyObject *strptime_result;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200715 _Py_IDENTIFIER(_strptime_time);
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000716
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000717 if (!strptime_module)
718 return NULL;
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200719 strptime_result = _PyObject_CallMethodId(strptime_module,
720 &PyId__strptime_time, "O", args);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000721 Py_DECREF(strptime_module);
722 return strptime_result;
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000723}
724
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000725
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000726PyDoc_STRVAR(strptime_doc,
Brett Cannon20def8b2003-07-01 05:16:08 +0000727"strptime(string, format) -> struct_time\n\
Martin v. Löwisb3cfc1d2001-12-02 12:27:43 +0000728\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000729Parse a string to a time tuple according to a format specification.\n\
Alexander Belopolskycf774542012-10-02 18:39:16 -0400730See the library reference manual for formatting codes (same as\n\
731strftime()).\n\
732\n" STRFTIME_FORMAT_CODES);
Guido van Rossumd3c46d52002-07-19 17:06:47 +0000733
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000734static PyObject *
735_asctime(struct tm *timeptr)
736{
737 /* Inspired by Open Group reference implementation available at
738 * http://pubs.opengroup.org/onlinepubs/009695399/functions/asctime.html */
Victor Stinner499dfcf2011-03-21 13:26:24 +0100739 static char wday_name[7][4] = {
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000740 "Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"
741 };
Victor Stinner499dfcf2011-03-21 13:26:24 +0100742 static char mon_name[12][4] = {
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000743 "Jan", "Feb", "Mar", "Apr", "May", "Jun",
744 "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
745 };
Victor Stinner499dfcf2011-03-21 13:26:24 +0100746 return PyUnicode_FromFormat(
747 "%s %s%3d %.2d:%.2d:%.2d %d",
748 wday_name[timeptr->tm_wday],
749 mon_name[timeptr->tm_mon],
750 timeptr->tm_mday, timeptr->tm_hour,
751 timeptr->tm_min, timeptr->tm_sec,
752 1900 + timeptr->tm_year);
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000753}
Guido van Rossum87ce7bb1998-06-09 16:30:31 +0000754
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000755static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000756time_asctime(PyObject *self, PyObject *args)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000757{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000758 PyObject *tup = NULL;
759 struct tm buf;
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000760
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000761 if (!PyArg_UnpackTuple(args, "asctime", 0, 1, &tup))
762 return NULL;
763 if (tup == NULL) {
764 time_t tt = time(NULL);
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100765 if (pylocaltime(&tt, &buf) == -1)
766 return NULL;
767
Alexander Belopolsky38e29962010-10-01 14:18:49 +0000768 } else if (!gettmarg(tup, &buf) || !checktm(&buf))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000769 return NULL;
Alexander Belopolskyb9588b52011-01-04 16:34:30 +0000770 return _asctime(&buf);
Guido van Rossum9e90a671993-06-24 11:10:19 +0000771}
772
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000773PyDoc_STRVAR(asctime_doc,
Thomas Woutersfe385252001-01-19 23:16:56 +0000774"asctime([tuple]) -> string\n\
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000775\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000776Convert a time tuple to a string, e.g. 'Sat Jun 06 16:26:11 1998'.\n\
777When the time tuple is not present, current time as returned by localtime()\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000778is used.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000779
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000780static PyObject *
Peter Schneider-Kamp416d4132000-07-10 12:15:54 +0000781time_ctime(PyObject *self, PyObject *args)
Guido van Rossum9e90a671993-06-24 11:10:19 +0000782{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000783 time_t tt;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100784 struct tm buf;
785 if (!parse_time_t_args(args, "|O:ctime", &tt))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000786 return NULL;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100787 if (pylocaltime(&tt, &buf) == -1)
Alexander Belopolsky5da468f2011-01-04 17:15:52 +0000788 return NULL;
Victor Stinnerc1b5d342012-01-27 00:08:48 +0100789 return _asctime(&buf);
Guido van Rossum9e90a671993-06-24 11:10:19 +0000790}
791
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000792PyDoc_STRVAR(ctime_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000793"ctime(seconds) -> string\n\
794\n\
795Convert a time in seconds since the Epoch to a string in local time.\n\
Thomas Woutersfe385252001-01-19 23:16:56 +0000796This is equivalent to asctime(localtime(seconds)). When the time tuple is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000797not present, current time as returned by localtime() is used.");
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000798
Guido van Rossum60cd8131998-03-06 17:16:21 +0000799#ifdef HAVE_MKTIME
Barry Warsaw9a2a8a81996-12-06 23:32:14 +0000800static PyObject *
Victor Stinner4195b5c2012-02-08 23:03:19 +0100801time_mktime(PyObject *self, PyObject *tup)
Guido van Rossum234f9421993-06-17 12:35:49 +0000802{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000803 struct tm buf;
804 time_t tt;
805 if (!gettmarg(tup, &buf))
806 return NULL;
Victor Stinner1ac42612014-02-21 09:27:17 +0100807#ifdef _AIX
808 /* year < 1902 or year > 2037 */
809 if (buf.tm_year < 2 || buf.tm_year > 137) {
810 /* Issue #19748: On AIX, mktime() doesn't report overflow error for
811 * timestamp < -2^31 or timestamp > 2**31-1. */
812 PyErr_SetString(PyExc_OverflowError,
813 "mktime argument out of range");
814 return NULL;
815 }
816#else
Alexander Belopolskyb7d40d12011-01-11 01:21:25 +0000817 buf.tm_wday = -1; /* sentinel; original value ignored */
Victor Stinner1ac42612014-02-21 09:27:17 +0100818#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000819 tt = mktime(&buf);
Alexander Belopolskyb7d40d12011-01-11 01:21:25 +0000820 /* Return value of -1 does not necessarily mean an error, but tm_wday
Ezio Melotti13925002011-03-16 11:05:33 +0200821 * cannot remain set to -1 if mktime succeeded. */
Victor Stinner93037492013-06-25 22:54:35 +0200822 if (tt == (time_t)(-1)
823#ifndef _AIX
824 /* Return value of -1 does not necessarily mean an error, but
825 * tm_wday cannot remain set to -1 if mktime succeeded. */
826 && buf.tm_wday == -1
827#else
828 /* on AIX, tm_wday is always sets, even on error */
829#endif
830 )
831 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000832 PyErr_SetString(PyExc_OverflowError,
833 "mktime argument out of range");
834 return NULL;
835 }
Victor Stinner4195b5c2012-02-08 23:03:19 +0100836 return PyFloat_FromDouble((double)tt);
Guido van Rossum234f9421993-06-17 12:35:49 +0000837}
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000838
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +0000839PyDoc_STRVAR(mktime_doc,
Guido van Rossum0ef577b1998-06-27 20:38:36 +0000840"mktime(tuple) -> floating point number\n\
841\n\
Alexander Belopolskyc142bba2012-06-13 22:15:26 -0400842Convert a time tuple in local time to seconds since the Epoch.\n\
843Note that mktime(gmtime(0)) will not generally return zero for most\n\
844time zones; instead the returned value will either be equal to that\n\
845of the timezone or altzone attributes on the time module.");
Guido van Rossum60cd8131998-03-06 17:16:21 +0000846#endif /* HAVE_MKTIME */
Guido van Rossum234f9421993-06-17 12:35:49 +0000847
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000848#ifdef HAVE_WORKING_TZSET
Martin v. Löwis1a214512008-06-11 05:26:20 +0000849static void PyInit_timezone(PyObject *module);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000850
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000851static PyObject *
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000852time_tzset(PyObject *self, PyObject *unused)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000853{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000854 PyObject* m;
Fred Drake9bb74322002-04-01 14:49:59 +0000855
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000856 m = PyImport_ImportModuleNoBlock("time");
857 if (m == NULL) {
858 return NULL;
859 }
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000860
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000861 tzset();
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000862
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000863 /* Reset timezone, altzone, daylight and tzname */
864 PyInit_timezone(m);
865 Py_DECREF(m);
Victor Stinner2ff51b82013-07-17 21:42:45 +0200866 if (PyErr_Occurred())
867 return NULL;
Tim Peters1b6f7a92004-06-20 02:50:16 +0000868
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000869 Py_INCREF(Py_None);
870 return Py_None;
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000871}
872
873PyDoc_STRVAR(tzset_doc,
R. David Murray4d55bf92010-12-14 00:55:46 +0000874"tzset()\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000875\n\
876Initialize, or reinitialize, the local timezone to the value stored in\n\
877os.environ['TZ']. The TZ environment variable should be specified in\n\
Neal Norwitzdc8e1942004-07-20 22:34:37 +0000878standard Unix timezone format as documented in the tzset man page\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +0000879(eg. 'US/Eastern', 'Europe/Amsterdam'). Unknown timezones will silently\n\
880fall back to UTC. If the TZ environment variable is not set, the local\n\
881timezone is set to the systems best guess of wallclock time.\n\
882Changing the TZ environment variable without calling tzset *may* change\n\
883the local timezone used by methods such as localtime, but this behaviour\n\
884should not be relied on.");
885#endif /* HAVE_WORKING_TZSET */
886
Victor Stinnerae586492014-09-02 23:18:25 +0200887static PyObject *
Victor Stinnerec895392012-04-29 02:41:27 +0200888pymonotonic(_Py_clock_info_t *info)
Victor Stinnerb94b2662012-01-18 01:50:21 +0100889{
Victor Stinnerae586492014-09-02 23:18:25 +0200890 _PyTime_timeval tv;
891 if (_PyTime_monotonic_info(&tv, info) < 0) {
892 assert(info != NULL);
Victor Stinner071eca32012-03-15 01:17:09 +0100893 return NULL;
894 }
Victor Stinnerae586492014-09-02 23:18:25 +0200895 return PyFloat_FromDouble((double)tv.tv_sec + tv.tv_usec * 1e-6);
Victor Stinner8b302012012-02-07 23:29:46 +0100896}
897
Victor Stinner071eca32012-03-15 01:17:09 +0100898static PyObject *
Victor Stinnerec895392012-04-29 02:41:27 +0200899time_monotonic(PyObject *self, PyObject *unused)
Victor Stinner071eca32012-03-15 01:17:09 +0100900{
Victor Stinnerec895392012-04-29 02:41:27 +0200901 return pymonotonic(NULL);
Victor Stinner071eca32012-03-15 01:17:09 +0100902}
903
Victor Stinnerec895392012-04-29 02:41:27 +0200904PyDoc_STRVAR(monotonic_doc,
905"monotonic() -> float\n\
Victor Stinner8b302012012-02-07 23:29:46 +0100906\n\
Victor Stinnerec895392012-04-29 02:41:27 +0200907Monotonic clock, cannot go backward.");
Victor Stinnerec919cc2012-03-15 00:58:32 +0100908
Victor Stinnerec895392012-04-29 02:41:27 +0200909static PyObject*
910perf_counter(_Py_clock_info_t *info)
911{
Victor Stinner54884492014-08-29 16:51:33 +0200912#ifdef WIN32_PERF_COUNTER
913 return win_perf_counter(info);
914#else
Victor Stinnerae586492014-09-02 23:18:25 +0200915 return pymonotonic(info);
Victor Stinner54884492014-08-29 16:51:33 +0200916#endif
Victor Stinnerec895392012-04-29 02:41:27 +0200917}
918
919static PyObject *
920time_perf_counter(PyObject *self, PyObject *unused)
921{
922 return perf_counter(NULL);
923}
924
925PyDoc_STRVAR(perf_counter_doc,
926"perf_counter() -> float\n\
927\n\
928Performance counter for benchmarking.");
929
930static PyObject*
931py_process_time(_Py_clock_info_t *info)
932{
933#if defined(MS_WINDOWS)
934 HANDLE process;
935 FILETIME creation_time, exit_time, kernel_time, user_time;
936 ULARGE_INTEGER large;
937 double total;
938 BOOL ok;
939
940 process = GetCurrentProcess();
941 ok = GetProcessTimes(process, &creation_time, &exit_time, &kernel_time, &user_time);
942 if (!ok)
943 return PyErr_SetFromWindowsErr(0);
944
945 large.u.LowPart = kernel_time.dwLowDateTime;
946 large.u.HighPart = kernel_time.dwHighDateTime;
947 total = (double)large.QuadPart;
948 large.u.LowPart = user_time.dwLowDateTime;
949 large.u.HighPart = user_time.dwHighDateTime;
950 total += (double)large.QuadPart;
951 if (info) {
952 info->implementation = "GetProcessTimes()";
953 info->resolution = 1e-7;
Benjamin Peterson49a69e42012-05-01 09:38:34 -0400954 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +0200955 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +0200956 }
957 return PyFloat_FromDouble(total * 1e-7);
958#else
959
960#if defined(HAVE_SYS_RESOURCE_H)
961 struct rusage ru;
962#endif
963#ifdef HAVE_TIMES
964 struct tms t;
965 static long ticks_per_second = -1;
966#endif
967
968#if defined(HAVE_CLOCK_GETTIME) \
969 && (defined(CLOCK_PROCESS_CPUTIME_ID) || defined(CLOCK_PROF))
970 struct timespec tp;
971#ifdef CLOCK_PROF
972 const clockid_t clk_id = CLOCK_PROF;
973 const char *function = "clock_gettime(CLOCK_PROF)";
974#else
975 const clockid_t clk_id = CLOCK_PROCESS_CPUTIME_ID;
976 const char *function = "clock_gettime(CLOCK_PROCESS_CPUTIME_ID)";
977#endif
978
979 if (clock_gettime(clk_id, &tp) == 0) {
980 if (info) {
981 struct timespec res;
982 info->implementation = function;
Benjamin Peterson49a69e42012-05-01 09:38:34 -0400983 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +0200984 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +0200985 if (clock_getres(clk_id, &res) == 0)
986 info->resolution = res.tv_sec + res.tv_nsec * 1e-9;
987 else
988 info->resolution = 1e-9;
989 }
990 return PyFloat_FromDouble(tp.tv_sec + tp.tv_nsec * 1e-9);
991 }
992#endif
993
994#if defined(HAVE_SYS_RESOURCE_H)
995 if (getrusage(RUSAGE_SELF, &ru) == 0) {
996 double total;
997 total = ru.ru_utime.tv_sec + ru.ru_utime.tv_usec * 1e-6;
998 total += ru.ru_stime.tv_sec + ru.ru_stime.tv_usec * 1e-6;
999 if (info) {
1000 info->implementation = "getrusage(RUSAGE_SELF)";
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001001 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001002 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001003 info->resolution = 1e-6;
1004 }
1005 return PyFloat_FromDouble(total);
1006 }
1007#endif
1008
1009#ifdef HAVE_TIMES
1010 if (times(&t) != (clock_t)-1) {
1011 double total;
1012
1013 if (ticks_per_second == -1) {
1014#if defined(HAVE_SYSCONF) && defined(_SC_CLK_TCK)
1015 ticks_per_second = sysconf(_SC_CLK_TCK);
1016 if (ticks_per_second < 1)
1017 ticks_per_second = -1;
1018#elif defined(HZ)
1019 ticks_per_second = HZ;
1020#else
1021 ticks_per_second = 60; /* magic fallback value; may be bogus */
1022#endif
1023 }
1024
1025 if (ticks_per_second != -1) {
1026 total = (double)t.tms_utime / ticks_per_second;
1027 total += (double)t.tms_stime / ticks_per_second;
1028 if (info) {
1029 info->implementation = "times()";
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001030 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001031 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001032 info->resolution = 1.0 / ticks_per_second;
1033 }
1034 return PyFloat_FromDouble(total);
1035 }
1036 }
1037#endif
1038
1039 return floatclock(info);
1040#endif
1041}
1042
1043static PyObject *
1044time_process_time(PyObject *self, PyObject *unused)
1045{
1046 return py_process_time(NULL);
1047}
1048
1049PyDoc_STRVAR(process_time_doc,
1050"process_time() -> float\n\
1051\n\
1052Process time for profiling: sum of the kernel and user-space CPU time.");
1053
1054
Victor Stinnerec895392012-04-29 02:41:27 +02001055static PyObject *
1056time_get_clock_info(PyObject *self, PyObject *args)
1057{
1058 char *name;
Victor Stinnerec895392012-04-29 02:41:27 +02001059 _Py_clock_info_t info;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001060 PyObject *obj = NULL, *dict, *ns;
Victor Stinnerec895392012-04-29 02:41:27 +02001061
1062 if (!PyArg_ParseTuple(args, "s:get_clock_info", &name))
1063 return NULL;
1064
1065#ifdef Py_DEBUG
1066 info.implementation = NULL;
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001067 info.monotonic = -1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001068 info.adjustable = -1;
Victor Stinnerec895392012-04-29 02:41:27 +02001069 info.resolution = -1.0;
1070#else
1071 info.implementation = "";
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001072 info.monotonic = 0;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001073 info.adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001074 info.resolution = 1.0;
1075#endif
1076
1077 if (strcmp(name, "time") == 0)
1078 obj = floattime(&info);
1079#ifdef PYCLOCK
1080 else if (strcmp(name, "clock") == 0)
1081 obj = pyclock(&info);
1082#endif
Victor Stinnerec895392012-04-29 02:41:27 +02001083 else if (strcmp(name, "monotonic") == 0)
1084 obj = pymonotonic(&info);
Victor Stinnerec895392012-04-29 02:41:27 +02001085 else if (strcmp(name, "perf_counter") == 0)
1086 obj = perf_counter(&info);
1087 else if (strcmp(name, "process_time") == 0)
1088 obj = py_process_time(&info);
1089 else {
1090 PyErr_SetString(PyExc_ValueError, "unknown clock");
1091 return NULL;
1092 }
1093 if (obj == NULL)
1094 return NULL;
1095 Py_DECREF(obj);
1096
Victor Stinnerbda4b882012-06-12 22:11:44 +02001097 dict = PyDict_New();
1098 if (dict == NULL)
Victor Stinnerec895392012-04-29 02:41:27 +02001099 return NULL;
1100
1101 assert(info.implementation != NULL);
1102 obj = PyUnicode_FromString(info.implementation);
1103 if (obj == NULL)
1104 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001105 if (PyDict_SetItemString(dict, "implementation", obj) == -1)
1106 goto error;
1107 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001108
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001109 assert(info.monotonic != -1);
1110 obj = PyBool_FromLong(info.monotonic);
Victor Stinnerec895392012-04-29 02:41:27 +02001111 if (obj == NULL)
1112 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001113 if (PyDict_SetItemString(dict, "monotonic", obj) == -1)
1114 goto error;
1115 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001116
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001117 assert(info.adjustable != -1);
1118 obj = PyBool_FromLong(info.adjustable);
Victor Stinnerec895392012-04-29 02:41:27 +02001119 if (obj == NULL)
1120 goto error;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001121 if (PyDict_SetItemString(dict, "adjustable", obj) == -1)
Victor Stinnerbda4b882012-06-12 22:11:44 +02001122 goto error;
1123 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001124
1125 assert(info.resolution > 0.0);
1126 assert(info.resolution <= 1.0);
1127 obj = PyFloat_FromDouble(info.resolution);
1128 if (obj == NULL)
1129 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001130 if (PyDict_SetItemString(dict, "resolution", obj) == -1)
1131 goto error;
1132 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001133
Victor Stinnerbda4b882012-06-12 22:11:44 +02001134 ns = _PyNamespace_New(dict);
1135 Py_DECREF(dict);
1136 return ns;
Victor Stinnerec895392012-04-29 02:41:27 +02001137
1138error:
Victor Stinnerbda4b882012-06-12 22:11:44 +02001139 Py_DECREF(dict);
1140 Py_XDECREF(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001141 return NULL;
1142}
1143
1144PyDoc_STRVAR(get_clock_info_doc,
1145"get_clock_info(name: str) -> dict\n\
1146\n\
1147Get information of the specified clock.");
Victor Stinner8b302012012-02-07 23:29:46 +01001148
Martin v. Löwisd218dc12008-04-07 03:17:54 +00001149static void
Martin v. Löwis1a214512008-06-11 05:26:20 +00001150PyInit_timezone(PyObject *m) {
1151 /* This code moved from PyInit_time wholesale to allow calling it from
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001152 time_tzset. In the future, some parts of it can be moved back
1153 (for platforms that don't HAVE_WORKING_TZSET, when we know what they
1154 are), and the extraneous calls to tzset(3) should be removed.
1155 I haven't done this yet, as I don't want to change this code as
1156 little as possible when introducing the time.tzset and time.tzsetwall
1157 methods. This should simply be a method of doing the following once,
1158 at the top of this function and removing the call to tzset() from
1159 time_tzset():
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001160
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001161 #ifdef HAVE_TZSET
1162 tzset()
1163 #endif
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001164
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001165 And I'm lazy and hate C so nyer.
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001166 */
Guido van Rossum10b164a2001-09-25 13:59:01 +00001167#if defined(HAVE_TZNAME) && !defined(__GLIBC__) && !defined(__CYGWIN__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001168 PyObject *otz0, *otz1;
1169 tzset();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001170 PyModule_AddIntConstant(m, "timezone", timezone);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001171#ifdef HAVE_ALTZONE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001172 PyModule_AddIntConstant(m, "altzone", altzone);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001173#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001174 PyModule_AddIntConstant(m, "altzone", timezone-3600);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001175#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001176 PyModule_AddIntConstant(m, "daylight", daylight);
Victor Stinner1b579672011-12-17 05:47:23 +01001177 otz0 = PyUnicode_DecodeLocale(tzname[0], "surrogateescape");
1178 otz1 = PyUnicode_DecodeLocale(tzname[1], "surrogateescape");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001179 PyModule_AddObject(m, "tzname", Py_BuildValue("(NN)", otz0, otz1));
Guido van Rossum10b164a2001-09-25 13:59:01 +00001180#else /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001181#ifdef HAVE_STRUCT_TM_TM_ZONE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001182 {
Guido van Rossum234f9421993-06-17 12:35:49 +00001183#define YEAR ((time_t)((365 * 24 + 6) * 3600))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001184 time_t t;
1185 struct tm *p;
1186 long janzone, julyzone;
1187 char janname[10], julyname[10];
1188 t = (time((time_t *)0) / YEAR) * YEAR;
1189 p = localtime(&t);
1190 janzone = -p->tm_gmtoff;
1191 strncpy(janname, p->tm_zone ? p->tm_zone : " ", 9);
1192 janname[9] = '\0';
1193 t += YEAR/2;
1194 p = localtime(&t);
1195 julyzone = -p->tm_gmtoff;
1196 strncpy(julyname, p->tm_zone ? p->tm_zone : " ", 9);
1197 julyname[9] = '\0';
Guido van Rossum10b164a2001-09-25 13:59:01 +00001198
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001199 if( janzone < julyzone ) {
1200 /* DST is reversed in the southern hemisphere */
1201 PyModule_AddIntConstant(m, "timezone", julyzone);
1202 PyModule_AddIntConstant(m, "altzone", janzone);
1203 PyModule_AddIntConstant(m, "daylight",
1204 janzone != julyzone);
1205 PyModule_AddObject(m, "tzname",
1206 Py_BuildValue("(zz)",
1207 julyname, janname));
1208 } else {
1209 PyModule_AddIntConstant(m, "timezone", janzone);
1210 PyModule_AddIntConstant(m, "altzone", julyzone);
1211 PyModule_AddIntConstant(m, "daylight",
1212 janzone != julyzone);
1213 PyModule_AddObject(m, "tzname",
1214 Py_BuildValue("(zz)",
1215 janname, julyname));
1216 }
1217 }
Guido van Rossume6a4b7b1997-10-08 15:27:56 +00001218#else
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001219#endif /* HAVE_STRUCT_TM_TM_ZONE */
Tim Peters26ae7cd2001-03-20 03:26:49 +00001220#ifdef __CYGWIN__
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001221 tzset();
1222 PyModule_AddIntConstant(m, "timezone", _timezone);
1223 PyModule_AddIntConstant(m, "altzone", _timezone-3600);
1224 PyModule_AddIntConstant(m, "daylight", _daylight);
1225 PyModule_AddObject(m, "tzname",
1226 Py_BuildValue("(zz)", _tzname[0], _tzname[1]));
Tim Peters26ae7cd2001-03-20 03:26:49 +00001227#endif /* __CYGWIN__ */
Guido van Rossum10b164a2001-09-25 13:59:01 +00001228#endif /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
Victor Stinnere0be4232011-10-25 13:06:09 +02001229
Victor Stinnerec895392012-04-29 02:41:27 +02001230#if defined(HAVE_CLOCK_GETTIME)
Victor Stinnere0be4232011-10-25 13:06:09 +02001231 PyModule_AddIntMacro(m, CLOCK_REALTIME);
Victor Stinnere0be4232011-10-25 13:06:09 +02001232#ifdef CLOCK_MONOTONIC
1233 PyModule_AddIntMacro(m, CLOCK_MONOTONIC);
1234#endif
1235#ifdef CLOCK_MONOTONIC_RAW
1236 PyModule_AddIntMacro(m, CLOCK_MONOTONIC_RAW);
1237#endif
Victor Stinner1470f352012-04-03 00:31:17 +02001238#ifdef CLOCK_HIGHRES
1239 PyModule_AddIntMacro(m, CLOCK_HIGHRES);
1240#endif
Victor Stinnere0be4232011-10-25 13:06:09 +02001241#ifdef CLOCK_PROCESS_CPUTIME_ID
1242 PyModule_AddIntMacro(m, CLOCK_PROCESS_CPUTIME_ID);
1243#endif
1244#ifdef CLOCK_THREAD_CPUTIME_ID
1245 PyModule_AddIntMacro(m, CLOCK_THREAD_CPUTIME_ID);
1246#endif
1247#endif /* HAVE_CLOCK_GETTIME */
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001248}
1249
1250
1251static PyMethodDef time_methods[] = {
Victor Stinner4195b5c2012-02-08 23:03:19 +01001252 {"time", time_time, METH_NOARGS, time_doc},
Victor Stinnerec895392012-04-29 02:41:27 +02001253#ifdef PYCLOCK
Victor Stinner4195b5c2012-02-08 23:03:19 +01001254 {"clock", time_clock, METH_NOARGS, clock_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001255#endif
Victor Stinnere0be4232011-10-25 13:06:09 +02001256#ifdef HAVE_CLOCK_GETTIME
Victor Stinner4195b5c2012-02-08 23:03:19 +01001257 {"clock_gettime", time_clock_gettime, METH_VARARGS, clock_gettime_doc},
Victor Stinner30d79472012-04-03 00:45:07 +02001258 {"clock_settime", time_clock_settime, METH_VARARGS, clock_settime_doc},
Victor Stinner4195b5c2012-02-08 23:03:19 +01001259 {"clock_getres", time_clock_getres, METH_VARARGS, clock_getres_doc},
Victor Stinnere0be4232011-10-25 13:06:09 +02001260#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001261 {"sleep", time_sleep, METH_VARARGS, sleep_doc},
1262 {"gmtime", time_gmtime, METH_VARARGS, gmtime_doc},
1263 {"localtime", time_localtime, METH_VARARGS, localtime_doc},
1264 {"asctime", time_asctime, METH_VARARGS, asctime_doc},
1265 {"ctime", time_ctime, METH_VARARGS, ctime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001266#ifdef HAVE_MKTIME
Victor Stinner4195b5c2012-02-08 23:03:19 +01001267 {"mktime", time_mktime, METH_O, mktime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001268#endif
1269#ifdef HAVE_STRFTIME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001270 {"strftime", time_strftime, METH_VARARGS, strftime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001271#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001272 {"strptime", time_strptime, METH_VARARGS, strptime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001273#ifdef HAVE_WORKING_TZSET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001274 {"tzset", time_tzset, METH_NOARGS, tzset_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001275#endif
Victor Stinnerec895392012-04-29 02:41:27 +02001276 {"monotonic", time_monotonic, METH_NOARGS, monotonic_doc},
Victor Stinnerec895392012-04-29 02:41:27 +02001277 {"process_time", time_process_time, METH_NOARGS, process_time_doc},
1278 {"perf_counter", time_perf_counter, METH_NOARGS, perf_counter_doc},
1279 {"get_clock_info", time_get_clock_info, METH_VARARGS, get_clock_info_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001280 {NULL, NULL} /* sentinel */
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001281};
1282
1283
1284PyDoc_STRVAR(module_doc,
1285"This module provides various functions to manipulate time values.\n\
1286\n\
1287There are two standard representations of time. One is the number\n\
1288of seconds since the Epoch, in UTC (a.k.a. GMT). It may be an integer\n\
1289or a floating point number (to represent fractions of seconds).\n\
1290The Epoch is system-defined; on Unix, it is generally January 1st, 1970.\n\
1291The actual value can be retrieved by calling gmtime(0).\n\
1292\n\
1293The other representation is a tuple of 9 integers giving local time.\n\
1294The tuple items are:\n\
Alexander Belopolsky03163ac2011-05-02 12:20:52 -04001295 year (including century, e.g. 1998)\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001296 month (1-12)\n\
1297 day (1-31)\n\
1298 hours (0-23)\n\
1299 minutes (0-59)\n\
1300 seconds (0-59)\n\
1301 weekday (0-6, Monday is 0)\n\
1302 Julian day (day in the year, 1-366)\n\
1303 DST (Daylight Savings Time) flag (-1, 0 or 1)\n\
1304If the DST flag is 0, the time is given in the regular time zone;\n\
1305if it is 1, the time is given in the DST time zone;\n\
1306if it is -1, mktime() should guess based on the date and time.\n\
1307\n\
1308Variables:\n\
1309\n\
1310timezone -- difference in seconds between UTC and local standard time\n\
1311altzone -- difference in seconds between UTC and local DST time\n\
1312daylight -- whether local time should reflect DST\n\
1313tzname -- tuple of (standard time zone name, DST time zone name)\n\
1314\n\
1315Functions:\n\
1316\n\
1317time() -- return current time in seconds since the Epoch as a float\n\
1318clock() -- return CPU time since process start as a float\n\
1319sleep() -- delay for a number of seconds given as a float\n\
1320gmtime() -- convert seconds since Epoch to UTC tuple\n\
1321localtime() -- convert seconds since Epoch to local time tuple\n\
1322asctime() -- convert time tuple to string\n\
1323ctime() -- convert time in seconds to string\n\
1324mktime() -- convert local time tuple to seconds since Epoch\n\
1325strftime() -- convert time tuple to string according to format specification\n\
1326strptime() -- parse string to time tuple according to format specification\n\
1327tzset() -- change the local timezone");
1328
1329
Martin v. Löwis1a214512008-06-11 05:26:20 +00001330
1331static struct PyModuleDef timemodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001332 PyModuleDef_HEAD_INIT,
1333 "time",
1334 module_doc,
1335 -1,
1336 time_methods,
1337 NULL,
1338 NULL,
1339 NULL,
1340 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00001341};
1342
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001343PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00001344PyInit_time(void)
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001345{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001346 PyObject *m;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001347 m = PyModule_Create(&timemodule);
1348 if (m == NULL)
1349 return NULL;
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001350
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001351 /* Set, or reset, module variables like time.timezone */
1352 PyInit_timezone(m);
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001353
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001354 if (!initialized) {
Victor Stinner1c8f0592013-07-22 22:24:54 +02001355 if (PyStructSequence_InitType2(&StructTimeType,
1356 &struct_time_type_desc) < 0)
1357 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001358 }
1359 Py_INCREF(&StructTimeType);
Alexander Belopolskyc142bba2012-06-13 22:15:26 -04001360#ifdef HAVE_STRUCT_TM_TM_ZONE
1361 PyModule_AddIntConstant(m, "_STRUCT_TM_ITEMS", 11);
1362#else
1363 PyModule_AddIntConstant(m, "_STRUCT_TM_ITEMS", 9);
1364#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001365 PyModule_AddObject(m, "struct_time", (PyObject*) &StructTimeType);
1366 initialized = 1;
1367 return m;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001368}
1369
Victor Stinner071eca32012-03-15 01:17:09 +01001370static PyObject*
Victor Stinnerec895392012-04-29 02:41:27 +02001371floattime(_Py_clock_info_t *info)
Victor Stinner4195b5c2012-02-08 23:03:19 +01001372{
1373 _PyTime_timeval t;
Victor Stinner00111242014-08-29 16:31:59 +02001374 if (_PyTime_gettimeofday_info(&t, info) < 0) {
1375 assert(info != NULL);
1376 return NULL;
1377 }
Victor Stinner70b2e1e2012-03-26 22:08:02 +02001378 return PyFloat_FromDouble((double)t.tv_sec + t.tv_usec * 1e-6);
Victor Stinner4195b5c2012-02-08 23:03:19 +01001379}
1380
1381
Guido van Rossumb6775db1994-08-01 11:34:53 +00001382/* Implement floatsleep() for various platforms.
1383 When interrupted (or when another error occurs), return -1 and
1384 set an exception; else return 0. */
1385
1386static int
Guido van Rossuma320fd31995-03-09 12:14:15 +00001387floatsleep(double secs)
Guido van Rossum426035c1991-02-19 12:27:35 +00001388{
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001389/* XXX Should test for MS_WINDOWS first! */
Skip Montanaroeb33e5a2007-08-17 12:57:41 +00001390#if defined(HAVE_SELECT) && !defined(__EMX__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001391 struct timeval t;
1392 double frac;
Victor Stinner48b1ce52011-07-01 13:50:09 +02001393 int err;
1394
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001395 frac = fmod(secs, 1.0);
1396 secs = floor(secs);
1397 t.tv_sec = (long)secs;
1398 t.tv_usec = (long)(frac*1000000.0);
1399 Py_BEGIN_ALLOW_THREADS
Victor Stinner48b1ce52011-07-01 13:50:09 +02001400 err = select(0, (fd_set *)0, (fd_set *)0, (fd_set *)0, &t);
1401 Py_END_ALLOW_THREADS
1402 if (err != 0) {
Guido van Rossum09cbb011999-11-08 15:32:27 +00001403#ifdef EINTR
Victor Stinner48b1ce52011-07-01 13:50:09 +02001404 if (errno == EINTR) {
1405 if (PyErr_CheckSignals())
1406 return -1;
1407 }
1408 else
Guido van Rossum09cbb011999-11-08 15:32:27 +00001409#endif
Victor Stinner48b1ce52011-07-01 13:50:09 +02001410 {
Victor Stinnera734af32014-07-31 13:07:17 +02001411 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001412 return -1;
1413 }
1414 }
Martin v. Löwis02af9642002-01-16 11:04:06 +00001415#elif defined(__WATCOMC__) && !defined(__QNX__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001416 /* XXX Can't interrupt this sleep */
1417 Py_BEGIN_ALLOW_THREADS
1418 delay((int)(secs * 1000 + 0.5)); /* delay() uses milliseconds */
1419 Py_END_ALLOW_THREADS
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00001420#elif defined(MS_WINDOWS)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001421 {
1422 double millisecs = secs * 1000.0;
1423 unsigned long ul_millis;
Tim Peters513a1cd2003-01-19 04:54:58 +00001424
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001425 if (millisecs > (double)ULONG_MAX) {
1426 PyErr_SetString(PyExc_OverflowError,
1427 "sleep length is too large");
1428 return -1;
1429 }
1430 Py_BEGIN_ALLOW_THREADS
1431 /* Allow sleep(0) to maintain win32 semantics, and as decreed
1432 * by Guido, only the main thread can be interrupted.
1433 */
1434 ul_millis = (unsigned long)millisecs;
Antoine Pitrou6dd381e2011-11-21 21:26:56 +01001435 if (ul_millis == 0 || !_PyOS_IsMainThread())
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001436 Sleep(ul_millis);
1437 else {
1438 DWORD rc;
Antoine Pitrou6dd381e2011-11-21 21:26:56 +01001439 HANDLE hInterruptEvent = _PyOS_SigintEvent();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001440 ResetEvent(hInterruptEvent);
Martin v. Löwisb26a9b12013-01-25 14:25:48 +01001441 rc = WaitForSingleObjectEx(hInterruptEvent, ul_millis, FALSE);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001442 if (rc == WAIT_OBJECT_0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001443 Py_BLOCK_THREADS
1444 errno = EINTR;
Victor Stinnera734af32014-07-31 13:07:17 +02001445 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001446 return -1;
1447 }
1448 }
1449 Py_END_ALLOW_THREADS
1450 }
Martin v. Löwis02af9642002-01-16 11:04:06 +00001451#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001452 /* XXX Can't interrupt this sleep */
1453 Py_BEGIN_ALLOW_THREADS
1454 sleep((int)secs);
1455 Py_END_ALLOW_THREADS
Martin v. Löwis02af9642002-01-16 11:04:06 +00001456#endif
Guido van Rossum98bf58f2001-10-18 20:34:25 +00001457
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001458 return 0;
Guido van Rossum80c9d881991-04-16 08:47:51 +00001459}