233 lines
9.4 KiB
Python
233 lines
9.4 KiB
Python
import sqlite3
|
||
|
||
|
||
def GetFilesDBPathInfo(db_path: str = "../src/db_ntfs_info.db",
|
||
table_name: str = "db_path",
|
||
files_path=None) -> list:
|
||
"""
|
||
根据传入的文件路径列表，在指定表中查询对应记录的 ID 和 Name 字段。
|
||
|
||
:param db_path: 数据库文件路径
|
||
:param table_name: 要查询的数据表名称
|
||
:param files_path: 文件的完整路径列表
|
||
:return: 查询结果列表，每项为 {'absolute_path': str, 'id': int, 'name': str}
|
||
"""
|
||
if files_path is None:
|
||
file_paths = []
|
||
results = []
|
||
|
||
# 连接数据库
|
||
conn = sqlite3.connect(db_path)
|
||
cursor = conn.cursor()
|
||
|
||
for path in files_path:
|
||
try:
|
||
# 使用字符串格式化插入表名，参数化查询只适用于值
|
||
sql = f"SELECT ID, Name FROM {table_name} WHERE Path = ?"
|
||
cursor.execute(sql, (path,))
|
||
row = cursor.fetchone()
|
||
if row:
|
||
results.append({
|
||
'absolute_path': path,
|
||
'id': row[0],
|
||
'name': row[1]
|
||
})
|
||
else:
|
||
print(f"未找到匹配记录：{path}")
|
||
except Exception as e:
|
||
print(f"查询失败：{path}，错误：{e}")
|
||
|
||
conn.close()
|
||
return results
|
||
|
||
|
||
# if __name__ == "__main__":
|
||
# 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"
|
||
# ]
|
||
#
|
||
# result = GetFilesDBPathInfo(files_path=test_files)
|
||
# for item in result:
|
||
# print(item)
|
||
|
||
|
||
def GetFilesDBNodeInfo(db_path: str = "../src/db_ntfs_info.db", table_name: str = "db_node",
|
||
path_records: list = None) -> list:
|
||
"""
|
||
根据 db_path 查询结果中的 ID 去 db_node 表中查找对应的 extent 分片信息。
|
||
|
||
:param db_path: 数据库文件路径
|
||
:param table_name: db_node 表名
|
||
:param path_records: 来自 get_db_path_info 的结果列表
|
||
:return: 包含文件分片信息的结果列表
|
||
"""
|
||
conn = sqlite3.connect(db_path)
|
||
cursor = conn.cursor()
|
||
|
||
results = []
|
||
|
||
for record in path_records:
|
||
path_id = record['id']
|
||
absolute_path = record['absolute_path']
|
||
name = record['name']
|
||
|
||
try:
|
||
# 查询 db_node 表中 PathID 对应的记录
|
||
cursor.execute(f"SELECT * FROM {table_name} WHERE PathID = ?", (path_id,))
|
||
row = cursor.fetchone()
|
||
|
||
if not row:
|
||
print(f"未找到 PathID={path_id} 在表 {table_name} 中的记录")
|
||
continue
|
||
|
||
# 获取字段索引（适用于按列名获取）
|
||
columns = [desc[0] for desc in cursor.description]
|
||
|
||
# 构建字典以便按列名访问
|
||
node_data = dict(zip(columns, row))
|
||
|
||
# 获取 ExtentCount
|
||
extent_count = node_data.get("ExtentCount", 0)
|
||
|
||
# 解析分片信息
|
||
fragments = []
|
||
for i in range(1, 5): # extent1 ~ extent4
|
||
loc = node_data.get(f"extent{i}_Location")
|
||
length = node_data.get(f"extent{i}_Length")
|
||
|
||
if loc is not None and length is not None and length > 0:
|
||
fragments.append({
|
||
"start_byte": loc,
|
||
"length": length
|
||
})
|
||
|
||
results.append({
|
||
"absolute_path": absolute_path,
|
||
"name": name,
|
||
"path_id": path_id,
|
||
"extent_count": extent_count,
|
||
"fragments": fragments
|
||
})
|
||
|
||
except Exception as e:
|
||
print(f"查询失败：PathID={path_id}, 错误：{e}")
|
||
|
||
conn.close()
|
||
return results
|
||
|
||
|
||
if __name__ == "__main__":
|
||
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"
|
||
]
|
||
|
||
# 第一步：获取 db_path 表中的 ID 和 Name
|
||
path_info = GetFilesDBPathInfo(files_path=test_files)
|
||
|
||
# 第二步：根据 PathID 查询 db_node 表中的分片信息
|
||
file_extents_info = GetFilesDBNodeInfo(path_records=path_info)
|
||
|
||
# 打印结果
|
||
for item in file_extents_info:
|
||
print(item)
|
||
|
||
|
||
def sort_fragments_by_start_byte(file_extents_list: list) -> list:
|
||
"""
|
||
对所有文件的分片按 start_byte 进行排序，并标注是第几个分片。
|
||
|
||
:param file_extents_list: get_file_extents_info 返回的结果列表
|
||
:return: 按 start_byte 排序后的片段列表，包含文件路径、文件名、第几个分片等信息
|
||
"""
|
||
all_fragments = []
|
||
|
||
for file_info in file_extents_list:
|
||
absolute_path = file_info['absolute_path']
|
||
filename = file_info['name']
|
||
extent_count = file_info['extent_count']
|
||
fragments = file_info['fragments']
|
||
|
||
# 对当前文件的片段排序（虽然通常已经是有序的）
|
||
sorted_fragments = sorted(fragments, key=lambda x: x['start_byte'])
|
||
|
||
# 添加片段索引信息
|
||
for idx, fragment in enumerate(sorted_fragments, start=1):
|
||
all_fragments.append({
|
||
'absolute_path': absolute_path,
|
||
'filename': filename,
|
||
'extent_count': extent_count,
|
||
'start_byte': fragment['start_byte'],
|
||
'length': fragment['length'],
|
||
'fragment_index': idx
|
||
})
|
||
|
||
# 全局排序：按 start_byte 排序所有片段
|
||
all_fragments.sort(key=lambda x: x['start_byte'])
|
||
|
||
return all_fragments
|
||
|
||
|
||
# if __name__ == "__main__":
|
||
# 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"
|
||
# ]
|
||
# test_files_sort = [
|
||
# {'absolute_path': 'CloudMusic/AGA - MIZU.mp3', 'name': 'AGA - MIZU.mp3', 'path_id': 6, 'extent_count': 1,
|
||
# 'fragments': [{'start_byte': 694849536, 'length': 8126464}]},
|
||
# {'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': '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': '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': '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': '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': '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': '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)
|
||
#
|
||
# # 根据文件片段先后排序
|
||
# 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)
|