change try/except to sys.module.get and a conditional

This commit is contained in:
Chris Gilling 2014-03-26 09:27:33 -07:00
parent 3388d82c1c
commit c3f4eb6d57
1 changed files with 3 additions and 3 deletions

View File

@ -1836,12 +1836,12 @@ def getfuncargnames(function, startindex=None):
if startindex is None:
startindex = inspect.ismethod(function) and 1 or 0
if realfunction != function:
try:
from mock import DEFAULT
mock_default = sys.modules.get('mock.DEFAULT')
if mock_default:
for patching in getattr(function, "patchings", []):
if not patching.attribute_name and patching.new is DEFAULT:
startindex += 1
except ImportError:
else:
startindex += len(getattr(function, "patchings", []))
function = realfunction
argnames = inspect.getargs(py.code.getrawcode(function))[0]