diff --git a/stubdoc/cli.py b/stubdoc/cli.py index 075184b..53083cc 100644 --- a/stubdoc/cli.py +++ b/stubdoc/cli.py @@ -121,7 +121,7 @@ def _validate_module_path_arg(module_path_arg: Optional[str]) -> None: if not os.path.isfile(module_path_arg): raise ValueError( f'Specified module not found: {module_path_arg}') - if not module_path_arg.endswith('.py'): + if not os.path.splitext(module_path_arg)[1] in ['.py', '.pyd']: raise ValueError( f'A non-python module path specified: {module_path_arg}') diff --git a/stubdoc/stubdoc.py b/stubdoc/stubdoc.py index 5b4e0e2..e2a3adf 100644 --- a/stubdoc/stubdoc.py +++ b/stubdoc/stubdoc.py @@ -406,7 +406,7 @@ def _read_module(module_path: str) -> ModuleType: file_name: str = os.path.basename(module_path) dir_path: str = module_path.replace(file_name, '', 1) sys.path.append(dir_path) - package_name: str = module_path.replace('.py', '') + package_name: str = os.path.splitext(module_path)[0] package_name = package_name.replace('/', '.') package_name = package_name.replace('\\', '.') while package_name.startswith('.'):