ungleich-cli/ungleich_dns.py

47 lines
1.6 KiB
Python
Raw Normal View History

2019-06-06 14:19:43 +00:00
import urllib.request
import argparse
2019-06-06 14:19:43 +00:00
import json
from pyotp import TOTP
class ungleichDNS(object):
def __init__(self, parser, parents):
self.parser = parser
self.parser['dns'] = self.parser['sub'].add_parser(
2019-05-12 10:42:28 +00:00
'dns',
help="Manage DNS entries @ ungleich",
parents=[parents])
2019-05-18 16:39:28 +00:00
self.parser['dns'].add_argument('--set-reverse', help='REQUIRED: IPv6 Address of your VM', required=True)
self.parser['dns'].add_argument('--user', help='Your stored user ID', required=True)
2019-05-18 16:39:28 +00:00
self.parser['dns'].add_argument('--name', help='Hostname', required=True)
self.parser['dns'].add_argument('--email', help='registered email', required=True)
2019-05-15 22:49:00 +00:00
self.parser['dns'].set_defaults(func=self._handle_dns)
2019-05-15 22:49:00 +00:00
def _handle_dns(self, args):
2019-05-30 09:24:09 +00:00
"""Reverse dns endpoint."""
f = open('users.json', 'r')
data = json.loads(f.read())[args.user]
2019-06-06 14:19:43 +00:00
url = 'https://dns.service.ungleich.ch'
req = urllib.request.Request(
url=url,
method='POST',
headers={
"Content-Type": "application/json"
},
data=json.dumps({
'username': data['name'],
'token': TOTP(data['seed']).now(),
2019-05-15 22:49:00 +00:00
'ipaddress': args.set_reverse,
'name': args.name,
'email': args.email,
'realm': data['realm']
2019-06-06 14:19:43 +00:00
}).encode('utf-8')
)
try:
response = urllib.request.urlopen(req).read()
except urllib.error.HTTPError as e:
error_message = e.read()
print(json.loads(error_message))