fix python 2 & 3 compat

b4bab658 · floriandeboissieu · 424f49fd · b4bab658
Commit b4bab658 authored Nov 12, 2019 by floriandeboissieu
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 2 deletions

generateDS.py generateDS.py +8 -2

No files found.
--- a/generateDS.py
+++ b/generateDS.py
@@ -6999,11 +6999,17 @@ def makeFile(outFileName):
            reply = input(
                'File %s exists.  Overwrite? (y/n): ' % outFileName)
            if reply == 'y':
-                outFile = open(outFileName, 'w', encoding="utf-8")
+                if sys.version_info[0] == 2:
+                    outFile = open(outFileName, 'w')
+                else:
+                    outFile = open(outFileName, 'w', encoding="utf-8")
            else:
                sys.exit('Exiting.  No output file.')
    else:
-        outFile = open(outFileName, 'w', encoding="utf-8")
+        if sys.version_info[0] == 2:
+            outFile = open(outFileName, 'w')
+        else:
+            outFile = open(outFileName, 'w', encoding="utf-8")
    return outFile