diff options
Diffstat (limited to 'sugar/logger.py')
| -rw-r--r-- | sugar/logger.py | 16 |
1 files changed, 12 insertions, 4 deletions
diff --git a/sugar/logger.py b/sugar/logger.py index 284b52c..56d66c6 100644 --- a/sugar/logger.py +++ b/sugar/logger.py @@ -29,10 +29,7 @@ class LogWriter: def __init__(self, module_id): self._module_id = module_id - logs_dir = os.path.join(env.get_profile_path(), 'logs') - if not os.path.isdir(logs_dir): - os.makedirs(logs_dir) - + logs_dir = _get_logs_dir() log_path = os.path.join(logs_dir, module_id + '.log') self._log_file = open(log_path, 'w') @@ -68,6 +65,12 @@ def __exception_handler(typ, exc, tb): _log_writer.write(logging.ERROR, trace.getvalue()) +def _get_logs_dir(): + logs_dir = os.path.join(env.get_profile_path(), 'logs') + if not os.path.isdir(logs_dir): + os.makedirs(logs_dir) + return logs_dir + def start(module_id): log_writer = LogWriter(module_id) @@ -78,3 +81,8 @@ def start(module_id): global _log_writer _log_writer = log_writer sys.excepthook = __exception_handler + +def cleanup(): + logs_dir = _get_logs_dir() + for f in os.listdir(logs_dir): + os.remove(os.path.join(logs_dir, f)) |