parse.py - mozsearch

Enable keyboard shortcuts

#!/usr/bin/env python3

# This Source Code Form is subject to the terms of the Mozilla Public

# License, v. 2.0. If a copy of the MPL was not distributed with this

# file, You can obtain one at http://mozilla.org/MPL/2.0/.

import sys

import yaml

def has_pkg_section(p, section):

    has_section = section in p.keys()

    if has_section:

        for pkg in p[section]:

            if type(pkg) == str:

                yield pkg

            else:

                yield from has_pkg_section(pkg, next(iter(pkg.keys())))

def iter_pkgs(part, all_pkgs, arch):

    for section in ["build-packages", "stage-packages"]:

        for pkg in has_pkg_section(part, section):

            if pkg not in all_pkgs:

                if ":" in pkg and pkg.split(":")[1] != arch:

                    continue

                all_pkgs.append(pkg)

def parse(yaml_file, arch):

    all_pkgs = []

    with open(yaml_file, "r") as inp:

        snap = yaml.safe_load(inp)

        parts = snap["parts"]

        for p in parts:

            iter_pkgs(parts[p], all_pkgs, arch)

    return " ".join(all_pkgs)

if __name__ == "__main__":

    print(parse(sys.argv[1], sys.argv[2]))