Make count and append_const actions use set_if in icinga conf

2022-12-26 17:45:38 -05:00 · 2022-12-26 17:45:38 -05:00 · b2524c8799
commit b2524c8799
parent a459b3cdbe
1 changed files with 8 additions and 1 deletions
--- a/check_home_assistant_state.py
+++ b/check_home_assistant_state.py
@ -124,7 +124,14 @@ class Icinga2ConfAction(argparse.Action):
            if action.required:
                print("      required = true")

-            if isinstance(action, argparse._StoreConstAction):
+            if isinstance(
+                action,
+                (
+                    argparse._StoreConstAction,
+                    argparse._CountAction,
+                    argparse._AppendConstAction,
+                ),
+            ):
                print(f'      set_if = "${command_name}_{icinga2_var}$"')
            else:
                print(f'      value = "${command_name}_{icinga2_var}$"')