Added support for comments, negative numbers and complex

datatypes.py

@@ -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,
 }