fix compile error

This commit is contained in:
rdb 2013-02-17 22:53:48 +00:00
parent 5846d8d8c4
commit eb6eeb6d02

View File

@ -277,8 +277,12 @@ ns_get_environment_variable(const string &var) const {
if (item != NULL) {
if (PyUnicode_Check(item)) {
Py_ssize_t size = PyUnicode_GetSize(item);
wchar_t *data = new wchar_t[size];
wchar_t *data = new wchar_t[size + 1];
#if PY_MAJOR_VERSION >= 3
if (PyUnicode_AsWideChar(item, data, size) != -1) {
#else
if (PyUnicode_AsWideChar((PyUnicodeObject*) item, data, size) != -1) {
#endif
wstring wstr (data, size);
main_dir = Filename::from_os_specific_w(wstr);
}