from files_utils.files_save import CopySingleFragmentFiles, CopyMultiFragmentFiles
from files_utils.files_sort import GetSortFragments

fragment_lists = {}
test_files = [
    r"CloudMusic\AGA - MIZU.mp3",
    r"CloudMusic\AGA - 一.mp3",
    r"CloudMusic\Aaron Zigman - Main Title.mp3",
    r"CloudMusic\Anson Seabra - Keep Your Head Up Princess.mp3",
    r"CloudMusic\Anthony Keyrouz,Romy Wave - Something Just Like This (feat. Romy Wave).mp3",
    r"CloudMusic\Ava Max - Sweet but Psycho.mp3",
    r"CloudMusic\Cecilia Cheung - Turn Into Fireworks and Fall for You.mp3",
    r"CloudMusic\Color Music Choir - Something Just Like This (Live).mp3"
]

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