menderbot.antlr_generated.PythonParserBase

Complexity

Total Complexity

6

Size/Duplication

Total Lines	34
Duplicated Lines	61.76 %

Importance

Changes

0

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):
        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)