Browse Source

Merge 5c8df7a9c5 into fe92b62f7a

pull/7/merge
Raphael Kargon 3 years ago
committed by GitHub
parent
commit
6a6f6f0b24
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
  1. +5
    -2
      importhook/finder.py

+ 5
- 2
importhook/finder.py View File

@ -29,7 +29,7 @@ def hook_finder(finder):
def wrapper(fullname, path, target=None):
logger.debug(f'{finder_name}.find_spec(fullname={fullname}, path={path}, target={target})')
spec = find_spec(fullname, path, target=target)
if spec is not None:
if spec is not None and spec.loader is not None:
spec.loader = HookLoader(spec.loader)
return spec
return wrapper
@ -39,7 +39,10 @@ def hook_finder(finder):
def wrapper(fullname, path):
logger.debug(f'{finder_name}.find_loader(fullname={fullname}, path={path})')
loader = find_loader(fullname, path)
return HookLoader(loader)
if loader is None:
return None
else:
return HookLoader(loader)
return wrapper
# Override the functions we care about


Loading…
Cancel
Save