import rapidjson
from datetime import timedelta
import tracemalloc

tracemalloc.start()

def object_hook(td):
    if '__td__' in td:
        return timedelta(td['__td__'])
    else:
        return td

def parse_json():
    data = []
    for i in range(1, 100):
        data.append({"name": f"a{i}", "timestamp": timedelta(seconds=i)})

    a = rapidjson.dumps(
        data,
        datetime_mode=rapidjson.DM_ISO8601,
        default=lambda td: {"__td__": td.total_seconds()} if isinstance(td, timedelta) else td,
    )

    data = rapidjson.loads(
        a,
        datetime_mode=rapidjson.DM_ISO8601,
        object_hook=object_hook
    )
    return data

print("before")
if tracemalloc.is_tracing():
    snapshot = tracemalloc.take_snapshot()
    top_stats = snapshot.statistics('lineno')
    print([str(x) for x in top_stats[:10]])

for _ in range(1000):
    parse_json()

print("after")
if tracemalloc.is_tracing():
    snapshot = tracemalloc.take_snapshot()
    top_stats = snapshot.statistics('lineno')
    print([str(x) for x in top_stats[:10]])