created file copier to copy files

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)