diff options
Diffstat (limited to 'src')
| -rwxr-xr-x | src/op_mode/powerctrl.py | 32 |
1 files changed, 12 insertions, 20 deletions
diff --git a/src/op_mode/powerctrl.py b/src/op_mode/powerctrl.py index 4210c3bbe..a70418801 100755 --- a/src/op_mode/powerctrl.py +++ b/src/op_mode/powerctrl.py @@ -15,16 +15,13 @@ # along with this program. If not, see <http://www.gnu.org/licenses/>. import os -import sys -import argparse import re +from argparse import ArgumentParser from datetime import datetime, timedelta, time as type_time, date as type_date -from vyos.util import ask_yes_no -from vyos.util import cmd -from vyos.util import call -from vyos.util import run -from vyos.util import STDOUT +from sys import exit + +from vyos.util import ask_yes_no, cmd, call, run, STDOUT systemd_sched_file = "/run/systemd/shutdown/scheduled" @@ -85,7 +82,7 @@ def cancel_shutdown(): try: cmd('/sbin/shutdown -c --no-wall') except OSError as e: - sys.exit("Could not cancel a reboot or poweroff: %s" % e) + exit("Could not cancel a reboot or poweroff: %s" % e) message = "Scheduled %s has been cancelled %s" % ( output['MODE'], timenow) run(f'wall {message}') @@ -97,7 +94,7 @@ def execute_shutdown(time, reboot=True, ask=True): if not ask: action = "reboot" if reboot else "poweroff" if not ask_yes_no("Are you sure you want to %s this system?" % action): - sys.exit(0) + exit(0) action = "-r" if reboot else "-P" @@ -115,8 +112,7 @@ def execute_shutdown(time, reboot=True, ask=True): if ts: cmd(f'/sbin/shutdown {action} {time[0]}', stderr=STDOUT) else: - sys.exit( - "Invalid time \"{0}\". The valid format is HH:MM".format(time[0])) + exit("Invalid time \"{0}\". The valid format is HH:MM".format(time[0])) elif len(time) == 2: # Assume it's date and time ts = parse_time(time[0]) @@ -128,14 +124,11 @@ def execute_shutdown(time, reboot=True, ask=True): cmd('/sbin/shutdown {action} {t2}', stderr=STDOUT) else: if not ts: - sys.exit( - "Invalid time \"{0}\". The valid format is HH:MM".format(time[0])) + exit("Invalid time \"{0}\". The valid format is HH:MM".format(time[0])) else: - sys.exit( - "Invalid time \"{0}\". A valid format is YYYY-MM-DD [HH:MM]".format(time[1])) + exit("Invalid time \"{0}\". A valid format is YYYY-MM-DD [HH:MM]".format(time[1])) else: - sys.exit( - "Could not decode date and time. Valids formats are HH:MM or YYYY-MM-DD HH:MM") + exit("Could not decode date and time. Valids formats are HH:MM or YYYY-MM-DD HH:MM") check_shutdown() @@ -153,7 +146,7 @@ def chk_vyatta_based_reboots(): def main(): - parser = argparse.ArgumentParser() + parser = ArgumentParser() parser.add_argument("--yes", "-y", help="Do not ask for confirmation", action="store_true", @@ -188,8 +181,7 @@ def main(): if args.check: check_shutdown() except KeyboardInterrupt: - sys.exit("Interrupted") - + exit("Interrupted") if __name__ == "__main__": main() |