Commit 66f82370 authored by Ka-Ping Yee's avatar Ka-Ping Yee

Also accept .so as an extension for module files.

parent fbc8f7e2
...@@ -127,14 +127,10 @@ def stripid(text): ...@@ -127,14 +127,10 @@ def stripid(text):
def modulename(path): def modulename(path):
"""Return the Python module name for a given path, or None.""" """Return the Python module name for a given path, or None."""
filename = os.path.basename(path) filename = os.path.basename(path)
if lower(filename[-3:]) == '.py': for ending in ['.py', '.pyc', '.pyd', '.pyo',
return filename[:-3] 'module.so', 'module.so.1', '.so']:
elif lower(filename[-4:]) in ['.pyc', '.pyd', '.pyo']: if len(filename) > len(ending) and filename[-len(ending):] == ending:
return filename[:-4] return filename[:-len(ending)]
elif lower(filename[-11:]) == 'module.so':
return filename[:-11]
elif lower(filename[-13:]) == 'module.so.1':
return filename[:-13]
class DocImportError(Exception): class DocImportError(Exception):
"""Class for errors while trying to import something to document it.""" """Class for errors while trying to import something to document it."""
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment