2023-04-08 20:07:28 +00:00
|
|
|
from dataclasses import dataclass
|
|
|
|
from enum import auto, Enum
|
|
|
|
from typing import Any
|
|
|
|
|
|
|
|
|
|
|
|
class Datatype(Enum):
|
|
|
|
NONE = auto()
|
2023-04-08 21:12:17 +00:00
|
|
|
NULL = auto()
|
2023-04-09 10:58:58 +00:00
|
|
|
COMMENT = auto()
|
2023-04-08 20:07:28 +00:00
|
|
|
STRING = auto()
|
|
|
|
INTEGER = auto()
|
|
|
|
FLOAT = auto()
|
2023-04-09 10:58:58 +00:00
|
|
|
COMPLEX = auto()
|
2023-04-08 20:07:28 +00:00
|
|
|
|
|
|
|
|
2023-04-09 15:04:56 +00:00
|
|
|
@dataclass
|
|
|
|
class Data:
|
|
|
|
type: Datatype
|
|
|
|
value: Any
|
|
|
|
|
|
|
|
|
2023-04-08 20:07:28 +00:00
|
|
|
@dataclass
|
|
|
|
class Variable:
|
|
|
|
name: str
|
|
|
|
type: Datatype
|
2023-04-09 15:04:56 +00:00
|
|
|
value: Data
|
|
|
|
|
|
|
|
def __setattr__(self, name, value):
|
|
|
|
if name == "value" and isinstance(value, Data):
|
|
|
|
if self.type == value.type:
|
|
|
|
object.__setattr__(self, name, value)
|
|
|
|
else:
|
|
|
|
raise ValueError(
|
|
|
|
"Type inequality; cannot assign value of "
|
|
|
|
f"type {value.type} to variable of type {self.type}"
|
|
|
|
)
|
|
|
|
else:
|
|
|
|
object.__setattr__(self, name, value)
|
2023-04-08 21:12:17 +00:00
|
|
|
|
|
|
|
|
|
|
|
NONE_TYPE = Variable(None, Datatype.NONE, None)
|
|
|
|
|
|
|
|
VAR_MATCH = {
|
|
|
|
r"null": Datatype.NULL,
|
2023-04-09 10:58:58 +00:00
|
|
|
r"\/\/.+": Datatype.COMMENT,
|
2023-04-08 21:12:17 +00:00
|
|
|
r"\".+\"": Datatype.STRING,
|
2023-04-09 10:58:58 +00:00
|
|
|
r"-?\d+\.?\d{0,}?\+\d+\.?\d{0,}?i": Datatype.COMPLEX,
|
|
|
|
r"-?\d+\.\d{0,}": Datatype.FLOAT,
|
|
|
|
r"-?\d+": Datatype.INTEGER,
|
2023-04-08 21:12:17 +00:00
|
|
|
}
|