mirror of
https://github.com/jpetazzo/container.training.git
synced 2026-02-14 17:49:59 +00:00
71 lines
2.0 KiB
Python
Executable File
71 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""
|
|
There are two ways to use this script:
|
|
|
|
1. Pass a domain as the 1st argument, and IP addresses then.
|
|
It will configure the domain with the listed IP addresses.
|
|
Example: ./map-dns.py open-duck.site 1.2.3.4 2.3.4.5 3.4.5.6
|
|
|
|
2. Pass two files names as argument, in which case the first
|
|
file should contain a list of domains, and the second a list of
|
|
groups of IP addresses, with one group per line.
|
|
There should be more domains than groups of addresses.
|
|
Example: ./map-dns.py domains.txt tags/2020-08-15-jp/clusters.txt
|
|
|
|
In both cases, the domains should be configured to use GANDI LiveDNS.
|
|
"""
|
|
import os
|
|
import requests
|
|
import sys
|
|
import yaml
|
|
|
|
# This can be tweaked if necessary.
|
|
config_file = os.path.join(
|
|
os.environ["HOME"], ".config/gandi/config.yaml")
|
|
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.
|
|
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('#') ]
|
|
clusters_file = sys.argv[2]
|
|
lines = open(clusters_file).read().split('\n')
|
|
clusters = [line.split() for line in lines]
|
|
else:
|
|
domains = [domain_or_domain_file]
|
|
clusters = [sys.argv[2:]]
|
|
|
|
# Now, do the work.
|
|
while domains and clusters:
|
|
domain = domains.pop(0)
|
|
cluster = clusters.pop(0)
|
|
print(f"{domain} => {cluster}")
|
|
zone = ""
|
|
node = 0
|
|
for ip in cluster:
|
|
node += 1
|
|
zone += f"@ 300 IN A {ip}\n"
|
|
zone += f"* 300 IN A {ip}\n"
|
|
zone += f"node{node} 300 IN A {ip}\n"
|
|
r = requests.put(
|
|
f"{apiurl}/{domain}/records",
|
|
headers={
|
|
"x-api-key": apikey,
|
|
"content-type": "text/plain",
|
|
},
|
|
data=zone)
|
|
print(r.text)
|
|
|
|
#r = requests.get(
|
|
# f"{apiurl}/{domain}/records",
|
|
# headers={"x-api-key": apikey},
|
|
# )
|
|
|
|
if domains:
|
|
print(f"Good, we have {len(domains)} domains left.")
|
|
|
|
if clusters:
|
|
print(f"Crap, we have {len(clusters)} clusters left.")
|