file_organizer_copy.py

import os
import shutil

def file_organizer(source, image_destination, doc_destination):
  # Create directories if it doesn't exist
  if image_destination is not None:
    os.makedirs(image_destination, exist_ok=True)
  if doc_destination is not None:
    os.makedirs(doc_destination, exist_ok=True)


  for file in os.listdir(source):
    
    # Skip directories
    if os.path.isdir(os.path.join(source, file)):
      continue

    if file.endswith(".jpg") and image_destination is not None:
      shutil.copy(os.path.join(source, file), image_destination)
    elif file.endswith(".pdf") and doc_destination is not None:
      shutil.copy(os.path.join(source, file), doc_destination)
  
  print("Files successfully arranged✅")

  

file_organizer("C:\\Users\\user\\Desktop\\file organizer", None, "C:\\Users\\user\\Downloads\\Favour")