1#!/usr/bin/env python
2
3r"""
4Set the auto_boot policy according to the caller's wishes.
5"""
6
7import os
8import sys
9
10save_dir_path = sys.path.pop(0)
11
12modules = ['gen_arg', 'gen_print', 'gen_valid', 'gen_misc', 'gen_cmd', 'gen_plug_in_utils', 'gen_call_robot']
13for module in modules:
14    exec("from " + module + " import *")
15
16sys.path.insert(0, save_dir_path)
17
18
19# Set exit_on_error for gen_valid functions.
20set_exit_on_error(True)
21
22parser = argparse.ArgumentParser(
23    usage='%(prog)s [OPTIONS]',
24    description="%(prog)s will set the auto_boot policy according to the"
25        + " user's wishes.",
26    formatter_class=argparse.RawTextHelpFormatter,
27    prefix_chars='-+')
28
29
30# Populate stock_list with options we want.
31stock_list = [("test_mode", get_plug_default("test_mode", 0)),
32              ("quiet", get_plug_default("quiet", 0)),
33              ("debug", get_plug_default("debug", 0))]
34
35AUTO_REBOOT_DISABLE = "1"
36
37
38def validate_parms():
39
40    r"""
41    Validate program parameters, etc.  Return True or False (i.e. pass/fail) accordingly.
42    """
43
44    get_plug_vars()
45
46    valid_value(AUTOBOOT_OPENBMC_HOST)
47    global AUTO_REBOOT_DISABLE
48    if pgm_name == "cp_cleanup":
49        AUTO_REBOOT_DISABLE = 0
50    else:
51        valid_value(AUTO_REBOOT_DISABLE, valid_values=["0", "1"])
52        AUTO_REBOOT_DISABLE = int(AUTO_REBOOT_DISABLE)
53
54
55def main():
56
57    gen_setup()
58
59    set_term_options(term_requests='children')
60
61    print_plug_in_header()
62
63    if pgm_name == "cp_setup" or pgm_name == "cp_cleanup":
64        exit_not_master()
65
66    init_robot_out_parms(get_plug_in_package_name() + "." + pgm_name + ".")
67
68    lib_file_path = init_robot_file_path("lib/utils.robot")
69
70    enable_auto_reboot = 1 - AUTO_REBOOT_DISABLE
71    print_var(enable_auto_reboot)
72    keyword_string = "Set Auto Reboot Setting  ${%i}" % enable_auto_reboot
73
74    cmd_buf = create_robot_cmd_string("extended/run_keyword.robot", OPENBMC_HOST, SSH_PORT, HTTPS_PORT,
75                                      REST_USERNAME, REST_PASSWORD, OPENBMC_USERNAME, OPENBMC_PASSWORD,
76                                      REDFISH_SUPPORT_TRANS_STATE, keyword_string, lib_file_path, quiet,
77                                      test_mode, debug, outputdir, output, log, report)
78
79    if not robot_cmd_fnc(cmd_buf):
80        print_error_report("Robot command execution failed.")
81        exit(1)
82
83
84main()
85