48 lines
2.3 KiB
Python
48 lines
2.3 KiB
Python
import requests
|
|
import logging
|
|
|
|
class CloudflareAPIException(Exception):
|
|
def __init__(self, message, response) -> None:
|
|
full_message = f'Cloudflare request failed (code: {response.status_code}): {message}\n\nServer body:\n{response.text}'
|
|
super().__init__(full_message)
|
|
|
|
class CloudFlare:
|
|
def __init__(self, api_token):
|
|
self.headers = {'Content-Type': 'application/json', 'Authorization': f'Bearer {api_token}'}
|
|
|
|
def get_zone_id(self, domain):
|
|
response = requests.get(f'https://api.cloudflare.com/client/v4/zones?name={domain}', headers=self.headers)
|
|
|
|
if response.status_code != 200:
|
|
raise CloudflareAPIException('Something went wrong requesting the zone of the domain on Cloudflare...', response)
|
|
|
|
zone_id = response.json()['result'][0]['id']
|
|
return zone_id
|
|
|
|
def get_record_id(self, zoneId, record):
|
|
logging.debug('Getting record id...')
|
|
response = requests.get(f'https://api.cloudflare.com/client/v4/zones/{zoneId}/dns_records?name={record}', headers=self.headers)
|
|
|
|
if response.status_code != 200:
|
|
raise CloudflareAPIException('Something went wrong requesting the record id of the domain name on Cloudflare...', response)
|
|
|
|
return response.json()['result'][0]['id']
|
|
|
|
|
|
def change_record(self, subdomain, zoneId, recordId, targetIp):
|
|
logging.info(f'Changing record for {subdomain} to point to {targetIp}...')
|
|
response = requests.put(f'https://api.cloudflare.com/client/v4/zones/{zoneId}/dns_records/{recordId}',
|
|
headers=self.headers,
|
|
json={
|
|
'content': targetIp,
|
|
'name': subdomain,
|
|
'proxied': False,
|
|
'type': 'A',
|
|
'comment': 'Is set to change with my DDNS script (on the Raspberry Pi).',
|
|
'ttl': 1 # Meaning "Automatic", see: https://developers.cloudflare.com/api/operations/dns-records-for-a-zone-update-dns-record
|
|
})
|
|
|
|
if response.status_code != 200:
|
|
raise CloudflareAPIException('Something went wrong updating the dns record...', response)
|
|
|
|
logging.info('Succesfully updated the record ✅!')
|