Little simplification in pygments_token_to_classname and PygmentsLexer code.

Author: jonathanslenders

prompt_toolkit/lexers/pygments.py

@@ -12,6 +12,7 @@
 
 from prompt_toolkit.filters import to_filter
 from prompt_toolkit.formatted_text.utils import split_lines
+from prompt_toolkit.styles.pygments import pygments_token_to_classname
 from .base import Lexer, SimpleLexer
 
 __all__ = [
@@ -116,9 +117,7 @@ class _TokenCache(dict):
  ``class:pygments,pygments.A,pygments.A.B,pygments.A.B.C``
  """
  def __missing__(self, key):
- parts = ('pygments',) + key
-
- result = ('class:' + '.'.join(parts)).lower()
+ result = 'class:' + pygments_token_to_classname(key)
  self[key] = result
  return result
 

prompt_toolkit/styles/pygments.py

@@ -57,7 +57,5 @@ def pygments_token_to_classname(token):
  (Our Pygments lexer will also turn the tokens that pygments produces in a
  prompt_toolkit list of fragments that match these styling rules.)
  """
- if token:
- return 'pygments.' + '.'.join(token).lower()
- else:
- return 'pygments'
+ parts = ('pygments', ) + token
+ return '.'.join(parts).lower()