added directory size tutorial

Author: x4nth055

README.md

@@ -56,6 +56,7 @@ This is a repository of all the tutorials of [The Python Code](https://www.thepy
     - [How to Automate your VPS or Dedicated Server Management in Python](https://www.thepythoncode.com/article/automate-veesp-server-management-in-python). ([code](general/automating-server-management))
     - [How to Execute BASH Commands in a Remote Machine in Python](https://www.thepythoncode.com/article/executing-bash-commands-remotely-in-python). ([code](general/execute-ssh-commands))
     - [How to Convert Python Files into Executables](https://www.thepythoncode.com/article/building-python-files-into-stand-alone-executables-using-pyinstaller)
+    - [How to Get the Size of Directories in Python](https://www.thepythoncode.com/article/get-directory-size-in-bytes-using-python). ([code](general/calculate-directory-size))
 
 - ### [Web Scraping](https://www.thepythoncode.com/topic/web-scraping)
     - [How to Access Wikipedia in Python](https://www.thepythoncode.com/article/access-wikipedia-python). ([code](web-scraping/wikipedia-extractor))

general/calculate-directory-size/README.md

@@ -0,0 +1,8 @@
+# [How to Get the Size of Directories in Python](https://www.thepythoncode.com/article/get-directory-size-in-bytes-using-python)
+To run this:
+- `pip3 install -r requirements.txt`
+- To get the size of subdirectories of `/root/home` directory:
+    ```
+    python get_directory_size.py /root/home
+    ```
+    This will show a pie chart that indicates the size of each subdirectory.

general/calculate-directory-size/get_directory_size.py

@@ -0,0 +1,67 @@
+import os
+import matplotlib.pyplot as plt
+
+
+def get_size_format(b, factor=1024, suffix="B"):
+    """
+    Scale bytes to its proper byte format
+    e.g:
+        1253656 => '1.20MB'
+        1253656678 => '1.17GB'
+    """
+    for unit in ["", "K", "M", "G", "T", "P", "E", "Z"]:
+        if b < factor:
+            return f"{b:.2f}{unit}{suffix}"
+        b /= factor
+    return f"{b:.2f}Y{suffix}"
+
+
+def get_directory_size(directory):
+    """Returns the `directory` size in bytes."""
+    total = 0
+    try:
+        # print("[+] Getting the size of", directory)
+        for entry in os.scandir(directory):
+            if entry.is_file():
+                # if it's a file, use stat() function
+                total += entry.stat().st_size
+            elif entry.is_dir():
+                # if it's a directory, recursively call this function
+                total += get_directory_size(entry.path)
+    except NotADirectoryError:
+        # if `directory` isn't a directory, get the file size then
+        return os.path.getsize(directory)
+    except PermissionError:
+        # if for whatever reason we can't open the folder, return 0
+        return 0
+    return total
+
+
+def plot_pie(sizes, names):
+    """Plots a pie where `sizes` is the wedge sizes and `names` """
+    plt.pie(sizes, labels=names, autopct=lambda pct: f"{pct:.2f}%")
+    plt.title("Different Sub-directory sizes in bytes")
+    plt.show()
+
+
+if __name__ == "__main__":
+    import sys
+    folder_path = sys.argv[1]
+
+    directory_sizes = []
+    names = []
+    # iterate over all the directories inside this path
+    for directory in os.listdir(folder_path):
+        directory = os.path.join(folder_path, directory)
+        # get the size of this directory (folder)
+        directory_size = get_directory_size(directory)
+        if directory_size == 0:
+            continue
+        directory_sizes.append(directory_size)
+        names.append(os.path.basename(directory) + ": " + get_size_format(directory_size))
+
+    print("[+] Total directory size:", get_size_format(sum(directory_sizes)))
+    plot_pie(directory_sizes, names)
+    
+
+

general/calculate-directory-size/requirements.txt

@@ -0,0 +1 @@
+matplotlib