Add create folders function

This commit is contained in:
Alireza Adli 2024-03-26 19:22:32 -04:00
parent ecc668f719
commit 7bbd538d2f

View File

@ -12,3 +12,22 @@ def find_shp_files(root_folder):
shp_files.append(new_file_name)
return shp_files
def create_folders(directory, num_folders):
"""
Create a specified number of folders in the given directory.
Args:
- directory (str): The directory where folders will be created.
- num_folders (int): The number of folders to create.
"""
# Check if the directory exists, if not, create it
if not os.path.exists(directory):
os.makedirs(directory)
# Create folders
for i in range(num_folders):
folder_name = f"layer_{i}"
folder_path = os.path.join(directory, folder_name)
os.makedirs(folder_path)
print(f"Created folder: {folder_path}")