-
Notifications
You must be signed in to change notification settings - Fork 99
/
kci_rootfs.py
executable file
·158 lines (129 loc) · 5.22 KB
/
kci_rootfs.py
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
#!/usr/bin/env python3
#
# Copyright (C) 2019 Collabora Limited
# Author: Michal Galka <[email protected]>
#
# This module is free software; you can redistribute it and/or modify it under
# the terms of the GNU Lesser General Public License as published by the Free
# Software Foundation; either version 2.1 of the License, or (at your option)
# any later version.
#
# This library is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
# details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this library; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
import sys
from pathlib import Path
sys.path.append(str(Path(__file__).parent.parent.parent))
sys.path.append(str(Path(__file__).parent))
from kernelci.legacy.cli import Args, Command, parse_opts # noqa: E402
import kernelci.config # noqa: E402
import kernelci.rootfs # noqa: E402
import kernelci.storage # noqa: E402
# -----------------------------------------------------------------------------
# Commands
#
class cmd_validate(Command):
help = "Validate the YAML configuration"
opt_args = [Args.verbose]
def __call__(self, config_data, args, **kwargs):
# ToDo: Use jsonschema
entries = [
'rootfs_configs',
]
err = kernelci.config.validate_yaml(args.yaml_config, entries)
if err:
print(err)
return False
return True
class cmd_list_configs(Command):
help = "List all rootfs config names"
opt_args = [Args.rootfs_type]
def __call__(self, config_data, args, **kwargs):
rootfs_configs = config_data['rootfs_configs']
if args.rootfs_type is None:
config_names = rootfs_configs
else:
config_names = (name
for name, config_data in rootfs_configs.items()
if config_data.rootfs_type == args.rootfs_type)
for config_name in config_names:
print(config_name)
return True
class cmd_list_variants(Command):
help = "List all rootfs variants"
append_action = {
'action': 'append'
}
arch_arg = Args.arch.copy()
arch_arg.update(append_action)
rootfs_type_arg = Args.rootfs_type.copy()
rootfs_type_arg.update(append_action)
opt_args = [Args.rootfs_config, arch_arg, rootfs_type_arg]
def __call__(self, configs, args, **kwargs):
rootfs_configs = configs['rootfs_configs']
arch_requested = set(args.arch) if args.arch else set()
if args.rootfs_config and args.rootfs_type:
print("--rootfs-config and --rootfs-type can't be used at once")
return False
if args.rootfs_config:
config = rootfs_configs.get(args.rootfs_config)
if not config:
print("{} : invalid config entry".format(args.rootfs_config))
return False
build_configs = [config]
elif args.rootfs_type:
build_configs = list(
config
for config in rootfs_configs.values()
if config.rootfs_type in args.rootfs_type
)
else:
build_configs = list(rootfs_configs.values())
for config in build_configs:
arch_available = set(config.arch_list)
arch_set = (arch_requested & arch_available
if arch_requested
else arch_available)
for arch in arch_set:
print(' '.join([config.name, arch, config.rootfs_type]))
return True
class cmd_build(Command):
help = "Build a rootfs image"
args = [Args.rootfs_config, Args.arch, Args.output]
opt_args = [Args.data_path]
def __call__(self, config_data, args, **kwargs):
config_name = args.rootfs_config
config = config_data['rootfs_configs'].get(config_name)
if not config:
print("{} invalid. Check 'kci_rootfs list_configs' for valid \
entries".format(config_name))
return False
data_path = (
args.data_path or 'config/rootfs/{}'.format(config.rootfs_type)
)
return kernelci.rootfs.build(config_name, config, data_path,
args.arch, args.output)
class cmd_upload(Command):
help = "Upload a rootfs image"
args = [Args.rootfs_dir, Args.upload_path, Args.storage_config]
opt_args = [Args.storage_cred]
def __call__(self, configs, args):
storage_conf = configs['storage_configs'][args.storage_config]
storage = kernelci.storage.get_storage(storage_conf, args.storage_cred)
kernelci.rootfs.upload(storage, args.upload_path, args.rootfs_dir)
return True
# -----------------------------------------------------------------------------
# Main
#
def main():
opts = parse_opts("kci_rootfs", globals())
configs = kernelci.config.load(opts.get_yaml_configs())
status = opts.command(configs, opts)
sys.exit(0 if status is True else 1)
if __name__ == '__main__':
main()