logging fixes (#817)

Co-authored-by: Joshua Davis <[email protected]>

hammer/logging/logging.py

@@ -5,6 +5,7 @@
 from functools import reduce
 from enum import Enum
 from typing import Callable, Iterable, List, NamedTuple, Type, Optional
+import atexit 
 
 class Level(Enum):
     """
@@ -38,7 +39,6 @@ def with_default_callbacks(cls):
     cls.add_callback(cls.callback_buffering)
     return cls
 
-
 class HammerVLSIFileLogger:
     """A file logger for HammerVLSILogging."""
 
@@ -51,6 +51,8 @@ def __init__(self, output_path: str, format_msg_callback: Optional[Callable[[Ful
         """
         self._file = open(output_path, "a")
         self._format_msg_callback = format_msg_callback
+        # close file when code exits 
+        atexit.register(self.close)
 
     def __enter__(self):
         return self
@@ -59,7 +61,8 @@ def close(self) -> None:
         """
         Close this file logger.
         """
-        self._file.close()
+        if not self._file.closed:
+            self._file.close()
 
     def __exit__(self, exc_type, exc_value, traceback):
         self.close()

hammer/vlsi/driver.py

@@ -65,12 +65,11 @@ def __init__(self, options: HammerDriverOptions, extra_project_config: dict = {}
         file_logger = HammerVLSIFileLogger(options.log_file)
         HammerVLSILogging.add_callback(file_logger.callback)
         self.log = HammerVLSILogging.context()  # type: HammerVLSILoggingContext
-
+        
         # Create a new hammer database.
         self.database = hammer_config.HammerDatabase()  # type: hammer_config.HammerDatabase
 
         self.log.info("Loading hammer-vlsi libraries and reading settings")
-
         # Store the run dir (this should already be canonicalized by the CLI driver).
         self.obj_dir = options.obj_dir  # type: str