Source code for pycropml.transpiler.antlr_py.grammars.PythonParserBase

import sys
from enum import Enum
from typing import TextIO
from antlr4 import Parser, TokenStream



class PythonVersion(Enum):
    Autodetect = 0
    Python2 = 2
    Python3 = 3




class PythonParserBase(Parser):
    def __init__(self, input_stream: TokenStream, output: TextIO = sys.stdout):
        super().__init__(input_stream, output)
        self.__version = PythonVersion.Autodetect

    @property
    def version(self) -> PythonVersion:
        return self.__version

    @version.setter
    def version(self, version: PythonVersion):
        if isinstance(version, PythonVersion):
            self.__version = version
        else:
            self.__version = PythonVersion(version)



    def CheckVersion(self, version: int) -> bool:
        return self.__version == PythonVersion.Autodetect or version == self.__version.value




    def SetVersion(self, required_version: int) -> None:
        self.__version = PythonVersion(required_version)