Adds headers and footers to log files

1 files changed, 9 insertions, 0 deletions

diff --git a/base/util.py b/base/util.py
index d4a6475..ec72594 100644
--- a/base/util.py
+++ b/base/util.py
@@ -19,6 +19,7 @@
 from __future__ import with_statement
 
 from contextlib import contextmanager
+from datetime import datetime
 from os import chdir, getcwd, remove
 from os import symlink as sym
 from os.path import abspath, lexists, isdir, islink, isfile
@@ -37,6 +38,14 @@ def pwd(dir):
     log.debug('changing dir to %s' % old_dir)
     chdir(old_dir)
 
+@contextmanager
+def log_file(fname):
+    with file(fname, 'a') as fout:
+        fout.write("-- Beginning log of %s at %s --\n" % (fname, datetime.now().isoformat(' ')))
+        fout.flush()
+        yield fout
+        fout.write("-- Ending log of %s at %s --\n" % (fname, datetime.now().isoformat(' ')))
+
 def rm(tgt):
     if isdir(tgt):
         rmtree(tgt)