1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
#!/usr/bin/python
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)
|
