beat_config.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# Copyright (c) 2022 Battelle Energy Alliance, LLC.  All rights reserved.

from __future__ import print_function

import argparse
import os
import platform
import sys
from beat_common import *

###################################################################################################
ScriptName = os.path.basename(__file__)

###################################################################################################
# main
def main():

    # extract arguments from the command line
    # print (sys.argv[1:]);
    parser = argparse.ArgumentParser(
        description='Beat configure script', add_help=False, usage='{} <arguments>'.format(ScriptName)
    )
    parser.add_argument(
        '-v', '--verbose', dest='debug', type=str2bool, nargs='?', const=True, default=False, help="Verbose output"
    )
    parser.add_argument(
        '-b', '--beat', required=True, dest='beatName', metavar='<STR>', type=str, default=None, help='Beat name'
    )
    parser.add_argument(
        '-c',
        '--config-file',
        required=False,
        dest='configFile',
        metavar='<STR>',
        type=str,
        default=None,
        help='Beat YML file to configure',
    )
    parser.add_argument(
        '-d',
        '--defaults',
        dest='acceptDefault',
        type=str2bool,
        nargs='?',
        const=True,
        default=False,
        help="Accept defaults to prompts without user interaction",
    )

    try:
        parser.error = parser.exit
        args = parser.parse_args()
    except SystemExit:
        parser.print_help()
        exit(2)

    if args.debug:
        eprint(os.path.join(ScriptPath, ScriptName))
        eprint("Arguments: {}".format(sys.argv[1:]))
        eprint("Arguments: {}".format(args))
    else:
        sys.tracebacklimit = 0

    args.beatName = args.beatName.lower()
    if not args.beatName.endswith('beat'):
        args.beatName = args.beatName + 'beat'

    if args.configFile is None:
        args.configFile = args.beatName + '.yml'

    installerPlatform = platform.system()
    if installerPlatform == PLATFORM_LINUX:
        Beatbox = LinuxBeatbox(debug=args.debug, ymlFileSpec=args.configFile, beatName=args.beatName)
    elif installerPlatform == PLATFORM_MAC:
        Beatbox = MacBeatbox(debug=args.debug, ymlFileSpec=args.configFile, beatName=args.beatName)
    elif installerPlatform == PLATFORM_WINDOWS:
        Beatbox = WindowsBeatbox(debug=args.debug, ymlFileSpec=args.configFile, beatName=args.beatName)

    success = False
    if hasattr(Beatbox, 'configure_beat_yml'):
        success = Beatbox.configure_beat_yml()
    if hasattr(Beatbox, 'configure_keystore'):
        success = Beatbox.configure_keystore()

    return success


if __name__ == '__main__':
    main()