RF: refactor fix_longtable utility for codec

matthew-brett · matthew-brett · commit caf7356a6efe · 2017-03-13T16:03:43.000-07:00
Specify utf8 codec for tex file processing.
diff --git a/tools/fix_longtable.py b/tools/fix_longtable.py
@@ -3,6 +3,7 @@
 """
 import re
 import sys
+import codecs
 
 lt_LL = re.compile(
     r"longtable}{(L+)}")
@@ -16,8 +17,9 @@ def replacer(match):
     raise RuntimeError("Enter path to tex file only")
 file_path = sys.argv[1]
 
-unfixed_tex = open(file_path,'rt').readlines()
-write_file = open(file_path, 'wt')
-for line in unfixed_tex:
-    line = lt_LL.sub(replacer, line, 1)
-    write_file.write(line)
+with codecs.open(file_path, 'r', encoding='utf8') as fobj:
+    unfixed_tex = fobj.readlines()
+with codecs.open(file_path, 'w', encoding='utf8') as fobj:
+    for line in unfixed_tex:
+        line = lt_LL.sub(replacer, line, 1)
+        fobj.write(line)
