Files
OpenNetworkLinux/tools/onlplatform.py
Wataru Ishida 224fe6a7cf [wip] tools: python3
Signed-off-by: Wataru Ishida <ishida@nel-america.com>
2020-02-18 01:43:13 -08:00

86 lines
2.6 KiB
Python
Executable File

#!/usr/bin/python3
"""onlplatform.py
Extract install file requirements from the platform YAML file and/or
the platform package metadata.
"""
import sys, os
import itertools
import argparse
toolsdir = os.path.dirname(os.path.abspath(__file__))
sys.path.append(toolsdir)
onldir = os.path.dirname(toolsdir)
onlpydir = os.path.join(onldir, "packages/base/all/vendor-config-onl/src/python")
sys.path.append(onlpydir)
import onl.YamlUtils
from onlpm import *
# glob import is required here so pickle load load properly
pm = defaultPm()
ap = argparse.ArgumentParser("ONL Platform Data Extractor.")
ap.add_argument("platform", help="Platform name")
ap.add_argument("arch", help="Architecture")
ap.add_argument("key", help="Lookup key.")
ops = ap.parse_args()
def extractKey(platform, arch, key):
pkg = "onl-platform-config-%s:%s" % (platform, arch,)
basename = "%s.yml" % platform
pm.require(pkg, force=False, build_missing=False)
platformConfigPath = pm.opr.get_file(pkg, basename)
if arch in ('amd64',):
pkg = "onl-vendor-config-onl:all"
basename = "platform-config-defaults-x86-64.yml"
subkey = 'grub'
else:
pkg = "onl-vendor-config-onl:all"
basename = "platform-config-defaults-uboot.yml"
subkey = 'flat_image_tree'
pm.require(pkg, force=False, build_missing=False)
defaultConfigPath = pm.opr.get_file(pkg, basename)
platformConf = onl.YamlUtils.merge(defaultConfigPath, platformConfigPath)
resource = platformConf[platform][subkey][key]
if type(resource) == dict:
pkg = resource['package']
basename = resource['=']
else:
pkg, sep, basename = resource.partition(',')
if not sep:
raise ValueError("resource missing package declaration: %s" % resource)
pkg = pkg.strip()
basename = basename.strip()
pm.require(pkg, force=False, build_missing=False)
resourcePath = pm.opr.get_file(pkg, basename)
return resourcePath
def extractVendor(platform, arch):
pkg = "onl-platform-config-%s:%s" % (platform, arch,)
l = pm.opr.lookup_all(pkg)
if not l:
raise SystemExit("cannot find package %s:%s"
% (platform, arch,))
l = [x for x in pm.package_groups if pkg in x]
l = list(itertools.chain(*[x.prerequisite_packages() for x in l]))
l = [x for x in l if x.startswith('onl-vendor-config-')]
return "\n".join(l)
if ops.key in ('kernel', 'initrd', 'dtb', 'itb',):
print((extractKey(ops.platform, ops.arch, ops.key)))
sys.exit(0)
if ops.key == 'vendor':
print((extractVendor(ops.platform, ops.arch)))
sys.exit(0)
raise SystemExit("invalid key %s" % ops.key)