variables.py

from dataclasses import dataclass, field
from enum import Enum
from typing import Optional

from gboml.ast.base import GBOMLObject
from gboml.ast.path import VarOrParam
from gboml.ast.rvalue import RValue


class VarScope(Enum):
    internal = "internal"
    external = "external"


class VarType(Enum):
    continuous = "continuous"
    integer = "integer"
    binary = "binary"


class DefinitionType(Enum):
    constant = "="
    expression = "<-"

@dataclass
class Definition(GBOMLObject):
    name: str
    type: DefinitionType
    value: RValue
    tags: list[str] = field(default_factory=list)


@dataclass
class VariableDefinition(GBOMLObject):
    scope: VarScope
    type: VarType
    name: VarOrParam
    import_from: Optional[VarOrParam] = None
    tags: list[str] = field(default_factory=list)


@dataclass
class ScopeChange(GBOMLObject):
    var: str
    scope: VarScope