From 5dbc4b50ff5348b558458bb28c03a18fc48264c3 Mon Sep 17 00:00:00 2001 From: Alex Weiss Date: Mon, 27 Jun 2016 10:36:50 -0400 Subject: [PATCH] it is possible for getmodule to return None (#3) --- harrison/__init__.py | 2 +- harrison/util/inspectlib.py | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/harrison/__init__.py b/harrison/__init__.py index 7df8ac1..18bb4aa 100644 --- a/harrison/__init__.py +++ b/harrison/__init__.py @@ -1,4 +1,4 @@ from harrison.profile import profile from harrison.timer import Timer -__version__ = '1.1.0' +__version__ = '1.1.1' diff --git a/harrison/util/inspectlib.py b/harrison/util/inspectlib.py index 282b72e..69e7a77 100644 --- a/harrison/util/inspectlib.py +++ b/harrison/util/inspectlib.py @@ -34,7 +34,11 @@ def stack_frame_info(stacklevel): raise ValueError('A stacklevel less than 1 is pointless') frame, filename, line_number, function_name, _, _ = inspect.stack()[stacklevel] - module_name = inspect.getmodule(frame).__name__ + module = inspect.getmodule(frame) # it is possible for getmodule to return None + if module is not None: + module_name = module.__name__ + else: + module_name = "" return Where( filename=filename,