diff options
| author | Takeshi KOMIYA <i.tkomiya@gmail.com> | 2019-12-27 01:29:43 +0900 |
|---|---|---|
| committer | Takeshi KOMIYA <i.tkomiya@gmail.com> | 2019-12-27 01:29:43 +0900 |
| commit | df31bc3ac8ce22d395ad147c64a6f68f836085f2 (patch) | |
| tree | c6377f8d11303398576d85a4a7bf848df05e932c /sphinx/pycode | |
| parent | 69d67e70509fb8286d8282924bc0078957bc0207 (diff) | |
| download | sphinx-git-df31bc3ac8ce22d395ad147c64a6f68f836085f2.tar.gz | |
refactor: pycode: do not call endswith twice
Diffstat (limited to 'sphinx/pycode')
| -rw-r--r-- | sphinx/pycode/__init__.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/sphinx/pycode/__init__.py b/sphinx/pycode/__init__.py index 1106d3fdb..6c8cd725e 100644 --- a/sphinx/pycode/__init__.py +++ b/sphinx/pycode/__init__.py @@ -56,12 +56,11 @@ class ModuleAnalyzer: # all methods for getting filename failed, so raise... raise PycodeError('no source found for module %r' % modname) filename = path.normpath(path.abspath(filename)) - lfilename = filename.lower() - if lfilename.endswith('.pyo') or lfilename.endswith('.pyc'): + if filename.lower().endswith(('.pyo', '.pyc')): filename = filename[:-1] if not path.isfile(filename) and path.isfile(filename + 'w'): filename += 'w' - elif not (lfilename.endswith('.py') or lfilename.endswith('.pyw')): + elif not filename.lower().endswith(('.py', '.pyw')): raise PycodeError('source is not a .py file: %r' % filename) elif ('.egg' + path.sep) in filename: pat = '(?<=\\.egg)' + re.escape(path.sep) |
