evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add 'bitwise_xor' filter to jinja templates (#104942)

Browse source 
Co-authored-by: Robert Resch <robert@resch.dev>
This commit is contained in:
Lars R 2024-01-10 09:40:52 +01:00 • committed by GitHub
parent 554c27a31a
commit bf6b9175a1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 22 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
homeassistant/helpers/template.py
View file

@ -2099,6 +2099,11 @@ def bitwise_or(first_value, second_value):
return first_value | second_value
def bitwise_xor(first_value, second_value):
"""Perform a bitwise xor operation."""
return first_value ^ second_value
def struct_pack(value: Any | None, format_string: str) -> bytes | None:
"""Pack an object into a bytes object."""
try:
@ -2462,6 +2467,7 @@ class TemplateEnvironment(ImmutableSandboxedEnvironment):
self.filters["regex_findall_index"] = regex_findall_index
self.filters["bitwise_and"] = bitwise_and
self.filters["bitwise_or"] = bitwise_or
self.filters["bitwise_xor"] = bitwise_xor
self.filters["pack"] = struct_pack
self.filters["unpack"] = struct_unpack
self.filters["ord"] = ord

16
tests/helpers/test_template.py
View file

@ -2405,6 +2405,22 @@ def test_bitwise_or(hass: HomeAssistant) -> None:
assert tpl.async_render() == 8 | 2
@pytest.mark.parametrize(
("value", "xor_value", "expected"),
[(8, 8, 0), (10, 2, 8), (0x8000, 0xFAFA, 31482), (True, False, 1), (True, True, 0)],
)
def test_bitwise_xor(
hass: HomeAssistant, value: Any, xor_value: Any, expected: int
) -> None:
"""Test bitwise_xor method."""
assert (
template.Template("{{ value | bitwise_xor(xor_value) }}", hass).async_render(
{"value": value, "xor_value": xor_value}
)
== expected
)
def test_pack(hass: HomeAssistant, caplog: pytest.LogCaptureFixture) -> None:
"""Test struct pack method."""