#!/usr/bin/python
"""
Createrepo_c doesn't support indexing module metadata.
This script indexes all yaml files found in a target directory,
concatenates them and injects into repodata as "modules" mdtype.
"""
import os
import argparse
import subprocess
import tempfile
import gi
gi.require_version('Modulemd', '1.0')
from gi.repository import Modulemd
def get_parser():
"""
Construct argument parser.
:returns: ArgumentParser object with arguments set up.
:rtype: argparse.ArgumentParser
"""
parser = argparse.ArgumentParser(
description="Scan directory for modulemd yaml files and inject them into repodata.",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
)
parser.add_argument(
"path",
metavar="directory_to_index",
)
return parser
def index_modulemd_files(repo_path):
result = []
for fn in sorted(os.listdir(repo_path)):
if not fn.endswith(".yaml"):
continue
yaml_path = os.path.join(repo_path, fn)
mmd = Modulemd.objects_from_file_ext(yaml_path)
result.append(mmd[0][0])
return result
def modify_repo(repo_path, modules):
tmp = tempfile.mkdtemp()
path = os.path.join(tmp, "modules.yaml")
for module in modules:
Modulemd.dump(modules, path)
subprocess.check_call(["modifyrepo_c", "--mdtype=modules", path,
os.path.join(repo_path, "repodata")])
os.unlink(path)
os.rmdir(tmp)
if __name__ == "__main__":
parser = get_parser()
args = parser.parse_args()
modules = index_modulemd_files(args.path)
modify_repo(args.path, modules)