Fix problem when exec'ing a string with a coding

skrah · Aug 11, 2007 · f7f28fc · f7f28fc
1 parent 7ac9712
commit f7f28fc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/Lib/test/test_coding.py b/Lib/test/test_coding.py
@@ -21,6 +21,11 @@ def verify_bad_module(self, module_name):
         fp.close()
         self.assertRaises(SyntaxError, compile, text, filename, 'exec')
 
+    def test_exec_valid_coding(self):
+        d = {}
+        exec('# coding: cp949\na = 5\n', d)
+        self.assertEqual(d['a'], 5)
+
 def test_main():
     test.test_support.run_unittest(CodingTest)
 

diff --git a/Parser/tokenizer.c b/Parser/tokenizer.c
@@ -632,7 +632,7 @@ decode_str(const char *str, struct tok_state *tok)
 				"unknown encoding: %s", tok->enc);
 			return error_ret(tok);
 		}
-		str = PyString_AsString(utf8);
+		str = PyBytes_AsString(utf8);
 	}
 	assert(tok->decoding_buffer == NULL);
 	tok->decoding_buffer = utf8; /* CAUTION */