Update updateip.py

add CIDR support validate interface IPv4 address

Author: gregbo

restconf_update_ipaddress/updateip.py

@@ -26,6 +26,7 @@
 import requests
 import sys
 import os
+import ipaddress
 from argparse import ArgumentParser
 from collections import OrderedDict
 from getpass import getpass
@@ -97,7 +98,7 @@ def configure_ip_address(url_base, interface, ip, username, password):
 
 
 # Retrieve and print the current configuration of an interface
-def print_interface_details(url_base, interface, username, password):
+def print_interface_details(url_base, interface, username, password, cidr):
  url = url_base + "/interface={i}".format(i=interface)
 
  # this statement performs a GET on the specified url
@@ -116,10 +117,17 @@ def print_interface_details(url_base, interface, username, password):
  # return the json as text
  print("Name: ", intf[0]["name"])
  try:
+ netmask = intf[0]["ietf-ip:ipv4"]["address"][0]["netmask"]
+ if cidr:
+ nma = ipaddress.ip_address(netmask)
+ netmask = str("{0:b}".format(nma._ip).count('1'))
  print("IP Address: ", intf[0]["ietf-ip:ipv4"]["address"][0]["ip"], "/",
-  intf[0]["ietf-ip:ipv4"]["address"][0]["netmask"])
+ netmask)
  except KeyError:
  print("IP Address: UNCONFIGURED")
+ except Exception as e:
+ print(e, file=sys.stderr)
+ sys.exit(1)
  print()
 
  return(intf)
@@ -142,12 +150,23 @@ def interface_selection(interfaces, mgmt_if):
  return(sel)
 
 
-# Asks the user to provide an IP address and Mask. Data is NOT validated.
-def get_ip_info():
+# Asks the user to provide an IP address and Mask.
+def get_ip_info(cidr):
  # Ask User for IP and Mask
  ip = {}
- ip["address"] = input("What IP address do you want to set? ")
- ip["mask"] = input("What Subnet Mask do you want to set? ")
+ try:
+ if cidr:
+ (ipa_t, ipl) = input("What IP address/prefixlen do you want to set? ").split('/')
+ ipa = ipaddress.ip_address(ipa_t)
+ ip["address"] = ipa.compressed
+ ip["mask"] = ipa._make_netmask(int(ipl))[0].compressed
+ else:
+ ipa_t = input("What IP address do you want to set? ")
+ ip["address"] = ipaddress.ip_address(ipa_t).compressed
+ ip["mask"] = input("What Subnet Mask do you want to set? ")
+ except Exception as e:
+ print(e, file=sys.stderr)
+ sys.exit(1)
  return(ip)
 
 
@@ -170,6 +189,8 @@ def main():
  help='management interface', default='GigabitEthernet1')
  parser.add_argument('--port', '-P', type=int,
  help='sandbox web port', default=443)
+ parser.add_argument('--cidr', help='use CIDR format for interface IP',
+ action='store_true')
  args = parser.parse_args()
 
  password = os.getenv('DEVNET_RESTCONF_PASSWORD')
@@ -196,18 +217,18 @@ def main():
  # Print Starting Interface Details
  print("Starting Interface Configuration")
  print_interface_details(url_base, selected_interface, args.username,
- password)
+ password, args.cidr)
 
  # As User for IP Address to set
- ip = get_ip_info()
+ ip = get_ip_info(args.cidr)
 
  # Configure interface
  configure_ip_address(url_base, selected_interface, ip, args.username, password)
 
  # Print Ending Interface Details
  print("Ending Interface Configuration")
  print_interface_details(url_base, selected_interface, args.username,
- password)
+ password, args.cidr)
 
 if __name__ == '__main__':
  sys.exit(main())