mtl_gis_oo/basic_functions.py

import os
import glob
import time


def find_shp_files(root_folder):
  shp_files = []
  # Sort folders alphabetically
  for foldername, _, _ in sorted(os.walk(root_folder)):
    for filename in sorted(glob.glob(os.path.join(foldername, '*.shp'))):
      new_file_name = filename.replace('\\', r'/')
      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}")