blob: 5f6290d98ff3160f1bab6ddb3f62a0c7d4b67711 [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, */
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 Stinner4bfb4602015-03-27 22:27:24 +0100890 _PyTime_t t;
891 double d;
892 if (_PyTime_GetMonotonicClockWithInfo(&t, info) < 0) {
Victor Stinnerae586492014-09-02 23:18:25 +0200893 assert(info != NULL);
Victor Stinner071eca32012-03-15 01:17:09 +0100894 return NULL;
895 }
Victor Stinner4bfb4602015-03-27 22:27:24 +0100896 d = _PyTime_AsSecondsDouble(t);
897 return PyFloat_FromDouble(d);
Victor Stinner8b302012012-02-07 23:29:46 +0100898}
899
Victor Stinner071eca32012-03-15 01:17:09 +0100900static PyObject *
Victor Stinnerec895392012-04-29 02:41:27 +0200901time_monotonic(PyObject *self, PyObject *unused)
Victor Stinner071eca32012-03-15 01:17:09 +0100902{
Victor Stinnerec895392012-04-29 02:41:27 +0200903 return pymonotonic(NULL);
Victor Stinner071eca32012-03-15 01:17:09 +0100904}
905
Victor Stinnerec895392012-04-29 02:41:27 +0200906PyDoc_STRVAR(monotonic_doc,
907"monotonic() -> float\n\
Victor Stinner8b302012012-02-07 23:29:46 +0100908\n\
Victor Stinnerec895392012-04-29 02:41:27 +0200909Monotonic clock, cannot go backward.");
Victor Stinnerec919cc2012-03-15 00:58:32 +0100910
Victor Stinnerec895392012-04-29 02:41:27 +0200911static PyObject*
912perf_counter(_Py_clock_info_t *info)
913{
Victor Stinner54884492014-08-29 16:51:33 +0200914#ifdef WIN32_PERF_COUNTER
915 return win_perf_counter(info);
916#else
Victor Stinnerae586492014-09-02 23:18:25 +0200917 return pymonotonic(info);
Victor Stinner54884492014-08-29 16:51:33 +0200918#endif
Victor Stinnerec895392012-04-29 02:41:27 +0200919}
920
921static PyObject *
922time_perf_counter(PyObject *self, PyObject *unused)
923{
924 return perf_counter(NULL);
925}
926
927PyDoc_STRVAR(perf_counter_doc,
928"perf_counter() -> float\n\
929\n\
930Performance counter for benchmarking.");
931
932static PyObject*
933py_process_time(_Py_clock_info_t *info)
934{
935#if defined(MS_WINDOWS)
936 HANDLE process;
937 FILETIME creation_time, exit_time, kernel_time, user_time;
938 ULARGE_INTEGER large;
939 double total;
940 BOOL ok;
941
942 process = GetCurrentProcess();
943 ok = GetProcessTimes(process, &creation_time, &exit_time, &kernel_time, &user_time);
944 if (!ok)
945 return PyErr_SetFromWindowsErr(0);
946
947 large.u.LowPart = kernel_time.dwLowDateTime;
948 large.u.HighPart = kernel_time.dwHighDateTime;
949 total = (double)large.QuadPart;
950 large.u.LowPart = user_time.dwLowDateTime;
951 large.u.HighPart = user_time.dwHighDateTime;
952 total += (double)large.QuadPart;
953 if (info) {
954 info->implementation = "GetProcessTimes()";
955 info->resolution = 1e-7;
Benjamin Peterson49a69e42012-05-01 09:38:34 -0400956 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +0200957 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +0200958 }
959 return PyFloat_FromDouble(total * 1e-7);
960#else
961
962#if defined(HAVE_SYS_RESOURCE_H)
963 struct rusage ru;
964#endif
965#ifdef HAVE_TIMES
966 struct tms t;
967 static long ticks_per_second = -1;
968#endif
969
970#if defined(HAVE_CLOCK_GETTIME) \
971 && (defined(CLOCK_PROCESS_CPUTIME_ID) || defined(CLOCK_PROF))
972 struct timespec tp;
973#ifdef CLOCK_PROF
974 const clockid_t clk_id = CLOCK_PROF;
975 const char *function = "clock_gettime(CLOCK_PROF)";
976#else
977 const clockid_t clk_id = CLOCK_PROCESS_CPUTIME_ID;
978 const char *function = "clock_gettime(CLOCK_PROCESS_CPUTIME_ID)";
979#endif
980
981 if (clock_gettime(clk_id, &tp) == 0) {
982 if (info) {
983 struct timespec res;
984 info->implementation = function;
Benjamin Peterson49a69e42012-05-01 09:38:34 -0400985 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +0200986 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +0200987 if (clock_getres(clk_id, &res) == 0)
988 info->resolution = res.tv_sec + res.tv_nsec * 1e-9;
989 else
990 info->resolution = 1e-9;
991 }
992 return PyFloat_FromDouble(tp.tv_sec + tp.tv_nsec * 1e-9);
993 }
994#endif
995
996#if defined(HAVE_SYS_RESOURCE_H)
997 if (getrusage(RUSAGE_SELF, &ru) == 0) {
998 double total;
999 total = ru.ru_utime.tv_sec + ru.ru_utime.tv_usec * 1e-6;
1000 total += ru.ru_stime.tv_sec + ru.ru_stime.tv_usec * 1e-6;
1001 if (info) {
1002 info->implementation = "getrusage(RUSAGE_SELF)";
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001003 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001004 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001005 info->resolution = 1e-6;
1006 }
1007 return PyFloat_FromDouble(total);
1008 }
1009#endif
1010
1011#ifdef HAVE_TIMES
1012 if (times(&t) != (clock_t)-1) {
1013 double total;
1014
1015 if (ticks_per_second == -1) {
1016#if defined(HAVE_SYSCONF) && defined(_SC_CLK_TCK)
1017 ticks_per_second = sysconf(_SC_CLK_TCK);
1018 if (ticks_per_second < 1)
1019 ticks_per_second = -1;
1020#elif defined(HZ)
1021 ticks_per_second = HZ;
1022#else
1023 ticks_per_second = 60; /* magic fallback value; may be bogus */
1024#endif
1025 }
1026
1027 if (ticks_per_second != -1) {
1028 total = (double)t.tms_utime / ticks_per_second;
1029 total += (double)t.tms_stime / ticks_per_second;
1030 if (info) {
1031 info->implementation = "times()";
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001032 info->monotonic = 1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001033 info->adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001034 info->resolution = 1.0 / ticks_per_second;
1035 }
1036 return PyFloat_FromDouble(total);
1037 }
1038 }
1039#endif
1040
1041 return floatclock(info);
1042#endif
1043}
1044
1045static PyObject *
1046time_process_time(PyObject *self, PyObject *unused)
1047{
1048 return py_process_time(NULL);
1049}
1050
1051PyDoc_STRVAR(process_time_doc,
1052"process_time() -> float\n\
1053\n\
1054Process time for profiling: sum of the kernel and user-space CPU time.");
1055
1056
Victor Stinnerec895392012-04-29 02:41:27 +02001057static PyObject *
1058time_get_clock_info(PyObject *self, PyObject *args)
1059{
1060 char *name;
Victor Stinnerec895392012-04-29 02:41:27 +02001061 _Py_clock_info_t info;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001062 PyObject *obj = NULL, *dict, *ns;
Victor Stinnerec895392012-04-29 02:41:27 +02001063
1064 if (!PyArg_ParseTuple(args, "s:get_clock_info", &name))
1065 return NULL;
1066
1067#ifdef Py_DEBUG
1068 info.implementation = NULL;
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001069 info.monotonic = -1;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001070 info.adjustable = -1;
Victor Stinnerec895392012-04-29 02:41:27 +02001071 info.resolution = -1.0;
1072#else
1073 info.implementation = "";
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001074 info.monotonic = 0;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001075 info.adjustable = 0;
Victor Stinnerec895392012-04-29 02:41:27 +02001076 info.resolution = 1.0;
1077#endif
1078
1079 if (strcmp(name, "time") == 0)
1080 obj = floattime(&info);
1081#ifdef PYCLOCK
1082 else if (strcmp(name, "clock") == 0)
1083 obj = pyclock(&info);
1084#endif
Victor Stinnerec895392012-04-29 02:41:27 +02001085 else if (strcmp(name, "monotonic") == 0)
1086 obj = pymonotonic(&info);
Victor Stinnerec895392012-04-29 02:41:27 +02001087 else if (strcmp(name, "perf_counter") == 0)
1088 obj = perf_counter(&info);
1089 else if (strcmp(name, "process_time") == 0)
1090 obj = py_process_time(&info);
1091 else {
1092 PyErr_SetString(PyExc_ValueError, "unknown clock");
1093 return NULL;
1094 }
1095 if (obj == NULL)
1096 return NULL;
1097 Py_DECREF(obj);
1098
Victor Stinnerbda4b882012-06-12 22:11:44 +02001099 dict = PyDict_New();
1100 if (dict == NULL)
Victor Stinnerec895392012-04-29 02:41:27 +02001101 return NULL;
1102
1103 assert(info.implementation != NULL);
1104 obj = PyUnicode_FromString(info.implementation);
1105 if (obj == NULL)
1106 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001107 if (PyDict_SetItemString(dict, "implementation", obj) == -1)
1108 goto error;
1109 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001110
Benjamin Peterson49a69e42012-05-01 09:38:34 -04001111 assert(info.monotonic != -1);
1112 obj = PyBool_FromLong(info.monotonic);
Victor Stinnerec895392012-04-29 02:41:27 +02001113 if (obj == NULL)
1114 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001115 if (PyDict_SetItemString(dict, "monotonic", obj) == -1)
1116 goto error;
1117 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001118
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001119 assert(info.adjustable != -1);
1120 obj = PyBool_FromLong(info.adjustable);
Victor Stinnerec895392012-04-29 02:41:27 +02001121 if (obj == NULL)
1122 goto error;
Victor Stinner2b89fdf2012-06-12 22:46:37 +02001123 if (PyDict_SetItemString(dict, "adjustable", obj) == -1)
Victor Stinnerbda4b882012-06-12 22:11:44 +02001124 goto error;
1125 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001126
1127 assert(info.resolution > 0.0);
1128 assert(info.resolution <= 1.0);
1129 obj = PyFloat_FromDouble(info.resolution);
1130 if (obj == NULL)
1131 goto error;
Victor Stinnerbda4b882012-06-12 22:11:44 +02001132 if (PyDict_SetItemString(dict, "resolution", obj) == -1)
1133 goto error;
1134 Py_CLEAR(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001135
Victor Stinnerbda4b882012-06-12 22:11:44 +02001136 ns = _PyNamespace_New(dict);
1137 Py_DECREF(dict);
1138 return ns;
Victor Stinnerec895392012-04-29 02:41:27 +02001139
1140error:
Victor Stinnerbda4b882012-06-12 22:11:44 +02001141 Py_DECREF(dict);
1142 Py_XDECREF(obj);
Victor Stinnerec895392012-04-29 02:41:27 +02001143 return NULL;
1144}
1145
1146PyDoc_STRVAR(get_clock_info_doc,
1147"get_clock_info(name: str) -> dict\n\
1148\n\
1149Get information of the specified clock.");
Victor Stinner8b302012012-02-07 23:29:46 +01001150
Martin v. Löwisd218dc12008-04-07 03:17:54 +00001151static void
Martin v. Löwis1a214512008-06-11 05:26:20 +00001152PyInit_timezone(PyObject *m) {
1153 /* This code moved from PyInit_time wholesale to allow calling it from
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001154 time_tzset. In the future, some parts of it can be moved back
1155 (for platforms that don't HAVE_WORKING_TZSET, when we know what they
1156 are), and the extraneous calls to tzset(3) should be removed.
1157 I haven't done this yet, as I don't want to change this code as
1158 little as possible when introducing the time.tzset and time.tzsetwall
1159 methods. This should simply be a method of doing the following once,
1160 at the top of this function and removing the call to tzset() from
1161 time_tzset():
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001162
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001163 #ifdef HAVE_TZSET
1164 tzset()
1165 #endif
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001166
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001167 And I'm lazy and hate C so nyer.
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001168 */
Guido van Rossum10b164a2001-09-25 13:59:01 +00001169#if defined(HAVE_TZNAME) && !defined(__GLIBC__) && !defined(__CYGWIN__)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001170 PyObject *otz0, *otz1;
1171 tzset();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001172 PyModule_AddIntConstant(m, "timezone", timezone);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001173#ifdef HAVE_ALTZONE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001174 PyModule_AddIntConstant(m, "altzone", altzone);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001175#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001176 PyModule_AddIntConstant(m, "altzone", timezone-3600);
Guido van Rossumb6775db1994-08-01 11:34:53 +00001177#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001178 PyModule_AddIntConstant(m, "daylight", daylight);
Victor Stinner1b579672011-12-17 05:47:23 +01001179 otz0 = PyUnicode_DecodeLocale(tzname[0], "surrogateescape");
1180 otz1 = PyUnicode_DecodeLocale(tzname[1], "surrogateescape");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001181 PyModule_AddObject(m, "tzname", Py_BuildValue("(NN)", otz0, otz1));
Guido van Rossum10b164a2001-09-25 13:59:01 +00001182#else /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001183#ifdef HAVE_STRUCT_TM_TM_ZONE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001184 {
Guido van Rossum234f9421993-06-17 12:35:49 +00001185#define YEAR ((time_t)((365 * 24 + 6) * 3600))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001186 time_t t;
1187 struct tm *p;
1188 long janzone, julyzone;
1189 char janname[10], julyname[10];
1190 t = (time((time_t *)0) / YEAR) * YEAR;
1191 p = localtime(&t);
1192 janzone = -p->tm_gmtoff;
1193 strncpy(janname, p->tm_zone ? p->tm_zone : " ", 9);
1194 janname[9] = '\0';
1195 t += YEAR/2;
1196 p = localtime(&t);
1197 julyzone = -p->tm_gmtoff;
1198 strncpy(julyname, p->tm_zone ? p->tm_zone : " ", 9);
1199 julyname[9] = '\0';
Guido van Rossum10b164a2001-09-25 13:59:01 +00001200
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001201 if( janzone < julyzone ) {
1202 /* DST is reversed in the southern hemisphere */
1203 PyModule_AddIntConstant(m, "timezone", julyzone);
1204 PyModule_AddIntConstant(m, "altzone", janzone);
1205 PyModule_AddIntConstant(m, "daylight",
1206 janzone != julyzone);
1207 PyModule_AddObject(m, "tzname",
1208 Py_BuildValue("(zz)",
1209 julyname, janname));
1210 } else {
1211 PyModule_AddIntConstant(m, "timezone", janzone);
1212 PyModule_AddIntConstant(m, "altzone", julyzone);
1213 PyModule_AddIntConstant(m, "daylight",
1214 janzone != julyzone);
1215 PyModule_AddObject(m, "tzname",
1216 Py_BuildValue("(zz)",
1217 janname, julyname));
1218 }
1219 }
Guido van Rossume6a4b7b1997-10-08 15:27:56 +00001220#else
Martin v. Löwis60a5d722002-10-16 20:28:25 +00001221#endif /* HAVE_STRUCT_TM_TM_ZONE */
Tim Peters26ae7cd2001-03-20 03:26:49 +00001222#ifdef __CYGWIN__
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001223 tzset();
1224 PyModule_AddIntConstant(m, "timezone", _timezone);
1225 PyModule_AddIntConstant(m, "altzone", _timezone-3600);
1226 PyModule_AddIntConstant(m, "daylight", _daylight);
1227 PyModule_AddObject(m, "tzname",
1228 Py_BuildValue("(zz)", _tzname[0], _tzname[1]));
Tim Peters26ae7cd2001-03-20 03:26:49 +00001229#endif /* __CYGWIN__ */
Guido van Rossum10b164a2001-09-25 13:59:01 +00001230#endif /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
Victor Stinnere0be4232011-10-25 13:06:09 +02001231
Victor Stinnerec895392012-04-29 02:41:27 +02001232#if defined(HAVE_CLOCK_GETTIME)
Victor Stinnere0be4232011-10-25 13:06:09 +02001233 PyModule_AddIntMacro(m, CLOCK_REALTIME);
Victor Stinnere0be4232011-10-25 13:06:09 +02001234#ifdef CLOCK_MONOTONIC
1235 PyModule_AddIntMacro(m, CLOCK_MONOTONIC);
1236#endif
1237#ifdef CLOCK_MONOTONIC_RAW
1238 PyModule_AddIntMacro(m, CLOCK_MONOTONIC_RAW);
1239#endif
Victor Stinner1470f352012-04-03 00:31:17 +02001240#ifdef CLOCK_HIGHRES
1241 PyModule_AddIntMacro(m, CLOCK_HIGHRES);
1242#endif
Victor Stinnere0be4232011-10-25 13:06:09 +02001243#ifdef CLOCK_PROCESS_CPUTIME_ID
1244 PyModule_AddIntMacro(m, CLOCK_PROCESS_CPUTIME_ID);
1245#endif
1246#ifdef CLOCK_THREAD_CPUTIME_ID
1247 PyModule_AddIntMacro(m, CLOCK_THREAD_CPUTIME_ID);
1248#endif
1249#endif /* HAVE_CLOCK_GETTIME */
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001250}
1251
1252
1253static PyMethodDef time_methods[] = {
Victor Stinner4195b5c2012-02-08 23:03:19 +01001254 {"time", time_time, METH_NOARGS, time_doc},
Victor Stinnerec895392012-04-29 02:41:27 +02001255#ifdef PYCLOCK
Victor Stinner4195b5c2012-02-08 23:03:19 +01001256 {"clock", time_clock, METH_NOARGS, clock_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001257#endif
Victor Stinnere0be4232011-10-25 13:06:09 +02001258#ifdef HAVE_CLOCK_GETTIME
Victor Stinner4195b5c2012-02-08 23:03:19 +01001259 {"clock_gettime", time_clock_gettime, METH_VARARGS, clock_gettime_doc},
Victor Stinner30d79472012-04-03 00:45:07 +02001260 {"clock_settime", time_clock_settime, METH_VARARGS, clock_settime_doc},
Victor Stinner4195b5c2012-02-08 23:03:19 +01001261 {"clock_getres", time_clock_getres, METH_VARARGS, clock_getres_doc},
Victor Stinnere0be4232011-10-25 13:06:09 +02001262#endif
Victor Stinnercb29f012015-03-27 13:31:18 +01001263 {"sleep", time_sleep, METH_O, sleep_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001264 {"gmtime", time_gmtime, METH_VARARGS, gmtime_doc},
1265 {"localtime", time_localtime, METH_VARARGS, localtime_doc},
1266 {"asctime", time_asctime, METH_VARARGS, asctime_doc},
1267 {"ctime", time_ctime, METH_VARARGS, ctime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001268#ifdef HAVE_MKTIME
Victor Stinner4195b5c2012-02-08 23:03:19 +01001269 {"mktime", time_mktime, METH_O, mktime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001270#endif
1271#ifdef HAVE_STRFTIME
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001272 {"strftime", time_strftime, METH_VARARGS, strftime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001273#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001274 {"strptime", time_strptime, METH_VARARGS, strptime_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001275#ifdef HAVE_WORKING_TZSET
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001276 {"tzset", time_tzset, METH_NOARGS, tzset_doc},
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001277#endif
Victor Stinnerec895392012-04-29 02:41:27 +02001278 {"monotonic", time_monotonic, METH_NOARGS, monotonic_doc},
Victor Stinnerec895392012-04-29 02:41:27 +02001279 {"process_time", time_process_time, METH_NOARGS, process_time_doc},
1280 {"perf_counter", time_perf_counter, METH_NOARGS, perf_counter_doc},
1281 {"get_clock_info", time_get_clock_info, METH_VARARGS, get_clock_info_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001282 {NULL, NULL} /* sentinel */
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001283};
1284
1285
1286PyDoc_STRVAR(module_doc,
1287"This module provides various functions to manipulate time values.\n\
1288\n\
1289There are two standard representations of time. One is the number\n\
1290of seconds since the Epoch, in UTC (a.k.a. GMT). It may be an integer\n\
1291or a floating point number (to represent fractions of seconds).\n\
1292The Epoch is system-defined; on Unix, it is generally January 1st, 1970.\n\
1293The actual value can be retrieved by calling gmtime(0).\n\
1294\n\
1295The other representation is a tuple of 9 integers giving local time.\n\
1296The tuple items are:\n\
Alexander Belopolsky03163ac2011-05-02 12:20:52 -04001297 year (including century, e.g. 1998)\n\
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001298 month (1-12)\n\
1299 day (1-31)\n\
1300 hours (0-23)\n\
1301 minutes (0-59)\n\
1302 seconds (0-59)\n\
1303 weekday (0-6, Monday is 0)\n\
1304 Julian day (day in the year, 1-366)\n\
1305 DST (Daylight Savings Time) flag (-1, 0 or 1)\n\
1306If the DST flag is 0, the time is given in the regular time zone;\n\
1307if it is 1, the time is given in the DST time zone;\n\
1308if it is -1, mktime() should guess based on the date and time.\n\
1309\n\
1310Variables:\n\
1311\n\
1312timezone -- difference in seconds between UTC and local standard time\n\
1313altzone -- difference in seconds between UTC and local DST time\n\
1314daylight -- whether local time should reflect DST\n\
1315tzname -- tuple of (standard time zone name, DST time zone name)\n\
1316\n\
1317Functions:\n\
1318\n\
1319time() -- return current time in seconds since the Epoch as a float\n\
1320clock() -- return CPU time since process start as a float\n\
1321sleep() -- delay for a number of seconds given as a float\n\
1322gmtime() -- convert seconds since Epoch to UTC tuple\n\
1323localtime() -- convert seconds since Epoch to local time tuple\n\
1324asctime() -- convert time tuple to string\n\
1325ctime() -- convert time in seconds to string\n\
1326mktime() -- convert local time tuple to seconds since Epoch\n\
1327strftime() -- convert time tuple to string according to format specification\n\
1328strptime() -- parse string to time tuple according to format specification\n\
1329tzset() -- change the local timezone");
1330
1331
Martin v. Löwis1a214512008-06-11 05:26:20 +00001332
1333static struct PyModuleDef timemodule = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001334 PyModuleDef_HEAD_INIT,
1335 "time",
1336 module_doc,
1337 -1,
1338 time_methods,
1339 NULL,
1340 NULL,
1341 NULL,
1342 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00001343};
1344
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001345PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00001346PyInit_time(void)
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001347{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001348 PyObject *m;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001349 m = PyModule_Create(&timemodule);
1350 if (m == NULL)
1351 return NULL;
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001352
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001353 /* Set, or reset, module variables like time.timezone */
1354 PyInit_timezone(m);
Guido van Rossumd11b62e2003-03-14 21:51:36 +00001355
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001356 if (!initialized) {
Victor Stinner1c8f0592013-07-22 22:24:54 +02001357 if (PyStructSequence_InitType2(&StructTimeType,
1358 &struct_time_type_desc) < 0)
1359 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001360 }
1361 Py_INCREF(&StructTimeType);
Alexander Belopolskyc142bba2012-06-13 22:15:26 -04001362#ifdef HAVE_STRUCT_TM_TM_ZONE
1363 PyModule_AddIntConstant(m, "_STRUCT_TM_ITEMS", 11);
1364#else
1365 PyModule_AddIntConstant(m, "_STRUCT_TM_ITEMS", 9);
1366#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001367 PyModule_AddObject(m, "struct_time", (PyObject*) &StructTimeType);
1368 initialized = 1;
1369 return m;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001370}
1371
Victor Stinner071eca32012-03-15 01:17:09 +01001372static PyObject*
Victor Stinnerec895392012-04-29 02:41:27 +02001373floattime(_Py_clock_info_t *info)
Victor Stinner4195b5c2012-02-08 23:03:19 +01001374{
Victor Stinnera47b8812015-03-27 18:16:17 +01001375 _PyTime_t t;
1376 double d;
1377 if (_PyTime_GetSystemClockWithInfo(&t, info) < 0) {
Victor Stinner00111242014-08-29 16:31:59 +02001378 assert(info != NULL);
1379 return NULL;
1380 }
Victor Stinnera47b8812015-03-27 18:16:17 +01001381 d = _PyTime_AsSecondsDouble(t);
1382 return PyFloat_FromDouble(d);
Victor Stinner4195b5c2012-02-08 23:03:19 +01001383}
1384
1385
Victor Stinnercb29f012015-03-27 13:31:18 +01001386/* Implement pysleep() for various platforms.
Guido van Rossumb6775db1994-08-01 11:34:53 +00001387 When interrupted (or when another error occurs), return -1 and
1388 set an exception; else return 0. */
1389
1390static int
Victor Stinnercb29f012015-03-27 13:31:18 +01001391pysleep(_PyTime_t secs)
Guido van Rossum426035c1991-02-19 12:27:35 +00001392{
Victor Stinnercb29f012015-03-27 13:31:18 +01001393 _PyTime_t deadline, monotonic;
Victor Stinner79d68f92015-03-19 21:54:09 +01001394#ifndef MS_WINDOWS
1395 struct timeval timeout;
Victor Stinner79d68f92015-03-19 21:54:09 +01001396 int err = 0;
1397#else
Victor Stinnercb29f012015-03-27 13:31:18 +01001398 _PyTime_t millisecs;
Victor Stinner79d68f92015-03-19 21:54:09 +01001399 unsigned long ul_millis;
1400 DWORD rc;
1401 HANDLE hInterruptEvent;
Victor Stinner0c2fd892015-03-17 10:49:17 +01001402#endif
Victor Stinner79d68f92015-03-19 21:54:09 +01001403
Victor Stinnercb29f012015-03-27 13:31:18 +01001404 deadline = _PyTime_GetMonotonicClock() + secs;
Victor Stinner79d68f92015-03-19 21:54:09 +01001405
1406 do {
1407#ifndef MS_WINDOWS
Victor Stinner869e1772015-03-30 03:49:14 +02001408 if (_PyTime_AsTimeval(secs, &timeout, _PyTime_ROUND_CEILING) < 0)
Victor Stinnercb29f012015-03-27 13:31:18 +01001409 return -1;
Victor Stinner79d68f92015-03-19 21:54:09 +01001410
1411 Py_BEGIN_ALLOW_THREADS
1412 err = select(0, (fd_set *)0, (fd_set *)0, (fd_set *)0, &timeout);
1413 Py_END_ALLOW_THREADS
1414
1415 if (err == 0)
1416 break;
1417
1418 if (errno != EINTR) {
Victor Stinner0c2fd892015-03-17 10:49:17 +01001419 PyErr_SetFromErrno(PyExc_OSError);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001420 return -1;
1421 }
Victor Stinner79d68f92015-03-19 21:54:09 +01001422#else
Victor Stinner869e1772015-03-30 03:49:14 +02001423 millisecs = _PyTime_AsMilliseconds(secs, _PyTime_ROUND_CEILING);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001424 if (millisecs > (double)ULONG_MAX) {
1425 PyErr_SetString(PyExc_OverflowError,
1426 "sleep length is too large");
1427 return -1;
1428 }
Victor Stinner79d68f92015-03-19 21:54:09 +01001429
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001430 /* Allow sleep(0) to maintain win32 semantics, and as decreed
1431 * by Guido, only the main thread can be interrupted.
1432 */
1433 ul_millis = (unsigned long)millisecs;
Victor Stinner79d68f92015-03-19 21:54:09 +01001434 if (ul_millis == 0 || !_PyOS_IsMainThread()) {
1435 Py_BEGIN_ALLOW_THREADS
Victor Stinner0eac1302015-03-20 03:06:12 +01001436 Sleep(ul_millis);
Victor Stinner79d68f92015-03-19 21:54:09 +01001437 Py_END_ALLOW_THREADS
1438 break;
Victor Stinner945c82e2015-03-12 16:19:01 +01001439 }
Victor Stinner79d68f92015-03-19 21:54:09 +01001440
1441 hInterruptEvent = _PyOS_SigintEvent();
1442 ResetEvent(hInterruptEvent);
1443
1444 Py_BEGIN_ALLOW_THREADS
1445 rc = WaitForSingleObjectEx(hInterruptEvent, ul_millis, FALSE);
Victor Stinner0c2fd892015-03-17 10:49:17 +01001446 Py_END_ALLOW_THREADS
Victor Stinner79d68f92015-03-19 21:54:09 +01001447
1448 if (rc != WAIT_OBJECT_0)
1449 break;
Victor Stinner945c82e2015-03-12 16:19:01 +01001450#endif
Victor Stinner0c2fd892015-03-17 10:49:17 +01001451
Victor Stinner79d68f92015-03-19 21:54:09 +01001452 /* sleep was interrupted by SIGINT */
1453 if (PyErr_CheckSignals())
1454 return -1;
1455
Victor Stinnercb29f012015-03-27 13:31:18 +01001456 monotonic = _PyTime_GetMonotonicClock();
1457 secs = deadline - monotonic;
Victor Stinner6aa446c2015-03-30 21:33:51 +02001458 if (secs < 0)
Victor Stinner79d68f92015-03-19 21:54:09 +01001459 break;
1460 /* retry with the recomputed delay */
1461 } while (1);
1462
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001463 return 0;
Guido van Rossum80c9d881991-04-16 08:47:51 +00001464}