From f306749f683ec80e9af3f23bbfca93112d04d8b9 Mon Sep 17 00:00:00 2001 From: Jerome Petazzoni Date: Wed, 10 Mar 2021 19:45:23 +0100 Subject: [PATCH] =?UTF-8?q?=F0=9F=96=A8=EF=B8=8F=20Improve=20output=20in?= =?UTF-8?q?=20case=20no=20arg=20is=20provided?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- prepare-vms/map-dns.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/prepare-vms/map-dns.py b/prepare-vms/map-dns.py index 1d25eee9..58f54e1c 100755 --- a/prepare-vms/map-dns.py +++ b/prepare-vms/map-dns.py @@ -26,16 +26,16 @@ apiurl = "https://dns.api.gandi.net/api/v5/domains" apikey = yaml.safe_load(open(config_file))["apirest"]["key"] # Figure out if we're called for a bunch of domains, or just one. -first_arg = sys.argv[1] -if os.path.isfile(first_arg): - domains = open(first_arg).read().split() +domain_or_domain_file = sys.argv[1] +if os.path.isfile(domain_or_domain_file): + domains = open(domain_or_domain_file).read().split() domains = [ d for d in domains if not d.startswith('#') ] tag = sys.argv[2] ips = open(f"tags/{tag}/ips.txt").read().split() settings_file = f"tags/{tag}/settings.yaml" clustersize = yaml.safe_load(open(settings_file))["clustersize"] else: - domains = [first_arg] + domains = [domain_or_domain_file] ips = sys.argv[2:] clustersize = len(ips)