Moved compilation of regex to VAR_MATCH

master
Ruben Dahl 2023-04-09 20:12:31 +02:00
parent 44f88345a2
commit 5d1935023a
No known key found for this signature in database
GPG Key ID: D9B193810A18673E
1 changed files with 11 additions and 6 deletions

View File

@ -1,6 +1,7 @@
from dataclasses import dataclass from dataclasses import dataclass
from enum import auto, Enum from enum import auto, Enum
from typing import Any from typing import Any
import re
class Datatype(Enum): class Datatype(Enum):
@ -40,11 +41,15 @@ class Variable:
NONE_TYPE = Variable(None, Datatype.NONE, None) NONE_TYPE = Variable(None, Datatype.NONE, None)
STR_MATCH_BEGIN = re.compile(r"^\"")
STR_MATCH_END = re.compile(r"\"$")
VAR_ASSIGN_TOKEN = re.compile(r"[A-Za-z]+ ?= ?")
VAR_MATCH = { VAR_MATCH = {
r"null": Datatype.NULL, re.compile(r"null"): Datatype.NULL,
r"\/\/.+": Datatype.COMMENT, re.compile(r"\/\/.+"): Datatype.COMMENT,
r"\".+\"": Datatype.STRING, re.compile(r"\".+\""): Datatype.STRING,
r"-?\d+\.?\d{0,}?\+\d+\.?\d{0,}?i": Datatype.COMPLEX, re.compile(r"-?\d+\.?\d{0,}?\+\d+\.?\d{0,}?i"): Datatype.COMPLEX,
r"-?\d+\.\d{0,}": Datatype.FLOAT, re.compile(r"-?\d+\.\d{0,}"): Datatype.FLOAT,
r"-?\d+": Datatype.INTEGER, re.compile(r"-?\d+"): Datatype.INTEGER,
} }