ungleich-cli/ungleich_ripe.py

import argparse
import ipaddress
import json
import urllib.request
import pprint
import requests

# 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_object['source'] = "TEST"

        ripe_attributes = [{ "name": key, "value": value } for key, value in ripe_object.items() ]

        # Format according to API layout
        ripe_element = {}
        ripe_element['objects'] = { 
            "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'))

        r = requests.post(url, data=data, headers={
            "Content-Type": "application/json",
            "Accept": "application/json"
        })
        pprint.pprint(json.loads(r.content.decode("utf-8")))
        # print(r.content.decode("utf-8"))