Fix warnings

This commit is contained in:
Mark Haines 2015-12-04 11:34:05 +00:00
parent 48a2526d62
commit d6059bdd2a
2 changed files with 13 additions and 3 deletions

View file

@ -23,6 +23,7 @@ try:
import resource
RUSAGE_THREAD = 1
resource.getrusage(RUSAGE_THREAD)
def get_thread_resource_usage():
return resource.getrusage(RUSAGE_THREAD)
except:
@ -137,13 +138,12 @@ class LoggingContext(object):
return
if self.usage_start:
self.usage_end = get_thread_resource_usage()
self.usage_end = get_thread_resource_usage()
def get_resource_usage(self):
ru_utime = self.ru_utime
ru_stime = self.ru_stime
start = self.usage_start
if self.usage_start and threading.current_thread() is self.main_thread:
current = get_thread_resource_usage()
ru_utime += current.ru_utime - self.usage_start.ru_utime