almost finish

fake_main.py

@@ -1,19 +1,28 @@
+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 = {}
-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"
+mix_test_data = [
+    "CloudMusic\\AGA - MIZU.mp3",
+    "CloudMusic/AGA - 一.mp3",
+    "CloudMusic/Aaron Zigman - Main Title.mp3",
+    "CloudMusic/Anson Seabra - Keep Your Head Up Princess.mp3",
+    "CloudMusic/Anthony Keyrouz,Romy Wave - Something Just Like This (feat. Romy Wave).mp3",
+    "CloudMusic/Ava Max - Sweet but Psycho.mp3",
+    "CloudMusic\\",
+    "folder1/",
+    "CloudMusic/Cecilia Cheung - Turn Into Fireworks and Fall for You.mp3",
+    "CloudMusic/Color Music Choir - Something Just Like This (Live).mp3"
 ]
+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=test_files)
+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: