summaryrefslogtreecommitdiff
path: root/sphinx/pycode
diff options
context:
space:
mode:
authorTakeshi KOMIYA <i.tkomiya@gmail.com>2019-12-27 01:29:43 +0900
committerTakeshi KOMIYA <i.tkomiya@gmail.com>2019-12-27 01:29:43 +0900
commitdf31bc3ac8ce22d395ad147c64a6f68f836085f2 (patch)
treec6377f8d11303398576d85a4a7bf848df05e932c /sphinx/pycode
parent69d67e70509fb8286d8282924bc0078957bc0207 (diff)
downloadsphinx-git-df31bc3ac8ce22d395ad147c64a6f68f836085f2.tar.gz
refactor: pycode: do not call endswith twice
Diffstat (limited to 'sphinx/pycode')
-rw-r--r--sphinx/pycode/__init__.py5
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)