mirror of
https://github.com/python/cpython.git
synced 2024-11-25 02:44:06 +08:00
ec79b1ca21
svn+ssh://pythondev@svn.python.org/python/trunk ........ r75020 | brett.cannon | 2009-09-22 12:13:27 -0700 (Tue, 22 Sep 2009) | 1 line Fix whitespace. ........
1112 lines
29 KiB
C
1112 lines
29 KiB
C
|
|
/* Time module */
|
|
|
|
#include "Python.h"
|
|
#include "structseq.h"
|
|
#include "timefuncs.h"
|
|
|
|
#define TZNAME_ENCODING "utf-8"
|
|
|
|
#ifdef __APPLE__
|
|
#if defined(HAVE_GETTIMEOFDAY) && defined(HAVE_FTIME)
|
|
/*
|
|
* floattime falls back to ftime when getttimeofday fails because the latter
|
|
* might fail on some platforms. This fallback is unwanted on MacOSX because
|
|
* that makes it impossible to use a binary build on OSX 10.4 on earlier
|
|
* releases of the OS. Therefore claim we don't support ftime.
|
|
*/
|
|
# undef HAVE_FTIME
|
|
#endif
|
|
#endif
|
|
|
|
#include <ctype.h>
|
|
|
|
#ifdef HAVE_SYS_TYPES_H
|
|
#include <sys/types.h>
|
|
#endif /* HAVE_SYS_TYPES_H */
|
|
|
|
#ifdef QUICKWIN
|
|
#include <io.h>
|
|
#endif
|
|
|
|
#ifdef HAVE_FTIME
|
|
#include <sys/timeb.h>
|
|
#if !defined(MS_WINDOWS) && !defined(PYOS_OS2)
|
|
extern int ftime(struct timeb *);
|
|
#endif /* MS_WINDOWS */
|
|
#endif /* HAVE_FTIME */
|
|
|
|
#if defined(__WATCOMC__) && !defined(__QNX__)
|
|
#include <i86.h>
|
|
#else
|
|
#ifdef MS_WINDOWS
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <windows.h>
|
|
#include "pythread.h"
|
|
|
|
/* helper to allow us to interrupt sleep() on Windows*/
|
|
static HANDLE hInterruptEvent = NULL;
|
|
static BOOL WINAPI PyCtrlHandler(DWORD dwCtrlType)
|
|
{
|
|
SetEvent(hInterruptEvent);
|
|
/* allow other default handlers to be called.
|
|
Default Python handler will setup the
|
|
KeyboardInterrupt exception.
|
|
*/
|
|
return FALSE;
|
|
}
|
|
static long main_thread;
|
|
|
|
#if defined(__BORLANDC__)
|
|
/* These overrides not needed for Win32 */
|
|
#define timezone _timezone
|
|
#define tzname _tzname
|
|
#define daylight _daylight
|
|
#endif /* __BORLANDC__ */
|
|
#endif /* MS_WINDOWS */
|
|
#endif /* !__WATCOMC__ || __QNX__ */
|
|
|
|
#if defined(MS_WINDOWS) && !defined(__BORLANDC__)
|
|
/* Win32 has better clock replacement; we have our own version below. */
|
|
#undef HAVE_CLOCK
|
|
#undef TZNAME_ENCODING
|
|
#define TZNAME_ENCODING "mbcs"
|
|
#endif /* MS_WINDOWS && !defined(__BORLANDC__) */
|
|
|
|
#if defined(PYOS_OS2)
|
|
#define INCL_DOS
|
|
#define INCL_ERRORS
|
|
#include <os2.h>
|
|
#endif
|
|
|
|
#if defined(PYCC_VACPP)
|
|
#include <sys/time.h>
|
|
#endif
|
|
|
|
/* Forward declarations */
|
|
static int floatsleep(double);
|
|
static double floattime(void);
|
|
|
|
/* For Y2K check */
|
|
static PyObject *moddict;
|
|
|
|
/* Exposed in timefuncs.h. */
|
|
time_t
|
|
_PyTime_DoubleToTimet(double x)
|
|
{
|
|
time_t result;
|
|
double diff;
|
|
|
|
result = (time_t)x;
|
|
/* How much info did we lose? time_t may be an integral or
|
|
* floating type, and we don't know which. If it's integral,
|
|
* we don't know whether C truncates, rounds, returns the floor,
|
|
* etc. If we lost a second or more, the C rounding is
|
|
* unreasonable, or the input just doesn't fit in a time_t;
|
|
* call it an error regardless. Note that the original cast to
|
|
* time_t can cause a C error too, but nothing we can do to
|
|
* worm around that.
|
|
*/
|
|
diff = x - (double)result;
|
|
if (diff <= -1.0 || diff >= 1.0) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"timestamp out of range for platform time_t");
|
|
result = (time_t)-1;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
static PyObject *
|
|
time_time(PyObject *self, PyObject *unused)
|
|
{
|
|
double secs;
|
|
secs = floattime();
|
|
if (secs == 0.0) {
|
|
PyErr_SetFromErrno(PyExc_IOError);
|
|
return NULL;
|
|
}
|
|
return PyFloat_FromDouble(secs);
|
|
}
|
|
|
|
PyDoc_STRVAR(time_doc,
|
|
"time() -> floating point number\n\
|
|
\n\
|
|
Return the current time in seconds since the Epoch.\n\
|
|
Fractions of a second may be present if the system clock provides them.");
|
|
|
|
#ifdef HAVE_CLOCK
|
|
|
|
#ifndef CLOCKS_PER_SEC
|
|
#ifdef CLK_TCK
|
|
#define CLOCKS_PER_SEC CLK_TCK
|
|
#else
|
|
#define CLOCKS_PER_SEC 1000000
|
|
#endif
|
|
#endif
|
|
|
|
static PyObject *
|
|
time_clock(PyObject *self, PyObject *unused)
|
|
{
|
|
return PyFloat_FromDouble(((double)clock()) / CLOCKS_PER_SEC);
|
|
}
|
|
#endif /* HAVE_CLOCK */
|
|
|
|
#if defined(MS_WINDOWS) && !defined(__BORLANDC__)
|
|
/* Due to Mark Hammond and Tim Peters */
|
|
static PyObject *
|
|
time_clock(PyObject *self, PyObject *unused)
|
|
{
|
|
static LARGE_INTEGER ctrStart;
|
|
static double divisor = 0.0;
|
|
LARGE_INTEGER now;
|
|
double diff;
|
|
|
|
if (divisor == 0.0) {
|
|
LARGE_INTEGER freq;
|
|
QueryPerformanceCounter(&ctrStart);
|
|
if (!QueryPerformanceFrequency(&freq) || freq.QuadPart == 0) {
|
|
/* Unlikely to happen - this works on all intel
|
|
machines at least! Revert to clock() */
|
|
return PyFloat_FromDouble(((double)clock()) /
|
|
CLOCKS_PER_SEC);
|
|
}
|
|
divisor = (double)freq.QuadPart;
|
|
}
|
|
QueryPerformanceCounter(&now);
|
|
diff = (double)(now.QuadPart - ctrStart.QuadPart);
|
|
return PyFloat_FromDouble(diff / divisor);
|
|
}
|
|
|
|
#define HAVE_CLOCK /* So it gets included in the methods */
|
|
#endif /* MS_WINDOWS && !defined(__BORLANDC__) */
|
|
|
|
#ifdef HAVE_CLOCK
|
|
PyDoc_STRVAR(clock_doc,
|
|
"clock() -> floating point number\n\
|
|
\n\
|
|
Return the CPU time or real time since the start of the process or since\n\
|
|
the first call to clock(). This has as much precision as the system\n\
|
|
records.");
|
|
#endif
|
|
|
|
static PyObject *
|
|
time_sleep(PyObject *self, PyObject *args)
|
|
{
|
|
double secs;
|
|
if (!PyArg_ParseTuple(args, "d:sleep", &secs))
|
|
return NULL;
|
|
if (floatsleep(secs) != 0)
|
|
return NULL;
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
PyDoc_STRVAR(sleep_doc,
|
|
"sleep(seconds)\n\
|
|
\n\
|
|
Delay execution for a given number of seconds. The argument may be\n\
|
|
a floating point number for subsecond precision.");
|
|
|
|
static PyStructSequence_Field struct_time_type_fields[] = {
|
|
{"tm_year", NULL},
|
|
{"tm_mon", NULL},
|
|
{"tm_mday", NULL},
|
|
{"tm_hour", NULL},
|
|
{"tm_min", NULL},
|
|
{"tm_sec", NULL},
|
|
{"tm_wday", NULL},
|
|
{"tm_yday", NULL},
|
|
{"tm_isdst", NULL},
|
|
{0}
|
|
};
|
|
|
|
static PyStructSequence_Desc struct_time_type_desc = {
|
|
"time.struct_time",
|
|
NULL,
|
|
struct_time_type_fields,
|
|
9,
|
|
};
|
|
|
|
static int initialized;
|
|
static PyTypeObject StructTimeType;
|
|
|
|
static PyObject *
|
|
tmtotuple(struct tm *p)
|
|
{
|
|
PyObject *v = PyStructSequence_New(&StructTimeType);
|
|
if (v == NULL)
|
|
return NULL;
|
|
|
|
#define SET(i,val) PyStructSequence_SET_ITEM(v, i, PyLong_FromLong((long) val))
|
|
|
|
SET(0, p->tm_year + 1900);
|
|
SET(1, p->tm_mon + 1); /* Want January == 1 */
|
|
SET(2, p->tm_mday);
|
|
SET(3, p->tm_hour);
|
|
SET(4, p->tm_min);
|
|
SET(5, p->tm_sec);
|
|
SET(6, (p->tm_wday + 6) % 7); /* Want Monday == 0 */
|
|
SET(7, p->tm_yday + 1); /* Want January, 1 == 1 */
|
|
SET(8, p->tm_isdst);
|
|
#undef SET
|
|
if (PyErr_Occurred()) {
|
|
Py_XDECREF(v);
|
|
return NULL;
|
|
}
|
|
|
|
return v;
|
|
}
|
|
|
|
static PyObject *
|
|
structtime_totuple(PyObject *t)
|
|
{
|
|
PyObject *x = NULL;
|
|
unsigned int i;
|
|
PyObject *v = PyTuple_New(9);
|
|
if (v == NULL)
|
|
return NULL;
|
|
|
|
for (i=0; i<9; i++) {
|
|
x = PyStructSequence_GET_ITEM(t, i);
|
|
Py_INCREF(x);
|
|
PyTuple_SET_ITEM(v, i, x);
|
|
}
|
|
|
|
if (PyErr_Occurred()) {
|
|
Py_XDECREF(v);
|
|
return NULL;
|
|
}
|
|
|
|
return v;
|
|
}
|
|
|
|
static PyObject *
|
|
time_convert(double when, struct tm * (*function)(const time_t *))
|
|
{
|
|
struct tm *p;
|
|
time_t whent = _PyTime_DoubleToTimet(when);
|
|
|
|
if (whent == (time_t)-1 && PyErr_Occurred())
|
|
return NULL;
|
|
errno = 0;
|
|
p = function(&whent);
|
|
if (p == NULL) {
|
|
#ifdef EINVAL
|
|
if (errno == 0)
|
|
errno = EINVAL;
|
|
#endif
|
|
return PyErr_SetFromErrno(PyExc_ValueError);
|
|
}
|
|
return tmtotuple(p);
|
|
}
|
|
|
|
/* Parse arg tuple that can contain an optional float-or-None value;
|
|
format needs to be "|O:name".
|
|
Returns non-zero on success (parallels PyArg_ParseTuple).
|
|
*/
|
|
static int
|
|
parse_time_double_args(PyObject *args, char *format, double *pwhen)
|
|
{
|
|
PyObject *ot = NULL;
|
|
|
|
if (!PyArg_ParseTuple(args, format, &ot))
|
|
return 0;
|
|
if (ot == NULL || ot == Py_None)
|
|
*pwhen = floattime();
|
|
else {
|
|
double when = PyFloat_AsDouble(ot);
|
|
if (PyErr_Occurred())
|
|
return 0;
|
|
*pwhen = when;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
static PyObject *
|
|
time_gmtime(PyObject *self, PyObject *args)
|
|
{
|
|
double when;
|
|
if (!parse_time_double_args(args, "|O:gmtime", &when))
|
|
return NULL;
|
|
return time_convert(when, gmtime);
|
|
}
|
|
|
|
PyDoc_STRVAR(gmtime_doc,
|
|
"gmtime([seconds]) -> (tm_year, tm_mon, tm_mday, tm_hour, tm_min,\n\
|
|
tm_sec, tm_wday, tm_yday, tm_isdst)\n\
|
|
\n\
|
|
Convert seconds since the Epoch to a time tuple expressing UTC (a.k.a.\n\
|
|
GMT). When 'seconds' is not passed in, convert the current time instead.");
|
|
|
|
static PyObject *
|
|
time_localtime(PyObject *self, PyObject *args)
|
|
{
|
|
double when;
|
|
if (!parse_time_double_args(args, "|O:localtime", &when))
|
|
return NULL;
|
|
return time_convert(when, localtime);
|
|
}
|
|
|
|
PyDoc_STRVAR(localtime_doc,
|
|
"localtime([seconds]) -> (tm_year,tm_mon,tm_mday,tm_hour,tm_min,\n\
|
|
tm_sec,tm_wday,tm_yday,tm_isdst)\n\
|
|
\n\
|
|
Convert seconds since the Epoch to a time tuple expressing local time.\n\
|
|
When 'seconds' is not passed in, convert the current time instead.");
|
|
|
|
static int
|
|
gettmarg(PyObject *args, struct tm *p)
|
|
{
|
|
int y;
|
|
PyObject *t = NULL;
|
|
|
|
memset((void *) p, '\0', sizeof(struct tm));
|
|
|
|
if (PyTuple_Check(args)) {
|
|
t = args;
|
|
Py_INCREF(t);
|
|
}
|
|
else if (Py_TYPE(args) == &StructTimeType) {
|
|
t = structtime_totuple(args);
|
|
}
|
|
else {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"Tuple or struct_time argument required");
|
|
return 0;
|
|
}
|
|
|
|
if (t == NULL || !PyArg_ParseTuple(t, "iiiiiiiii",
|
|
&y,
|
|
&p->tm_mon,
|
|
&p->tm_mday,
|
|
&p->tm_hour,
|
|
&p->tm_min,
|
|
&p->tm_sec,
|
|
&p->tm_wday,
|
|
&p->tm_yday,
|
|
&p->tm_isdst)) {
|
|
Py_XDECREF(t);
|
|
return 0;
|
|
}
|
|
Py_DECREF(t);
|
|
|
|
if (y < 1900) {
|
|
PyObject *accept = PyDict_GetItemString(moddict,
|
|
"accept2dyear");
|
|
if (accept == NULL || !PyLong_CheckExact(accept) ||
|
|
!PyObject_IsTrue(accept)) {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"year >= 1900 required");
|
|
return 0;
|
|
}
|
|
if (69 <= y && y <= 99)
|
|
y += 1900;
|
|
else if (0 <= y && y <= 68)
|
|
y += 2000;
|
|
else {
|
|
PyErr_SetString(PyExc_ValueError,
|
|
"year out of range");
|
|
return 0;
|
|
}
|
|
}
|
|
p->tm_year = y - 1900;
|
|
p->tm_mon--;
|
|
p->tm_wday = (p->tm_wday + 1) % 7;
|
|
p->tm_yday--;
|
|
return 1;
|
|
}
|
|
|
|
#ifdef HAVE_STRFTIME
|
|
#ifdef HAVE_WCSFTIME
|
|
#define time_char wchar_t
|
|
#define format_time wcsftime
|
|
#define time_strlen wcslen
|
|
#else
|
|
#define time_char char
|
|
#define format_time strftime
|
|
#define time_strlen strlen
|
|
#endif
|
|
|
|
static PyObject *
|
|
time_strftime(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *tup = NULL;
|
|
struct tm buf;
|
|
const time_char *fmt;
|
|
PyObject *format, *tmpfmt;
|
|
size_t fmtlen, buflen;
|
|
time_char *outbuf = 0;
|
|
size_t i;
|
|
|
|
memset((void *) &buf, '\0', sizeof(buf));
|
|
|
|
/* Will always expect a unicode string to be passed as format.
|
|
Given that there's no str type anymore in py3k this seems safe.
|
|
*/
|
|
if (!PyArg_ParseTuple(args, "U|O:strftime", &format, &tup))
|
|
return NULL;
|
|
|
|
if (tup == NULL) {
|
|
time_t tt = time(NULL);
|
|
buf = *localtime(&tt);
|
|
} else if (!gettmarg(tup, &buf))
|
|
return NULL;
|
|
|
|
/* Checks added to make sure strftime() does not crash Python by
|
|
indexing blindly into some array for a textual representation
|
|
by some bad index (fixes bug #897625).
|
|
|
|
Also support values of zero from Python code for arguments in which
|
|
that is out of range by forcing that value to the lowest value that
|
|
is valid (fixed bug #1520914).
|
|
|
|
Valid ranges based on what is allowed in struct tm:
|
|
|
|
- tm_year: [0, max(int)] (1)
|
|
- tm_mon: [0, 11] (2)
|
|
- tm_mday: [1, 31]
|
|
- tm_hour: [0, 23]
|
|
- tm_min: [0, 59]
|
|
- tm_sec: [0, 60]
|
|
- tm_wday: [0, 6] (1)
|
|
- tm_yday: [0, 365] (2)
|
|
- tm_isdst: [-max(int), max(int)]
|
|
|
|
(1) gettmarg() handles bounds-checking.
|
|
(2) Python's acceptable range is one greater than the range in C,
|
|
thus need to check against automatic decrement by gettmarg().
|
|
*/
|
|
if (buf.tm_mon == -1)
|
|
buf.tm_mon = 0;
|
|
else if (buf.tm_mon < 0 || buf.tm_mon > 11) {
|
|
PyErr_SetString(PyExc_ValueError, "month out of range");
|
|
return NULL;
|
|
}
|
|
if (buf.tm_mday == 0)
|
|
buf.tm_mday = 1;
|
|
else if (buf.tm_mday < 0 || buf.tm_mday > 31) {
|
|
PyErr_SetString(PyExc_ValueError, "day of month out of range");
|
|
return NULL;
|
|
}
|
|
if (buf.tm_hour < 0 || buf.tm_hour > 23) {
|
|
PyErr_SetString(PyExc_ValueError, "hour out of range");
|
|
return NULL;
|
|
}
|
|
if (buf.tm_min < 0 || buf.tm_min > 59) {
|
|
PyErr_SetString(PyExc_ValueError, "minute out of range");
|
|
return NULL;
|
|
}
|
|
if (buf.tm_sec < 0 || buf.tm_sec > 61) {
|
|
PyErr_SetString(PyExc_ValueError, "seconds out of range");
|
|
return NULL;
|
|
}
|
|
/* tm_wday does not need checking of its upper-bound since taking
|
|
``% 7`` in gettmarg() automatically restricts the range. */
|
|
if (buf.tm_wday < 0) {
|
|
PyErr_SetString(PyExc_ValueError, "day of week out of range");
|
|
return NULL;
|
|
}
|
|
if (buf.tm_yday == -1)
|
|
buf.tm_yday = 0;
|
|
else if (buf.tm_yday < 0 || buf.tm_yday > 365) {
|
|
PyErr_SetString(PyExc_ValueError, "day of year out of range");
|
|
return NULL;
|
|
}
|
|
/* Normalize tm_isdst just in case someone foolishly implements %Z
|
|
based on the assumption that tm_isdst falls within the range of
|
|
[-1, 1] */
|
|
if (buf.tm_isdst < -1)
|
|
buf.tm_isdst = -1;
|
|
else if (buf.tm_isdst > 1)
|
|
buf.tm_isdst = 1;
|
|
|
|
#ifdef HAVE_WCSFTIME
|
|
tmpfmt = PyBytes_FromStringAndSize(NULL,
|
|
sizeof(wchar_t) * (PyUnicode_GetSize(format)+1));
|
|
if (!tmpfmt)
|
|
return NULL;
|
|
/* This assumes that PyUnicode_AsWideChar doesn't do any UTF-16
|
|
expansion. */
|
|
if (PyUnicode_AsWideChar((PyUnicodeObject*)format,
|
|
(wchar_t*)PyBytes_AS_STRING(tmpfmt),
|
|
PyUnicode_GetSize(format)+1) == (size_t)-1)
|
|
/* This shouldn't fail. */
|
|
Py_FatalError("PyUnicode_AsWideChar failed");
|
|
format = tmpfmt;
|
|
fmt = (wchar_t*)PyBytes_AS_STRING(format);
|
|
#else
|
|
/* Convert the unicode string to an ascii one */
|
|
format = PyUnicode_AsEncodedString(format, TZNAME_ENCODING, NULL);
|
|
if (format == NULL)
|
|
return NULL;
|
|
fmt = PyBytes_AS_STRING(format);
|
|
#endif
|
|
|
|
#if defined(MS_WINDOWS) && defined(HAVE_WCSFTIME)
|
|
/* check that the format string contains only valid directives */
|
|
for(outbuf = wcschr(fmt, L'%');
|
|
outbuf != NULL;
|
|
outbuf = wcschr(outbuf+2, L'%'))
|
|
{
|
|
if (outbuf[1]=='#')
|
|
++outbuf; /* not documented by python, */
|
|
if (outbuf[1]=='\0' ||
|
|
!wcschr(L"aAbBcdfHIjmMpSUwWxXyYzZ%", outbuf[1]))
|
|
{
|
|
PyErr_SetString(PyExc_ValueError, "Invalid format string");
|
|
return 0;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
fmtlen = time_strlen(fmt);
|
|
|
|
/* I hate these functions that presume you know how big the output
|
|
* will be ahead of time...
|
|
*/
|
|
for (i = 1024; ; i += i) {
|
|
outbuf = (time_char *)PyMem_Malloc(i*sizeof(time_char));
|
|
if (outbuf == NULL) {
|
|
Py_DECREF(format);
|
|
return PyErr_NoMemory();
|
|
}
|
|
buflen = format_time(outbuf, i, fmt, &buf);
|
|
if (buflen > 0 || i >= 256 * fmtlen) {
|
|
/* If the buffer is 256 times as long as the format,
|
|
it's probably not failing for lack of room!
|
|
More likely, the format yields an empty result,
|
|
e.g. an empty format, or %Z when the timezone
|
|
is unknown. */
|
|
PyObject *ret;
|
|
#ifdef HAVE_WCSFTIME
|
|
ret = PyUnicode_FromWideChar(outbuf, buflen);
|
|
#else
|
|
ret = PyUnicode_Decode(outbuf, buflen,
|
|
TZNAME_ENCODING, NULL);
|
|
#endif
|
|
PyMem_Free(outbuf);
|
|
Py_DECREF(format);
|
|
return ret;
|
|
}
|
|
PyMem_Free(outbuf);
|
|
#if defined _MSC_VER && _MSC_VER >= 1400 && defined(__STDC_SECURE_LIB__)
|
|
/* VisualStudio .NET 2005 does this properly */
|
|
if (buflen == 0 && errno == EINVAL) {
|
|
PyErr_SetString(PyExc_ValueError, "Invalid format string");
|
|
Py_DECREF(format);
|
|
return 0;
|
|
}
|
|
#endif
|
|
}
|
|
}
|
|
|
|
#undef time_char
|
|
#undef format_time
|
|
|
|
PyDoc_STRVAR(strftime_doc,
|
|
"strftime(format[, tuple]) -> string\n\
|
|
\n\
|
|
Convert a time tuple to a string according to a format specification.\n\
|
|
See the library reference manual for formatting codes. When the time tuple\n\
|
|
is not present, current time as returned by localtime() is used.");
|
|
#endif /* HAVE_STRFTIME */
|
|
|
|
static PyObject *
|
|
time_strptime(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *strptime_module = PyImport_ImportModuleNoBlock("_strptime");
|
|
PyObject *strptime_result;
|
|
|
|
if (!strptime_module)
|
|
return NULL;
|
|
strptime_result = PyObject_CallMethod(strptime_module,
|
|
"_strptime_time", "O", args);
|
|
Py_DECREF(strptime_module);
|
|
return strptime_result;
|
|
}
|
|
|
|
PyDoc_STRVAR(strptime_doc,
|
|
"strptime(string, format) -> struct_time\n\
|
|
\n\
|
|
Parse a string to a time tuple according to a format specification.\n\
|
|
See the library reference manual for formatting codes (same as strftime()).");
|
|
|
|
|
|
static PyObject *
|
|
time_asctime(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *tup = NULL;
|
|
struct tm buf;
|
|
char *p;
|
|
if (!PyArg_UnpackTuple(args, "asctime", 0, 1, &tup))
|
|
return NULL;
|
|
if (tup == NULL) {
|
|
time_t tt = time(NULL);
|
|
buf = *localtime(&tt);
|
|
} else if (!gettmarg(tup, &buf))
|
|
return NULL;
|
|
p = asctime(&buf);
|
|
if (p[24] == '\n')
|
|
p[24] = '\0';
|
|
return PyUnicode_FromString(p);
|
|
}
|
|
|
|
PyDoc_STRVAR(asctime_doc,
|
|
"asctime([tuple]) -> string\n\
|
|
\n\
|
|
Convert a time tuple to a string, e.g. 'Sat Jun 06 16:26:11 1998'.\n\
|
|
When the time tuple is not present, current time as returned by localtime()\n\
|
|
is used.");
|
|
|
|
static PyObject *
|
|
time_ctime(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *ot = NULL;
|
|
time_t tt;
|
|
char *p;
|
|
|
|
if (!PyArg_UnpackTuple(args, "ctime", 0, 1, &ot))
|
|
return NULL;
|
|
if (ot == NULL || ot == Py_None)
|
|
tt = time(NULL);
|
|
else {
|
|
double dt = PyFloat_AsDouble(ot);
|
|
if (PyErr_Occurred())
|
|
return NULL;
|
|
tt = _PyTime_DoubleToTimet(dt);
|
|
if (tt == (time_t)-1 && PyErr_Occurred())
|
|
return NULL;
|
|
}
|
|
p = ctime(&tt);
|
|
if (p == NULL) {
|
|
PyErr_SetString(PyExc_ValueError, "unconvertible time");
|
|
return NULL;
|
|
}
|
|
if (p[24] == '\n')
|
|
p[24] = '\0';
|
|
return PyUnicode_FromString(p);
|
|
}
|
|
|
|
PyDoc_STRVAR(ctime_doc,
|
|
"ctime(seconds) -> string\n\
|
|
\n\
|
|
Convert a time in seconds since the Epoch to a string in local time.\n\
|
|
This is equivalent to asctime(localtime(seconds)). When the time tuple is\n\
|
|
not present, current time as returned by localtime() is used.");
|
|
|
|
#ifdef HAVE_MKTIME
|
|
static PyObject *
|
|
time_mktime(PyObject *self, PyObject *tup)
|
|
{
|
|
struct tm buf;
|
|
time_t tt;
|
|
if (!gettmarg(tup, &buf))
|
|
return NULL;
|
|
tt = mktime(&buf);
|
|
if (tt == (time_t)(-1)) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"mktime argument out of range");
|
|
return NULL;
|
|
}
|
|
return PyFloat_FromDouble((double)tt);
|
|
}
|
|
|
|
PyDoc_STRVAR(mktime_doc,
|
|
"mktime(tuple) -> floating point number\n\
|
|
\n\
|
|
Convert a time tuple in local time to seconds since the Epoch.");
|
|
#endif /* HAVE_MKTIME */
|
|
|
|
#ifdef HAVE_WORKING_TZSET
|
|
static void PyInit_timezone(PyObject *module);
|
|
|
|
static PyObject *
|
|
time_tzset(PyObject *self, PyObject *unused)
|
|
{
|
|
PyObject* m;
|
|
|
|
m = PyImport_ImportModuleNoBlock("time");
|
|
if (m == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
tzset();
|
|
|
|
/* Reset timezone, altzone, daylight and tzname */
|
|
PyInit_timezone(m);
|
|
Py_DECREF(m);
|
|
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
PyDoc_STRVAR(tzset_doc,
|
|
"tzset(zone)\n\
|
|
\n\
|
|
Initialize, or reinitialize, the local timezone to the value stored in\n\
|
|
os.environ['TZ']. The TZ environment variable should be specified in\n\
|
|
standard Unix timezone format as documented in the tzset man page\n\
|
|
(eg. 'US/Eastern', 'Europe/Amsterdam'). Unknown timezones will silently\n\
|
|
fall back to UTC. If the TZ environment variable is not set, the local\n\
|
|
timezone is set to the systems best guess of wallclock time.\n\
|
|
Changing the TZ environment variable without calling tzset *may* change\n\
|
|
the local timezone used by methods such as localtime, but this behaviour\n\
|
|
should not be relied on.");
|
|
#endif /* HAVE_WORKING_TZSET */
|
|
|
|
static void
|
|
PyInit_timezone(PyObject *m) {
|
|
/* This code moved from PyInit_time wholesale to allow calling it from
|
|
time_tzset. In the future, some parts of it can be moved back
|
|
(for platforms that don't HAVE_WORKING_TZSET, when we know what they
|
|
are), and the extraneous calls to tzset(3) should be removed.
|
|
I haven't done this yet, as I don't want to change this code as
|
|
little as possible when introducing the time.tzset and time.tzsetwall
|
|
methods. This should simply be a method of doing the following once,
|
|
at the top of this function and removing the call to tzset() from
|
|
time_tzset():
|
|
|
|
#ifdef HAVE_TZSET
|
|
tzset()
|
|
#endif
|
|
|
|
And I'm lazy and hate C so nyer.
|
|
*/
|
|
#if defined(HAVE_TZNAME) && !defined(__GLIBC__) && !defined(__CYGWIN__)
|
|
PyObject *otz0, *otz1;
|
|
tzset();
|
|
#ifdef PYOS_OS2
|
|
PyModule_AddIntConstant(m, "timezone", _timezone);
|
|
#else /* !PYOS_OS2 */
|
|
PyModule_AddIntConstant(m, "timezone", timezone);
|
|
#endif /* PYOS_OS2 */
|
|
#ifdef HAVE_ALTZONE
|
|
PyModule_AddIntConstant(m, "altzone", altzone);
|
|
#else
|
|
#ifdef PYOS_OS2
|
|
PyModule_AddIntConstant(m, "altzone", _timezone-3600);
|
|
#else /* !PYOS_OS2 */
|
|
PyModule_AddIntConstant(m, "altzone", timezone-3600);
|
|
#endif /* PYOS_OS2 */
|
|
#endif
|
|
PyModule_AddIntConstant(m, "daylight", daylight);
|
|
otz0 = PyUnicode_Decode(tzname[0], strlen(tzname[0]), TZNAME_ENCODING, NULL);
|
|
otz1 = PyUnicode_Decode(tzname[1], strlen(tzname[1]), TZNAME_ENCODING, NULL);
|
|
PyModule_AddObject(m, "tzname", Py_BuildValue("(NN)", otz0, otz1));
|
|
#else /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
|
|
#ifdef HAVE_STRUCT_TM_TM_ZONE
|
|
{
|
|
#define YEAR ((time_t)((365 * 24 + 6) * 3600))
|
|
time_t t;
|
|
struct tm *p;
|
|
long janzone, julyzone;
|
|
char janname[10], julyname[10];
|
|
t = (time((time_t *)0) / YEAR) * YEAR;
|
|
p = localtime(&t);
|
|
janzone = -p->tm_gmtoff;
|
|
strncpy(janname, p->tm_zone ? p->tm_zone : " ", 9);
|
|
janname[9] = '\0';
|
|
t += YEAR/2;
|
|
p = localtime(&t);
|
|
julyzone = -p->tm_gmtoff;
|
|
strncpy(julyname, p->tm_zone ? p->tm_zone : " ", 9);
|
|
julyname[9] = '\0';
|
|
|
|
if( janzone < julyzone ) {
|
|
/* DST is reversed in the southern hemisphere */
|
|
PyModule_AddIntConstant(m, "timezone", julyzone);
|
|
PyModule_AddIntConstant(m, "altzone", janzone);
|
|
PyModule_AddIntConstant(m, "daylight",
|
|
janzone != julyzone);
|
|
PyModule_AddObject(m, "tzname",
|
|
Py_BuildValue("(zz)",
|
|
julyname, janname));
|
|
} else {
|
|
PyModule_AddIntConstant(m, "timezone", janzone);
|
|
PyModule_AddIntConstant(m, "altzone", julyzone);
|
|
PyModule_AddIntConstant(m, "daylight",
|
|
janzone != julyzone);
|
|
PyModule_AddObject(m, "tzname",
|
|
Py_BuildValue("(zz)",
|
|
janname, julyname));
|
|
}
|
|
}
|
|
#else
|
|
#endif /* HAVE_STRUCT_TM_TM_ZONE */
|
|
#ifdef __CYGWIN__
|
|
tzset();
|
|
PyModule_AddIntConstant(m, "timezone", _timezone);
|
|
PyModule_AddIntConstant(m, "altzone", _timezone-3600);
|
|
PyModule_AddIntConstant(m, "daylight", _daylight);
|
|
PyModule_AddObject(m, "tzname",
|
|
Py_BuildValue("(zz)", _tzname[0], _tzname[1]));
|
|
#endif /* __CYGWIN__ */
|
|
#endif /* !HAVE_TZNAME || __GLIBC__ || __CYGWIN__*/
|
|
}
|
|
|
|
|
|
static PyMethodDef time_methods[] = {
|
|
{"time", time_time, METH_NOARGS, time_doc},
|
|
#ifdef HAVE_CLOCK
|
|
{"clock", time_clock, METH_NOARGS, clock_doc},
|
|
#endif
|
|
{"sleep", time_sleep, METH_VARARGS, sleep_doc},
|
|
{"gmtime", time_gmtime, METH_VARARGS, gmtime_doc},
|
|
{"localtime", time_localtime, METH_VARARGS, localtime_doc},
|
|
{"asctime", time_asctime, METH_VARARGS, asctime_doc},
|
|
{"ctime", time_ctime, METH_VARARGS, ctime_doc},
|
|
#ifdef HAVE_MKTIME
|
|
{"mktime", time_mktime, METH_O, mktime_doc},
|
|
#endif
|
|
#ifdef HAVE_STRFTIME
|
|
{"strftime", time_strftime, METH_VARARGS, strftime_doc},
|
|
#endif
|
|
{"strptime", time_strptime, METH_VARARGS, strptime_doc},
|
|
#ifdef HAVE_WORKING_TZSET
|
|
{"tzset", time_tzset, METH_NOARGS, tzset_doc},
|
|
#endif
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
|
|
PyDoc_STRVAR(module_doc,
|
|
"This module provides various functions to manipulate time values.\n\
|
|
\n\
|
|
There are two standard representations of time. One is the number\n\
|
|
of seconds since the Epoch, in UTC (a.k.a. GMT). It may be an integer\n\
|
|
or a floating point number (to represent fractions of seconds).\n\
|
|
The Epoch is system-defined; on Unix, it is generally January 1st, 1970.\n\
|
|
The actual value can be retrieved by calling gmtime(0).\n\
|
|
\n\
|
|
The other representation is a tuple of 9 integers giving local time.\n\
|
|
The tuple items are:\n\
|
|
year (four digits, e.g. 1998)\n\
|
|
month (1-12)\n\
|
|
day (1-31)\n\
|
|
hours (0-23)\n\
|
|
minutes (0-59)\n\
|
|
seconds (0-59)\n\
|
|
weekday (0-6, Monday is 0)\n\
|
|
Julian day (day in the year, 1-366)\n\
|
|
DST (Daylight Savings Time) flag (-1, 0 or 1)\n\
|
|
If the DST flag is 0, the time is given in the regular time zone;\n\
|
|
if it is 1, the time is given in the DST time zone;\n\
|
|
if it is -1, mktime() should guess based on the date and time.\n\
|
|
\n\
|
|
Variables:\n\
|
|
\n\
|
|
timezone -- difference in seconds between UTC and local standard time\n\
|
|
altzone -- difference in seconds between UTC and local DST time\n\
|
|
daylight -- whether local time should reflect DST\n\
|
|
tzname -- tuple of (standard time zone name, DST time zone name)\n\
|
|
\n\
|
|
Functions:\n\
|
|
\n\
|
|
time() -- return current time in seconds since the Epoch as a float\n\
|
|
clock() -- return CPU time since process start as a float\n\
|
|
sleep() -- delay for a number of seconds given as a float\n\
|
|
gmtime() -- convert seconds since Epoch to UTC tuple\n\
|
|
localtime() -- convert seconds since Epoch to local time tuple\n\
|
|
asctime() -- convert time tuple to string\n\
|
|
ctime() -- convert time in seconds to string\n\
|
|
mktime() -- convert local time tuple to seconds since Epoch\n\
|
|
strftime() -- convert time tuple to string according to format specification\n\
|
|
strptime() -- parse string to time tuple according to format specification\n\
|
|
tzset() -- change the local timezone");
|
|
|
|
|
|
|
|
static struct PyModuleDef timemodule = {
|
|
PyModuleDef_HEAD_INIT,
|
|
"time",
|
|
module_doc,
|
|
-1,
|
|
time_methods,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
NULL
|
|
};
|
|
|
|
PyMODINIT_FUNC
|
|
PyInit_time(void)
|
|
{
|
|
PyObject *m;
|
|
char *p;
|
|
m = PyModule_Create(&timemodule);
|
|
if (m == NULL)
|
|
return NULL;
|
|
|
|
/* Accept 2-digit dates unless PYTHONY2K is set and non-empty */
|
|
p = Py_GETENV("PYTHONY2K");
|
|
PyModule_AddIntConstant(m, "accept2dyear", (long) (!p || !*p));
|
|
/* Squirrel away the module's dictionary for the y2k check */
|
|
moddict = PyModule_GetDict(m);
|
|
Py_INCREF(moddict);
|
|
|
|
/* Set, or reset, module variables like time.timezone */
|
|
PyInit_timezone(m);
|
|
|
|
#ifdef MS_WINDOWS
|
|
/* Helper to allow interrupts for Windows.
|
|
If Ctrl+C event delivered while not sleeping
|
|
it will be ignored.
|
|
*/
|
|
main_thread = PyThread_get_thread_ident();
|
|
hInterruptEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
|
|
SetConsoleCtrlHandler( PyCtrlHandler, TRUE);
|
|
#endif /* MS_WINDOWS */
|
|
if (!initialized) {
|
|
PyStructSequence_InitType(&StructTimeType,
|
|
&struct_time_type_desc);
|
|
}
|
|
Py_INCREF(&StructTimeType);
|
|
PyModule_AddObject(m, "struct_time", (PyObject*) &StructTimeType);
|
|
initialized = 1;
|
|
return m;
|
|
}
|
|
|
|
|
|
/* Implement floattime() for various platforms */
|
|
|
|
static double
|
|
floattime(void)
|
|
{
|
|
/* There are three ways to get the time:
|
|
(1) gettimeofday() -- resolution in microseconds
|
|
(2) ftime() -- resolution in milliseconds
|
|
(3) time() -- resolution in seconds
|
|
In all cases the return value is a float in seconds.
|
|
Since on some systems (e.g. SCO ODT 3.0) gettimeofday() may
|
|
fail, so we fall back on ftime() or time().
|
|
Note: clock resolution does not imply clock accuracy! */
|
|
#ifdef HAVE_GETTIMEOFDAY
|
|
{
|
|
struct timeval t;
|
|
#ifdef GETTIMEOFDAY_NO_TZ
|
|
if (gettimeofday(&t) == 0)
|
|
return (double)t.tv_sec + t.tv_usec*0.000001;
|
|
#else /* !GETTIMEOFDAY_NO_TZ */
|
|
if (gettimeofday(&t, (struct timezone *)NULL) == 0)
|
|
return (double)t.tv_sec + t.tv_usec*0.000001;
|
|
#endif /* !GETTIMEOFDAY_NO_TZ */
|
|
}
|
|
|
|
#endif /* !HAVE_GETTIMEOFDAY */
|
|
{
|
|
#if defined(HAVE_FTIME)
|
|
struct timeb t;
|
|
ftime(&t);
|
|
return (double)t.time + (double)t.millitm * (double)0.001;
|
|
#else /* !HAVE_FTIME */
|
|
time_t secs;
|
|
time(&secs);
|
|
return (double)secs;
|
|
#endif /* !HAVE_FTIME */
|
|
}
|
|
}
|
|
|
|
|
|
/* Implement floatsleep() for various platforms.
|
|
When interrupted (or when another error occurs), return -1 and
|
|
set an exception; else return 0. */
|
|
|
|
static int
|
|
floatsleep(double secs)
|
|
{
|
|
/* XXX Should test for MS_WINDOWS first! */
|
|
#if defined(HAVE_SELECT) && !defined(__EMX__)
|
|
struct timeval t;
|
|
double frac;
|
|
frac = fmod(secs, 1.0);
|
|
secs = floor(secs);
|
|
t.tv_sec = (long)secs;
|
|
t.tv_usec = (long)(frac*1000000.0);
|
|
Py_BEGIN_ALLOW_THREADS
|
|
if (select(0, (fd_set *)0, (fd_set *)0, (fd_set *)0, &t) != 0) {
|
|
#ifdef EINTR
|
|
if (errno != EINTR) {
|
|
#else
|
|
if (1) {
|
|
#endif
|
|
Py_BLOCK_THREADS
|
|
PyErr_SetFromErrno(PyExc_IOError);
|
|
return -1;
|
|
}
|
|
}
|
|
Py_END_ALLOW_THREADS
|
|
#elif defined(__WATCOMC__) && !defined(__QNX__)
|
|
/* XXX Can't interrupt this sleep */
|
|
Py_BEGIN_ALLOW_THREADS
|
|
delay((int)(secs * 1000 + 0.5)); /* delay() uses milliseconds */
|
|
Py_END_ALLOW_THREADS
|
|
#elif defined(MS_WINDOWS)
|
|
{
|
|
double millisecs = secs * 1000.0;
|
|
unsigned long ul_millis;
|
|
|
|
if (millisecs > (double)ULONG_MAX) {
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
"sleep length is too large");
|
|
return -1;
|
|
}
|
|
Py_BEGIN_ALLOW_THREADS
|
|
/* Allow sleep(0) to maintain win32 semantics, and as decreed
|
|
* by Guido, only the main thread can be interrupted.
|
|
*/
|
|
ul_millis = (unsigned long)millisecs;
|
|
if (ul_millis == 0 ||
|
|
main_thread != PyThread_get_thread_ident())
|
|
Sleep(ul_millis);
|
|
else {
|
|
DWORD rc;
|
|
ResetEvent(hInterruptEvent);
|
|
rc = WaitForSingleObject(hInterruptEvent, ul_millis);
|
|
if (rc == WAIT_OBJECT_0) {
|
|
/* Yield to make sure real Python signal
|
|
* handler called.
|
|
*/
|
|
Sleep(1);
|
|
Py_BLOCK_THREADS
|
|
errno = EINTR;
|
|
PyErr_SetFromErrno(PyExc_IOError);
|
|
return -1;
|
|
}
|
|
}
|
|
Py_END_ALLOW_THREADS
|
|
}
|
|
#elif defined(PYOS_OS2)
|
|
/* This Sleep *IS* Interruptable by Exceptions */
|
|
Py_BEGIN_ALLOW_THREADS
|
|
if (DosSleep(secs * 1000) != NO_ERROR) {
|
|
Py_BLOCK_THREADS
|
|
PyErr_SetFromErrno(PyExc_IOError);
|
|
return -1;
|
|
}
|
|
Py_END_ALLOW_THREADS
|
|
#elif defined(PLAN9)
|
|
{
|
|
double millisecs = secs * 1000.0;
|
|
if (millisecs > (double)LONG_MAX) {
|
|
PyErr_SetString(PyExc_OverflowError, "sleep length is too large");
|
|
return -1;
|
|
}
|
|
/* This sleep *CAN BE* interrupted. */
|
|
Py_BEGIN_ALLOW_THREADS
|
|
if(sleep((long)millisecs) < 0){
|
|
Py_BLOCK_THREADS
|
|
PyErr_SetFromErrno(PyExc_IOError);
|
|
return -1;
|
|
}
|
|
Py_END_ALLOW_THREADS
|
|
}
|
|
#else
|
|
/* XXX Can't interrupt this sleep */
|
|
Py_BEGIN_ALLOW_THREADS
|
|
sleep((int)secs);
|
|
Py_END_ALLOW_THREADS
|
|
#endif
|
|
|
|
return 0;
|
|
}
|