up/ОАиП/2024-25/26гр/Задача_1759618/find_and_copy.py

"""
задача: есть папка с файлами *.jpg, необюходимо найти файлы в другой папке с таким же названием, но с расширением .arw и скопировать их в другую папку. 
"""

import os, shutil
import random

# директория, в которой находится этот файл
BASE_DIR = os.path.dirname(__file__)
print(BASE_DIR)

# создаем папку с .jpg файлами
jpg_dir = os.path.join(BASE_DIR, "jpg")
os.makedirs(jpg_dir, exist_ok=True)

# создаем папку с .arw файлами
arw_dir = os.path.join(BASE_DIR, "arw")
os.makedirs(arw_dir, exist_ok=True)

# создаем папку для совпадающих файлов
copy_dir = os.path.join(BASE_DIR, "copy")
os.makedirs(copy_dir, exist_ok=True)

# создадим несколько файлов jpg и arw
for i in range(0,6):
    random_filename = str(random.randint(1,10))
    random_jpg_path = os.path.join(jpg_dir, f"{random_filename}.jpg")
    f = open(random_jpg_path, "w")
    f.close()
    random_filename = str(random.randint(1,10))
    random_arw_path = os.path.join(arw_dir, f"{random_filename}.arw")
    f = open(random_arw_path, "w")
    f.close()



# Получаем список jpg-файлов
jpg_files = os.listdir(jpg_dir)
print(jpg_files)

# Получаем список arw-файлов
arw_files = os.listdir(arw_dir)
print(arw_files)

for jpg_file in jpg_files:
    jpg_basename, ext = os.path.splitext(jpg_file)
    for arw_file in arw_files:
        arw_basename, ext = os.path.splitext(arw_file)

        if jpg_basename == arw_basename:
            # куда копировать
            copy_path = os.path.join(copy_dir, f"{arw_basename}.arw")
            # что копировать
            arw_path = os.path.join(arw_dir, f"{arw_basename}.arw")

            shutil.copy(arw_path, copy_path)

print("success")