File size: 5,615 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 |
"""
pygments.lexers.vyper
~~~~~~~~~~~~~~~~~~~~~
Lexer for the Vyper Smart Contract language.
:copyright: Copyright 2006-2025 by the Pygments team, see AUTHORS.
:license: BSD, see LICENSE for details.
"""
from pygments.lexer import RegexLexer, bygroups, words
from pygments.token import (Comment, String, Name, Keyword, Number,
Operator, Punctuation, Text, Whitespace)
__all__ = ['VyperLexer']
class VyperLexer(RegexLexer):
"""For the Vyper smart contract language.
"""
name = 'Vyper'
aliases = ['vyper']
filenames = ['*.vy']
url = "https://vyper.readthedocs.io"
version_added = '2.17'
tokens = {
'root': [
# Whitespace
(r'\s+', Whitespace),
# Line continuations
(r'(\\)(\n|\r\n|\r)', bygroups(Text, Whitespace)),
# Comments - inline and multiline
(r'#.*$', Comment.Single),
(r'\"\"\"', Comment.Multiline, 'multiline-comment'),
# Strings - single and double
(r"'", String.Single, 'single-string'),
(r'"', String.Double, 'double-string'),
# Functions (working)
(r'(def)(\s+)([a-zA-Z_][a-zA-Z0-9_]*)',
bygroups(Keyword, Whitespace, Name.Function)),
# Event and Struct
(r'(event|struct|interface|log)(\s+)([a-zA-Z_][a-zA-Z0-9_]*)',
bygroups(Keyword, Whitespace, Name.Class)),
# Imports
(r'(from)(\s+)(vyper\.\w+)(\s+)(import)(\s+)(\w+)',
bygroups(Keyword, Whitespace, Name.Namespace, Whitespace,
Keyword, Whitespace, Name.Class)),
# Numeric Literals
(r'\b0x[0-9a-fA-F]+\b', Number.Hex),
(r'\b(\d{1,3}(?:_\d{3})*|\d+)\b', Number.Integer),
(r'\b\d+\.\d*\b', Number.Float),
# Keywords
(words(('def', 'event', 'pass', 'return', 'for', 'while', 'if', 'elif',
'else', 'assert', 'raise', 'import', 'in', 'struct', 'implements',
'interface', 'from', 'indexed', 'log', 'extcall', 'staticcall'),
prefix=r'\b', suffix=r'\b'), Keyword),
# Visibility and State Mutability
(words(('public', 'private', 'view', 'pure', 'constant',
'immutable', 'nonpayable'), prefix=r'\b', suffix=r'\b'),
Keyword.Declaration),
# Built-in Functions
(words(('bitwise_and', 'bitwise_not', 'bitwise_or', 'bitwise_xor', 'shift',
'create_minimal_proxy_to', 'create_copy_of', 'create_from_blueprint',
'ecadd', 'ecmul', 'ecrecover', 'keccak256', 'sha256', 'concat', 'convert',
'uint2str', 'extract32', 'slice', 'abs', 'ceil', 'floor', 'max', 'max_value',
'min', 'min_value', 'pow_mod256', 'sqrt', 'isqrt', 'uint256_addmod',
'uint256_mulmod', 'unsafe_add', 'unsafe_sub', 'unsafe_mul', 'unsafe_div',
'as_wei_value', 'blockhash', 'empty', 'len', 'method_id', '_abi_encode',
'_abi_decode', 'print', 'range'), prefix=r'\b', suffix=r'\b'),
Name.Builtin),
# Built-in Variables and Attributes
(words(('msg.sender', 'msg.value', 'block.timestamp', 'block.number', 'msg.gas'),
prefix=r'\b', suffix=r'\b'),
Name.Builtin.Pseudo),
(words(('uint', 'uint8', 'uint16', 'uint32', 'uint64', 'uint128', 'uint256',
'int', 'int8', 'int16', 'int32', 'int64', 'int128', 'int256', 'bool',
'decimal', 'bytes', 'bytes1', 'bytes2', 'bytes3', 'bytes4', 'bytes5',
'bytes6', 'bytes7', 'bytes8', 'bytes9', 'bytes10', 'bytes11',
'bytes12', 'bytes13', 'bytes14', 'bytes15', 'bytes16', 'bytes17',
'bytes18', 'bytes19', 'bytes20', 'bytes21', 'bytes22', 'bytes23',
'bytes24', 'bytes25', 'bytes26', 'bytes27', 'bytes28', 'bytes29',
'bytes30', 'bytes31', 'bytes32', 'string', 'String', 'address',
'enum', 'struct'), prefix=r'\b', suffix=r'\b'),
Keyword.Type),
# indexed keywords
(r'\b(indexed)\b(\s*)(\()(\s*)(\w+)(\s*)(\))',
bygroups(Keyword, Whitespace, Punctuation, Whitespace,
Keyword.Type, Punctuation)),
# Operators and Punctuation
(r'(\+|\-|\*|\/|<=?|>=?|==|!=|=|\||&|%)', Operator),
(r'[.,:;()\[\]{}]', Punctuation),
# Other variable names and types
(r'@[\w.]+', Name.Decorator),
(r'__\w+__', Name.Magic), # Matches double underscores followed by word characters
(r'EMPTY_BYTES32', Name.Constant),
(r'\bERC20\b', Name.Class),
(r'\bself\b', Name.Attribute),
(r'Bytes\[\d+\]', Keyword.Type),
# Generic names and variables
(r'\b[a-zA-Z_]\w*\b:', Name.Variable),
(r'\b[a-zA-Z_]\w*\b', Name),
],
'multiline-comment': [
(r'\"\"\"', Comment.Multiline, '#pop'),
(r'[^"]+', Comment.Multiline),
(r'\"', Comment.Multiline)
],
'single-string': [
(r"[^\\']+", String.Single),
(r"'", String.Single, '#pop'),
(r'\\.', String.Escape),
],
'double-string': [
(r'[^\\"]+', String.Double),
(r'"', String.Double, '#pop'),
(r'\\.', String.Escape),
]
}
|