Python resource-based coarse memory growth observer

gistfile1.py

import contextlib
import resource
import typing

memory_profiler_data = dict()

@contextlib.contextmanager
def record_memory(name: str) -> typing.ContextManager:
    initial_rusage = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
    assert name not in memory_profiler_data, f"{name}'s memory usage has already been recorded, can't record again"
    try:
        yield
    finally:
        memory_profiler_data[name] = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss - initial_rusage


def main():
    with record_memory('operation1'):
        do_stuff_for_operation1()

    with record_memory('operation2'):
        do_stuff_for_operation2()

    print(memory_profiler_data)