tracemalloc-report.py

#!/usr/bin/env python3
import sys
from tracemalloc import Snapshot

def main():
    # todo: argparser
    fil = sys.argv[1]
    num = 10
    if len(sys.argv) > 2:
        num = int(sys.argv[2])

    snap = Snapshot.load(fil)

    topmem(snap, num)

    if snap.traceback_limit != 1:
        tracemem(snap, range(num))


def topmem(snap, num):
    top_stats = snap.statistics('lineno')

    print(f"[ Top {num} ]")
    for stat in top_stats[:num]:
        print(stat)


def tracemem(snap, nums):
    top_stats = snap.statistics('traceback')
    limit = snap.traceback_limit
    print(f"[ Frame limit {limit} ]")

    # todo: add filter cli args/config
    for num in nums:
        # pick the biggest memory block
        stat = top_stats[num]
        print(f"[ Top {num} ]")
        print("%s memory blocks: %.1f KiB" % (stat.count, stat.size / 1024))
        for line in stat.traceback.format():
            print(line)



if __name__ == "__main__":
    main()