CMS/check_home_assistant_state
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: CMS:75da5babd0cc371825c2a456244a039b413b255a
Branches Tags
CMS:main
CMS:0.4.0
CMS:0.3.0
CMS:0.2.0
CMS:0.1.0
...
compare: CMS:37d091c3daacca94feefe3dfb3fa14a131f24f17
Branches Tags
CMS:main
CMS:0.4.0
CMS:0.3.0
CMS:0.2.0
CMS:0.1.0

2 Commits
75da5babd0 ... 37d091c3da

Author SHA1 Message Date
Adam Goldsmith
37d091c3da Allow filtering by domain 2024-12-19 18:50:50 -05:00
Adam Goldsmith
e3d9312a1e Make device-class not required 2024-12-19 18:50:43 -05:00
1 changed files with 17 additions and 4 deletions
Show Stats Expand all files Collapse all files

21
check_home_assistant_state.py
View File

@ -66,6 +66,7 @@ class AttributeFilter:
class Entities(nagiosplugin.Resource): class Entities(nagiosplugin.Resource):
url: str url: str
token: str token: str
domain: str
device_class: str device_class: str
numeric: bool numeric: bool
area: str area: str
@ -148,7 +149,13 @@ class Entities(nagiosplugin.Resource):
) )
def probe(self): def probe(self):
template_filter = f"states|selectattr('attributes.device_class', 'eq', '{ self.device_class }')" template_filter = "states"
if self.domain:
template_filter += f"|selectattr('domain', 'eq', '{ self.domain }')"
if self.device_class:
template_filter += (
f"|selectattr('attributes.device_class', 'eq', '{ self.device_class }')"
)
for f in self.filters: for f in self.filters:
op = "ne" if f.negated else "eq" op = "ne" if f.negated else "eq"
template_filter += ( template_filter += (
@ -347,11 +354,17 @@ def main():
common_args = shared_parser.add_argument_group( common_args = shared_parser.add_argument_group(
"Common Arguments", "Arguments shared between metric types" "Common Arguments", "Arguments shared between metric types"
) )
common_args.add_argument(
"--domain",
type=str,
required=False,
help="domain of entities to monitor",
)
common_args.add_argument( common_args.add_argument(
"-d", "-d",
"--device-class", "--device-class",
type=str, type=str,
required=True, required=False,
help="device class of entities to monitor", help="device class of entities to monitor",
) )
common_args.add_argument( common_args.add_argument(
@ -469,7 +482,7 @@ def main():
max=args.max, max=args.max,
**parsed_common_args, **parsed_common_args,
), ),
ScalarOrUnknownContext(args.device_class, args.warning, args.critical), ScalarOrUnknownContext("scalar_entities", args.warning, args.critical),
) )
elif args.subparser_name == "text": elif args.subparser_name == "text":
check = nagiosplugin.Check( check = nagiosplugin.Check(
@ -479,7 +492,7 @@ def main():
numeric=False, numeric=False,
**parsed_common_args, **parsed_common_args,
), ),
RegexContext(args.device_class, args.ok, args.warning, args.critical), RegexContext("text_entities", args.ok, args.warning, args.critical),
) )
check.main(args.verbose) check.main(args.verbose)