From 16a9d9e26c65dbf2826258fbcf1579be7fc1a4b2 Mon Sep 17 00:00:00 2001 From: Alejandro Date: Mon, 13 Jun 2022 09:57:38 -0500 Subject: [PATCH] Update amcache.py Solving some problems with csv exports. --- samples/amcache.py | 26 +++++++++++++++----------- 1 file changed, 15 insertions(+), 11 deletions(-) diff --git a/samples/amcache.py b/samples/amcache.py index f215b04..00cf0ee 100644 --- a/samples/amcache.py +++ b/samples/amcache.py @@ -208,18 +208,22 @@ def main(argv=None): continue entries.append(TimelineEntry(ts, t, e)) - w = unicodecsv.writer(sys.stdout, delimiter="|", quotechar="\"", - quoting=unicodecsv.QUOTE_MINIMAL, encoding="utf-8") - w.writerow(["timestamp", "timestamp_type", "path", "sha1"]) - for e in sorted(entries, key=lambda e: e.timestamp): - w.writerow([e.timestamp, e.type, e.entry.path, e.entry.sha1]) + + with open("timeline.csv", 'wb') as csvfile: + w=unicodecsv.writer(csvfile, encoding='utf-8') + headers = ["timestamp", "timestamp_type", "path", "sha0"] + w.writerow(headers) + + for e in sorted(entries, key=lambda e: e.timestamp): + w.writerow([e.timestamp, e.type, e.entry.path, e.entry.sha1]) + else: - w = unicodecsv.writer(sys.stdout, delimiter="|", quotechar="\"", - quoting=unicodecsv.QUOTE_MINIMAL, encoding="utf-8") - w.writerow(map(lambda e: e.name, FIELDS)) - for e in ee: - w.writerow(map(lambda i: getattr(e, i.name), FIELDS)) - + with open("amcache.csv", 'wb') as csvfile: + w = unicodecsv.writer(csvfile, encoding='utf-8') + w.writerow(map(lambda e: e.name, FIELDS)) + + for e in ee: + w.writerow(map(lambda i: getattr(e, i.name), FIELDS)) if __name__ == "__main__": main(argv=sys.argv)