Source code for sourced.ml.core.utils.bblfsh

from distutils.version import StrictVersion

from bblfsh import BblfshClient


# this is versions of bblfsh server, not python package

[docs]BBLFSH_VERSION_LOW = "2.2"


[docs]BBLFSH_VERSION_HIGH = "3.0"




[docs]def check_version(host: str = "0.0.0.0", port: str = "9432") -> bool:
    """
    Check if the bblfsh server version matches module requirements.

    :param host: bblfsh server host.
    :param port: bblfsh server port.
    :return: True if bblfsh version specified matches requirements.
    """
    # get version and remove leading 'v'
    version = StrictVersion(BblfshClient("%s:%s" % (host, port)).version().version.lstrip("v"))
    return StrictVersion(BBLFSH_VERSION_LOW) <= version < StrictVersion(BBLFSH_VERSION_HIGH)