Initial commit
This commit is contained in:
commit
81f5ef665c
309
ddns.py
Executable file
309
ddns.py
Executable file
@ -0,0 +1,309 @@
|
||||
#!/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
|
||||
|
||||
homefilepath = Path.home()
|
||||
filepath = homefilepath.joinpath('.config/ddns')
|
||||
database = filepath.joinpath('ddns.db')
|
||||
app_version = '0.1'
|
||||
|
||||
|
||||
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)
|
||||
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:
|
||||
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,
|
||||
ip4_server text,
|
||||
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,
|
||||
name text)''')
|
||||
|
||||
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))
|
||||
print('Your API key has been added.')
|
||||
else:
|
||||
cursor.execute('UPDATE apikey SET api = ? WHERE id = 1',(api_value,))
|
||||
print('Your API key has been updated.')
|
||||
conn.commit()
|
||||
|
||||
|
||||
def add_domian(domain):
|
||||
cursor = conn.cursor()
|
||||
cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(domain,))
|
||||
count = cursor.fetchone()[0]
|
||||
if count == 0:
|
||||
cursor.execute('INSERT INTO domains values(?,?)', (None, domain,))
|
||||
print('The domain %s has been added to the DB' % (domain))
|
||||
conn.commit()
|
||||
else:
|
||||
print('Error: Domain name (%s) already in database!' % (domain))
|
||||
|
||||
|
||||
|
||||
def add_subdomain(sub,top):
|
||||
apikey = get_api()
|
||||
if apikey == None:
|
||||
print("Missing APIkey. Please add one!")
|
||||
else:
|
||||
ip = get_ip()
|
||||
if ip == None or 'urlopen error' in ip:
|
||||
print('Failed to get public IP. Do you have a typo in your URI? Error %s' % (ip))
|
||||
else:
|
||||
cursor = conn.cursor()
|
||||
cursor.execute('SELECT COUNT(*) FROM domains WHERE name like ?',(top,))
|
||||
count = cursor.fetchone()[0]
|
||||
if count == 0:
|
||||
print('Error: Top domain %s does not exist in the DB. Please add it with ddns -t %s ' % (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('The subdomain %s already exists for the domain %s.' % (sub,top))
|
||||
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'])
|
||||
cursor.execute('INSERT INTO subdomains values(?,?,?)',(domainid,topdomain_id,sub,))
|
||||
conn.commit()
|
||||
print('The subdomain %s for the domain %s has been added.' % (sub,top))
|
||||
else:
|
||||
return 'Error: %s ' % (str(response))
|
||||
|
||||
|
||||
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:
|
||||
print('Name : '+k['name']+ ' *')
|
||||
else:
|
||||
print('Name : '+k['name'])
|
||||
|
||||
else:
|
||||
print("Missing APIkey. Please add one!")
|
||||
|
||||
|
||||
|
||||
def list_sub_domains(domain):
|
||||
# TODO: Finish this :)
|
||||
apikey = get_api()
|
||||
cursor = conn.cursor()
|
||||
if apikey == None:
|
||||
print("Missing APIkey. Please add one!")
|
||||
else:
|
||||
cursor.execute('SELECT COUNT(*) FROM domains WHERE name LIKE ?',(domain,))
|
||||
count = cursor.fetchone()[0]
|
||||
if count == 0:
|
||||
print("Error: No such domain. Check spelling or use ddns -d to show all top domains.")
|
||||
else:
|
||||
cursor.execute('SELECT * FROM domains WHERE name LIKE ?', (domain,))
|
||||
row = cursor.fetchone()
|
||||
print(row)
|
||||
|
||||
print("List -> " + domain)
|
||||
|
||||
|
||||
def domaininfo(domain):
|
||||
print("domaininfo")
|
||||
|
||||
|
||||
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:
|
||||
ipserver = 'No IP resolvers in DB'
|
||||
else:
|
||||
cursor.execute('SELECT * FROM ipservers')
|
||||
ipserver = cursor.fetchall()[0][1]
|
||||
|
||||
if API == None:
|
||||
API = 'API key not stored in DB'
|
||||
|
||||
print('Current info.')
|
||||
print('==========================================')
|
||||
print('API key : %s' % (API))
|
||||
print('IP resolver : %s' % (ipserver))
|
||||
print('App version : %s' % (app_version))
|
||||
print('')
|
||||
print('IPv6 is not supported and not listed here.')
|
||||
|
||||
|
||||
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))
|
||||
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))
|
||||
conn.commit()
|
||||
|
||||
|
||||
|
||||
def updateip():
|
||||
print("updateip")
|
||||
|
||||
|
||||
|
||||
|
||||
# Commandline arguments
|
||||
conn = connect_database()
|
||||
|
||||
parser = argparse.ArgumentParser(prog='ddns',
|
||||
description='Application to use domains from DigitalOcean account as dynamic DNS domain(s).\r\nThe app only supports IP4. IPv6 is planned for a later release!',
|
||||
epilog='Making Selfhosting easier...')
|
||||
|
||||
parser.add_argument('-a', '--api', help='Add/Change API key.',
|
||||
nargs=1, metavar=('APIkey'), required=False, action="append")
|
||||
|
||||
parser.add_argument('-l', '--list', help='List subdomains for supplied domain.',
|
||||
nargs=1, metavar=('domain'), required=False, action="append")
|
||||
|
||||
parser.add_argument('-d', '--domains', help='List top domains on your DigitalOcean account.',
|
||||
required=False, action="store_true")
|
||||
|
||||
parser.add_argument('-c', '--current', help='List the current IP address for the sub-domain given',
|
||||
required=False, nargs=1, action="append")
|
||||
|
||||
parser.add_argument('-t', '--top', help='Add a new domain from your DigitalOcean account to use as a dynamic DNS domain',
|
||||
required=False, nargs=1, metavar=('domain'), action='append')
|
||||
|
||||
parser.add_argument('-s', '--sub', help='Add a new subdomain to dynamic DNS DO domain',
|
||||
required=False, nargs=2, metavar=('subdomain', 'domain'), action='append')
|
||||
|
||||
parser.add_argument('-i', '--info', help='Show current config info',
|
||||
required=False, action='store_true')
|
||||
|
||||
parser.add_argument('-p', '--ipserver', help='Set IP server lookup to use. Indicate 4 or 6 for IP type.',
|
||||
required=False, nargs=2, metavar=('ip4.iurl.no', '4'), action="append")
|
||||
args = vars(parser.parse_args())
|
||||
|
||||
if args['list']:
|
||||
list_sub_domains(args['list'][0][0])
|
||||
elif args['domains']:
|
||||
show_all_top_domains()
|
||||
elif args['current']:
|
||||
domaininfo(args['current'][0][0])
|
||||
elif args['top']:
|
||||
add_domian(args['top'][0][0])
|
||||
elif args['sub']:
|
||||
add_subdomain(args['sub'][0][0],args['sub'][0][1])
|
||||
#add_subdomain(args['sub'][0][0])
|
||||
elif args['info']:
|
||||
show_current_info()
|
||||
elif args['ipserver']:
|
||||
ip_server(args['ipserver'][0][0],args['ipserver'][0][1])
|
||||
elif args['api']:
|
||||
api(args['api'][0][0])
|
||||
else:
|
||||
get_ip()
|
||||
# get_api()
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user