aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/ipython/py2/IPython/core/completerlib.py
diff options
context:
space:
mode:
authorshadchin <shadchin@yandex-team.ru>2022-02-10 16:44:39 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:44:39 +0300
commite9656aae26e0358d5378e5b63dcac5c8dbe0e4d0 (patch)
tree64175d5cadab313b3e7039ebaa06c5bc3295e274 /contrib/python/ipython/py2/IPython/core/completerlib.py
parent2598ef1d0aee359b4b6d5fdd1758916d5907d04f (diff)
downloadydb-e9656aae26e0358d5378e5b63dcac5c8dbe0e4d0.tar.gz
Restoring authorship annotation for <shadchin@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'contrib/python/ipython/py2/IPython/core/completerlib.py')
-rw-r--r--contrib/python/ipython/py2/IPython/core/completerlib.py110
1 files changed, 55 insertions, 55 deletions
diff --git a/contrib/python/ipython/py2/IPython/core/completerlib.py b/contrib/python/ipython/py2/IPython/core/completerlib.py
index d31357d0e1..e736ca73d1 100644
--- a/contrib/python/ipython/py2/IPython/core/completerlib.py
+++ b/contrib/python/ipython/py2/IPython/core/completerlib.py
@@ -19,7 +19,7 @@ from __future__ import print_function
# Stdlib imports
import glob
import inspect
-import itertools
+import itertools
import os
import re
import sys
@@ -45,8 +45,8 @@ from IPython.utils.py3compat import string_types
# FIXME: this should be pulled in with the right call via the component system
from IPython import get_ipython
-from __res import importer
-
+from __res import importer
+
#-----------------------------------------------------------------------------
# Globals and constants
#-----------------------------------------------------------------------------
@@ -71,51 +71,51 @@ magic_run_re = re.compile(r'.*(\.ipy|\.ipynb|\.py[w]?)$')
# Local utilities
#-----------------------------------------------------------------------------
-arcadia_rootmodules_cache = None
-arcadia_modules_cache = None
-
-
-def arcadia_init_cache():
- global arcadia_rootmodules_cache, arcadia_modules_cache
- arcadia_rootmodules_cache = set()
- arcadia_modules_cache = {}
-
- all_modules = itertools.chain(
- sys.builtin_module_names,
- importer.memory
- )
-
- for name in all_modules:
- path = name.split('.')
- arcadia_rootmodules_cache.add(path[0])
-
- prefix = path[0]
- for element in path[1:]:
- if element == '__init__':
- continue
-
- arcadia_modules_cache.setdefault(prefix, set()).add(element)
- prefix += '.' + element
-
- arcadia_rootmodules_cache = sorted(arcadia_rootmodules_cache)
- arcadia_modules_cache = {k: sorted(v) for k, v in arcadia_modules_cache.items()}
-
-
-def arcadia_module_list(mod):
- if arcadia_modules_cache is None:
- arcadia_init_cache()
-
- return arcadia_modules_cache.get(mod, ())
-
-
-def arcadia_get_root_modules():
- if arcadia_rootmodules_cache is None:
- arcadia_init_cache()
-
- return arcadia_rootmodules_cache
-
-
-
+arcadia_rootmodules_cache = None
+arcadia_modules_cache = None
+
+
+def arcadia_init_cache():
+ global arcadia_rootmodules_cache, arcadia_modules_cache
+ arcadia_rootmodules_cache = set()
+ arcadia_modules_cache = {}
+
+ all_modules = itertools.chain(
+ sys.builtin_module_names,
+ importer.memory
+ )
+
+ for name in all_modules:
+ path = name.split('.')
+ arcadia_rootmodules_cache.add(path[0])
+
+ prefix = path[0]
+ for element in path[1:]:
+ if element == '__init__':
+ continue
+
+ arcadia_modules_cache.setdefault(prefix, set()).add(element)
+ prefix += '.' + element
+
+ arcadia_rootmodules_cache = sorted(arcadia_rootmodules_cache)
+ arcadia_modules_cache = {k: sorted(v) for k, v in arcadia_modules_cache.items()}
+
+
+def arcadia_module_list(mod):
+ if arcadia_modules_cache is None:
+ arcadia_init_cache()
+
+ return arcadia_modules_cache.get(mod, ())
+
+
+def arcadia_get_root_modules():
+ if arcadia_rootmodules_cache is None:
+ arcadia_init_cache()
+
+ return arcadia_rootmodules_cache
+
+
+
def module_list(path):
"""
Return the list containing the names of the modules available in the given
@@ -196,7 +196,7 @@ def get_root_modules():
rootmodules.extend(modules)
if store:
ip.db['rootmodules_cache'] = rootmodules_cache
- rootmodules = list(set(rootmodules))
+ rootmodules = list(set(rootmodules))
return rootmodules
@@ -216,8 +216,8 @@ def try_import(mod, only_modules=False):
for module in mods[1:]:
m = getattr(m, module)
- filename = getattr(m, '__file__', '')
- m_is_init = '__init__' in (filename or '') or filename == mod
+ filename = getattr(m, '__file__', '')
+ m_is_init = '__init__' in (filename or '') or filename == mod
completions = []
if (not hasattr(m, '__file__')) or (not only_modules) or m_is_init:
@@ -226,10 +226,10 @@ def try_import(mod, only_modules=False):
completions.extend(getattr(m, '__all__', []))
if m_is_init:
- completions.extend(arcadia_module_list(mod))
+ completions.extend(arcadia_module_list(mod))
completions = {c for c in completions if isinstance(c, string_types)}
completions.discard('__init__')
- return sorted(completions)
+ return sorted(completions)
#-----------------------------------------------------------------------------
@@ -278,10 +278,10 @@ def module_completion(line):
# 'from xy<tab>' or 'import xy<tab>'
if nwords < 3 and (words[0] in {'%aimport', 'import', 'from'}) :
if nwords == 1:
- return arcadia_get_root_modules()
+ return arcadia_get_root_modules()
mod = words[1].split('.')
if len(mod) < 2:
- return arcadia_get_root_modules()
+ return arcadia_get_root_modules()
completion_list = try_import('.'.join(mod[:-1]), True)
return ['.'.join(mod[:-1] + [el]) for el in completion_list]