import itertools

from files_utils.files_save import CopySingleFragmentFiles, CopyMultiFragmentFiles
from files_utils.files_sort import GetSortFragments
from files_utils.folders_sort import ClassifyFilesAndFolders, ScanMultiFolders

fragment_lists = {}
target_path = r"Z:\test_files"
mix_test_data = [
    "test-copy"
]
classify_files_and_folders = ClassifyFilesAndFolders(mix_test_data)
files_list = classify_files_and_folders["files"]
folders_files_list = ScanMultiFolders(classify_files_and_folders["folders"])
merged_list = list(itertools.chain(files_list, folders_files_list))

sort_fragments = GetSortFragments(db_path="./src/db_ntfs_info.db", files_list=merged_list)
for item in sort_fragments:
    extent_count = item['extent_count']
    if extent_count == 1:
        CopySingleFragmentFiles(item, target_path=target_path)
    elif extent_count > 1:
        CopyMultiFragmentFiles(item, fragment_lists=fragment_lists, target_path=target_path)