ddns/ddns.py

650 lines
25 KiB
Python
Raw Normal View History

2023-03-06 14:01:07 +01:00
#!/usr/bin/python3
import sqlite3
from pathlib import Path
from sqlite3 import Error
import urllib.request
import json
import logging
import argparse
import requests
import os
import time
2023-03-19 17:23:11 +01:00
from datetime import datetime
2023-03-07 17:49:38 +01:00
from string import ascii_letters, digits
from rich import print
from argparse import RawTextHelpFormatter
2023-03-06 14:01:07 +01:00
homefilepath = Path.home()
filepath = homefilepath.joinpath('.config/ddns')
database = filepath.joinpath('ddns.db')
logfile = filepath.joinpath('ddns.log')
2023-04-05 10:44:00 +02:00
logging.basicConfig(filename=logfile,level=logging.INFO,format='%(message)s')
2023-04-15 17:12:43 +02:00
app_version = '0.5.1'
2023-03-06 14:01:07 +01:00
def get_ip():
cursor = conn.cursor()
cursor.execute('SELECT COUNT(ip4_server) FROM ipservers')
count = cursor.fetchone()[0]
if count != 0:
cursor.execute('SELECT ip4_server from ipservers')
server = cursor.fetchone()
server = server[0]
try:
current_ip = urllib.request.urlopen(server).read().decode('utf-8')
return current_ip
except Exception as e:
error = str(e)
2023-03-23 17:42:15 +01:00
logging.error(time.strftime("%Y-%m-%d %H:%M") + ' - Error : ' + str(e))
2023-03-06 14:01:07 +01:00
return error
else:
return None
def connect_database():
Path(filepath).mkdir(parents=True, exist_ok=True)
conn = None
try:
conn = sqlite3.connect(database)
except Error as e:
2023-03-23 17:42:15 +01:00
logging.error(time.strftime("%Y-%m-%d %H:%M") + ' - Error : ' + str(e))
2023-03-06 14:01:07 +01:00
print(e)
finally:
if conn:
c = conn.cursor()
c.execute('''CREATE TABLE IF NOT EXISTS apikey
(id integer NOT NULL PRIMARY KEY,
api text NOT NULL)''')
c.execute('''CREATE TABLE IF NOT EXISTS ipservers
(id integer NOT NULL PRIMARY KEY,
2023-03-19 17:23:11 +01:00
ip4_server text NOT NULL,
2023-03-06 14:01:07 +01:00
ip6_server text)''')
c.execute('''CREATE TABLE IF NOT EXISTS domains
(id integer PRIMARY KEY,
name text NOT NULL)''')
c.execute('''CREATE TABLE IF NOT EXISTS subdomains
(id integer PRIMARY KEY,
main_id integer NOT NULL,
2023-03-19 17:23:11 +01:00
name text NOT NULL,
2023-03-07 17:49:38 +01:00
current_ip4 text NOT NULL,
current_ip6 text NULL)''')
2023-03-06 14:01:07 +01:00
return conn
def get_api():
cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM apikey')
count = cursor.fetchone()[0]
if count == 0:
return None
else:
cursor.execute('SELECT * FROM apikey')
rows = cursor.fetchone()
return rows[1]
def api(api_value):
cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM apikey')
count = cursor.fetchone()[0]
if count == 0:
cursor.execute('INSERT INTO apikey values(?,?)', (1, api_value))
2023-03-23 17:42:15 +01:00
logging.info(time.strftime("%Y-%m-%d %H:%M") + ' - Info : API key added')
2023-03-06 14:01:07 +01:00
print('Your API key has been added.')
else:
cursor.execute('UPDATE apikey SET api = ? WHERE id = 1',(api_value,))
2023-03-23 17:42:15 +01:00
logging.info(time.strftime("%Y-%m-%d %H:%M") + ' - Info : API key updated')
2023-03-06 14:01:07 +01:00
print('Your API key has been updated.')
conn.commit()
def add_domian(domain):
2023-03-07 17:49:38 +01:00
apikey = get_api()
2023-03-06 14:01:07 +01:00
cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(domain,))
count = cursor.fetchone()[0]
2023-03-07 17:49:38 +01:00
if count != 0:
print('[red]Error:[/red] Domain name (%s) already in database!' % (domain))
2023-03-06 14:01:07 +01:00
else:
2023-03-07 17:49:38 +01:00
if apikey != None:
headers = {'Authorization': 'Bearer ' + apikey, "Content-Type": "application/json"}
response = requests.get('https://api.digitalocean.com/v2/domains/' + domain, headers=headers)
response_data = response.json()
if 'id' in response_data:
print('[red]Error: [/red]The domain does not exist in your DigitalOcean account.\nPlease add the domain from your control panel [b]https://cloud.digitalocean.com/networking/domains/[/b]')
else:
cursor.execute('INSERT INTO domains values(?,?)', (None, domain,))
print('The domain [b]%s[/b] has been added to the DB' % (domain))
2023-03-23 17:42:15 +01:00
logging.info(time.strftime("%Y-%m-%d %H:%M") + ' - Info : Domain %s added' %(domain))
2023-03-07 17:49:38 +01:00
conn.commit()
2023-03-06 14:01:07 +01:00
2023-03-07 17:49:38 +01:00
def add_subdomain(domain):
now = datetime.now().strftime("%Y-%m-%d %H:%M")
2023-04-05 10:44:00 +02:00
if set(domain).difference(ascii_letters + '.' + digits + '-' + '@'):
2023-03-07 17:49:38 +01:00
print('[red]Error:[/red] Give the domain name in simple form e.g. [b]test.domain.com[/b]')
else:
parts = domain.split('.')
2023-04-15 17:11:35 +02:00
if len(parts) > 3:
top = parts[1] + '.' + parts[2] + '.' + parts[3]
sub = parts[0]
else:
sub = parts[0]
top = parts[1] + '.' + parts[2]
2023-03-07 17:49:38 +01:00
apikey = get_api()
if apikey == None:
print("[red]Error:[/red] Missing APIkey. Please add one!")
else:
ip = get_ip()
if ip == None or 'urlopen error' in ip:
print('[red]Error:[/red] Failed to get public IP. Do you have a typo in your URI? [red]Error %s.[/red]' % (ip))
else:
cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(top,))
count = cursor.fetchone()[0]
if count == 0:
print('[red]Error:[/red] Top domain [bold]%s[/bold] does not exist in the DB. Please add it with [i]ddns -t %s[/i].' % (top,top))
else:
2023-04-15 17:11:35 +02:00
cursor.execute('SELECT id,name FROM domains WHERE name LIKE ?',(top,))
topdomain = cursor.fetchone()
topdomain_id = topdomain[0]
topdomain_name = topdomain[1]
2023-03-07 17:49:38 +01:00
cursor.execute('SELECT count(*) FROM subdomains WHERE main_id LIKE ? AND name like ?',(topdomain_id,sub,))
count = cursor.fetchone()[0]
if count != 0:
print('[red]Error:[/red] [bold]%s[/bold] already exists.' % (domain))
else:
data = {'name': sub,'data': ip,'type': "A",'ttl': 3600}
headers = {'Authorization': 'Bearer ' + apikey, "Content-Type": "application/json"}
response = requests.post('https://api.digitalocean.com/v2/domains/' + top + '/records',
data=json.dumps(data), headers=headers)
if str(response) == '<Response [201]>':
if response != 'Fail':
response_data = response.json()
domainid = str(response_data['domain_record']['id'])
2023-04-05 10:44:00 +02:00
cursor.execute('INSERT INTO subdomains values(?,?,?,?,?,?,?,?,?)',(domainid,topdomain_id,sub,ip,None,now,now,now,1,))
2023-03-07 17:49:38 +01:00
conn.commit()
print('The domain %s has been added.' % (domain))
2023-03-23 17:42:15 +01:00
logging.info(time.strftime("%Y-%m-%d %H:%M") + ' - Info : subdomain %s added'%(domain))
2023-03-07 17:49:38 +01:00
else:
return '[red]Error: %s [/red]' % (str(response))
def remove_subdomain(domain):
2023-04-05 10:44:00 +02:00
if set(domain).difference(ascii_letters + '.' + digits + '-' + '@'):
2023-03-07 17:49:38 +01:00
print('[red]Error:[/red] Give the domain name in simple form e.g. [b]test.domain.com[/b]')
2023-03-06 14:01:07 +01:00
else:
2023-03-07 17:49:38 +01:00
parts = domain.split('.')
2023-04-15 17:11:35 +02:00
if len(parts) > 3:
top = parts[1] + '.' + parts[2] + '.' + parts[3]
sub = parts[0]
else:
sub = parts[0]
top = parts[1] + '.' + parts[2]
2023-03-07 17:49:38 +01:00
cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(top,))
count = cursor.fetchone()[0]
if count == 0:
print('[red]Error:[/red] Top domain [bold]%s[/bold] does not exist in the DB. So I\'m giving up!.' % (top))
2023-03-06 14:01:07 +01:00
else:
2023-03-07 17:49:38 +01:00
cursor.execute('SELECT COUNT(*) FROM subdomains WHERE name like ? and main_id=(SELECT id from domains WHERE name=?)',(sub,top,))
2023-03-06 14:01:07 +01:00
count = cursor.fetchone()[0]
if count == 0:
2023-03-07 17:49:38 +01:00
print('[red]Error:[/red] Domain [bold]%s[/bold] does not exist in the DB. So I\'m giving up!.' % (domain))
2023-03-06 14:01:07 +01:00
else:
2023-03-07 17:49:38 +01:00
apikey = get_api()
if apikey == None:
print("[red]Error:[/red] Missing APIkey. Please add one!")
2023-03-06 14:01:07 +01:00
else:
2023-03-07 17:49:38 +01:00
cursor.execute('SELECT id FROM subdomains WHERE name like ? and main_id=(SELECT id from domains WHERE name=?)',(sub,top,))
subdomain_id = str(cursor.fetchone()[0])
2023-03-06 14:01:07 +01:00
headers = {'Authorization': 'Bearer ' + apikey, "Content-Type": "application/json"}
2023-03-07 17:49:38 +01:00
response = requests.delete('https://api.digitalocean.com/v2/domains/'+top+'/records/' + subdomain_id, headers=headers)
if str(response) == '<Response [204]>':
cursor.execute('DELETE from subdomains where id=?',(subdomain_id,))
2023-03-23 17:42:15 +01:00
logging.info(time.strftime("%Y-%m-%d %H:%M") + ' - Info : Subdomain %s removed' %(domain))
2023-03-07 17:49:38 +01:00
conn.commit()
2023-03-06 14:01:07 +01:00
else:
2023-03-07 17:49:38 +01:00
print('[red]Error: [/red]An error occurred! Please try again later!')
2023-03-06 14:01:07 +01:00
2023-04-05 10:44:00 +02:00
def edit_subdomain(domain):
if set(domain).difference(ascii_letters + '.' + digits + '-' + '@'):
print('[red]Error:[/red] Give the domain name in simple form e.g. [b]test.domain.com[/b]')
else:
parts = domain.split('.')
2023-04-15 17:11:35 +02:00
if len(parts) > 3:
top = parts[1] + '.' + parts[2] + '.' + parts[3]
sub = parts[0]
else:
sub = parts[0]
top = parts[1] + '.' + parts[2]
2023-04-05 10:44:00 +02:00
cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(top,))
count = cursor.fetchone()[0]
if count == 0:
print('[red]Error:[/red] Top domain [bold]%s[/bold] does not exist in the DB. So I\'m giving up!.' % (top))
else:
cursor.execute('SELECT COUNT(*) FROM subdomains WHERE name like ? and main_id=(SELECT id from domains WHERE name=?)',(sub,top,))
count = cursor.fetchone()[0]
if count == 0:
print('[red]Error:[/red] Domain [bold]%s[/bold] does not exist in the DB. So I\'m giving up!.' % (domain))
else:
apikey = get_api()
if apikey == None:
print("[red]Error:[/red] Missing APIkey. Please add one!")
else:
cursor.execute('SELECT id,active FROM subdomains WHERE name like ? and main_id=(SELECT id from domains WHERE name=?)',(sub,top,))
domain_info = cursor.fetchone()
subdomain_id = str(domain_info[0])
status = domain_info[1]
if status == 1:
status = 0
else:
status = 1
cursor.execute('UPDATE subdomains SET active = ? WHERE id = ?',(status,subdomain_id,))
logging.info(time.strftime("%Y-%m-%d %H:%M") + ' - Info : Status for domain %s changed' %(domain))
print('Status for domain %s changed' %(domain))
conn.commit()
2023-03-06 14:01:07 +01:00
def show_all_top_domains():
cursor = conn.cursor()
apikey = get_api()
if apikey != None:
req = urllib.request.Request('https://api.digitalocean.com/v2/domains/?per_page=200')
req.add_header('Content-Type', 'application/json')
req.add_header('Authorization', 'Bearer ' + apikey)
current = urllib.request.urlopen(req)
remote = current.read().decode('utf-8')
remoteData = json.loads(remote)
print('Domains in database are marked with a [*]')
print('================================================')
for k in remoteData["domains"]:
cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(k['name'],))
count = cursor.fetchone()[0]
if count != 0:
2023-03-07 17:49:38 +01:00
print('Name : [bold]'+k['name']+ ' [*][/bold]')
2023-03-06 14:01:07 +01:00
else:
print('Name : '+k['name'])
else:
2023-03-07 17:49:38 +01:00
print("[red]Error:[/red] Missing APIkey. Please add one!")
2023-03-06 14:01:07 +01:00
def list_sub_domains(domain):
apikey = get_api()
cursor = conn.cursor()
if apikey == None:
2023-03-07 17:49:38 +01:00
print("[red]Error:[/red] Missing APIkey. Please add one!")
2023-03-06 14:01:07 +01:00
else:
cursor.execute('SELECT COUNT(*) FROM domains WHERE name LIKE ?',(domain,))
count = cursor.fetchone()[0]
if count == 0:
2023-03-07 17:49:38 +01:00
print("[red]Error: [/red]No such domain. Check spelling or use ddns -d to show all top domains.")
2023-03-06 14:01:07 +01:00
else:
2023-03-28 12:12:04 +02:00
print('\n\nCurrent sub domains for [b]%s[/b]\n\n' % (domain))
2023-04-05 10:44:00 +02:00
print('Domain\t\t\t\tCreated\t\t\tUpdated\t\t\tChecked\t\t\tActive')
print('==================================================================================================================')
2023-03-07 17:49:38 +01:00
cursor.execute('SELECT id FROM domains WHERE name LIKE ?', (domain,))
topdomain_id = cursor.fetchone()[0]
cursor.execute('SELECT COUNT(*) FROM subdomains WHERE main_id LIKE ?',(topdomain_id,))
count = cursor.fetchone()[0]
if count == 0:
print('[red]Error:[/red] No sub domains for [b]%s[/b]' % (domain))
else:
2023-04-05 10:44:00 +02:00
cursor.execute('SELECT name,last_updated,last_checked,created,active FROM subdomains WHERE main_id LIKE ?',(topdomain_id,) )
2023-03-07 17:49:38 +01:00
subdomains = cursor.fetchall()
for i in subdomains:
2023-04-05 10:44:00 +02:00
if i[4] == 1:
active = 'True'
else:
active = 'False'
topdomain = i[0]+'.'+domain
topdomain = "{:<25}".format(topdomain)
2023-04-05 10:44:00 +02:00
print(topdomain+'\t'+i[3]+'\t'+i[1]+'\t'+i[2]+'\t'+active)
2023-03-07 17:49:38 +01:00
print('\n')
def list_do_sub_domains(domain):
apikey = get_api()
cursor = conn.cursor()
if apikey == None:
print("[red]Error:[/red] Missing APIkey. Please add one!")
else:
req = urllib.request.Request('https://api.digitalocean.com/v2/domains/'+domain+'/records?type="A"/?per_page=200')
req.add_header('Content-Type', 'application/json')
req.add_header('Authorization', 'Bearer ' + apikey)
current = urllib.request.urlopen(req)
remote = current.read().decode('utf-8')
remoteData = json.loads(remote)
print('Domains in your DigitalOcean account not in ddns DB for [b]%s[/b]' % (domain))
2023-03-07 19:05:02 +01:00
print('===================================================================')
2023-03-07 17:49:38 +01:00
for k in remoteData["domain_records"]:
if k['type'] == 'A':
cursor.execute('SELECT COUNT(*) FROM subdomains WHERE id like ?',(str(k['id']),))
count = cursor.fetchone()[0]
if count == 0:
print(k['name']+'.'+domain+'\t\tID : '+str(k['id']))
2023-03-07 17:49:38 +01:00
2023-03-06 14:01:07 +01:00
def domaininfo(domain):
2023-03-07 17:49:38 +01:00
apikey = get_api()
local_ip = get_ip()
cursor = conn.cursor()
2023-04-05 10:44:00 +02:00
if set(domain).difference(ascii_letters + '.' + digits + '@' + '-'):
2023-03-07 17:49:38 +01:00
print('[red]Error:[/red]. Give the domain name in simple form e.g. [bold]test.domain.com[/bold]')
else:
parts = domain.split('.')
topdomain = parts[1]+'.'+parts[2]
cursor.execute('SELECT id FROM domains WHERE name like ?', (topdomain,))
domainid = cursor.fetchone()[0]
cursor.execute('SELECT * FROM subdomains WHERE main_id like ?', (domainid,))
domains = cursor.fetchall()
if local_ip != domains[0][3]:
localip = '[red]%s[/red]' % (local_ip)
else:
localip = local_ip
print ('The domain [bold]%s[/bold] has the IP [bold]%s[/bold]. Your public IP is [bold]%s[/bold]' % (domain,domains[0][3],localip))
2023-03-06 14:01:07 +01:00
def show_current_info():
ipserver = None
API = get_api()
cursor = conn.cursor()
cursor.execute('SELECT COUNT(ip4_server) FROM ipservers')
count = cursor.fetchone()[0]
if count == 0:
2023-03-07 17:49:38 +01:00
ipserver = '[red]Error:[/red] No IP resolvers in DB'
2023-03-06 14:01:07 +01:00
else:
cursor.execute('SELECT * FROM ipservers')
ipservers = cursor.fetchall()
ip4server = ipservers[0][1]
ip6server = ipservers[0][2]
2023-03-06 14:01:07 +01:00
if API == None:
2023-03-07 17:49:38 +01:00
API = '[red]Error:[/red] API key not stored in DB'
cursor.execute('SELECT COUNT(*) FROM domains')
topdomains = cursor.fetchone()[0]
cursor.execute('SELECT COUNT(*) FROM subdomains')
subdomains = cursor.fetchone()[0]
2023-03-06 14:01:07 +01:00
2023-03-23 17:42:15 +01:00
print('\n[b]ddns[/b] - a DigitalOcean dynamic DNS solution.')
2023-03-08 10:28:09 +01:00
print('===================================================')
print('API key : [b]%s[/b]' % (API))
print('IP v4 resolver : [b]%s[/b]' % (ip4server))
print('IP v6 resolver : [b]%s[/b]' % (ip6server))
print('Logfile : [b]%s[/b]' % (logfile))
2023-03-08 10:28:09 +01:00
print('Top domains : [b]%s[/b]' % (topdomains))
print('sub domains : [b]%s[/b]' % (subdomains))
2023-03-06 14:01:07 +01:00
print('')
2023-03-08 10:28:09 +01:00
print('App version : [b]%s[/b] (https://gitlab.pm/rune/ddns)' % (app_version))
print('')
print('[i]IPv6 is not supported and not listed here.[/i]')
2023-03-06 14:01:07 +01:00
def ip_server(ipserver, ip_type):
cursor = conn.cursor()
if ip_type == '4':
cursor.execute('SELECT COUNT(ip4_server) FROM ipservers')
count = cursor.fetchone()[0]
if count == 0:
cursor.execute('INSERT INTO ipservers values(?,?,?)', (None, ipserver,None))
conn.commit()
print('New IP resolver (%s) for ipv%s added.' % (ipserver, ip_type))
else:
cursor.execute('UPDATE ipservers SET ip4_server = ? WHERE id = 1',(ipserver,))
print('IP resolver (%s) for ipv%s updated.' % (ipserver, ip_type))
2023-03-23 17:42:15 +01:00
logging.info(time.strftime("%Y-%m-%d %H:%M")+' - Info : IP resolver (%s) for ipv%s updated.' % (ipserver, ip_type))
2023-03-06 14:01:07 +01:00
conn.commit()
elif ip_type == '6':
cursor.execute('SELECT COUNT(ip6_server) FROM ipservers')
count = cursor.fetchone()[0]
if count == 0:
cursor.execute('INSERT INTO ipservers values(?,?,?)', (None, None,ipserver))
conn.commit()
print('New IP resolver (%s) for ipv%s added. \n\r This IP version is not supported.' % (ipserver, ip_type))
else:
cursor.execute('UPDATE ipservers SET ip6_server = ? WHERE id = 1',(ipserver,))
print('IP resolver (%s) for ipv%s updated. \n\r This IP version is not supported.' % (ipserver, ip_type))
2023-03-23 17:42:15 +01:00
logging.info(time.strftime("%Y-%m-%d %H:%M") + ' - Info : IP resolver (%s) for ipv%s updated.' % (ipserver, ip_type))
2023-03-06 14:01:07 +01:00
conn.commit()
def updateip(force):
2023-03-07 17:49:38 +01:00
apikey = get_api()
current_ip = get_ip()
cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM subdomains')
count = cursor.fetchone()[0]
now = datetime.now().strftime("%d-%m-%Y %H:%M")
2023-03-23 17:42:15 +01:00
updated = None
2023-03-07 17:49:38 +01:00
if count == 0:
print('[red]Error: [/red]There are no dynamic domains active.'\
' Start by adding a new domain with [i]ddns -s test.example.com[/i]')
else:
2023-04-05 10:44:00 +02:00
cursor.execute('SELECT id,active FROM subdomains')
2023-03-07 17:49:38 +01:00
rows = cursor.fetchall()
for i in rows:
cursor.execute('SELECT name FROM domains WHERE id like (SELECT main_id from subdomains WHERE id = ?)',(i[0],))
2023-04-05 10:44:00 +02:00
domain_info = cursor.fetchone()
domain_name = str(domain_info[0])
domain_status = i[1]
2023-03-07 17:49:38 +01:00
subdomain_id = str(i[0])
# Chek if an update is required
2023-04-05 10:44:00 +02:00
if domain_status == 1:
req = urllib.request.Request('https://api.digitalocean.com/v2/domains/' + domain_name + '/records/' + subdomain_id)
req.add_header('Content-Type', 'application/json')
req.add_header('Authorization', 'Bearer ' + apikey)
current = urllib.request.urlopen(req)
remote = current.read().decode('utf-8')
remoteData = json.loads(remote)
remoteIP4 = remoteData['domain_record']['data']
domainname = str(remoteData['domain_record']['name'])
if remoteIP4 != current_ip or force == True and domain_status == 1:
updated = True
data = {'type': 'A', 'data': current_ip}
headers = {'Authorization': 'Bearer ' + apikey, "Content-Type": "application/json"}
response = requests.patch('https://api.digitalocean.com/v2/domains/'+domain_name+'/records/' + subdomain_id, data=json.dumps(data), headers=headers)
if str(response) != '<Response [200]>':
logging.error(time.strftime("%Y-%m-%d %H:%M")+' - Error updating ('+str(domain_name)+') : ' + str(response.content))
else:
cursor.execute('UPDATE subdomains SET current_ip4=? WHERE id = ?',(current_ip,subdomain_id,))
cursor.execute('UPDATE subdomains SET last_updated=? WHERE id = ?',(now,subdomain_id,))
cursor.execute('UPDATE subdomains SET last_checked=? WHERE id = ?',(now,subdomain_id,))
conn.commit()
else:
2023-03-23 17:42:15 +01:00
cursor.execute('UPDATE subdomains SET last_checked=? WHERE id = ?',(now,subdomain_id,))
conn.commit()
2023-03-23 17:42:15 +01:00
if updated == None:
logging.info(time.strftime("%Y-%m-%d %H:%M") + ' - Info : No updated necessary')
else:
logging.info(time.strftime("%Y-%m-%d %H:%M") + ' - Info : Updates done. Use ddns -l domain.com to check domain')
2023-03-06 14:01:07 +01:00
2023-03-07 17:49:38 +01:00
def local_add_subdomain(domain,domainid):
now = datetime.now().strftime("%Y-%m-%d %H:%M")
2023-04-05 10:44:00 +02:00
if set(domain).difference(ascii_letters + '.' + digits + '-' + '@'):
2023-03-07 17:49:38 +01:00
print('[red]Error:[/red] Give the domain name in simple form e.g. [b]test.domain.com[/b]')
else:
parts = domain.split('.')
2023-04-15 17:11:35 +02:00
if len(parts) > 3:
top = parts[1] + '.' + parts[2] + '.' + parts[3]
sub = parts[0]
else:
sub = parts[0]
top = parts[1] + '.' + parts[2]
2023-03-07 17:49:38 +01:00
apikey = get_api()
if apikey == None:
print("[red]Error:[/red] Missing APIkey. Please add one!")
else:
ip = get_ip()
if ip == None or 'urlopen error' in ip:
print('[red]Error:[/red] Failed to get public IP. Do you have a typo in your URI? [red]Error %s.[/red]' % (ip))
else:
cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(top,))
count = cursor.fetchone()[0]
if count == 0:
print('[red]Error:[/red] Top domain [bold]%s[/bold] does not exist in the DB. Please add it with [i]ddns -t %s[/i].' % (top,top))
else:
cursor.execute('SELECT id FROM domains WHERE name LIKE ?',(top,))
topdomain_id = cursor.fetchone()
topdomain_id = topdomain_id[0]
cursor.execute('SELECT count(*) FROM subdomains WHERE main_id LIKE ? AND name like ?',(topdomain_id,sub,))
count = cursor.fetchone()[0]
if count != 0:
print('[red]Error:[/red] [bold]%s[/bold] already exists.' % (domain))
else:
2023-04-05 10:44:00 +02:00
cursor.execute('INSERT INTO subdomains values(?,?,?,?,?,?,?,?,?)',(domainid,topdomain_id,sub,ip,None,now,now,now,1,))
2023-03-07 17:49:38 +01:00
conn.commit()
print('The domain %s has been added.' % (domain))
2023-04-05 10:44:00 +02:00
def show_log():
log_file = open(logfile, 'r')
content = log_file.read()
print(content)
log_file.close()
2023-03-06 14:01:07 +01:00
2023-03-19 17:23:11 +01:00
def updatedb():
# Update DB with new column 20.03.23
# Add last updated field for subdomains
2023-03-28 12:12:04 +02:00
new_column = 'last_updated'
2023-03-19 17:23:11 +01:00
info = conn.execute("PRAGMA table_info('subdomains')").fetchall()
2023-03-28 12:12:04 +02:00
if not any(new_column in word for word in info):
2023-03-20 14:57:15 +01:00
add_column = "ALTER TABLE subdomains ADD COLUMN last_updated text default 'N/A'"
2023-03-19 17:23:11 +01:00
conn.execute(add_column)
conn.commit()
2023-03-23 17:42:15 +01:00
logging.info(time.strftime("%Y-%m-%d %H:%M")+' - Info : Database updated')
2023-03-19 17:23:11 +01:00
2023-03-28 12:12:04 +02:00
new_column = 'last_checked'
info = conn.execute("PRAGMA table_info('subdomains')").fetchall()
2023-03-28 12:12:04 +02:00
if not any(new_column in word for word in info):
add_column = "ALTER TABLE subdomains ADD COLUMN last_checked text default 'N/A'"
conn.execute(add_column)
2023-03-23 17:42:15 +01:00
conn.commit()
logging.info(time.strftime("%Y-%m-%d %H:%M") + ' - Info : Database updated')
2023-03-28 12:12:04 +02:00
new_column = 'created'
info = conn.execute("PRAGMA table_info('subdomains')").fetchall()
if not any(new_column in word for word in info):
2023-03-28 12:16:45 +02:00
add_column = "ALTER TABLE subdomains ADD COLUMN created text default '[b]Unknown Info[/b]'"
2023-03-28 12:12:04 +02:00
conn.execute(add_column)
conn.commit()
logging.info(time.strftime("%Y-%m-%d %H:%M") + ' - Info : Database updated')
2023-04-05 10:44:00 +02:00
new_column = 'active'
info = conn.execute("PRAGMA table_info('subdomains')").fetchall()
if not any(new_column in word for word in info):
add_column = "ALTER TABLE subdomains ADD COLUMN active integer default 1"
conn.execute(add_column)
conn.commit()
logging.info(time.strftime("%Y-%m-%d %H:%M") + ' - Info : Database updated')
2023-03-19 17:23:11 +01:00
2023-03-06 14:01:07 +01:00
# Commandline arguments
conn = connect_database()
2023-03-19 17:23:11 +01:00
updatedb()
2023-03-23 17:42:15 +01:00
2023-03-06 14:01:07 +01:00
parser = argparse.ArgumentParser(prog='ddns',
2023-03-07 17:49:38 +01:00
description='Application to use domains from DigitalOcean account as dynamic '\
'DNS domain(s).\nThe app only supports IP4. IPv6 is planned for a later release!'\
2023-03-23 17:42:15 +01:00
'\nYou\'ll always find the latest version on https://gitlab.pm/rune/ddns\n\n'\
'For bugs, suggestions, pull requests visit https://gitlab.pm/rune/ddns/issues',
2023-03-07 17:49:38 +01:00
formatter_class=RawTextHelpFormatter,
2023-03-06 14:01:07 +01:00
epilog='Making Selfhosting easier...')
2023-04-05 10:44:00 +02:00
parser.add_argument('-a', '--api', help='Add/Change API key.\n\n',
2023-03-06 14:01:07 +01:00
nargs=1, metavar=('APIkey'), required=False, action="append")
2023-04-05 10:44:00 +02:00
parser.add_argument('-f', '--force', help='Force update of IP address for all domains.\n\n',
required=False, action="store_true")
2023-04-05 10:44:00 +02:00
parser.add_argument('-l', '--list', help='List subdomains for supplied domain.\n\n',
2023-03-06 14:01:07 +01:00
nargs=1, metavar=('domain'), required=False, action="append")
2023-04-05 10:44:00 +02:00
parser.add_argument('-o', '--serverdomains', help='List subdomains for supplied domain not in ddns DB.\n\n',
2023-03-07 17:49:38 +01:00
nargs=1, metavar=('domain'), required=False, action="append")
2023-04-05 10:44:00 +02:00
parser.add_argument('-d', '--domains', help='List top domains in your DigitalOcean account.\n\n',
2023-03-06 14:01:07 +01:00
required=False, action="store_true")
2023-04-05 10:44:00 +02:00
parser.add_argument('-c', '--current', help='List the current IP address for the sub-domain given\n\n',
2023-03-06 14:01:07 +01:00
required=False, nargs=1, action="append")
2023-04-05 10:44:00 +02:00
parser.add_argument('-t', '--top', help='Add a new domain from your DigitalOcean account to use as a dynamic DNS domain\n\n',
2023-03-06 14:01:07 +01:00
required=False, nargs=1, metavar=('domain'), action='append')
2023-04-05 10:44:00 +02:00
parser.add_argument('-s', '--sub', help='Add a new subdomain to your DigitalOcean account and use as dynamic DNS.\n\n\n',
2023-03-07 17:49:38 +01:00
required=False, nargs=1, metavar=('domain'), action='append')
2023-04-05 10:44:00 +02:00
parser.add_argument('-k', '--local', help='Add an existing DigitalOcean subdomain to your ddns DB and use as dynamic DNS.\n\n',
2023-03-07 17:49:38 +01:00
required=False, nargs=2, metavar=('domain','domainid'), action='append')
2023-04-05 10:44:00 +02:00
parser.add_argument('-r', '--remove', help='Remove a subdomain from your DigitalOcean account and ddns.\n\n',
2023-03-07 17:49:38 +01:00
required=False, nargs=1, metavar=('domain'), action='append')
2023-03-06 14:01:07 +01:00
2023-04-05 10:44:00 +02:00
parser.add_argument('-v', '--version', help='Show current version and config info\n\n',
2023-03-06 14:01:07 +01:00
required=False, action='store_true')
2023-04-05 10:44:00 +02:00
parser.add_argument('-q', '--log', help=argparse.SUPPRESS, required=False, action='store_true')
parser.add_argument('-p', '--ipserver', help='Sets or updates IP server lookup to use. Indicate 4 or 6 for IP type.\n\n',
2023-03-06 14:01:07 +01:00
required=False, nargs=2, metavar=('ip4.iurl.no', '4'), action="append")
2023-04-05 10:44:00 +02:00
parser.add_argument('-e', '--edit', help='Changes domain from active to inactive or the other way around...',
required=False, nargs=1, metavar=('test.example.com'), action="append")
2023-03-06 14:01:07 +01:00
args = vars(parser.parse_args())
if args['list']:
list_sub_domains(args['list'][0][0])
elif args['domains']:
show_all_top_domains()
2023-03-07 17:49:38 +01:00
elif args['serverdomains']:
list_do_sub_domains(args['serverdomains'][0][0])
2023-03-06 14:01:07 +01:00
elif args['current']:
domaininfo(args['current'][0][0])
elif args['top']:
add_domian(args['top'][0][0])
elif args['sub']:
2023-03-07 17:49:38 +01:00
add_subdomain(args['sub'][0][0])
2023-03-07 19:05:02 +01:00
elif args['version']:
2023-03-06 14:01:07 +01:00
show_current_info()
elif args['force']:
updateip(True)
2023-04-05 10:44:00 +02:00
elif args['log']:
show_log()
2023-03-06 14:01:07 +01:00
elif args['ipserver']:
ip_server(args['ipserver'][0][0],args['ipserver'][0][1])
elif args['api']:
api(args['api'][0][0])
2023-03-07 17:49:38 +01:00
elif args['remove']:
remove_subdomain(args['remove'][0][0])
2023-04-05 10:44:00 +02:00
elif args['edit']:
edit_subdomain(args['edit'][0][0])
2023-03-07 17:49:38 +01:00
elif args['local']:
local_add_subdomain(args['local'][0][0],args['local'][0][1])
2023-03-06 14:01:07 +01:00
else:
updateip(None)
2023-03-28 12:12:04 +02:00