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

            yield pkg

def iter_pkgs(part, all_pkgs):

    for pkg in has_pkg_section(part, "build-packages"):

        if pkg not in all_pkgs:

            all_pkgs.append(pkg)

    for pkg in has_pkg_section(part, "stage-packages"):

        if pkg not in all_pkgs:

            all_pkgs.append(pkg)

def parse(yaml_file):

    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)

    return " ".join(all_pkgs)

if __name__ == "__main__":

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