diff --git a/check_home_assistant_state.py b/check_home_assistant_state.py index 47d50ab..6a45dbe 100755 --- a/check_home_assistant_state.py +++ b/check_home_assistant_state.py @@ -56,6 +56,14 @@ class Entities(nagiosplugin.Resource): @nagiosplugin.guarded def main(): argp = argparse.ArgumentParser(description=__doc__) + argp.add_argument( + "-t", "--token", required=True, type=str, help="API token for Home Assistant" + ) + argp.add_argument( + "-u", "--url", required=True, type=str, help="URL for Home Assistant" + ) + argp.add_argument("-v", "--verbose", action="count", default=0) + argp.add_argument( "-w", "--warning", @@ -70,13 +78,7 @@ def main(): required=True, help="return critical if battery percentage is outside RANGE", ) - argp.add_argument( - "-t", "--token", required=True, type=str, help="API token for Home Assistant" - ) - argp.add_argument( - "-u", "--url", required=True, type=str, help="URL for Home Assistant" - ) - argp.add_argument("-v", "--verbose", action="count", default=0) + args = argp.parse_args() check = nagiosplugin.Check(