66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
import os
|
|
import re
|
|
import sys
|
|
|
|
from rope.base import utils
|
|
from rope.base.utils import pycompat
|
|
|
|
|
|
def _stdlib_path():
|
|
if pycompat.PY2:
|
|
from distutils import sysconfig
|
|
|
|
return sysconfig.get_python_lib(standard_lib=True, plat_specific=True)
|
|
elif pycompat.PY3:
|
|
import inspect
|
|
|
|
return os.path.dirname(inspect.getsourcefile(inspect))
|
|
|
|
|
|
@utils.cached(1)
|
|
def standard_modules():
|
|
return python_modules() | dynload_modules()
|
|
|
|
|
|
@utils.cached(1)
|
|
def python_modules():
|
|
result = set()
|
|
lib_path = _stdlib_path()
|
|
if os.path.exists(lib_path):
|
|
for name in os.listdir(lib_path):
|
|
path = os.path.join(lib_path, name)
|
|
if os.path.isdir(path):
|
|
if "-" not in name:
|
|
result.add(name)
|
|
else:
|
|
if name.endswith(".py"):
|
|
result.add(name[:-3])
|
|
return result
|
|
|
|
|
|
def normalize_so_name(name):
|
|
"""
|
|
Handle different types of python installations
|
|
"""
|
|
if "cpython" in name:
|
|
return os.path.splitext(os.path.splitext(name)[0])[0]
|
|
# XXX: Special handling for Fedora python2 distribution
|
|
# See: https://github.com/python-rope/rope/issues/211
|
|
if name == "timemodule.so":
|
|
return "time"
|
|
return os.path.splitext(name)[0]
|
|
|
|
|
|
@utils.cached(1)
|
|
def dynload_modules():
|
|
result = set(sys.builtin_module_names)
|
|
dynload_path = os.path.join(_stdlib_path(), "lib-dynload")
|
|
if os.path.exists(dynload_path):
|
|
for name in os.listdir(dynload_path):
|
|
path = os.path.join(dynload_path, name)
|
|
if os.path.isfile(path):
|
|
if name.endswith(".dll"):
|
|
result.add(normalize_so_name(name))
|
|
if name.endswith(".so"):
|
|
result.add(normalize_so_name(name))
|
|
return result
|