Format codebase w/ black.

[kiosk.git] / file_writer.py

diff --git a/file_writer.py b/file_writer.py
index aba7d8f25f5b0efda0df5d5ef55548af0aeb0c26..988d0a03ac49378c06079b6901cb29dc11fa5633 100644 (file)
--- a/file_writer.py
+++ b/file_writer.py
@@ -1,22 +1,23 @@
 import constants
 import os
 
+
 def remove_tricky_unicode(x):
     try:
-        x = x.decode('utf-8')
-        x = x.replace(u"\u2018", "'").replace(u"\u2019", "'")
-        x = x.replace(u"\u201c", '"').replace(u"\u201d", '"')
-        x = x.replace(u"\u2e3a", "-").replace(u"\u2014", "-")
+        x = x.decode("utf-8")
+        x = x.replace("\u2018", "'").replace("\u2019", "'")
+        x = x.replace("\u201c", '"').replace("\u201d", '"')
+        x = x.replace("\u2e3a", "-").replace("\u2014", "-")
     except:
         pass
     return x
 
+
 class file_writer:
     def __init__(self, filename):
-        self.full_filename = os.path.join(constants.pages_dir,
-                                          filename)
-        self.f = open(self.full_filename, 'w')
-        self.xforms = [ remove_tricky_unicode ]
+        self.full_filename = os.path.join(constants.pages_dir, filename)
+        self.f = open(self.full_filename, "wb")
+        self.xforms = [remove_tricky_unicode]
 
     def add_xform(self, xform):
         self.xforms.append(xform)
@@ -24,7 +25,7 @@ class file_writer:
     def write(self, data):
         for xform in self.xforms:
             data = xform(data)
-        self.f.write(data.encode('utf-8'))
+        self.f.write(data.encode("utf-8"))
 
     def done(self):
         self.f.close()
@@ -32,11 +33,12 @@ class file_writer:
     def close(self):
         self.done()
 
+
 # Test
-#def toupper(x):
+# def toupper(x):
 #    return x.upper()
 #
-#fw = file_writer("test")
-#fw.add_xform(toupper)
-#fw.write(u"This is a \u201ctest\u201d. \n")
-#fw.done()
+# fw = file_writer("test")
+# fw.add_xform(toupper)
+# fw.write(u"This is a \u201ctest\u201d. \n")
+# fw.done()