From 3105c04deef32286aa13068167e96c07ca71c52a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mois=C3=A9s=20L=C3=B3pez?= Date: Thu, 17 May 2018 16:52:40 -0500 Subject: [PATCH] [REF] ProfilerContext: Support lines parameter --- vmprof/profiler.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/vmprof/profiler.py b/vmprof/profiler.py index 403abd5e..16a634e2 100644 --- a/vmprof/profiler.py +++ b/vmprof/profiler.py @@ -11,7 +11,7 @@ class VMProfError(Exception): class ProfilerContext(object): done = False - def __init__(self, name, period, memory, native, real_time): + def __init__(self, name, period, memory, native, real_time, lines=None): if name is None: self.tmpfile = tempfile.NamedTemporaryFile("w+b", delete=False) else: @@ -21,10 +21,12 @@ def __init__(self, name, period, memory, native, real_time): self.memory = memory self.native = native self.real_time = real_time + self.lines = bool(lines) def __enter__(self): vmprof.enable(self.tmpfile.fileno(), self.period, self.memory, - native=self.native, real_time=self.real_time) + native=self.native, real_time=self.real_time, + lines=self.lines) def __exit__(self, type, value, traceback): vmprof.disable() @@ -56,8 +58,8 @@ class Profiler(object): def __init__(self): self._lib_cache = {} - def measure(self, name=None, period=0.001, memory=False, native=False, real_time=False): - self.ctx = ProfilerContext(name, period, memory, native, real_time) + def measure(self, name=None, period=0.001, memory=False, native=False, real_time=False, lines=False): + self.ctx = ProfilerContext(name, period, memory, native, real_time, lines) return self.ctx def get_stats(self):