64 lines
2.2 KiB
Python
Executable File
64 lines
2.2 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright (c) 2021 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 local execution 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 config file')
|
|
|
|
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, 'beat_run'): success = Beatbox.beat_run()
|
|
|
|
return success
|
|
|
|
if __name__ == '__main__':
|
|
main()
|