optimize db_path memory
This commit is contained in:
@@ -4,6 +4,7 @@ import sqlite3
|
||||
from datetime import datetime
|
||||
|
||||
from ntfs_utils.mft_analyze import GetFile80hPattern, GetFragmentData, ExtractSequenceHexValues, hex_list_to_int
|
||||
from ntfs_utils.main import volume_letter
|
||||
|
||||
|
||||
# 工具函数:获取文件扩展名
|
||||
@@ -131,11 +132,10 @@ def GetFragmentLength(fragment):
|
||||
|
||||
|
||||
# 主函数:将 db_path 数据导入 db_node
|
||||
def InsertNodeDataToDB(volume_letter: str, db_path='../src/db_ntfs_info.db', table_name='db_node', batch_size=20):
|
||||
def InsertNodeDataToDB(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(':'):
|
||||
@@ -254,5 +254,4 @@ def InsertNodeDataToDB(volume_letter: str, db_path='../src/db_ntfs_info.db', tab
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
volume_letter_test = "Y"
|
||||
InsertNodeDataToDB(volume_letter=volume_letter_test)
|
||||
InsertNodeDataToDB()
|
||||
|
Reference in New Issue
Block a user