106 lines
4.3 KiB
Python
Executable file
106 lines
4.3 KiB
Python
Executable file
#!./.venv/bin/python
|
|
import argparse
|
|
import logging
|
|
import CloudFlare
|
|
import requests
|
|
|
|
logging.basicConfig()
|
|
logging.getLogger().setLevel(logging.INFO)
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
def get_config():
|
|
"""Parse command line arguments, extracting the config file name,
|
|
returning the union of config file and command line arguments
|
|
|
|
:return: dict of config file settings and command line arguments
|
|
"""
|
|
|
|
# TODO update description
|
|
parser = argparse.ArgumentParser(
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
description='''\
|
|
Can update all old_ip to new_ip.
|
|
When auto_sync is enable:
|
|
Check zone and name on cloudflare and compare with public ip, update all old ip with new ip if different.
|
|
You need your profile in file ~/.cloudflare/cloudflare.cfg
|
|
''',
|
|
epilog='''\
|
|
'''
|
|
)
|
|
parser.add_argument('--profile', required=True,
|
|
help="The profile of CloudFlare, check ~/.cloudflare/cloudflare.cfg.")
|
|
parser.add_argument('--old_ip', required=False, help="Ip to search to update.")
|
|
parser.add_argument('--new_ip', required=False, help="Ip to replace the old ip.")
|
|
parser.add_argument('--dns_name', required=False, help="DNS name to output his ip and sync.")
|
|
parser.add_argument('--zone_name', required=False, help="The CloudFlare zone name to check.")
|
|
parser.add_argument('--delete', action="store_true")
|
|
parser.add_argument('--debug', action="store_true")
|
|
parser.add_argument('--raw', action="store_true")
|
|
parser.add_argument('--auto_sync', action="store_true",
|
|
help="Need this to use DNS name and zone name to sync public ip to CloudFlare.")
|
|
args = parser.parse_args()
|
|
return args
|
|
|
|
|
|
class ManageCloudFlare:
|
|
def __init__(self, profile, debug=False, raw=False):
|
|
self.cf = CloudFlare.CloudFlare(profile=profile, debug=debug, raw=raw)
|
|
self.raw = raw
|
|
|
|
@staticmethod
|
|
def get_public_ip():
|
|
r = requests.get(r'https://api.ipify.org')
|
|
ip = r.text
|
|
return ip
|
|
|
|
def get_ip_cloudflare(self, zone_name, name):
|
|
"""Return ip from a zone and domain"""
|
|
zone = self.cf.zones.get(params={'name': zone_name})
|
|
lst_zone = [a.get("id") for a in zone]
|
|
for zone_id in lst_zone:
|
|
lst_result_dns = self.cf.zones.dns_records.get(zone_id)
|
|
for result_dns in lst_result_dns:
|
|
if result_dns.get("type") == "A" and result_dns.get("name") == name:
|
|
return result_dns.get("content")
|
|
|
|
def edit_cloudflare(self, old_ip, delete=False, new_ip=""):
|
|
"""Change ip of zone and name"""
|
|
|
|
# TODO support smaller range
|
|
params = {'per_page': 500}
|
|
zones = self.cf.zones.get(params=params)
|
|
result = zones.get("result") if self.raw else zones
|
|
lst_zone = [a.get("id") for a in result]
|
|
for zone_id in lst_zone:
|
|
# dns_records = cf.zones.dns_records.export.get(zone_id)
|
|
result_dns = self.cf.zones.dns_records.get(zone_id)
|
|
dns_records = result_dns.get("result") if self.raw else result_dns
|
|
for dns in dns_records:
|
|
if dns.get("content") == old_ip:
|
|
if delete:
|
|
_logger.info(f"Delete {dns.get('name')}")
|
|
r = self.cf.zones.dns_records.delete(zone_id, dns.get("id"))
|
|
elif new_ip:
|
|
_logger.info(f"Update {dns.get('name')} with ip {new_ip}, old ip was {old_ip}")
|
|
dns_record = {'name': dns.get("name"), 'type': dns.get("type"), 'content': new_ip}
|
|
r = self.cf.zones.dns_records.post(zone_id, data=dns_record)
|
|
r = self.cf.zones.dns_records.delete(zone_id, dns.get("id"))
|
|
|
|
|
|
def main():
|
|
config = get_config()
|
|
cl = ManageCloudFlare(config.profile, debug=config.debug)
|
|
if config.auto_sync:
|
|
configured_ip = cl.get_ip_cloudflare(config.zone_name, config.dns_name)
|
|
actual_ip = cl.get_public_ip()
|
|
if configured_ip != actual_ip:
|
|
cl.edit_cloudflare(configured_ip, new_ip=actual_ip)
|
|
else:
|
|
_logger.info(f"Nothing to do, same ip {actual_ip}")
|
|
else:
|
|
cl.edit_cloudflare(config.old_ip, delete=config.delete, new_ip=config.new_ip)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|