blob: f5fa18a5c8c9aefa6768a3164514b55e4f5eecef [file] [log] [blame]
George Keishinge7e91712021-09-03 11:28:44 -05001#!/usr/bin/env python3
Michael Walsh7423c012016-10-04 10:27:21 -05002
George Keishing09679892022-12-08 08:21:52 -06003import os
George Keishinge635ddc2022-12-08 07:38:02 -06004import sys
Patrick Williams20f38712022-12-08 06:18:26 -06005
George Keishing36efbc02018-12-12 10:18:23 -06006try:
7 import __builtin__
8except ImportError:
9 import builtins as __builtin__
10
Michael Walsh7423c012016-10-04 10:27:21 -050011
Michael Walsh410b1782019-10-22 15:56:18 -050012# python puts the program's directory path in sys.path[0]. In other words, the user ordinarily has no way
13# to override python's choice of a module from its own dir. We want to have that ability in our environment.
14# However, we don't want to break any established python modules that depend on this behavior. So, we'll
15# save the value from sys.path[0], delete it, import our modules and then restore sys.path to its original
16# value.
Michael Walsh7423c012016-10-04 10:27:21 -050017
18save_path_0 = sys.path[0]
19del sys.path[0]
20
Patrick Williams20f38712022-12-08 06:18:26 -060021from gen_arg import * # NOQA
22from gen_plug_in import * # NOQA
23from gen_print import * # NOQA
George Keishing37c58c82022-12-08 07:42:54 -060024
Michael Walsh7423c012016-10-04 10:27:21 -050025# Restore sys.path[0].
26sys.path.insert(0, save_path_0)
27
28
Michael Walsh7423c012016-10-04 10:27:21 -050029# Create parser object to process command line parameters and args.
30
31# Create parser object.
32parser = argparse.ArgumentParser(
Patrick Williams20f38712022-12-08 06:18:26 -060033 usage="%(prog)s [OPTIONS] [PLUG_IN_DIR_PATHS]",
Joy Onyerikwu004ad3c2018-06-11 16:29:56 -050034 description="%(prog)s will validate the plug-in packages passed to it."
Patrick Williams20f38712022-12-08 06:18:26 -060035 + " It will also print a list of the absolute plug-in"
36 + " directory paths for use by the calling program.",
Michael Walshd0741f82017-12-21 14:04:21 -060037 formatter_class=argparse.ArgumentDefaultsHelpFormatter,
Patrick Williams20f38712022-12-08 06:18:26 -060038 prefix_chars="-+",
39)
Michael Walsh7423c012016-10-04 10:27:21 -050040
41# Create arguments.
42parser.add_argument(
Patrick Williams20f38712022-12-08 06:18:26 -060043 "plug_in_dir_paths",
44 nargs="?",
Michael Walsh7423c012016-10-04 10:27:21 -050045 default="",
Patrick Williams20f38712022-12-08 06:18:26 -060046 help=plug_in_dir_paths_help_text + default_string,
47)
Michael Walsh7423c012016-10-04 10:27:21 -050048
49parser.add_argument(
Patrick Williams20f38712022-12-08 06:18:26 -060050 "--mch_class", default="obmc", help=mch_class_help_text + default_string
51)
Michael Walsh7423c012016-10-04 10:27:21 -050052
Michael Walsh410b1782019-10-22 15:56:18 -050053# The stock_list will be passed to gen_get_options. We populate it with the names of stock parm options we
54# want. These stock parms are pre-defined by gen_get_options.
Michael Walsh7423c012016-10-04 10:27:21 -050055stock_list = [("test_mode", 0), ("quiet", 1), ("debug", 0)]
56
Michael Walsh7423c012016-10-04 10:27:21 -050057
Patrick Williams20f38712022-12-08 06:18:26 -060058def exit_function(signal_number=0, frame=None):
Michael Walsh7423c012016-10-04 10:27:21 -050059 r"""
Michael Walsh410b1782019-10-22 15:56:18 -050060 Execute whenever the program ends normally or with the signals that we catch (i.e. TERM, INT).
Michael Walsh7423c012016-10-04 10:27:21 -050061 """
62
63 dprint_executing()
64 dprint_var(signal_number)
65
66 qprint_pgm_footer()
67
Michael Walsh7423c012016-10-04 10:27:21 -050068
Michael Walsh7423c012016-10-04 10:27:21 -050069def signal_handler(signal_number, frame):
Michael Walsh7423c012016-10-04 10:27:21 -050070 r"""
Michael Walsh410b1782019-10-22 15:56:18 -050071 Handle signals. Without a function to catch a SIGTERM or SIGINT, our program would terminate immediately
72 with return code 143 and without calling our exit_function.
Michael Walsh7423c012016-10-04 10:27:21 -050073 """
74
Michael Walsh410b1782019-10-22 15:56:18 -050075 # Our convention is to set up exit_function with atexit.registr() so there is no need to explicitly call
76 # exit_function from here.
Michael Walsh7423c012016-10-04 10:27:21 -050077
78 dprint_executing()
79
Michael Walsh410b1782019-10-22 15:56:18 -050080 # Calling exit prevents us from returning to the code that was running when we received the signal.
Michael Walsh7423c012016-10-04 10:27:21 -050081 exit(0)
82
Michael Walsh7423c012016-10-04 10:27:21 -050083
Michael Walsh7423c012016-10-04 10:27:21 -050084def validate_parms():
Michael Walsh7423c012016-10-04 10:27:21 -050085 r"""
86 Validate program parameters, etc. Return True or False accordingly.
87 """
88
89 gen_post_validation(exit_function, signal_handler)
90
91 return True
92
Michael Walsh7423c012016-10-04 10:27:21 -050093
Michael Walsh7423c012016-10-04 10:27:21 -050094def main():
Michael Walsh7423c012016-10-04 10:27:21 -050095 r"""
Michael Walsh410b1782019-10-22 15:56:18 -050096 This is the "main" function. The advantage of having this function vs just doing this in the true
97 mainline is that you can:
Michael Walsh7423c012016-10-04 10:27:21 -050098 - Declare local variables
99 - Use "return" instead of "exit".
100 - Indent 4 chars like you would in any function.
Michael Walsh410b1782019-10-22 15:56:18 -0500101 This makes coding more consistent, i.e. it's easy to move code from here into a function and vice versa.
Michael Walsh7423c012016-10-04 10:27:21 -0500102 """
103
104 if not gen_get_options(parser, stock_list):
105 return False
106
107 if not validate_parms():
108 return False
109
110 qprint_pgm_header()
111
112 # Access program parameter globals.
113 global plug_in_dir_paths
114 global mch_class
115
Patrick Williams20f38712022-12-08 06:18:26 -0600116 plug_in_packages_list = return_plug_in_packages_list(
117 plug_in_dir_paths, mch_class
118 )
Michael Walshc2762f62019-05-17 15:21:35 -0500119 qprint_var(plug_in_packages_list)
Michael Walsh7423c012016-10-04 10:27:21 -0500120
Michael Walsh410b1782019-10-22 15:56:18 -0500121 # As stated in the help text, this program must print the full paths of each selected plug in.
Michael Walsh7423c012016-10-04 10:27:21 -0500122 for plug_in_dir_path in plug_in_packages_list:
123 print(plug_in_dir_path)
124
125 return True
126
Michael Walsh7423c012016-10-04 10:27:21 -0500127
Michael Walsh7423c012016-10-04 10:27:21 -0500128# Main
129
130if not main():
131 exit(1)