| |
| /* 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 |
| static PyObject * |
| time_strftime(PyObject *self, PyObject *args) |
| { |
| PyObject *tup = NULL; |
| struct tm buf; |
| const char *fmt; |
| PyObject *format; |
| size_t fmtlen, buflen; |
| 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; |
| } |
| if (buf.tm_isdst < -1 || buf.tm_isdst > 1) { |
| PyErr_SetString(PyExc_ValueError, |
| "daylight savings flag out of range"); |
| return NULL; |
| } |
| |
| /* 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); |
| |
| #ifdef MS_WINDOWS |
| /* check that the format string contains only valid directives */ |
| for(outbuf = strchr(fmt, '%'); |
| outbuf != NULL; |
| outbuf = strchr(outbuf+2, '%')) |
| { |
| if (outbuf[1]=='#') |
| ++outbuf; /* not documented by python, */ |
| if (outbuf[1]=='\0' || |
| !strchr("aAbBcdfHIjmMpSUwWxXyYzZ%", outbuf[1])) |
| { |
| PyErr_SetString(PyExc_ValueError, "Invalid format string"); |
| return 0; |
| } |
| } |
| #endif |
| |
| fmtlen = 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 = (char *)PyMem_Malloc(i); |
| if (outbuf == NULL) { |
| Py_DECREF(format); |
| return PyErr_NoMemory(); |
| } |
| buflen = strftime(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; |
| ret = PyUnicode_Decode(outbuf, buflen, |
| TZNAME_ENCODING, NULL); |
| 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 |
| } |
| } |
| |
| 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; |
| } |