diff --git a/powerdnsadmin/lib/utils.py b/powerdnsadmin/lib/utils.py index 5a62ee3..9e7cf20 100644 --- a/powerdnsadmin/lib/utils.py +++ b/powerdnsadmin/lib/utils.py @@ -119,16 +119,12 @@ def fetch_json(remote_url, def display_record_name(data): - # Check that the data argument is a tuple containing two elements - if isinstance(data, Iterable) and len(data) == 2: - record_name, domain_name = data - if record_name == domain_name: - return '@' - else: - return record_name + record_name, domain_name = data + if record_name == domain_name: + return '@' else: - # If data is not a tuple of length 2, return an empty string - return '' + return re.sub('\.{}$'.format(domain_name), '', record_name) + def display_master_name(data): """