Add Dig interface

(cherry picked from commit 7347abd01f)
This commit is contained in:
Michal Nowak 2025-02-07 11:26:04 +01:00
parent a946528023
commit 8cb3eb5d9d
No known key found for this signature in database

View file

@ -65,6 +65,18 @@ def cmd(
return exc
class Dig:
def __init__(self, base_params: str = ""):
self.base_params = base_params
def __call__(self, params: str) -> str:
"""Run the dig command with the given parameters and return the decoded output."""
return cmd(
[os.environ.get("DIG")] + f"{self.base_params} {params}".split(),
log_stdout=True,
).stdout.decode("utf-8")
def retry_with_timeout(func, timeout, delay=1, msg=None):
start_time = time.time()
while time.time() < start_time + timeout: