From 27f58a0ede2e3ef7a1c55ffe466a2a5ab73f9d50 Mon Sep 17 00:00:00 2001 From: Dan Puttick Date: Fri, 17 Mar 2017 15:14:18 -0400 Subject: [PATCH] Put log dir creation code in separate method --- kittengroomer/helpers.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/kittengroomer/helpers.py b/kittengroomer/helpers.py index 318f1ca..5feadea 100644 --- a/kittengroomer/helpers.py +++ b/kittengroomer/helpers.py @@ -272,10 +272,7 @@ class GroomerLogger(object): def __init__(self, root_dir_path, debug=False): self.root_dir = root_dir_path - self.log_dir_path = os.path.join(root_dir_path, 'logs') - if os.path.exists(self.log_dir_path): - shutil.rmtree(self.log_dir_path) - os.makedirs(self.log_dir_path) + self.log_dir_path = self._make_log_dir(root_dir_path) self.log_processing = os.path.join(self.log_dir_path, 'processing.log') self.log_content = os.path.join(self.log_dir_path, 'content.log') twiggy.quick_setup(file=self.log_processing) @@ -287,6 +284,13 @@ class GroomerLogger(object): self.log_debug_err = os.devnull self.log_debug_out = os.devnull + def _make_log_dir(self, root_dir_path): + log_dir_path = os.path.join(root_dir_path, 'logs') + if os.path.exists(log_dir_path): + shutil.rmtree(log_dir_path) + os.makedirs(log_dir_path) + return log_dir_path + def tree(self, base_dir, padding=' '): """Write a graphical tree to the log for `base_dir`.""" with open(self.log_content, 'ab') as lf: