diff --git a/Pihole.py b/Pihole.py index 7451d0c..1369b85 100644 --- a/Pihole.py +++ b/Pihole.py @@ -123,10 +123,7 @@ class Pihole: data = { 'action': 'add' } data['domain'] = domain data['ip'] = ip - if reload: - data['reload'] = 'true' - else: - data['reload'] = 'false' + data['reload'] = str(reload).lower() self._log.info(f'try to add new {domain=} for {ip=}') return self._query('customdns', data) @@ -145,10 +142,7 @@ class Pihole: data = { 'action': 'delete' } data['domain'] = domain data['ip'] = ip - if reload: - data['reload'] = 'true' - else: - data['reload'] = 'false' + data['reload'] = str(reload).lower() self._log.info(f'try to delete {domain=} for {ip=}') return self._query('customdns', data) @@ -177,10 +171,8 @@ class Pihole: data = { 'action': 'add' } data['domain'] = domain data['target'] = target - if reload: - data['reload'] = 'true' - else: - data['reload'] = 'false' + data['reload'] = str(reload).lower() + self._log.info(f'try to add {domain=} cname for {target=}') return self._query('customcname', data) @@ -199,11 +191,8 @@ class Pihole: data = { 'action': 'delete' } data['domain'] = domain data['target'] = target - if reload: - data['reload'] = 'true' - else: - data['reload'] = 'false' - + data['reload'] = str(reload).lower() + self._log.info(f'try to delete {domain=} cname for {target=}') return self._query('customcname', data)