From bc94aad72d065dcc96ea1d8410d7931e778bb407 Mon Sep 17 00:00:00 2001
From: admin <admin>
Date: Sun, 19 May 2024 15:16:29 -0400
Subject: [PATCH] added get_subdirectories function

---
 CountFolderFiles.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/CountFolderFiles.py b/CountFolderFiles.py
index d38a95c..a4da81c 100644
--- a/CountFolderFiles.py
+++ b/CountFolderFiles.py
@@ -15,6 +15,13 @@ def append_to_folder_name(folder_path, suffix):
     except FileNotFoundError:
         print(f"Error: Folder '{folder_path}' not found.")
 
+def get_subdirectories(folder_path):
+    subdirectories = []
+    for root, dirs, files in os.walk(folder_path):
+        for dir_name in dirs:
+            subdirectories.append(os.path.join(root, dir_name))
+    return subdirectories
+
 def count_subdirectories(directory):
     subdirectory_count = 0
     for _, dirnames, _ in os.walk(directory):
@@ -24,8 +31,8 @@ def count_subdirectories(directory):
 def count_folder_files(folder_path):
     try:
         # If no subfolders exist, return the number of files in the folder
-        subdirectories = count_subdirectories(folder_path)
-        if subdirectories == 0:
+        num_subdirectories = count_subdirectories(folder_path)
+        if num_subdirectories == 0:
         
             # Count the number of files in the folder
             file_count = sum(1 for entry in os.scandir(folder_path) if entry.is_file())
@@ -33,7 +40,8 @@ def count_folder_files(folder_path):
             return file_count
         # If subdirectories do exist, return the number of files in each subdirectory then this directory
         else:
-            print(">0")
+            subdirectories = get_subdirectories(folder_path)
+            print(subdirectories)
             
             
     except FileNotFoundError: