tor

commit b4b8fa4899fcde9983f66a6310878ea47186e5eb
parent 6da8c6e9a544922c3e04ac460ca7fcbdc8a6eaf3
Author: Nick Mathewson <nickm@torproject.org>
Date:   Tue, 28 Aug 2018 16:25:15 -0400

Set the file encoding in checkIncludes.py with Python3

Diffstat:
M
scripts/maint/checkIncludes.py
 | 
11
+++++++++--

1 file changed, 9 insertions(+), 2 deletions(-)
diff --git a/scripts/maint/checkIncludes.py b/scripts/maint/checkIncludes.py
@@ -26,6 +26,13 @@ import sys
 # Global: Have there been any errors?
 trouble = False
 
+if sys.version_info[0] <= 2:
+    def open_file(fname):
+        return open(fname, 'r')
+else:
+    def open_file(fname):
+        return open(fname, 'r', encoding='utf-8')
+
 def err(msg):
     """ Declare that an error has happened, and remember that there has
         been an error. """
@@ -70,7 +77,7 @@ class Rules(object):
                         include, lineno, context))
 
     def applyToFile(self, fname):
-        with open(fname, 'r') as f:
+        with open_file(fname) as f:
             #print(fname)
             self.applyToLines(iter(f), " of {}".format(fname))
 
@@ -82,7 +89,7 @@ class Rules(object):
 def load_include_rules(fname):
     """ Read a rules file from 'fname', and return it as a Rules object. """
     result = Rules(os.path.split(fname)[0])
-    with open(fname, 'r') as f:
+    with open_file(fname) as f:
         for line in f:
             line = line.strip()
             if line.startswith("#") or not line: