bazarr/libs/pygments/lexers/q.py

236 lines
7.5 KiB
Python

"""
pygments.lexers.q
~~~~~~~~~~~~~~~~~
Lexer for the Q programming language.
:copyright: Copyright 2006-2022 by the Pygments team, see AUTHORS.
:license: BSD, see LICENSE for details.
"""
from pygments.lexer import RegexLexer, words, include, bygroups, inherit
from pygments.token import (
Comment,
Keyword,
Name,
Number,
Operator,
Punctuation,
String,
Text,
Whitespace,
Literal,
Generic,
)
__all__ = ["KLexer", "QLexer"]
class KLexer(RegexLexer):
"""
For `K <https://code.kx.com/>`_ source code.
.. versionadded:: 2.12
"""
name = "K"
aliases = ["k"]
filenames = ["*.k"]
tokens = {
"whitespace": [
# hashbang script
(r"^#!.*", Comment.Hashbang),
# Comments
(r"^/\s*\n", Comment.Multiline, "comments"),
(r"(?<!\S)/.*", Comment.Single),
# Whitespace
(r"\s+", Whitespace),
# Strings
(r"\"", String.Double, "strings"),
],
"root": [
include("whitespace"),
include("keywords"),
include("declarations"),
],
"keywords": [
(
words(
("abs", "acos", "asin", "atan", "avg", "bin",
"binr", "by", "cor", "cos", "cov", "dev",
"delete", "div", "do", "enlist", "exec", "exit",
"exp", "from", "getenv", "hopen", "if", "in",
"insert", "last", "like", "log", "max", "min",
"prd", "select", "setenv", "sin", "sqrt", "ss",
"sum", "tan", "update", "var", "wavg", "while",
"within", "wsum", "xexp"),
suffix=r"\b",
),
Operator.Word,
),
],
"declarations": [
# Timing
(r"^\\ts?", Comment.Preproc),
(
r"^(\\\w\s+[^/\n]*?)(/.*)",
bygroups(Comment.Preproc, Comment.Single),
),
# Generic System Commands
(r"^\\\w.*", Comment.Preproc),
# Prompt
(r"^[a-zA-Z]\)", Generic.Prompt),
# Function Names
(
r"([.]?[a-zA-Z][\w.]*)(\s*)([-.~=!@#$%^&*_+|,<>?/\\:']?:)(\s*)(\{)",
bygroups(Name.Function, Whitespace, Operator, Whitespace, Punctuation),
"functions",
),
# Variable Names
(
r"([.]?[a-zA-Z][\w.]*)(\s*)([-.~=!@#$%^&*_+|,<>?/\\:']?:)",
bygroups(Name.Variable, Whitespace, Operator),
),
# Functions
(r"\{", Punctuation, "functions"),
# Parentheses
(r"\(", Punctuation, "parentheses"),
# Brackets
(r"\[", Punctuation, "brackets"),
# Errors
(r"'`([a-zA-Z][\w.]*)?", Name.Exception),
# File Symbols
(r"`:([a-zA-Z/][\w./]*)?", String.Symbol),
# Symbols
(r"`([a-zA-Z][\w.]*)?", String.Symbol),
# Numbers
include("numbers"),
# Variable Names
(r"[a-zA-Z][\w.]*", Name),
# Operators
(r"[-=+*#$%@!~^&:.,<>'\\|/?_]", Operator),
# Punctuation
(r";", Punctuation),
],
"functions": [
include("root"),
(r"\}", Punctuation, "#pop"),
],
"parentheses": [
include("root"),
(r"\)", Punctuation, "#pop"),
],
"brackets": [
include("root"),
(r"\]", Punctuation, "#pop"),
],
"numbers": [
# Binary Values
(r"[01]+b", Number.Bin),
# Nulls/Infinities
(r"0[nNwW][cefghijmndzuvtp]?", Number),
# Timestamps
(
(
r"(?:[0-9]{4}[.][0-9]{2}[.][0-9]{2}|[0-9]+)"
"D(?:[0-9](?:[0-9](?::[0-9]{2}"
"(?::[0-9]{2}(?:[.][0-9]*)?)?)?)?)?"
),
Literal.Date,
),
# Datetimes
(
(
r"[0-9]{4}[.][0-9]{2}"
"(?:m|[.][0-9]{2}(?:T(?:[0-9]{2}:[0-9]{2}"
"(?::[0-9]{2}(?:[.][0-9]*)?)?)?)?)"
),
Literal.Date,
),
# Times
(
(r"[0-9]{2}:[0-9]{2}(?::[0-9]{2}(?:[.][0-9]{1,3})?)?"),
Literal.Date,
),
# GUIDs
(
r"[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}",
Number.Hex,
),
# Byte Vectors
(r"0x[0-9a-fA-F]+", Number.Hex),
# Floats
(
r"([0-9]*[.]?[0-9]+|[0-9]+[.]?[0-9]*)[eE][+-]?[0-9]+[ef]?",
Number.Float,
),
(r"([0-9]*[.][0-9]+|[0-9]+[.][0-9]*)[ef]?", Number.Float),
(r"[0-9]+[ef]", Number.Float),
# Characters
(r"[0-9]+c", Number),
# Integers
(r"[0-9]+[ihtuv]", Number.Integer),
# Long Integers
(r"[0-9]+[jnp]?", Number.Integer.Long),
],
"comments": [
(r"[^\\]+", Comment.Multiline),
(r"^\\", Comment.Multiline, "#pop"),
(r"\\", Comment.Multiline),
],
"strings": [
(r'[^"\\]+', String.Double),
(r"\\.", String.Escape),
(r'"', String.Double, "#pop"),
],
}
class QLexer(KLexer):
"""
For `Q <https://code.kx.com/>`_ source code.
.. versionadded:: 2.12
"""
name = "Q"
aliases = ["q"]
filenames = ["*.q"]
tokens = {
"root": [
(
words(
("aj", "aj0", "ajf", "ajf0", "all", "and", "any",
"asc", "asof", "attr", "avgs", "ceiling", "cols",
"count", "cross", "csv", "cut", "deltas", "desc",
"differ", "distinct", "dsave", "each", "ej",
"ema", "eval", "except", "fby", "fills", "first",
"fkeys", "flip", "floor", "get", "group", "gtime",
"hclose", "hcount", "hdel", "hsym", "iasc",
"idesc", "ij", "ijf", "inter", "inv", "key",
"keys", "lj", "ljf", "load", "lower", "lsq",
"ltime", "ltrim", "mavg", "maxs", "mcount", "md5",
"mdev", "med", "meta", "mins", "mmax", "mmin",
"mmu", "mod", "msum", "neg", "next", "not",
"null", "or", "over", "parse", "peach", "pj",
"prds", "prior", "prev", "rand", "rank", "ratios",
"raze", "read0", "read1", "reciprocal", "reval",
"reverse", "rload", "rotate", "rsave", "rtrim",
"save", "scan", "scov", "sdev", "set", "show",
"signum", "ssr", "string", "sublist", "sums",
"sv", "svar", "system", "tables", "til", "trim",
"txf", "type", "uj", "ujf", "ungroup", "union",
"upper", "upsert", "value", "view", "views", "vs",
"where", "wj", "wj1", "ww", "xasc", "xbar",
"xcol", "xcols", "xdesc", "xgroup", "xkey",
"xlog", "xprev", "xrank"),
suffix=r"\b",
),
Name.Builtin,
),
inherit,
],
}