close duplicated file discriptor in pygraphillion

takemaru · Feb 22, 2017 · 91acc36 · 91acc36
1 parent b26929a
commit 91acc36
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/pygraphillion.cc b/src/pygraphillion.cc
@@ -817,7 +817,7 @@ static PyObject* setset_dump(PySetsetObject* self, PyObject* obj) {
   self->ss->dump(fp);
   Py_END_ALLOW_THREADS;
 #if IS_PY3 == 1
-  fflush(fp);
+  fclose(fp);
 #else
   PyFile_DecUseCount(file);
 #endif
@@ -846,7 +846,9 @@ static PyObject* setset_load(PySetsetObject* self, PyObject* obj) {
       PySetset_Type.tp_alloc(&PySetset_Type, 0));
   ret->ss = new setset(setset::load(fp));
   Py_END_ALLOW_THREADS;
-#if IS_PY3 == 0
+#if IS_PY3 == 1
+  fclose(fp);
+#else
   PyFile_DecUseCount(file);
 #endif
   return reinterpret_cast<PyObject*>(ret);