jeans/LibreELEC.tv
jeans/LibreELEC.tv
1
0
Fork 0
mirror of https://github.com/LibreELEC/LibreELEC.tv.git synced 2025-07-29 13:46:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

uboot_helper: convert to use argparse

Browse Source
This commit is contained in:
Lukas Rusak 2021-01-18 19:40:04 -08:00
parent c3d5c939fd
commit fb54b12cfd
1 changed files with 51 additions and 61 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

112
scripts/uboot_helper
View File

@ -3,7 +3,7 @@
# SPDX-License-Identifier: GPL-2.0 # SPDX-License-Identifier: GPL-2.0
# Copyright (C) 2017-present Team LibreELEC (https://libreelec.tv) # Copyright (C) 2017-present Team LibreELEC (https://libreelec.tv)
import sys import argparse
# When adding new devices to the list please keep them in alphabetical order # When adding new devices to the list please keep them in alphabetical order
# board-name should contain 'dashes' (-) not 'underscores' (_) and shouldn't contain capitals # board-name should contain 'dashes' (-) not 'underscores' (_) and shouldn't contain capitals
@ -342,77 +342,67 @@ devices = \
}, },
} }
def usage(PROJECT=None, SOC=None, FILE=sys.stdout): choices = ['dtb', 'config', 'crust_config']
print('Usage: %s <project> <soc> <board-name> dtb|config|crust_config' % sys.argv[0], file=FILE)
print(' %s <project> <soc>' % sys.argv[0], file=FILE)
print(' %s <project>' % sys.argv[0], file=FILE)
print('', file=FILE)
if PROJECT and PROJECT not in devices: class OptionsAction(argparse.Action):
PROJECT = SOC = None
if PROJECT and SOC and SOC not in devices[PROJECT]:
SOC = None
print('Projects:', file=FILE) def __call__(self, parser, namespace, values, option_string=None):
print('', file=FILE) message = None
for project in sorted(devices):
if PROJECT is None or PROJECT == project:
print(' %s:' % project, file=FILE)
for soc in sorted(devices[project]):
if SOC is None or SOC == soc:
print(' %s:' % soc, file=FILE)
for board in sorted(devices[project][soc]):
print(' %s' % board, file=FILE)
print('', file=FILE)
print('', file=FILE)
def exit_error(msg, PROJECT=None, SOC=None): if self.dest == 'project':
print(msg, file=sys.stderr) if values is None:
print('', file=sys.stderr) print(' '.join(project for project in sorted(devices)))
usage(PROJECT=PROJECT, SOC=SOC, FILE=sys.stderr) parser.exit()
sys.exit(1)
# Basic argument validation if values not in devices.keys():
if len(sys.argv) == 2 and sys.argv[1] in ['help', 'usage']: message = "invalid choice: {0!r} (choose from {1})".format(values, ', '.join([repr(project) for project in devices]))
usage()
sys.exit(0)
if len(sys.argv) > 1 and sys.argv[1] not in devices: if self.dest == 'soc':
exit_error('Invalid project: %s' % sys.argv[1]) project = getattr(namespace, 'project')
if len(sys.argv) > 2 and sys.argv[2] not in devices[sys.argv[1]]: if values is None:
exit_error('Invalid soc: %s' % sys.argv[2], PROJECT=sys.argv[1]) print(' '.join(soc for soc in sorted(devices[project])))
parser.exit()
if len(sys.argv) > 3 and sys.argv[3] not in devices[sys.argv[1]][sys.argv[2]]: if values not in devices[project].keys():
exit_error('Invalid board-name: %s' % sys.argv[3], PROJECT=sys.argv[1], SOC=sys.argv[2]) message = "invalid choice: {0!r} (choose from {1})".format(values, ', '.join([repr(soc) for soc in devices[project]]))
if len(sys.argv) == 4: if self.dest == 'board':
exit_error('Invalid option: must specify dtb, config or crust_config', PROJECT=sys.argv[1], SOC=sys.argv[2]) project = getattr(namespace, 'project')
elif len(sys.argv) > 4 and sys.argv[4] not in ['dtb', 'config', 'crust_config']: soc = getattr(namespace, 'soc')
exit_error('Invalid option: %s' % sys.argv[4], PROJECT=sys.argv[1], SOC=sys.argv[2])
if len(sys.argv) > 5: if values is None:
exit_error('Invalid number of arguments: %s' % ' '.join(sys.argv[1:]), PROJECT=sys.argv[1], SOC=sys.argv[2]) print(' '.join(board for board in sorted(devices[project][soc])))
parser.exit()
# Get dtb, u-boot or crust config for a given project, soc, and board if values not in devices[project][soc].keys():
# ./scripts/uboot_helper project device board-name dtb|config|crust_config message = "invalid choice: {0!r} (choose from {1})".format(values, ', '.join([repr(board) for board in devices[project][soc]]))
if len(sys.argv) == 5:
if sys.argv[4] in devices[sys.argv[1]][sys.argv[2]][sys.argv[3]]:
print(devices[sys.argv[1]][sys.argv[2]][sys.argv[3]][sys.argv[4]])
# List boards supported by a given project and soc if self.dest == 'value':
# ./scripts/uboot_helper project device project = getattr(namespace, 'project')
elif len(sys.argv) == 3: soc = getattr(namespace, 'soc')
print(' '.join([board for board in sorted(devices[sys.argv[1]][sys.argv[2]])])) board = getattr(namespace, 'board')
# List socs supported by a given project if values not in choices:
# ./scripts/uboot_helper project message = "invalid choice: {0!r} (choose from {1})".format(values, ', '.join([repr(value) for value in choices]))
elif len(sys.argv) == 2:
print(' '.join([soc for soc in sorted(devices[sys.argv[1]])]))
# List projects if message is not None:
# ./scripts/uboot_helper raise argparse.ArgumentError(self, message)
elif len(sys.argv) == 1:
print(' '.join([project for project in sorted(devices)]))
sys.exit(0) setattr(namespace, self.dest, values)
parser = argparse.ArgumentParser(description='Script to help with u-boot configuration')
parser.add_argument('project', nargs='?', action=OptionsAction)
parser.add_argument('soc', nargs='?', action=OptionsAction)
parser.add_argument('board', nargs='?', action=OptionsAction)
parser.add_argument('value', nargs='?', metavar='|'.join(choices), action=OptionsAction)
options = parser.parse_args()
print(devices[options.project][options.soc][options.board][options.value])
parser.exit()