mirror of
https://github.com/eronnen/procmon-parser
synced 2026-06-08 14:05:59 +00:00
eronnen
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import argparse
|
|
import glob
|
|
import time
|
|
from csv import DictReader
|
|
from itertools import chain
|
|
|
|
from tests.test_logs_format import check_pml_equals_csv
|
|
from procmon_parser import ProcmonLogsReader
|
|
|
|
|
|
def manual_test_pml_equals_csv_local(pml_path, csv_path):
|
|
start = time.time()
|
|
csv_reader_local = DictReader(open(csv_path, "r", encoding="utf-8-sig"))
|
|
pml_readers = []
|
|
for logfile_path in [pml_path] + glob.glob("{0}-*.{1}".format(*pml_path.rsplit('.', 1))):
|
|
pml_readers.append(ProcmonLogsReader(open(logfile_path, "rb")))
|
|
|
|
loaded = time.time()
|
|
print("Loading readers took {} seconds".format(loaded - start))
|
|
pml_reader = chain(*pml_readers)
|
|
check_pml_equals_csv(csv_reader_local, pml_reader)
|
|
print("Reading events took {} seconds".format(time.time() - loaded))
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--pml-path", type=str, help="Path to PML file")
|
|
parser.add_argument("--csv-path", type=str, help="Path to CSV file converted from the PML")
|
|
args = parser.parse_args()
|
|
manual_test_pml_equals_csv_local(args.pml_path, args.csv_path)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|