Merge pull request #113 from adityacp/file_based_questions

File based questions

1 files changed, 33 insertions, 0 deletions

diff --git a/yaksh/file_utils.py b/yaksh/file_utils.py
new file mode 100644
index 0000000..8f6f6e5
--- /dev/null
+++ b/yaksh/file_utils.py
@@ -0,0 +1,33 @@
+import shutil
+import os
+import zipfile
+
+
+def copy_files(file_paths):
+    """ Copy Files to current directory, takes
+    tuple with file paths and extract status"""
+
+    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 and zipfile.is_zipfile(file_name):
+            unzip = zipfile.ZipFile(file_name)
+            for zip_files in unzip.namelist():
+                files.append(zip_files)
+            unzip.extractall()
+            unzip.close()
+    return files
+
+
+def delete_files(files):
+    """ Delete Files from current directory """
+
+    for file in files:
+        if os.path.exists(file):
+            if os.path.isfile(file):
+                os.remove(file)
+            else:
+                shutil.rmtree(file)