From 59cde8cfd5de1b5b097fd6c30a06e64d752cfbc0 Mon Sep 17 00:00:00 2001 From: Dan Puttick Date: Thu, 9 Mar 2017 13:48:07 -0500 Subject: [PATCH] Move safe_copy to FileBase --- bin/filecheck.py | 2 +- kittengroomer/helpers.py | 30 ++++++++++++++---------------- 2 files changed, 15 insertions(+), 17 deletions(-) diff --git a/bin/filecheck.py b/bin/filecheck.py index acc160d..157c83f 100644 --- a/bin/filecheck.py +++ b/bin/filecheck.py @@ -499,7 +499,7 @@ class KittenGroomerFileCheck(KittenGroomerBase): if file.is_recursive: self.process_archive(file) elif file.should_copy: - self.safe_copy(file.src_path, file.dst_path) + file.safe_copy() file.set_property('copied', True) file.write_log() if hasattr(file, 'tempdir_path'): diff --git a/kittengroomer/helpers.py b/kittengroomer/helpers.py index cf6309f..f422be5 100644 --- a/kittengroomer/helpers.py +++ b/kittengroomer/helpers.py @@ -201,6 +201,20 @@ class FileBase(object): path, filename = os.path.split(self.dst_path) self.dst_path = os.path.join(path, '{}.bin'.format(filename)) + def safe_copy(self, src=None, dst=None): + """Copy a file and create directory if needed.""" + if src is None: + src = self.src_path + if dst is None: + dst = self.dst_path + try: + dst_path, filename = os.path.split(dst) + if not os.path.exists(dst_path): + os.makedirs(dst_path) + shutil.copy(src, dst) + except Exception as e: + self.add_error(e, '') + def force_ext(self, ext): """If dst_path does not end in ext, changes it and edits _file_props.""" if not self.dst_path.endswith(ext): @@ -281,9 +295,6 @@ class GroomerLogger(object): # return a sublog for the file pass - def add_event(self, event, log_level): - pass - class KittenGroomerBase(object): """Base object responsible for copy/sanitization process.""" @@ -311,19 +322,6 @@ class KittenGroomerBase(object): if not os.path.exists(directory): os.makedirs(directory) - def safe_copy(self, src=None, dst=None): - """Copy a file and create directory if needed.""" - if src is None: - src = self.cur_file.src_path - if dst is None: - dst = self.cur_file.dst_path - try: - dst_path, filename = os.path.split(dst) - self.safe_mkdir(dst_path) - shutil.copy(src, dst) - except Exception as e: - self.add_error(e, '') - def list_all_files(self, directory): """Generator yielding path to all of the files in a directory tree.""" for root, dirs, files in os.walk(directory):