Update utils.py

This should fix the error you were experiencing, as it will now only attempt to process the `data` argument if it is a tuple containing two elements. If the `data` argument is not in the expected format, the function will simply return an empty string instead of raising an exception.
This commit is contained in:
Bernward Sanchez 2022-12-09 08:15:13 +08:00 committed by GitHub
parent 0bc7f2765b
commit 3e68044420
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -119,12 +119,16 @@ def fetch_json(remote_url,
def display_record_name(data):
record_name, domain_name = data
if record_name == domain_name:
return '@'
# 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
else:
return re.sub('\.{}$'.format(domain_name), '', record_name)
# If data is not a tuple of length 2, return an empty string
return ''
def display_master_name(data):
"""