2017-10-16 48 views
0

問題:如何將數據插入文件內的JSON數組?

我在文件上的元的項目,收集來自一個JSON文件的文件夾,並將其存儲文件信息的工作。

我的空JSON文件看起來像這樣

db.json

{ 
    "samples" : [] 
} 

現在我的想法是掃描每個文件並插入每個記錄到樣品陣列這樣

分貝.json

{ 
    "samples" : [ 
{ 
    "file_name": "ba18524fe5d5c6e99b99c7da66dbbfa094ab8f61", 
    "file_new_location": "/home/admin/", 
    "file_path": "/home/admin/vt_files/ba18524fe5d5c6e99b99c7da66dbbfa094ab8f61", 
    "file_source": "VT", 
    "md5": "149c4ac4ba0863607e033d6a5721fee7", 
    "mime": "application_x-dosexec", 
    "severity": "high", 
    "sha256": "50bc946af55f0a5deb66f935229046c1ec74fb2233d71efa3ae064a073a771f5" 
} 
] 
} 

但我的問題是,當我嘗試插入記錄時,Iam無法插入到樣本中,而是在一行中。

for file_original_name in files: 
    file_meta = {} 
    file_meta['file_name'] = file_original_name 
    file_meta['file_path'] = os.path.join(root,file_original_name) 
    file_meta['file_source'] = source 
    file_meta['severity'] = severity 
    file_meta['md5'] = md5(file_meta['file_path']) 
    file_meta['sha256'] = sha256(file_meta['file_path']) 
    mime = magic.Magic(mime=True) 
    file_meta['mime'] = mime.from_file(file_meta['file_path']).replace('/','_') 
    file_meta['file_new_location'] = '{}/{}/{}/{}'.format(default_root,severity,file_meta['mime'],file_original_name) 
    try: 
     extracted_new_dir = os.path.join(default_root,severity,file_meta['mime']) 
     if not os.path.isdir(extracted_new_dir): 
      os.makedirs(extracted_new_dir) 
     if delete: 
      move(file_meta['file_path'],file_meta['file_new_location']) 
     else: 
      copyfile(file_meta['file_path'],file_meta['file_new_location']) 
     print json.dumps(file_meta,sort_keys=True,indent=4) 
    except Exception as e: 
     print e 
    result[file_original_name] = (file_meta) 
    with open('db.json','a') as f: 
     json.dump(result , f) 

有關如何解決此問題的任何建議?

+0

@Sraw我要插入到數組。 –

+0

我認爲你需要首先加載整個json文件,然後像修改'dict/list'一樣修改它,最後再轉儲它。 – Sraw

回答

0

儘量先格式化結果,它寫入文件之前:

result = json.dumps(r, indent=4) 
with open('db.json','a') as f: 
    f.write(result)