keywords should be part of the api package

jedi/api/__init__.py

@@ -23,7 +23,7 @@ from jedi import settings
 from jedi import common
 from jedi import cache
 from jedi import interpret
-from jedi import keywords
+from jedi.api import keywords
 from jedi.api import classes
 from jedi.evaluate import Evaluator, filter_private_variable
 from jedi.evaluate import representation as er

jedi/api/classes.py

@@ -9,13 +9,13 @@ import functools
 from jedi._compatibility import unicode, next
 from jedi import settings
 from jedi import common
-from jedi.parser import representation as pr
 from jedi import cache
+from jedi.parser import representation as pr
 from jedi.evaluate import representation as er
 from jedi.evaluate import iterable
 from jedi.evaluate import imports
 from jedi.evaluate import compiled
-from jedi import keywords
+from jedi.api import keywords
 
 
 def clear_caches():

jedi/api/keywords.py

@@ -0,0 +1,100 @@
+import pydoc
+import keyword
+
+from jedi._compatibility import is_py3k
+from jedi import common
+from jedi.evaluate import compiled
+
+try:
+    from pydoc_data import topics as pydoc_topics
+except ImportError:
+    # Python 2.6
+    import pydoc_topics
+
+if is_py3k:
+    keys = keyword.kwlist
+else:
+    keys = keyword.kwlist + ['None', 'False', 'True']
+
+
+def keywords(string='', pos=(0, 0), all=False):
+    if all:
+        return set([Keyword(k, pos) for k in keys])
+    if string in keys:
+        return set([Keyword(string, pos)])
+    return set()
+
+
+def keyword_names(*args, **kwargs):
+    kwds = []
+    for k in keywords(*args, **kwargs):
+        start = k.start_pos
+        kwds.append(KeywordName(k, k.name, start))
+    return kwds
+
+
+def get_operator(string, pos):
+    return Keyword(string, pos)
+
+
+class KeywordName(object):
+    def __init__(self, parent, name, start_pos):
+        self.parent = parent
+        self.names = [name]
+        self.start_pos = start_pos
+
+    @property
+    def end_pos(self):
+        return self.start_pos[0], self.start_pos[1] + len(self.name)
+
+
+class Keyword(object):
+    def __init__(self, name, pos):
+        self.name = name
+        self.start_pos = pos
+        self.parent = compiled.builtin
+
+    def get_parent_until(self):
+        return self.parent
+
+    @property
+    def names(self):
+        """ For a `parsing.Name` like comparision """
+        return [self.name]
+
+    @property
+    def docstr(self):
+        return imitate_pydoc(self.name)
+
+    def __repr__(self):
+        return '<%s: %s>' % (type(self).__name__, self.name)
+
+
+def imitate_pydoc(string):
+    """
+    It's not possible to get the pydoc's without starting the annoying pager
+    stuff.
+    """
+    # str needed because of possible unicode stuff in py2k (pydoc doesn't work
+    # with unicode strings)
+    string = str(string)
+    h = pydoc.help
+    with common.ignored(KeyError):
+        # try to access symbols
+        string = h.symbols[string]
+        string, _, related = string.partition(' ')
+
+    get_target = lambda s: h.topics.get(s, h.keywords.get(s))
+    while isinstance(string, str):
+        string = get_target(string)
+
+    try:
+        # is a tuple now
+        label, related = string
+    except TypeError:
+        return ''
+
+    try:
+        return pydoc_topics.topics[label] if pydoc_topics else ''
+    except KeyError:
+        return ''