1  import sys
       2  from pathlib import Path
       3  
       4  CPYTHON_ROOT = Path(__file__).resolve().parent.parent.parent.parent
       5  sys.path.append(str(CPYTHON_ROOT / "Parser"))
       6  
       7  from pygments.lexer import RegexLexer, bygroups, include, words
       8  from pygments.token import (Comment, Keyword, Name, Operator,
       9                              Punctuation, Text)
      10  
      11  from asdl import builtin_types
      12  from sphinx.highlighting import lexers
      13  
      14  class ESC[4;38;5;81mASDLLexer(ESC[4;38;5;149mRegexLexer):
      15      name = "ASDL"
      16      aliases = ["asdl"]
      17      filenames = ["*.asdl"]
      18      _name = r"([^\W\d]\w*)"
      19      _text_ws = r"(\s*)"
      20  
      21      tokens = {
      22          "ws": [
      23              (r"\n", Text),
      24              (r"\s+", Text),
      25              (r"--.*?$", Comment.Singleline),
      26          ],
      27          "root": [
      28              include("ws"),
      29              (
      30                  r"(module)" + _text_ws + _name,
      31                  bygroups(Keyword, Text, Name.Tag),
      32              ),
      33              (
      34                  r"(\w+)(\*\s|\?\s|\s)(\w+)",
      35                  bygroups(Name.Builtin.Pseudo, Operator, Name),
      36              ),
      37              (words(builtin_types), Name.Builtin),
      38              (r"attributes", Name.Builtin),
      39              (
      40                  _name + _text_ws + "(=)",
      41                  bygroups(Name, Text, Operator),
      42              ),
      43              (_name, Name.Class),
      44              (r"\|", Operator),
      45              (r"{|}|\(|\)", Punctuation),
      46              (r".", Text),
      47          ],
      48      }
      49  
      50  
      51  def setup(app):
      52      lexers["asdl"] = ASDLLexer()
      53      return {'version': '1.0', 'parallel_read_safe': True}