Merge pull request #2421 from atom-community/python-encoding

fix: set utf-8 encoding for Python
atom-community · May 11, 2021 · de01edd · de01edd
2 parents a672214 + eb4700d
commit de01edd
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/lib/grammars/python.js b/lib/grammars/python.js
@@ -2,10 +2,20 @@
 
 import GrammarUtils from "../grammar-utils"
 
+// https://github.com/atom-community/atom-script/issues/214#issuecomment-418766763
+let encodingSet = false
+function setEncoding() {
+  if (!encodingSet) {
+    process.env.PYTHONIOENCODING = "utf-8"
+    encodingSet = true
+  }
+}
+
 export const Python = {
   "Selection Based": {
     command: "python",
     args(context) {
+      setEncoding()
       const code = context.getCode()
       const tmpFile = GrammarUtils.createTempFileWithCode(code)
       return ["-u", tmpFile]
@@ -15,6 +25,7 @@ export const Python = {
   "File Based": {
     command: "python",
     args({ filepath }) {
+      setEncoding()
       return ["-u", filepath]
     },
   },
@@ -26,6 +37,7 @@ export const Sage = {
   "Selection Based": {
     command: "sage",
     args(context) {
+      setEncoding()
       const code = context.getCode()
       const tmpFile = GrammarUtils.createTempFileWithCode(code)
       return [tmpFile]
@@ -35,6 +47,7 @@ export const Sage = {
   "File Based": {
     command: "sage",
     args({ filepath }) {
+      setEncoding()
       return [filepath]
     },
   },

diff --git a/spec/fixtures/issue_1166.py b/spec/fixtures/issue_1166.py
@@ -0,0 +1 @@
+print ('汉语/漢語')