diff options
| -rw-r--r-- | yaksh/copy_delete_files.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/yaksh/copy_delete_files.py b/yaksh/copy_delete_files.py new file mode 100644 index 0000000..f34b944 --- /dev/null +++ b/yaksh/copy_delete_files.py @@ -0,0 +1,29 @@ +import shutil +import os +import zipfile + + +class CopyDeleteFiles(object): + + def copy_files(self, file_paths): + files = [] + for src in file_paths: + file_path, extract = src + file_name = os.path.basename(file_path) + files.append(file_name) + shutil.copy(file_path, os.getcwd()) + if extract: + unzip = zipfile.ZipFile(file_name) + for zip_files in unzip.namelist(): + files.append(zip_files) + unzip.extractall() + unzip.close() + return files + + def delete_files(self, files): + for content in files: + if os.path.exists(content): + if os.path.isfile(content): + os.remove(content) + else: + shutil.rmtree(content) |