46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
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
|
|
|
|
|
|
def main():
|
|
volume_letter = 'Z'
|
|
|
|
# 初始化 db_config 表
|
|
config_data = GetNTFSBootInfo(volume_letter)
|
|
InsertInfoToDBConfig(config_data)
|
|
|
|
# 初始化 db_device 表
|
|
device_data = ScanSpecialVolumes(volume_letter)
|
|
InsertVolumesToDB([device_data])
|
|
|
|
# 初始化 db_user 表
|
|
user_list = ["Copier"]
|
|
InsertUserToDB(user_list)
|
|
|
|
# 初始化 db_group 表
|
|
group_name_list = ["Copier"]
|
|
InsertGroupToDB(group_name_list)
|
|
|
|
# 初始化 db_path 表
|
|
scanned_data = ScanVolume(volume_letter)
|
|
InsertPathDataToDB(scanned_data)
|
|
|
|
# 初始化 db_extend_name 表
|
|
common_extensions = [
|
|
"txt", "log", "csv", "xls", "xlsx", "doc", "docx",
|
|
"ppt", "pptx", "pdf", "jpg", "jpeg", "png", "gif",
|
|
"bmp", "mp3", "wav", "mp4", "avi", "mkv", "mov",
|
|
"exe", "dll", "bat", "ini", "reg", "zip", "rar", "7z",
|
|
"json", "xml", "html", "css", "js", "py", "java", "cpp"
|
|
]
|
|
count = InsertExtensionsToDB(common_extensions)
|
|
print(f"共插入 {count} 个新扩展名。")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|