ungleich-cli/ungleich_ripe.py

79 lines
2.8 KiB
Python
Executable file

import argparse
import ipaddress
import json
import urllib.request
import pprint
# RIPE_URL = "https://rest.db.ripe.net/{source}/{objecttype}/{key}"
RIPE_URL = "https://rest.db.ripe.net/ripe"
RIPE_URL = "https://rest-test.db.ripe.net/test"
class ungleichRIPE(object):
def __init__(self, parser, parents):
self.parser = parser
self.parser['ripe-add-route6'] = self.parser['sub'].add_parser(
'ripe-add-route6',
parents=[parents],
help="Create route6 object in the ripe database")
self.parser['ripe-add-route6'].add_argument('--network',
required=True)
self.parser['ripe-add-route6'].add_argument('--description',
required=True)
self.parser['ripe-add-route6'].add_argument('--password',
required=True,
help="Password for accessing the RIPE rest API")
self.parser['ripe-add-route6'].set_defaults(func=self.route6_add)
def route6_add(self, args):
try:
net = ipaddress.IPv6Network(args.network)
except Exception as e:
print("Sorry, {} does not look like an IPv6 network: {}".format(args.network, e))
raise
url = "{}/route6/?password={}".format(RIPE_URL, args.password)
ripe_object = {}
ripe_object['route6'] = args.network
ripe_object['origin'] = "AS209898"
ripe_object['descr'] = args.description
ripe_object['mnt-by'] = "mnt-ungleich"
ripe_attributes = [{ "name": key, "value": value } for key, value in ripe_object.items() ]
# Format according to API layout
ripe_element = {}
ripe_element['objects'] = []
ripe_element['objects'].append(
{ "object":
[
{
"attributes": {
"attribute": ripe_attributes
}
}
]
}
)
data = json.dumps(ripe_element).encode('utf-8')
# debug
pprint.pprint(ripe_element)
method = 'POST'
req = urllib.request.Request(url=url,
data=data,
method='POST',
headers={
"Content-Type": "application/json",
"Accept": "application/json"
})
print("Adding a v6 route object at {} for {} with {} req={}".format(url, args.network, data, str(req)))
with urllib.request.urlopen(req) as f:
print(f.read().decode('utf-8'))