river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-08-25 20:37:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix csv encoding problem

Browse Source
This commit is contained in:
cokeBeer
2022-09-27 18:44:18 +08:00
parent 714b4120f1
commit 7162e40d2e
2 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
common/utils.py
+2
View File
@@ -330,6 +330,8 @@ def export_all_results(path, name, fmt, datas):
row_list.append(Record(keys, values)) row_list.append(Record(keys, values))
rows = RecordCollection(iter(row_list)) rows = RecordCollection(iter(row_list))
content = rows.export(fmt) content = rows.export(fmt)
if fmt == 'csv':
content = '\ufeff' + content
save_to_file(path, content) save_to_file(path, content)
export.py
+2
View File
@@ -60,6 +60,8 @@ def do_export(fmt, path, rows, show, domain, target):
if show: if show:
print(rows.dataset) print(rows.dataset)
data = rows.export(fmt) data = rows.export(fmt)
if fmt == 'csv':
data = '\ufeff' + data
utils.save_to_file(path, data) utils.save_to_file(path, data)
logger.log('ALERT', f'The subdomain result for {domain}: {path}') logger.log('ALERT', f'The subdomain result for {domain}: {path}')
data = rows.as_dict() data = rows.as_dict()