X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=file_writer.py;h=9f31945f2b14e8ce17170178aeac6c1185efa1e3;hb=bfde32ea8f021da27fb2cdf535efb0e9c465d6a2;hp=fced44939cddc73681d6008f94ee589caf97a8e2;hpb=5c39d86ebc075ccb7be98b1dfab8040b72ff9134;p=kiosk.git

diff --git a/file_writer.py b/file_writer.py
index fced449..9f31945 100644
--- a/file_writer.py
+++ b/file_writer.py
@@ -1,20 +1,26 @@
 #!/usr/bin/env python3
 
-import constants
+import logging
 import os
 from uuid import uuid4
 
+import kiosk_constants
+
+
+logger = logging.getLogger(__file__)
+
 
 class file_writer:
     """Helper context to write a pages file."""
 
     def __init__(self, filename: str, *, transformations=[]):
         temp = "temp-" + str(uuid4())
-        self.temp_filename = os.path.join(constants.pages_dir, temp)
-        self.full_filename = os.path.join(constants.pages_dir, filename)
+        self.temp_filename = os.path.join(kiosk_constants.pages_dir, temp)
+        self.full_filename = os.path.join(kiosk_constants.pages_dir, filename)
         self.xforms = [file_writer.remove_tricky_unicode]
         self.xforms.extend(transformations)
         self.f = None
+        logger.info(f"Writing {self.temp_filename}...")
 
     @staticmethod
     def remove_tricky_unicode(x: str) -> str:
@@ -37,6 +43,7 @@ class file_writer:
 
     def __exit__(self, exc_type, exc_value, exc_traceback):
         self.close()
+        logger.info(f"Copying {self.temp_filename} to {self.full_filename}")
         cmd = f'/bin/mv -f {self.temp_filename} "{self.full_filename}"'
         os.system(cmd)
         print(cmd)
@@ -49,8 +56,8 @@ class file_writer:
 
 
 # Test
-#def toupper(x):
+# def toupper(x):
 #   return x.upper()
 
-#with file_writer("test", transformations=[toupper]) as fw:
+# with file_writer("test", transformations=[toupper]) as fw:
 #   fw.write(u"Another test!!")