Get Insights Core version from Insights Client RPM

get_core_version_from_insights_client_rpm.py

from contextlib import contextmanager
from os.path import join
from rpmfile import open as rpm_open
from sys import argv
from zipfile import ZipFile

EGG_PATH = join(".", "etc", "insights-client", "rpm.egg")
VERSION_PATH = join("insights", "VERSION")


@contextmanager
def get_egg(rpm_path):
    with rpm_open(rpm_path) as rpm_file:
        yield rpm_file.extractfile(EGG_PATH)


def get_version(egg):
    zip_file = ZipFile(egg)
    version_bytes = zip_file.read(VERSION_PATH)
    version_string = version_bytes.decode("utf-8")
    return version_string.rstrip()


def main(rpm_path):
    with get_egg(rpm_path) as egg:
        version = get_version(egg)
    print(version)


if __name__ == "__main__":
    if len(argv) == 2:
        rpm_path_arg = argv[1]
    else:
        raise ValueError("Invalid argument count.")

    main(rpm_path_arg)

requirements.txt

rpmfile ~= "1.0.8"