From 5dbf6bee57f539af588779639286fa456e066614 Mon Sep 17 00:00:00 2001
From: krasi <krasilnikoff.tihon@gmail.com>
Date: Sat, 11 May 2024 19:23:42 +0300
Subject: [PATCH] write script to compress folder and files

---
 README.md   |  5 ++++-
 zipfiles.py | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 56 insertions(+), 1 deletion(-)
 create mode 100644 zipfiles.py

diff --git a/README.md b/README.md
index 9124d4a..77b92e3 100644
--- a/README.md
+++ b/README.md
@@ -1,2 +1,5 @@
-# Write_script_to_compress_folder_and_files
+# write script to compress folder and files
 
+python zipfiles.py file_name(или имя папки)
+пример: Python zipfiles.py test.txt python zipfiles.py ./test (папка)
+Сжатый файл("filename.zip") будет сгенерирован после запуска программы
\ No newline at end of file
diff --git a/zipfiles.py b/zipfiles.py
new file mode 100644
index 0000000..dda3242
--- /dev/null
+++ b/zipfiles.py
@@ -0,0 +1,52 @@
+import zipfile
+import sys
+import os
+
+
+# compress file function
+def zip_file(file_path):
+    compress_file = zipfile.ZipFile(file_path + '.zip', 'w')
+    compress_file.write(path, compress_type=zipfile.ZIP_DEFLATED)
+    compress_file.close()
+
+
+# Declare the function to return all file paths of the particular directory
+def retrieve_file_paths(dir_name):
+    # setup file paths variable
+    file_paths = []
+
+    # Read all directory, subdirectories and file lists
+    for root, directories, files in os.walk(dir_name):
+        for filename in files:
+            # Create the full file path by using os module.
+            file_path = os.path.join(root, filename)
+            file_paths.append(file_path)
+
+    # return all paths
+    return file_paths
+
+
+def zip_dir(dir_path, file_paths):
+    # write files and folders to a zipfile
+    compress_dir = zipfile.ZipFile(dir_path + '.zip', 'w')
+    with compress_dir:
+        # write each file separately
+        for file in file_paths:
+            compress_dir.write(file)
+
+
+if __name__ == "__main__":
+    path = sys.argv[1]
+
+    if os.path.isdir(path):
+        files_path = retrieve_file_paths(path)
+        # print the list of files to be zipped
+        print('The following list of files will be zipped:')
+        for file_name in files_path:
+            print(file_name)
+        zip_dir(path, files_path)
+    elif os.path.isfile(path):
+        print('The %s will be zipped:' % path)
+        zip_file(path)
+    else:
+        print('a special file(socket,FIFO,device file), please input file or dir')
\ No newline at end of file