import os
import shutil
from pathlib import Path

TEST_FOLDER = ".test_folder"
TEST_IMAGES = "tests/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)