1*4882a593Smuzhiyun#!/usr/bin/env python3 2*4882a593Smuzhiyun 3*4882a593Smuzhiyun# This script has subcommands which operate against your bitbake layers, either 4*4882a593Smuzhiyun# displaying useful information, or acting against them. 5*4882a593Smuzhiyun# See the help output for details on available commands. 6*4882a593Smuzhiyun 7*4882a593Smuzhiyun# Copyright (C) 2011 Mentor Graphics Corporation 8*4882a593Smuzhiyun# Copyright (C) 2011-2015 Intel Corporation 9*4882a593Smuzhiyun# 10*4882a593Smuzhiyun# SPDX-License-Identifier: GPL-2.0-only 11*4882a593Smuzhiyun# 12*4882a593Smuzhiyun 13*4882a593Smuzhiyunimport logging 14*4882a593Smuzhiyunimport os 15*4882a593Smuzhiyunimport sys 16*4882a593Smuzhiyunimport argparse 17*4882a593Smuzhiyunimport warnings 18*4882a593Smuzhiyunwarnings.simplefilter("default") 19*4882a593Smuzhiyun 20*4882a593Smuzhiyunbindir = os.path.dirname(__file__) 21*4882a593Smuzhiyuntopdir = os.path.dirname(bindir) 22*4882a593Smuzhiyunsys.path[0:0] = [os.path.join(topdir, 'lib')] 23*4882a593Smuzhiyun 24*4882a593Smuzhiyunimport bb.tinfoil 25*4882a593Smuzhiyunimport bb.msg 26*4882a593Smuzhiyun 27*4882a593Smuzhiyunlogger = bb.msg.logger_create('bitbake-layers', sys.stdout) 28*4882a593Smuzhiyun 29*4882a593Smuzhiyundef main(): 30*4882a593Smuzhiyun parser = argparse.ArgumentParser( 31*4882a593Smuzhiyun description="BitBake layers utility", 32*4882a593Smuzhiyun epilog="Use %(prog)s <subcommand> --help to get help on a specific command", 33*4882a593Smuzhiyun add_help=False) 34*4882a593Smuzhiyun parser.add_argument('-d', '--debug', help='Enable debug output', action='store_true') 35*4882a593Smuzhiyun parser.add_argument('-q', '--quiet', help='Print only errors', action='store_true') 36*4882a593Smuzhiyun parser.add_argument('-F', '--force', help='Force add without recipe parse verification', action='store_true') 37*4882a593Smuzhiyun parser.add_argument('--color', choices=['auto', 'always', 'never'], default='auto', help='Colorize output (where %(metavar)s is %(choices)s)', metavar='COLOR') 38*4882a593Smuzhiyun 39*4882a593Smuzhiyun global_args, unparsed_args = parser.parse_known_args() 40*4882a593Smuzhiyun 41*4882a593Smuzhiyun # Help is added here rather than via add_help=True, as we don't want it to 42*4882a593Smuzhiyun # be handled by parse_known_args() 43*4882a593Smuzhiyun parser.add_argument('-h', '--help', action='help', default=argparse.SUPPRESS, 44*4882a593Smuzhiyun help='show this help message and exit') 45*4882a593Smuzhiyun subparsers = parser.add_subparsers(title='subcommands', metavar='<subcommand>') 46*4882a593Smuzhiyun subparsers.required = True 47*4882a593Smuzhiyun 48*4882a593Smuzhiyun if global_args.debug: 49*4882a593Smuzhiyun logger.setLevel(logging.DEBUG) 50*4882a593Smuzhiyun elif global_args.quiet: 51*4882a593Smuzhiyun logger.setLevel(logging.ERROR) 52*4882a593Smuzhiyun 53*4882a593Smuzhiyun # Need to re-run logger_create with color argument 54*4882a593Smuzhiyun # (will be the same logger since it has the same name) 55*4882a593Smuzhiyun bb.msg.logger_create('bitbake-layers', output=sys.stdout, 56*4882a593Smuzhiyun color=global_args.color, 57*4882a593Smuzhiyun level=logger.getEffectiveLevel()) 58*4882a593Smuzhiyun 59*4882a593Smuzhiyun plugins = [] 60*4882a593Smuzhiyun tinfoil = bb.tinfoil.Tinfoil(tracking=True) 61*4882a593Smuzhiyun tinfoil.logger.setLevel(logger.getEffectiveLevel()) 62*4882a593Smuzhiyun try: 63*4882a593Smuzhiyun tinfoil.prepare(True) 64*4882a593Smuzhiyun for path in ([topdir] + 65*4882a593Smuzhiyun tinfoil.config_data.getVar('BBPATH').split(':')): 66*4882a593Smuzhiyun pluginpath = os.path.join(path, 'lib', 'bblayers') 67*4882a593Smuzhiyun bb.utils.load_plugins(logger, plugins, pluginpath) 68*4882a593Smuzhiyun 69*4882a593Smuzhiyun registered = False 70*4882a593Smuzhiyun for plugin in plugins: 71*4882a593Smuzhiyun if hasattr(plugin, 'register_commands'): 72*4882a593Smuzhiyun registered = True 73*4882a593Smuzhiyun plugin.register_commands(subparsers) 74*4882a593Smuzhiyun if hasattr(plugin, 'tinfoil_init'): 75*4882a593Smuzhiyun plugin.tinfoil_init(tinfoil) 76*4882a593Smuzhiyun 77*4882a593Smuzhiyun if not registered: 78*4882a593Smuzhiyun logger.error("No commands registered - missing plugins?") 79*4882a593Smuzhiyun sys.exit(1) 80*4882a593Smuzhiyun 81*4882a593Smuzhiyun args = parser.parse_args(unparsed_args, namespace=global_args) 82*4882a593Smuzhiyun 83*4882a593Smuzhiyun if getattr(args, 'parserecipes', False): 84*4882a593Smuzhiyun tinfoil.config_data.disableTracking() 85*4882a593Smuzhiyun tinfoil.parse_recipes() 86*4882a593Smuzhiyun tinfoil.config_data.enableTracking() 87*4882a593Smuzhiyun 88*4882a593Smuzhiyun return args.func(args) 89*4882a593Smuzhiyun finally: 90*4882a593Smuzhiyun tinfoil.shutdown() 91*4882a593Smuzhiyun 92*4882a593Smuzhiyun 93*4882a593Smuzhiyunif __name__ == "__main__": 94*4882a593Smuzhiyun try: 95*4882a593Smuzhiyun ret = main() 96*4882a593Smuzhiyun except bb.BBHandledException: 97*4882a593Smuzhiyun ret = 1 98*4882a593Smuzhiyun except Exception: 99*4882a593Smuzhiyun ret = 1 100*4882a593Smuzhiyun import traceback 101*4882a593Smuzhiyun traceback.print_exc() 102*4882a593Smuzhiyun sys.exit(ret) 103