123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179 |
- #!/usr/bin/env python2
- #
- # Copyright (c) 2017-2018 Joe Clarke <jclarke@cisco.com>
- # All rights reserved.
- #
- # Redistribution and use in source and binary forms, with or without
- # modification, are permitted provided that the following conditions
- # are met:
- # 1. Redistributions of source code must retain the above copyright
- # notice, this list of conditions and the following disclaimer.
- # 2. Redistributions in binary form must reproduce the above copyright
- # notice, this list of conditions and the following disclaimer in the
- # documentation and/or other materials provided with the distribution.
- #
- # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
- # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- # ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
- # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
- # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
- # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
- # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
- # SUCH DAMAGE.
- # NOTE: This script is now obsolete. We no longer use Prime Infra.
- import requests
- from requests.packages.urllib3.exceptions import InsecureRequestWarning
- requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
- import json
- import sys
- import re
- import CLEUCreds
- PI = "10.100.253.22"
- DNS_BASE = "https://dc1-dns.ciscolive.local:8443/web-services/rest/resource/"
- DOMAIN = "ciscolive.local"
- CNR_HEADERS = {"authorization": CLEUCreds.JCLARKE_BASIC, "accept": "application/json", "content-type": "application/json"}
- PAGE_SIZE = 1000
- def get_devs():
- global PI, PAGE_SIZE, DOMAIN
- url = "https://{}/webacs/api/v1/data/Devices.json?.full=true&.maxResults={}".format(PI, PAGE_SIZE)
- headers = {"Connection": "close"}
- devices = []
- done = False
- first = 0
- while not done:
- code = 401
- i = 0
- nurl = url + "&.firstResult=" + str(first * PAGE_SIZE)
- while code != 200 and i < 10:
- response = requests.request("GET", nurl, auth=(CLEUCreds.PI_USER, CLEUCreds.PI_PASS), headers=headers, verify=False)
- code = response.status_code
- if code != 200:
- i += 1
- time.sleep(3)
- if code == 200:
- j = json.loads(response.text)
- if int(j["queryResponse"]["@last"]) + 1 == int(j["queryResponse"]["@count"]):
- done = True
- else:
- first += 1
- for dev in j["queryResponse"]["entity"]:
- dev_dic = {}
- if "deviceName" in dev["devicesDTO"]:
- dev_dic["name"] = dev["devicesDTO"]["deviceName"]
- else:
- continue
- if not re.search(r"^0", dev_dic["name"]):
- continue
- dev_dic["ip"] = dev["devicesDTO"]["ipAddress"]
- nparts = dev_dic["name"].split("-")
- if len(nparts) == 3:
- dev_dic["aliases"] = []
- dev_dic["name"] = dev_dic["name"].replace(".{}".format(DOMAIN), "")
- dev_dic["aliases"].append("-".join(nparts[0:2]) + ".{}.".format(DOMAIN))
- dev_dic["aliases"].append(nparts[2] + ".{}.".format(DOMAIN))
- devices.append(dev_dic)
- return devices
- def add_entry(url, hname, dev):
- global CNR_HEADERS, DOMAIN
- aliases = []
- if "aliases" in dev:
- aliases = dev["aliases"]
- try:
- host_obj = {"addrs": {"stringItem": [dev["ip"]]}, "aliases": {"stringItem": []}, "name": hname, "zoneOrigin": DOMAIN}
- for alias in aliases:
- host_obj["aliases"]["stringItem"].append(alias)
- response = requests.request("PUT", url, headers=CNR_HEADERS, json=host_obj, verify=False)
- response.raise_for_status()
- print("Added entry for {} ==> {} with aliases {}".format(hname, dev["ip"], str(aliases)))
- except Exception as e:
- sys.stderr.write("Error adding entry for {}: {}\n".format(hname, e))
- if __name__ == "__main__":
- devs = get_devs()
- for dev in devs:
- hname = dev["name"].replace(".{}".format(DOMAIN), "")
- url = DNS_BASE + "CCMHost" + "/{}".format(hname)
- response = requests.request("GET", url, headers=CNR_HEADERS, params={"zoneOrigin": DOMAIN}, verify=False)
- if response.status_code == 404:
- iurl = DNS_BASE + "CCMHost"
- response = requests.request(
- "GET", iurl, params={"zoneOrigin": DOMAIN, "addrs": dev["ip"] + "$"}, headers=CNR_HEADERS, verify=False
- )
- cur_entry = []
- if response.status_code != 404:
- cur_entry = response.json()
- if len(cur_entry) > 0:
- print("Found entry for {}: {}".format(dev["ip"], response.status_code))
- cur_entry = response.json()
- if len(cur_entry) > 1:
- print("ERROR: Found multiple entries for IP {}".format(dev["ip"]))
- continue
- print("Found old entry for IP {} => {}".format(dev["ip"], cur_entry[0]["name"]))
- durl = DNS_BASE + "CCMHost" + "/{}".format(cur_entry[0]["name"])
- try:
- response = requests.request("DELETE", durl, params={"zoneOrigin": DOMAIN}, headers=CNR_HEADERS, verify=False)
- response.raise_for_status()
- except Exception as e:
- sys.stderr.write("Failed to delete stale entry for {} ({})\n".format(cur_entry[0]["name"], dev["ip"]))
- continue
- add_entry(url, hname, dev)
- else:
- cur_entry = response.json()
- create_new = True
- for addr in cur_entry["addrs"]["stringItem"]:
- if addr == dev["ip"]:
- if "aliases" in dev and "aliases" in cur_entry:
- if (len(dev["aliases"]) > 0 and "stringItem" not in cur_entry["aliases"]) or (
- len(dev["aliases"]) != len(cur_entry["aliases"]["stringItem"])
- ):
- break
- common = set(dev["aliases"]) & set(cur_entry["aliases"]["stringItem"])
- if len(common) != len(dev["aliases"]):
- break
- create_new = False
- break
- elif ("aliases" in dev and "aliases" not in cur_entry) or ("aliases" in cur_entry and "aliases" not in dev):
- break
- else:
- create_new = False
- break
- if create_new:
- print("Deleting entry for {}".format(hname))
- try:
- response = requests.request("DELETE", url, headers=CNR_HEADERS, params={"zoneOrigin": DOMAIN}, verify=False)
- response.raise_for_status()
- except Exception as e:
- sys.stderr.write("Error deleting entry for {}: {}\n".format(hname, e))
- add_entry(url, hname, dev)
- else:
- print("Not creating a new entry for {} as it already exists".format(dev["name"]))
|