fix:修改pickle扫描方法 与其他统一

This commit is contained in:
2024-06-04 19:50:31 +08:00
parent 0f2fb3c925
commit ec30999d2c
2 changed files with 15 additions and 6 deletions

View File

@@ -142,11 +142,7 @@ def pickleDataDetection(filename: str, output_file=None):
pickscan = pickleScanner(file)
pickscan.load()
res = pickscan.output()
if output_file:
with open(output_file, "w") as file:
json.dump(res, file, indent=4)
else:
print(json.dumps(res))
return res
if __name__ == "__main__":