diff --git a/db_manage/clear_table_record.py b/db_manage/clear_table_record.py index fe9be28..89e8598 100644 --- a/db_manage/clear_table_record.py +++ b/db_manage/clear_table_record.py @@ -25,10 +25,10 @@ def ClearTableRecordsWithReset(db_path, table_name): if __name__ == '__main__': ClearTableRecordsWithReset(db_path='../src/db_ntfs_info.db', table_name='db_path') ClearTableRecordsWithReset(db_path='../src/db_ntfs_info.db', table_name='db_node') - ClearTableRecordsWithReset(db_path='../src/db_ntfs_info.db', table_name='db_device') - ClearTableRecordsWithReset(db_path='../src/db_ntfs_info.db', table_name='db_config') - ClearTableRecordsWithReset(db_path='../src/db_ntfs_info.db', table_name='db_user') - ClearTableRecordsWithReset(db_path='../src/db_ntfs_info.db', table_name='db_group') + # ClearTableRecordsWithReset(db_path='../src/db_ntfs_info.db', table_name='db_device') + # ClearTableRecordsWithReset(db_path='../src/db_ntfs_info.db', table_name='db_config') + # ClearTableRecordsWithReset(db_path='../src/db_ntfs_info.db', table_name='db_user') + # ClearTableRecordsWithReset(db_path='../src/db_ntfs_info.db', table_name='db_group') # ClearTableRecordsWithReset(db_path='../src/db_ntfs_info.db', table_name='db_extend_extent') - ClearTableRecordsWithReset(db_path='../src/db_ntfs_info.db', table_name='db_extend_name') + # ClearTableRecordsWithReset(db_path='../src/db_ntfs_info.db', table_name='db_extend_name') diff --git a/fake_main.py b/fake_main.py index b3a6895..7b1040f 100644 --- a/fake_main.py +++ b/fake_main.py @@ -3,14 +3,14 @@ from files_utils.files_sort import GetSortFragments fragment_lists = {} test_files = [ - r"Y:\CloudMusic\AGA - MIZU.mp3", - r"Y:\CloudMusic\AGA - 一.mp3", - r"Y:\CloudMusic\Aaron Zigman - Main Title.mp3", - r"Y:\CloudMusic\Anson Seabra - Keep Your Head Up Princess.mp3", - r"Y:\CloudMusic\Anthony Keyrouz,Romy Wave - Something Just Like This (feat. Romy Wave).mp3", - r"Y:\CloudMusic\Ava Max - Sweet but Psycho.mp3", - r"Y:\CloudMusic\Cecilia Cheung - Turn Into Fireworks and Fall for You.mp3", - r"Y:\CloudMusic\Color Music Choir - Something Just Like This (Live).mp3" + 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) diff --git a/files_utils/files_save.py b/files_utils/files_save.py index 57b4d76..f368d74 100644 --- a/files_utils/files_save.py +++ b/files_utils/files_save.py @@ -1,12 +1,9 @@ import os -def ExtractVolumeLetter(path: str) -> str: - """从绝对路径中提取盘符""" - drive = os.path.splitdrive(path)[0] - if not drive: - raise ValueError(f"无法从路径中提取盘符：{path}") - return drive[0].upper() # 返回 'Y' +def GetVolumeLetter() -> str: + from ntfs_utils.main import volume_letter + return volume_letter def CopySingleFragmentFiles(source_data_dict, target_path): @@ -29,7 +26,7 @@ def CopySingleFragmentFiles(source_data_dict, target_path): print("错误：缺少必要的文件信息") return - source_disk_path = ExtractVolumeLetter(absolute_path) + source_disk_path = GetVolumeLetter() target_file_path = os.path.join(target_path, file_name) try: @@ -105,7 +102,7 @@ def CopyMultiFragmentFiles( start_byte = item['start_byte'] length_byte = item['length'] - volume_letter = ExtractVolumeLetter(item['absolute_path']) + volume_letter = GetVolumeLetter() # 读取分片内容 fragment_content = ReadDiskBytes(volume_letter, start_byte, length_byte) diff --git a/ntfs_utils/db_node.py b/ntfs_utils/db_node.py index 6a75722..c928bf2 100644 --- a/ntfs_utils/db_node.py +++ b/ntfs_utils/db_node.py @@ -3,8 +3,7 @@ import os import sqlite3 from datetime import datetime -# 导入你的模块函数 -from mft_analyze import GetFile80hPattern, GetFragmentData, ExtractSequenceHexValues, hex_list_to_int +from ntfs_utils.mft_analyze import GetFile80hPattern, GetFragmentData, ExtractSequenceHexValues, hex_list_to_int # 工具函数：获取文件扩展名 @@ -132,11 +131,20 @@ def GetFragmentLength(fragment): # 主函数：将 db_path 数据导入 db_node -# 主函数：将 db_path 数据导入 db_node -def InsertNodeDataToDB(db_path='../src/db_ntfs_info.db', table_name='db_node'): +def InsertNodeDataToDB(volume_letter: str, db_path='../src/db_ntfs_info.db', table_name='db_node', batch_size=20): conn = sqlite3.connect(db_path) cursor = conn.cursor() + volume_letter = volume_letter.upper().strip() + if len(volume_letter) == 1: + volume_root = f"{volume_letter}:\\" + elif volume_letter.endswith(':'): + volume_root = f"{volume_letter}\\" + else: + volume_root = f"{volume_letter}:\\" # 支持 "Y" 或 "Y:" 输入 + + print(f"🔍 当前处理磁盘根目录：{volume_root}") + group_id = GetFirstGroupId(cursor) user_id = GetFirstUserId(cursor) device_id = GetDeviceId(cursor) @@ -144,8 +152,26 @@ def InsertNodeDataToDB(db_path='../src/db_ntfs_info.db', table_name='db_node'): cursor.execute("SELECT ID, Path, Name, ParentID FROM db_path") rows = cursor.fetchall() + insert_fields = [ + 'PathID', 'ParentID', 'NameHash', 'PathHash', + 'ExtendNameID', 'DirLayer', 'GroupID', 'UserID', + 'FileCreateTime', 'FileModifyTime', 'FileAccessTime', 'FileAuthTime', + 'FileSize', 'FileMode', 'FileHash', 'ExtentCount', + # extent 字段 + "extent1_DeviceID", "extent1_Location", "extent1_Length", + "extent2_DeviceID", "extent2_Location", "extent2_Length", + "extent3_DeviceID", "extent3_Location", "extent3_Length", + "extent4_DeviceID", "extent4_Location", "extent4_Length" + ] + insert_placeholders = ', '.join('?' * len(insert_fields)) + insert_sql = f"INSERT INTO {table_name} ({', '.join(insert_fields)}) VALUES ({insert_placeholders})" + + batch = [] + for row in rows: - path_id, full_path, name, parent_id = row + path_id, relative_path, name, parent_id = row + + full_path = os.path.join(volume_root, relative_path) # 检查是否已存在相同 PathID cursor.execute("SELECT COUNT(*) FROM db_node WHERE PathID = ?", (path_id,)) @@ -154,13 +180,10 @@ def InsertNodeDataToDB(db_path='../src/db_ntfs_info.db', table_name='db_node'): print(f"⚠️ PathID {path_id} 已存在，跳过插入") continue - # 获取文件的80h属性数据 try: file80h_pattern = GetFile80hPattern(full_path) fragments = GetFragmentData(file80h_pattern) - extent_count = min(len(fragments), 4) # 最多支持4个fragment - print(f"✅ 分片数量为: {extent_count}") - + extent_count = min(len(fragments), 4) except Exception as e: print(f"⚠️ 获取 ExtentCount 失败，使用默认值 0: {e}") fragments = [] @@ -168,10 +191,9 @@ def InsertNodeDataToDB(db_path='../src/db_ntfs_info.db', table_name='db_node'): # 计算字段 name_hash = hashlib.sha256(name.encode()).hexdigest() - dir_layer = GetDirLayer(full_path) + dir_layer = GetDirLayer(relative_path) extend_name_id = GetExtendNameId(name, cursor) - # ✅ 现在可以安全调用 GetFileSize(file80h_pattern) try: file_size = GetFileSize(file80h_pattern) except Exception as e: @@ -180,7 +202,7 @@ def InsertNodeDataToDB(db_path='../src/db_ntfs_info.db', table_name='db_node'): file_hash = GetFileHash(full_path) - # 获取文件的时间属性 + # 获取时间信息 file_times = GetFilesTime(full_path) create_time = file_times["FileCreateTime"] modify_time = file_times["FileModifyTime"] @@ -192,21 +214,7 @@ def InsertNodeDataToDB(db_path='../src/db_ntfs_info.db', table_name='db_node'): path_hash_result = cursor.fetchone() path_hash = path_hash_result[0] if path_hash_result else "" - # 构建插入语句字段和参数（保持原样） - fields = [ - 'PathID', 'ParentID', 'NameHash', 'PathHash', - 'ExtendNameID', 'DirLayer', 'GroupID', 'UserID', - 'FileCreateTime', 'FileModifyTime', 'FileAccessTime', 'FileAuthTime', - 'FileSize', 'FileMode', 'FileHash', 'ExtentCount' - ] - values = [ - path_id, parent_id, name_hash, path_hash, - extend_name_id, dir_layer, group_id, user_id, - create_time, modify_time, access_time, auth_time, - file_size, 'default', file_hash, extent_count - ] - - # 处理 Extent 片段字段 + # 构建 extent 字段 extent_data = [] for i in range(4): # 最多4个 extent if i < len(fragments): @@ -217,27 +225,34 @@ def InsertNodeDataToDB(db_path='../src/db_ntfs_info.db', table_name='db_node'): else: extent_data.extend([None, None, None]) - # 拼接字段和值 - extent_fields = [ - "extent1_DeviceID", "extent1_Location", "extent1_Length", - "extent2_DeviceID", "extent2_Location", "extent2_Length", - "extent3_DeviceID", "extent3_Location", "extent3_Length", - "extent4_DeviceID", "extent4_Location", "extent4_Length" + # 构建插入数据 + values = [ + path_id, parent_id, name_hash, path_hash, + extend_name_id, dir_layer, group_id, user_id, + create_time, modify_time, access_time, auth_time, + file_size, 'default', file_hash, extent_count, + *extent_data ] - fields += extent_fields - values += extent_data - # 构建 SQL 插入语句 - placeholders = ', '.join('?' * len(values)) - insert_sql = f"INSERT INTO {table_name} ({', '.join(fields)}) VALUES ({placeholders})" + batch.append(values) - # 执行插入 - cursor.execute(insert_sql, values) + # 批量插入 + if len(batch) >= batch_size: + cursor.executemany(insert_sql, batch) + conn.commit() + print(f"✅ 提交一批 {len(batch)} 条记录到 {table_name}") + batch.clear() + + # 插入剩余不足一批的数据 + if batch: + cursor.executemany(insert_sql, batch) + conn.commit() + print(f"✅ 提交最后一批 {len(batch)} 条记录到 {table_name}") - conn.commit() conn.close() print(f"✅ 数据已成功插入到 {table_name} 表") -if __name__ == '__main__': - InsertNodeDataToDB() +if __name__ == "__main__": + volume_letter_test = "Y" + InsertNodeDataToDB(volume_letter=volume_letter_test) diff --git a/ntfs_utils/db_path.py b/ntfs_utils/db_path.py index 5d5abf6..0c00d9b 100644 --- a/ntfs_utils/db_path.py +++ b/ntfs_utils/db_path.py @@ -27,23 +27,23 @@ def ScanVolume(volume_letter: str): """ 完整扫描指定磁盘的所有文件和目录，忽略 NTFS 元文件和系统文件夹， 并为每个节点分配 ParentID。 - - 返回: - list of dict：包含文件/目录信息的字典列表 """ + root_path = f"{volume_letter.upper()}:\\" if not os.path.exists(root_path): raise ValueError(f"磁盘 {root_path} 不存在") result = [] - path_to_id = {} # 用于记录路径到数据库 ID 的映射 - counter = 1 # 模拟数据库自增 ID + path_to_id = {} # 路径 -> ID 映射 + counter = 1 for root, dirs, files in os.walk(root_path, topdown=True, onerror=None, followlinks=False): # 过滤掉需要跳过的目录 dirs[:] = [d for d in dirs if not ShouldSkipPath(os.path.join(root, d))] - for entry in files + dirs: + entries = files + dirs + + for entry in entries: full_path = os.path.join(root, entry) if ShouldSkipPath(full_path): @@ -61,21 +61,35 @@ def ScanVolume(volume_letter: str): name = entry - # ✅ 修正点：对 Path 字段进行哈希 - path_hash = GenerateHash(full_path) + # 分离盘符并去除开头的 \ + _, relative_path = os.path.splitdrive(full_path) + relative_path = relative_path.lstrip("\\") + + # 如果是文件夹，Path 字段结尾加 '\\' + if is_dir and not relative_path.endswith("\\"): + relative_path += "\\" + + # ✅ 关键修改点：将所有 \ 替换为 \\ + relative_path = relative_path.replace("\\", "\\\\") + + path_hash = GenerateHash(relative_path) - # 计算 ContentSize（KB），小文件至少显示为 1 KB content_size = bytes_size // 1024 if content_size == 0 and bytes_size > 0: content_size = 1 - # 获取父目录路径 parent_path = os.path.dirname(full_path) - parent_id = path_to_id.get(parent_path, 0) # 默认为 0（根目录可能未录入） + _, parent_relative_path = os.path.splitdrive(parent_path) + parent_relative_path = parent_relative_path.lstrip("\\").rstrip("\\") # 去除首尾 \ + if os.path.isdir(parent_path) and not parent_relative_path.endswith("\\"): # 如果是目录，补 \ + parent_relative_path += "\\" + parent_relative_path = parent_relative_path.replace("\\", "\\\\") # 转换为双反斜杠 \\ + + parent_id = path_to_id.get(parent_relative_path, 0) item = { "ID": counter, - "Path": full_path, + "Path": relative_path, "Name": name, "PathHash": path_hash, "IsDir": is_dir, @@ -84,7 +98,7 @@ def ScanVolume(volume_letter: str): } result.append(item) - path_to_id[full_path] = counter + path_to_id[relative_path] = counter counter += 1 except Exception as e: @@ -163,7 +177,7 @@ def InsertPathDataToDB(data, db_path='../src/db_ntfs_info.db', table_name='db_pa # 示例主函数 def main(): - volume_letter = "Z" + volume_letter = "Y" print(f"🔍 开始全盘扫描磁盘 {volume_letter}:\\ ...") scanned_data = ScanVolume(volume_letter) diff --git a/ntfs_utils/__init__.py b/ntfs_utils/main.py similarity index 70% rename from ntfs_utils/__init__.py rename to ntfs_utils/main.py index 3938606..0fe468b 100644 --- a/ntfs_utils/__init__.py +++ b/ntfs_utils/main.py @@ -1,15 +1,15 @@ -from db_config import GetNTFSBootInfo, InsertInfoToDBConfig -from db_device import ScanSpecialVolumes, InsertVolumesToDB -from db_extend_name import InsertExtensionsToDB -from db_group import InsertGroupToDB -from db_path import GenerateHash, ShouldSkipPath, ScanVolume, InsertPathDataToDB -from db_user import InsertUserToDB -from db_node import InsertNodeDataToDB +from ntfs_utils.db_config import GetNTFSBootInfo, InsertInfoToDBConfig +from ntfs_utils.db_device import ScanSpecialVolumes, InsertVolumesToDB +from ntfs_utils.db_extend_name import InsertExtensionsToDB +from ntfs_utils.db_group import InsertGroupToDB +from ntfs_utils.db_node import InsertNodeDataToDB +from ntfs_utils.db_path import ScanVolume, InsertPathDataToDB +from ntfs_utils.db_user import InsertUserToDB + +volume_letter = 'Y' def main(): - volume_letter = 'Y' - # 初始化 db_config 表 config_data = GetNTFSBootInfo(volume_letter) InsertInfoToDBConfig(config_data) @@ -42,7 +42,7 @@ def main(): print(f"共插入 {count} 个新扩展名。") # 初始化 db_node 表 - InsertNodeDataToDB() + InsertNodeDataToDB(volume_letter) if __name__ == '__main__': diff --git a/ntfs_utils/mft_analyze.py b/ntfs_utils/mft_analyze.py index 9438d82..d69a39e 100644 --- a/ntfs_utils/mft_analyze.py +++ b/ntfs_utils/mft_analyze.py @@ -2,7 +2,7 @@ import os import pytsk3 -from db_config import GetNTFSBootInfo +from ntfs_utils.db_config import GetNTFSBootInfo def find_file_mft_entry(fs, target_path): diff --git a/test/files_sort.py b/test/files_sort.py index 85d4210..d1e6906 100644 --- a/test/files_sort.py +++ b/test/files_sort.py @@ -183,40 +183,50 @@ def sort_fragments_by_start_byte(file_extents_list: list) -> list: if __name__ == "__main__": test_files = [ - r"Y:\CloudMusic\AGA - MIZU.mp3", - r"Y:\CloudMusic\AGA - 一.mp3", - r"Y:\CloudMusic\Aaron Zigman - Main Title.mp3", - r"Y:\CloudMusic\Anson Seabra - Keep Your Head Up Princess.mp3", - r"Y:\CloudMusic\Anthony Keyrouz,Romy Wave - Something Just Like This (feat. Romy Wave).mp3", - r"Y:\CloudMusic\Ava Max - Sweet but Psycho.mp3", - r"Y:\CloudMusic\Cecilia Cheung - Turn Into Fireworks and Fall for You.mp3", - r"Y:\CloudMusic\Color Music Choir - Something Just Like This (Live).mp3" + 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" ] test_files_sort = [ - {'absolute_path': 'Y:\\CloudMusic\\AGA - MIZU.mp3', 'name': 'AGA - MIZU.mp3', 'path_id': 6, 'extent_count': 1, + {'absolute_path': 'CloudMusic\\AGA - MIZU.mp3', 'name': 'AGA - MIZU.mp3', 'path_id': 6, 'extent_count': 1, 'fragments': [{'start_byte': 694849536, 'length': 8126464}]}, - {'absolute_path': 'Y:\\CloudMusic\\AGA - 一.mp3', 'name': 'AGA - 一.mp3', 'path_id': 7, 'extent_count': 2, + {'absolute_path': 'CloudMusic\\AGA - 一.mp3', 'name': 'AGA - 一.mp3', 'path_id': 7, 'extent_count': 2, 'fragments': [{'start_byte': 702976000, 'length': 10870784}, {'start_byte': 23162880, 'length': 69632}]}, - {'absolute_path': 'Y:\\CloudMusic\\Aaron Zigman - Main Title.mp3', 'name': 'Aaron Zigman - Main Title.mp3', + {'absolute_path': 'CloudMusic\\Aaron Zigman - Main Title.mp3', 'name': 'Aaron Zigman - Main Title.mp3', 'path_id': 5, 'extent_count': 1, 'fragments': [{'start_byte': 687685632, 'length': 7163904}]}, - {'absolute_path': 'Y:\\CloudMusic\\Anson Seabra - Keep Your Head Up Princess.mp3', + {'absolute_path': 'CloudMusic\\Anson Seabra - Keep Your Head Up Princess.mp3', 'name': 'Anson Seabra - Keep Your Head Up Princess.mp3', 'path_id': 8, 'extent_count': 1, 'fragments': [{'start_byte': 713846784, 'length': 7970816}]}, - {'absolute_path': 'Y:\\CloudMusic\\Anthony Keyrouz,Romy Wave - Something Just Like This (feat. Romy Wave).mp3', + {'absolute_path': 'CloudMusic\\Anthony Keyrouz,Romy Wave - Something Just Like This (feat. Romy Wave).mp3', 'name': 'Anthony Keyrouz,Romy Wave - Something Just Like This (feat. Romy Wave).mp3', 'path_id': 9, 'extent_count': 1, 'fragments': [{'start_byte': 721817600, 'length': 9179136}]}, - {'absolute_path': 'Y:\\CloudMusic\\Ava Max - Sweet but Psycho.mp3', 'name': 'Ava Max - Sweet but Psycho.mp3', + {'absolute_path': 'CloudMusic\\Ava Max - Sweet but Psycho.mp3', 'name': 'Ava Max - Sweet but Psycho.mp3', 'path_id': 10, 'extent_count': 1, 'fragments': [{'start_byte': 731000832, 'length': 7938048}]}, - {'absolute_path': 'Y:\\CloudMusic\\Cecilia Cheung - Turn Into Fireworks and Fall for You.mp3', + {'absolute_path': 'CloudMusic\\Cecilia Cheung - Turn Into Fireworks and Fall for You.mp3', 'name': 'Cecilia Cheung - Turn Into Fireworks and Fall for You.mp3', 'path_id': 11, 'extent_count': 1, 'fragments': [{'start_byte': 738938880, 'length': 6791168}]}, - {'absolute_path': 'Y:\\CloudMusic\\Color Music Choir - Something Just Like This (Live).mp3', + {'absolute_path': 'CloudMusic\\Color Music Choir - Something Just Like This (Live).mp3', 'name': 'Color Music Choir - Something Just Like This (Live).mp3', 'path_id': 12, 'extent_count': 1, 'fragments': [{'start_byte': 745730048, 'length': 6193152}]}] - # path_info = GetFilesDBPathInfo(files_path=test_files) - # file_extents_data = GetFilesDBNodeInfo(path_records=path_info) - result = sort_fragments_by_start_byte(test_files_sort) - print(result) - # for item in result: - # print(item) + path_info = GetFilesDBPathInfo(files_path=test_files) + file_extents_data = GetFilesDBNodeInfo(path_records=path_info) + + # 根据文件片段先后排序 + single_fragment_result = sort_fragments_by_start_byte(file_extents_data) + + # 模拟多文件片段，根据文件片段先后排序 + multi_fragment_result = sort_fragments_by_start_byte(test_files_sort) + + print("单文件片段排序结果：") + for item in single_fragment_result: + print(item) + + print("\n多文件片段排序结果：") + for item in multi_fragment_result: + print(item) diff --git a/test/folders_save.py b/test/folders_save.py index e69de29..f56f2f7 100644 --- a/test/folders_save.py +++ b/test/folders_save.py @@ -0,0 +1,154 @@ +import sqlite3 + +from files_sort import GetFilesDBNodeInfo + + +def GetFolderID( + folder_path: str, + db_path: str = "../src/db_ntfs_info.db", + table_name: str = "db_path" +) -> int | None: + """ + 根据文件夹路径，查询数据库中该文件夹对应的 ID。 + + :param folder_path: 文件夹路径（如 r"CloudMusic\\"） + :param db_path: 数据库文件路径 + :param table_name: 要查询的数据表名称，默认为 'db_path' + :return: 成功则返回 ID（int），失败返回 None + """ + + conn = sqlite3.connect(db_path) + cursor = conn.cursor() + + try: + # 使用 table_name 构建 SQL 查询 + sql = f"SELECT ID FROM {table_name} WHERE Path = ?" + cursor.execute(sql, (folder_path,)) + result = cursor.fetchone() + + if result: + return result[0] + else: + print(f"未找到路径：{folder_path} 在表 {table_name} 中") + return None + + except sqlite3.Error as e: + print(f"数据库操作失败：{e}") + return None + + finally: + conn.close() + + +def GetNodeFragmentsByParentID( + parent_id: int, + db_path: str = "../src/db_ntfs_info.db", + table_name: str = "db_node" +) -> list: + """ + 根据 ParentID 查询 db_node 表中对应的文件/子目录的分片信息。 + + :param parent_id: 父节点 ID + :param db_path: 数据库文件路径 + :param table_name: 数据表名称 + :return: 包含 PathID、分片数量及分片信息的列表 + """ + conn = sqlite3.connect(db_path) + cursor = conn.cursor() + + # 构建查询语句（根据分片字段） + sql = f""" + SELECT + PathID, ExtentCount, + extent1_Location, extent1_Length, + extent2_Location, extent2_Length, + extent3_Location, extent3_Length, + extent4_Location, extent4_Length + FROM {table_name} + WHERE ParentID = ? + """ + + try: + cursor.execute(sql, (parent_id,)) + rows = cursor.fetchall() + except Exception as e: + print(f"数据库查询失败：{e}") + return [] + + results = [] + + for row in rows: + path_id = row[0] + extent_count = row[1] + + fragments = [] + for i in range(4): + location = row[2 + i * 2] + length = row[3 + i * 2] + + if location is not None and length is not None and length > 0: + fragments.append({ + 'start_byte': location, + 'length': length + }) + + results.append({ + 'path_id': path_id, + 'extent_count': extent_count, + 'fragments': fragments + }) + + conn.close() + return results + + +def GetSubPathsByParentID( + parent_id: int, + db_path: str = "../src/db_ntfs_info.db", + table_name: str = "db_path" +) -> list: + """ + 根据 ParentID 查询 db_path 表中对应的子项（文件/文件夹）。 + + :param parent_id: 父节点 ID + :param db_path: 数据库文件路径 + :param table_name: 数据表名称 + :return: 包含 ID、Path、Name 的字典列表 + """ + conn = sqlite3.connect(db_path) + cursor = conn.cursor() + + sql = f""" + SELECT ID, Path, Name + FROM {table_name} + WHERE ParentID = ? + """ + + try: + cursor.execute(sql, (parent_id,)) + rows = cursor.fetchall() + except Exception as e: + print(f"数据库查询失败：{e}") + return [] + + results = [] + for row in rows: + item = { + 'id': row[0], + 'absolute_path': row[1], # 因为 Path 已经是完整格式（如 CloudMusic\\AGA - MIZU.mp3） + 'name': row[2] + } + results.append(item) + + conn.close() + return results + + +if __name__ == "__main__": + test_folder_path = "pictures" + parent_id_test = GetFolderID(test_folder_path) + # node_data = GetNodeFragmentsByParentID(parent_id_test) + path_data = GetSubPathsByParentID(parent_id_test) + node_data = GetFilesDBNodeInfo(path_records=path_data) + for data in node_data: + print(data)