def test_update_domain(self):
domain_setup.main(self.host + self.domain, ['ipv4'])
assert (os.path.exists('settings.txt')), 'Settings file missing'
result = domain_update.main(self.host + self.domain)
assert (result in [
'A record up to date.', 'DNS record successfully updated.'
]), result
def main():
parser = argparse.ArgumentParser(description="Config domain for DynDNS")
parser.add_argument('action',
choices=['setup', 'update', 'status'],
help="action to be performed on domain(s)")
parser.add_argument('--config',
type=str,
default='settings.txt',
help="config file path")
parser.add_argument(
'--ignore-previous-ip',
action='store_true',
dest='ignore_previous_ip',
help=
"Update the IP even if no change detected. Don't use on regular update!"
)
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument('--domain', type=str, help="domain to keep up to date")
group.add_argument('--all',
action='store_true',
help="update all domains in config")
args = parser.parse_args()
action = args.action
domain = args.domain
all = args.all
settings = args.config
ignore_previous_ip = args.ignore_previous_ip
# validate domain
if domain and not validators.domain(domain) is True:
print("Domain is not valid.")
return
if all and action == 'setup':
print("Bulk setup not supported")
return
if ignore_previous_ip and action != 'update':
print("--ignore-previous-ip only valid with update action")
return
domains = []
if all:
try:
with open(settings, "r") as settings_file:
try:
config = json.load(settings_file)
except ValueError:
config = {}
except IOError:
print("Couldn't read setttings.")
domains = config.keys()
elif domain:
domains = [
domain,
]
for domain in domains:
if action == 'setup':
print(domain_setup.main(domain, settings))
elif action == 'update':
print(domain_update.main(domain, settings, ignore_previous_ip))
elif action == 'status':
print(domain_status.main(domain, settings))
def _setup_domain(domain):
domain_setup.main(domain, ['ipv4'])
def main():
parser = argparse.ArgumentParser(description="Config domain for DynDNS")
parser.add_argument('action',
choices=['setup', 'update', 'status', 'remove'],
help="action to be performed on "
"domain(s)")
parser.add_argument('--config',
type=str,
default='settings.txt',
help="config file path")
parser.add_argument('--backup_file',
default=None,
help="backup file path for remove domain")
parser.add_argument(
'--ignore-previous-ip',
action='store_true',
dest='ignore_previous_ip',
help=
"Update the IP even if no change detected. Don't use on regular update!"
)
parser.add_argument('--protocols',
nargs="*",
choices=["ipv4", "ipv6"],
default=["ipv4"],
help="Kind of IP to update",
dest="protocols")
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument('--domain', type=str, help="domain to keep up to date")
group.add_argument('--all',
action='store_true',
help="update all domains in config")
args = parser.parse_args()
action = args.action
domain = args.domain
allargs = args.all
settings = args.config
backup_file = args.backup_file
ignore_previous_ip = args.ignore_previous_ip
ignore_ipv4 = "ipv4" not in args.protocols
ignore_ipv6 = "ipv6" not in args.protocols
# validate domain
if domain and not validators.domain(domain) is True:
print("Domain is not valid.")
return
if allargs and action == 'setup':
print("Bulk setup not supported")
return
if backup_file and action != 'remove':
print("Backup file only supported on domain remove.")
return
if ignore_previous_ip and action != 'update':
print("--ignore-previous-ip only valid with update action")
return
protocols = ['IPv4', 'IPv6']
if ignore_ipv4:
protocols.remove('IPv4')
if ignore_ipv6:
protocols.remove('IPv6')
domains = []
if allargs:
try:
with open(settings, "r") as settings_file:
try:
config = json.load(settings_file)
except ValueError:
config = {}
except IOError:
print("Couldn't read setttings.")
domains = config.keys()
elif domain:
domains = [
domain,
]
for domain in domains:
if action == 'setup':
print(domain_setup.main(domain, protocols, settings))
elif action == 'update':
print(domain_update.main(domain, settings, ignore_previous_ip))
elif action == 'status':
print(domain_status.main(domain, settings))
elif action == 'remove':
print(domain_remove.main(domain, settings, backup_file))
def _setup_domain(self, domain):
domain_setup.main(domain)