45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
import os
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
from src.scan_folder import recursive_scan_folder
|
|
|
|
TEST_FOLDER = ".test_folder"
|
|
TEST_IMAGES = f"tests{os.sep}test_files"
|
|
TEST_TEMP_FOLDER = os.path.join(TEST_FOLDER, 'Temp')
|
|
TEST_IMAGE_FOLDER = os.path.join(TEST_FOLDER, 'Images')
|
|
|
|
|
|
def create_file(file):
|
|
Path(os.path.join(TEST_FOLDER, file)).touch()
|
|
|
|
|
|
def delete_folder():
|
|
shutil.rmtree(TEST_FOLDER, ignore_errors=True)
|
|
|
|
|
|
def create_folders():
|
|
delete_folder()
|
|
os.makedirs(os.path.join(TEST_FOLDER, '001', '001'))
|
|
os.makedirs(os.path.join(TEST_FOLDER, '002', '001'))
|
|
|
|
|
|
def copy_test_images():
|
|
delete_folder()
|
|
os.makedirs(TEST_TEMP_FOLDER)
|
|
os.makedirs(TEST_IMAGE_FOLDER)
|
|
shutil.copy(src=os.path.join(TEST_IMAGES, 'test_image_001.JPG'), dst=TEST_TEMP_FOLDER)
|
|
shutil.copy(src=os.path.join(TEST_IMAGES, 'test_image_002.JPG'), dst=TEST_TEMP_FOLDER)
|
|
|
|
|
|
def copy_images(brand:str, model:str):
|
|
delete_folder()
|
|
create_folders()
|
|
files = recursive_scan_folder(path=TEST_IMAGES)
|
|
for file in files:
|
|
file_name = file.split(os.sep)[2:][0]
|
|
file_name = file_name.split("_")
|
|
if file_name[0] == brand and file_name[1] == model:
|
|
shutil.copy(src=file, dst=TEST_FOLDER)
|
|
|