config to read/store user data in json file

This commit is contained in:
wcolmenares 2019-06-11 20:57:55 -04:00
parent 79abf872f2
commit ba6e83f925
3 changed files with 98 additions and 10 deletions

View file

@ -7,6 +7,7 @@ from ungleich_ripe import ungleichRIPE
from ungleich_account import Account_Create
from ungleich_weather import ungleichWeather
from ungleich_ssh_key import SSHKey
from ungleich_config import Ungleich_Config
VERSION = "0.0.4"
@ -17,7 +18,8 @@ class ungleichCLI(object):
# FIXME: make it generic
dns = ungleichDNS(self.parser, self.parser_parents)
ripe = ungleichRIPE(self.parser, self.parser_parents)
ripe = Account_Create(self.parser, self.parser_parents)
account = Account_Create(self.parser, self.parser_parents)
config = Ungleich_Config(self.parser, self.parser_parents)
SSHKey(self.parser, self.parser_parents)
ungleichWeather(self.parser, self.parser_parents)

82
ungleich_config.py Normal file
View file

@ -0,0 +1,82 @@
import argparse
import json
import urllib.request
class Ungleich_Config(object):
def __init__(self, parser, parents):
self.parser = parser
self.parser['add'] = self.parser['sub'].add_parser(
'config-add',
help="Add user to configuration file",
parents=[parents])
self.parser['list'] = self.parser['sub'].add_parser(
'config-list',
help="list users in configuration file",
parents=[parents])
self.parser['delete'] = self.parser['sub'].add_parser(
'config-delete',
help="remove a user in configuration file",
parents=[parents])
self.parser['add'].add_argument('--name', help='otp name', required=True)
self.parser['add'].add_argument('--realm', help='otp realm', required=True)
self.parser['add'].add_argument('--seed', help='otp seed', required=True)
self.parser['delete'].add_argument('--id', help='user ID to remove', required=True)
self.parser['list'].set_defaults(func=self._list_users)
self.parser['add'].set_defaults(func=self._add_user)
self.parser['delete'].set_defaults(func=self._delete_user)
def _add_user(self, args):
try:
f = open('users.json', 'r')
json_info = json.loads(f.read())
if len(json_info) > 0:
with open('users.json', 'w') as fp:
last_id = int(list(json_info.keys())[-1]) + 1
json_info[last_id] = {"name": args.name, "realm": args.realm, "seed": args.seed}
json.dump(json_info, fp)
print('User added.')
else:
with open('users.json', 'w') as fp:
d = {"name": args.name, "realm": args.realm, "seed": args.seed}
json.dump({"1": d}, fp)
print('User added.')
except FileNotFoundError:
with open('users.json', 'w') as fp:
d = {"name": args.name, "realm": args.realm, "seed": args.seed}
json.dump({"1": d}, fp)
print('User added.')
def _list_users(self, args):
try:
f = open('users.json', 'r')
json_info = json.loads(f.read())
for key, value in json_info.items():
print('{}: {}'.format(key, value['name']))
except FileNotFoundError:
print('No users loaded.')
def _delete_user(self, args):
try:
f = open('users.json', 'r')
json_info = json.loads(f.read())
try:
del json_info[args.id]
f = open('users.json', 'w')
x = 1
for i in list(json_info.keys()):
json_info[str(x)] = json_info.pop(i)
x+=1
f.write(json.dumps(json_info))
f.close()
except KeyError:
print("No user with such id.")
except FileNotFoundError:
print('No users loaded.')

View file

@ -1,7 +1,7 @@
import urllib.request
import argparse
import json
from pyotp import TOTP
class ungleichDNS(object):
def __init__(self, parser, parents):
@ -13,15 +13,16 @@ class ungleichDNS(object):
parents=[parents])
self.parser['dns'].add_argument('--set-reverse', help='REQUIRED: IPv6 Address of your VM', required=True)
self.parser['dns'].add_argument('--user', help='Your ungleich username', required=True)
self.parser['dns'].add_argument('--token', help='Your ungleich 6 digit OTP generated token', type=int, required=True)
self.parser['dns'].add_argument('--user', help='Your stored user ID', required=True)
self.parser['dns'].add_argument('--name', help='Hostname', required=True)
self.parser['dns'].add_argument('--email', help='registered email', required=True)
self.parser['dns'].add_argument('--realm', help='Otp realm', required=True)
self.parser['dns'].set_defaults(func=self._handle_dns)
def _handle_dns(self, args):
"""Reverse dns endpoint."""
f = open('users.json', 'r')
data = json.loads(f.read())[args.user]
url = 'https://dns.service.ungleich.ch'
req = urllib.request.Request(
url=url,
@ -30,13 +31,16 @@ class ungleichDNS(object):
"Content-Type": "application/json"
},
data=json.dumps({
'username': args.user,
'token': args.token,
'username': data['name'],
'token': TOTP(data['seed']).now(),
'ipaddress': args.set_reverse,
'name': args.name,
'email': args.email,
'realm': args.realm
'realm': data['realm']
}).encode('utf-8')
)
try:
response = urllib.request.urlopen(req).read()
print(json.loads(response))
except urllib.error.HTTPError as e:
error_message = e.read()
print(json.loads(error_message))