Added support for comments, negative numbers and complex

master
Ruben Dahl 2023-04-09 12:58:58 +02:00
parent 465bbe9c15
commit 252c8292dc
No known key found for this signature in database
GPG Key ID: D9B193810A18673E
1 changed files with 6 additions and 3 deletions

View File

@ -6,9 +6,11 @@ from typing import Any
class Datatype(Enum):
NONE = auto()
NULL = auto()
COMMENT = auto()
STRING = auto()
INTEGER = auto()
FLOAT = auto()
COMPLEX = auto()
@dataclass
@ -22,8 +24,9 @@ NONE_TYPE = Variable(None, Datatype.NONE, None)
VAR_MATCH = {
r"null": Datatype.NULL,
r"\/\/.+": Datatype.COMMENT,
r"\".+\"": Datatype.STRING,
r"\d+\.\d{0,}": Datatype.FLOAT,
r"\d+": Datatype.INTEGER,
# r".+": Datatype.NONE,
r"-?\d+\.?\d{0,}?\+\d+\.?\d{0,}?i": Datatype.COMPLEX,
r"-?\d+\.\d{0,}": Datatype.FLOAT,
r"-?\d+": Datatype.INTEGER,
}